From 2b48d32612e6ed4c33c199639939f86e24ebe2ed Mon Sep 17 00:00:00 2001
From: Timeo Williams <timwillie73@gmail.com>
Date: Wed, 25 Sep 2024 10:52:43 -0400
Subject: [PATCH] feat: added context for login vals to dynamically update
 NavBar options.

---
 electron.vite.config.ts              |  6 +--
 src/main/emailService.ts             | 34 ++++++++++---
 src/main/index.ts                    | 76 +++++++++++++++++-----------
 src/main/productivityUtils.ts        | 33 ++++++++++--
 src/main/worker.ts                   |  2 +-
 src/renderer/src/App.tsx             | 25 +++++----
 src/renderer/src/Login.tsx           |  5 +-
 src/renderer/src/Signup.tsx          |  6 +--
 src/renderer/src/lib/AuthContext.tsx | 26 ++++++++++
 src/renderer/src/lib/utils.ts        |  4 +-
 tsconfig.node.json                   |  3 +-
 tsconfig.web.json                    |  4 +-
 12 files changed, 163 insertions(+), 61 deletions(-)
 create mode 100644 src/renderer/src/lib/AuthContext.tsx

diff --git a/electron.vite.config.ts b/electron.vite.config.ts
index c777839..4b1d056 100644
--- a/electron.vite.config.ts
+++ b/electron.vite.config.ts
@@ -7,7 +7,7 @@ dotenv.config()
 
 export default defineConfig({
   main: {
-    plugins: [externalizeDepsPlugin(), bytecodePlugin()],
+    plugins: [externalizeDepsPlugin()],
     build: {
       outDir: 'out/main',
       rollupOptions: {
@@ -30,13 +30,13 @@ export default defineConfig({
     }
   },
   preload: {
-    plugins: [externalizeDepsPlugin(), bytecodePlugin()],
+    plugins: [externalizeDepsPlugin()],
     build: {
       outDir: 'out/preload',
       rollupOptions: {
         external: ['electron'],
         output: {
-          format: 'cjs'
+          format: 'es'
         }
       }
     }
diff --git a/src/main/emailService.ts b/src/main/emailService.ts
index 44b555f..c989f46 100644
--- a/src/main/emailService.ts
+++ b/src/main/emailService.ts
@@ -9,7 +9,7 @@ dotenv.config()
 
 interface TopSite {
   url: string
-  timeSpent: number
+  timeSpent: string
 }
 
 export class EmailService {
@@ -31,6 +31,11 @@ export class EmailService {
     schedule.scheduleJob('59 23 * * *', () => {
       this.sendDailySummary()
     })
+
+    if (!app.isPackaged) {
+      // For testing, send the email immediately
+      this.sendDailySummary()
+    }
   }
 
   private async sendDailySummary(): Promise<void> {
@@ -92,6 +97,15 @@ export class EmailService {
     }
   }
 
+  private formatTime(timeSpentInMinutes: number): string {
+    if (timeSpentInMinutes >= 60) {
+      const hours = Math.floor(timeSpentInMinutes / 60)
+      const minutes = timeSpentInMinutes % 60
+      return `${hours}h ${minutes}m`
+    }
+    return `${timeSpentInMinutes} minutes`
+  }
+
   public composeEmailBody(deepWorkHours: number, topSites: TopSite[]): string {
     return `
       <div style="font-family: Arial, sans-serif; color: #fff; background-color: #000; padding: 20px; border-radius: 8px; max-width: 80%; margin: auto;">
@@ -99,13 +113,13 @@ export class EmailService {
           <h1 style="color: #ecf0f1; margin: 0; font-style: italic; font-family: 'Montserrat', sans-serif;">deepFocus</h1>
         </div>
         <p style="font-size: 16px; margin-bottom: 20px;">Total Deep Work Hours: <strong>${deepWorkHours}</strong></p>
-        <h3 style="color: #ecf0f1; border-bottom: 2px solid #ecf0f1; padding-bottom: 10px; margin-top: 30px;">Top 5 Sites Visited</h3>
+        <h3 style="color: #ecf0f1; border-bottom: 2px solid #ecf0f1; padding-bottom: 10px; margin-top: 30px;">Top 3 Sites Visited</h3>
         <ul style="list-style-type: none; padding-left: 0;">
           ${topSites
             .map(
               (site) => `
                 <li style="background-color: #333; margin-bottom: 10px; padding: 10px; border-radius: 4px; border: 1px solid #555; color: #fff;">
-                  <strong>${site.url}</strong>: ${site.timeSpent} minutes
+                  <strong>${site.url}</strong>: ${site.timeSpent} 
                 </li>
               `
             )
@@ -123,18 +137,26 @@ export class EmailService {
       .filter((tracker) => !unproductiveSites?.includes(tracker.url))
       .reduce((total, tracker) => total + tracker.timeSpent, 0)
 
+    // Convert ms to hours and return rounded to 1 decimal place
     return parseFloat((productiveTime / (1000 * 60 * 60)).toFixed(1))
   }
 
   private async getTopSites(): Promise<TopSite[]> {
     const siteTimeTrackers = this.store.get('siteTimeTrackers', [])
 
+    const getTrimmedTitle = (title: string): string => {
+      const maxLength = 50
+      return title.length > maxLength ? title.slice(0, maxLength) + '...' : title
+    }
+
+    // Filter out trackers with zero time, then sort and slice to get top 3
     const topSites = siteTimeTrackers
+      .filter((tracker) => tracker.timeSpent > 0)
       .sort((a, b) => b.timeSpent - a.timeSpent)
-      .slice(0, 3)
+      .slice(0, 3) // Get top 3
       .map((tracker) => ({
-        url: tracker.url,
-        timeSpent: Math.round(tracker.timeSpent / (1000 * 60))
+        url: tracker.url || getTrimmedTitle(tracker.title || 'Unknown Title'),
+        timeSpent: this.formatTime(Math.round(tracker.timeSpent / (1000 * 60))) // Convert to minutes and format
       }))
 
     return topSites
diff --git a/src/main/index.ts b/src/main/index.ts
index 3fb2140..225b07b 100644
--- a/src/main/index.ts
+++ b/src/main/index.ts
@@ -10,21 +10,25 @@ import Store from 'electron-store'
 
 import { EmailService } from './emailService'
 import { StoreSchema, SiteTimeTracker, ExtendedResult } from './types'
-import { getUrlFromResult, formatTime, updateSiteTimeTracker } from './productivityUtils'
+import {
+  getUrlFromResult,
+  formatTime,
+  updateSiteTimeTracker,
+  getBaseURL
+} from './productivityUtils'
 
 export interface TypedStore extends Store<StoreSchema> {
   get<K extends keyof StoreSchema>(key: K): StoreSchema[K]
   get<K extends keyof StoreSchema>(key: K, defaultValue: StoreSchema[K]): StoreSchema[K]
   set(key: string, value: any): void
   delete<K extends keyof StoreSchema>(key: K): void
+  clear(): void
 }
 
 const store = new Store<StoreSchema>() as TypedStore
 let currentSiteTimeTrackers: SiteTimeTracker[] = []
 
 setupEnvironment()
-console.log('email', process.env.EMAIL)
-
 const emailService = new EmailService(process.env.EMAIL || '', store)
 
 // Initialize environment variables based on the environment
@@ -70,10 +74,12 @@ function resetCounters(type: 'daily' | 'weekly') {
   if (type === 'daily') {
     console.log('Resetting Daily Trackers')
     currentSiteTimeTrackers.forEach((tracker) => (tracker.timeSpent = 0))
+    console.log('currentSiteTimeTrackers after daily reset', currentSiteTimeTrackers)
+    console.log('store.get(lastResetDate)', store.get('lastResetDate'))
     store.set('lastResetDate', dayjs().format('YYYY-MM-DD'))
   } else if (type === 'weekly') {
     console.log('Resetting Weekly Trackers')
-    currentSiteTimeTrackers.length = 0
+    currentSiteTimeTrackers = []
   }
   store.set('siteTimeTrackers', currentSiteTimeTrackers)
 }
@@ -81,14 +87,17 @@ function resetCounters(type: 'daily' | 'weekly') {
 // Periodic saving of time trackers
 function setupPeriodicSave() {
   setInterval(() => store.set('siteTimeTrackers', currentSiteTimeTrackers), 5 * 60 * 1000)
+  console.log('setupPeriodicSave', store.get('siteTimeTrackers'))
 }
 
 // Monitor system idle time and user activity
 function startActivityMonitoring() {
   setInterval(async () => {
     const idleTime = powerMonitor.getSystemIdleTime()
-    if (idleTime > 60) return console.log(`System idle for ${idleTime} seconds.`)
-
+    if (idleTime > 60) {
+      console.log(`System idle for ${idleTime} seconds.`)
+      return
+    }
     try {
       const windowInfo = await activeWindow()
       if (windowInfo && windowInfo!.platform === 'macos') {
@@ -102,14 +111,16 @@ function startActivityMonitoring() {
     } catch (error) {
       console.error('Error getting active window:', error)
     }
-  }, 60000) // change back to 600000
+  }, 60000)
 }
 
 // Process activity data from active window
 function processActivityData(_windowInfoData: ExtendedResult | undefined) {
+  if (!_windowInfoData) return
+
   if (_windowInfoData?.siteTimeTracker) {
     console.log(
-      `Time spent on ${_windowInfoData.siteTimeTracker.title}: ${formatTime(_windowInfoData.siteTimeTracker.timeSpent)}. URL is ${_windowInfoData.url}`
+      `Time spent on ${_windowInfoData.siteTimeTracker.title}: ${formatTime(_windowInfoData.siteTimeTracker.timeSpent)}. URL is ${_windowInfoData?.url ? getBaseURL(_windowInfoData?.url) : `not defined`}`
     )
     console.log(
       `Last active timestamp: ${new Date(_windowInfoData.siteTimeTracker.lastActiveTimestamp).toISOString()}`
@@ -134,9 +145,6 @@ async function createWindow(): Promise<BrowserWindow> {
   mainWindow.on('ready-to-show', async () => {
     console.log('ready-to-show')
     mainWindow.show()
-    startActivityMonitoring()
-    currentSiteTimeTrackers = store.get('siteTimeTrackers', [])
-    emailService.scheduleEmailSend()
   })
 
   mainWindow.webContents.setWindowOpenHandler(({ url }) => {
@@ -160,32 +168,42 @@ app.whenReady().then(async () => {
   console.log('ready!')
   electronApp.setAppUserModelId('com.electron')
 
-  loadUserData()
-
-  const now = dayjs()
-  const lastResetDate = dayjs(store.get('lastResetDate'))
-
-  if (!lastResetDate.isSame(now, 'day') || now.diff(lastResetDate, 'hours') > 24) {
-    console.log('Missed daily reset from previous session, performing now.')
-    schedulerWorker.postMessage({ type: 'RESET_DAILY' })
-  }
-
-  await createWindow()
+  await createWindow().then(() => {
+    loadUserData()
+    handleDailyReset()
+    setupPeriodicSave()
+    setupIPCListeners()
+  })
 
-  setupPeriodicSave()
+  emailService.scheduleEmailSend()
+})
 
+function handleUserLogout() {
+  store.delete('user')
+  store.set('siteTimeTrackers', [])
+  schedulerWorker.postMessage({ type: 'RESET_TRACKERS' })
+}
+function setupIPCListeners() {
   ipcMain.on('send-user-data', (event, user) => {
     console.log('Received user data from frontend:', user, event.processId)
     handleUserData(user)
+    const savedUser = store.get('user')
+    if (savedUser) {
+      startActivityMonitoring()
+      currentSiteTimeTrackers = store.get('siteTimeTrackers', [])
+      emailService.scheduleEmailSend()
+    }
   })
   ipcMain.on('test-email-send', async () => await emailService.testEmailSend())
   ipcMain.on('logout-user', () => handleUserLogout())
-})
-
-function handleUserLogout() {
-  store.delete('user')
-  store.set('siteTimeTrackers', [])
-  schedulerWorker.postMessage({ type: 'RESET_TRACKERS' })
+}
+function handleDailyReset() {
+  const now = dayjs()
+  const lastResetDate = dayjs(store.get('lastResetDate'))
+  if (!lastResetDate.isSame(now, 'day') || now.diff(lastResetDate, 'hours') > 24) {
+    console.log('Missed daily reset from previous session, performing now.')
+    schedulerWorker.postMessage({ type: 'RESET_DAILY' })
+  }
 }
 
 app.on('before-quit', () => schedulerWorker.terminate())
diff --git a/src/main/productivityUtils.ts b/src/main/productivityUtils.ts
index 7273643..362b5fe 100644
--- a/src/main/productivityUtils.ts
+++ b/src/main/productivityUtils.ts
@@ -21,6 +21,16 @@ function getDomainFromUrl(url: string): string {
   return parsedUrl.hostname || ''
 }
 
+export function getBaseURL(url: string): string | null {
+  try {
+    const urlObj = new URL(url)
+    return `${urlObj.protocol}//${urlObj.hostname}` // This gives you the base URL
+  } catch (error) {
+    console.error('Invalid URL:', error)
+    return null
+  }
+}
+
 function isProductiveUrl(url: string): boolean {
   const domain = getDomainFromUrl(url)
   console.log(
@@ -63,6 +73,15 @@ export function formatUrl(input: string): string {
   }
 }
 
+function isValidURL(url: string): boolean {
+  try {
+    new URL(url) // URL constructor will throw an error if it's not a valid URL
+    return true
+  } catch (_) {
+    return false
+  }
+}
+
 export function formatTime(milliseconds: number): string {
   const seconds = Math.floor(milliseconds / 1000)
   const minutes = Math.floor(seconds / 60)
@@ -84,7 +103,15 @@ export function updateSiteTimeTracker(
   const currentTime = Date.now()
   const url = getUrlFromResult(windowInfo) || windowInfo.title
 
-  let tracker = timeTrackers.find((t) => t.url === url)
+  // Check if this is a URL or an app (if URL extraction fails)
+  let trackerKey = url
+  if (url && isValidURL(url)) {
+    trackerKey = getBaseURL(url) as string // Use base URL to track all paths under one domain
+  } else {
+    trackerKey = windowInfo.title || 'Unknown App'
+  }
+
+  let tracker = timeTrackers.find((t) => t.url === trackerKey)
   if (tracker) {
     console.log('Updating existing tracker')
     tracker.timeSpent += currentTime - tracker.lastActiveTimestamp
@@ -92,8 +119,8 @@ export function updateSiteTimeTracker(
   } else {
     console.log('Creating new tracker')
     tracker = {
-      url,
-      title: windowInfo.title,
+      url: trackerKey,
+      title: windowInfo.title || 'Unknown App',
       timeSpent: 0,
       lastActiveTimestamp: currentTime
     }
diff --git a/src/main/worker.ts b/src/main/worker.ts
index cd4e4da..f4c24eb 100644
--- a/src/main/worker.ts
+++ b/src/main/worker.ts
@@ -55,7 +55,7 @@ setInterval(() => {
   } else {
     console.log('Worker is running, but no username set yet.')
   }
-}, 30000)
+}, 60000)
 
 // Schedule daily reset at midnight
 schedule.scheduleJob('0 0 * * *', () => {
diff --git a/src/renderer/src/App.tsx b/src/renderer/src/App.tsx
index f02eb48..4af810d 100644
--- a/src/renderer/src/App.tsx
+++ b/src/renderer/src/App.tsx
@@ -1,6 +1,7 @@
-import { lazy, Suspense, onMount, createSignal } from 'solid-js'
+import { lazy, Suspense, onMount, createSignal, ComponentProps } from 'solid-js'
 import { Router, Route, A, useLocation } from '@solidjs/router'
 import { render } from 'solid-js/web'
+import { AuthProvider, useAuth } from './lib/AuthContext'
 
 import { sendUserToBackend } from './lib/utils'
 import './assets/main.css'
@@ -12,8 +13,8 @@ const Signup = lazy(() => import('./Signup'))
 const Versions = lazy(() => import('./Versions'))
 const HelloWorld = () => <h1>Hello World!</h1>
 
-const App = (props) => {
-  const [isLoggedIn, setIsLoggedIn] = createSignal(false)
+const App = (props: ComponentProps<typeof Router>) => {
+  const [isLoggedIn, setIsLoggedIn] = useAuth()
   const [isNewUser, setIsNewUser] = createSignal(true)
   const location = useLocation()
 
@@ -83,14 +84,16 @@ const App = (props) => {
 
 render(
   () => (
-    <Router root={App}>
-      <Suspense fallback={<div>Loading...</div>}>
-        <Route path="/" component={Versions} />
-        <Route path="/login" component={Login} />
-        <Route path="/hello-world" component={HelloWorld} />
-        <Route path="/signup" component={Signup} />
-      </Suspense>
-    </Router>
+    <AuthProvider>
+      <Router root={App}>
+        <Suspense fallback={<div>Loading...</div>}>
+          <Route path="/" component={Versions} />
+          <Route path="/login" component={Login} />
+          <Route path="/hello-world" component={HelloWorld} />
+          <Route path="/signup" component={Signup} />
+        </Suspense>
+      </Router>
+    </AuthProvider>
   ),
   document.getElementById('root') as HTMLElement
 )
diff --git a/src/renderer/src/Login.tsx b/src/renderer/src/Login.tsx
index 6cfbdeb..b236938 100644
--- a/src/renderer/src/Login.tsx
+++ b/src/renderer/src/Login.tsx
@@ -11,6 +11,7 @@ import { Grid } from './components/ui/grid'
 import { TextField, TextFieldInput, TextFieldLabel } from './components/ui/text-field'
 import User from './types'
 import { API_BASE_URL } from './config'
+import { useAuth } from './lib/AuthContext'
 
 import type { InferInput } from 'valibot'
 
@@ -27,6 +28,7 @@ function Login() {
   const [authForm, { Form, Field }] = createForm<AuthForm>()
   const [_loginError, setLoginError] = createSignal<null | string>(null)
   const navigate = useNavigate()
+  const [loggedIn, setIsLoggedIn] = useAuth()
 
   const handleSubmit: SubmitHandler<AuthForm> = async (values) => {
     try {
@@ -58,8 +60,7 @@ function Login() {
       localStorage.setItem('token', token)
       localStorage.setItem('user', JSON.stringify(user))
       sendUserToBackend(user)
-      window.location.reload()
-
+      setIsLoggedIn(true)
       navigate('/')
     } catch (error) {
       console.error('Login error:', error)
diff --git a/src/renderer/src/Signup.tsx b/src/renderer/src/Signup.tsx
index 287cb67..aba3f79 100644
--- a/src/renderer/src/Signup.tsx
+++ b/src/renderer/src/Signup.tsx
@@ -9,6 +9,7 @@ import { TextField, TextFieldInput, TextFieldLabel } from './components/ui/text-
 import { sendUserToBackend } from './lib/utils'
 import User from './types'
 import { API_BASE_URL } from './config'
+import { useAuth } from './lib/AuthContext'
 
 import type { SubmitHandler } from '@modular-forms/solid'
 import type { InferInput } from 'valibot'
@@ -28,6 +29,7 @@ function Signup() {
   const [authForm, { Form, Field }] = createForm<AuthForm>()
   const [signUpError, setSignUpError] = createSignal<null | string>(null)
   const navigate = useNavigate()
+  const [loggedIn, setIsLoggedIn] = useAuth()
 
   const handleSubmit: SubmitHandler<AuthForm> = async (values) => {
     try {
@@ -59,9 +61,7 @@ function Signup() {
       localStorage.setItem('token', token)
       localStorage.setItem('user', JSON.stringify(user))
       sendUserToBackend(user)
-      // Handle successful login (e.g., redirect to dashboard)
-      window.location.reload()
-
+      setIsLoggedIn(true)
       navigate('/')
       console.info('Navigating to home')
     } catch (error) {
diff --git a/src/renderer/src/lib/AuthContext.tsx b/src/renderer/src/lib/AuthContext.tsx
new file mode 100644
index 0000000..fa4c5c0
--- /dev/null
+++ b/src/renderer/src/lib/AuthContext.tsx
@@ -0,0 +1,26 @@
+import { RouterProps } from '@solidjs/router'
+import { createContext, useContext, createSignal, Accessor, Setter, JSX } from 'solid-js'
+
+// Create a context with two signals: `isLoggedIn` and `setIsLoggedIn`
+type AuthContextType = [Accessor<boolean>, Setter<boolean>]
+const AuthContext = createContext<AuthContextType>()
+
+export const AuthProvider = (props: {
+  children: number | boolean | Node | JSX.ArrayElement | (string & {}) | null | undefined
+}) => {
+  const [isLoggedIn, setIsLoggedIn] = createSignal(false) // Create the state here
+  return (
+    <AuthContext.Provider value={[isLoggedIn, setIsLoggedIn]}>
+      {props.children}
+    </AuthContext.Provider>
+  )
+}
+
+// Custom hook to access the auth context
+export const useAuth = (): AuthContextType => {
+  const context = useContext(AuthContext)
+  if (!context) {
+    throw new Error('useAuth must be used within an AuthProvider')
+  }
+  return context
+}
diff --git a/src/renderer/src/lib/utils.ts b/src/renderer/src/lib/utils.ts
index a543d11..6a5f0c1 100644
--- a/src/renderer/src/lib/utils.ts
+++ b/src/renderer/src/lib/utils.ts
@@ -1,13 +1,15 @@
 import type { ClassValue } from 'clsx'
 import { clsx } from 'clsx'
 import { twMerge } from 'tailwind-merge'
+import User from '../types'
 
 export function cn(...inputs: ClassValue[]): string {
   return twMerge(clsx(inputs))
 }
 
-export const sendUserToBackend = (user) => {
+export const sendUserToBackend = (user: User) => {
   console.log('Sending user data to backend:', user)
   const sanitizedUser = JSON.parse(JSON.stringify(user))
+  console.log('window is ', window)
   window?.electron.ipcRenderer.send('send-user-data', sanitizedUser)
 }
diff --git a/tsconfig.node.json b/tsconfig.node.json
index 7d89da0..617a43f 100644
--- a/tsconfig.node.json
+++ b/tsconfig.node.json
@@ -4,7 +4,8 @@
     "electron.vite.config.*",
     "src/main/**/*",
     "src/preload/**/*",
-    "src/main/*"
+    "src/main/*",
+    "src/main/**"
   ],
   "compilerOptions": {
     "composite": true,
diff --git a/tsconfig.web.json b/tsconfig.web.json
index eb90e9d..ef9aa7c 100644
--- a/tsconfig.web.json
+++ b/tsconfig.web.json
@@ -4,7 +4,9 @@
     "src/renderer/src/env.d.ts",
     "src/renderer/src/**/*",
     "src/renderer/src/**/*.tsx",
-    "src/preload/*.d.ts"
+    "src/preload/*.d.ts",
+    "src/renderer/src/**/*.ts",  
+    "src/renderer/src/**/*.tsx"  
   ],
   "compilerOptions": {
     "composite": true,