feat: smi

Author: sheepzh

src/content-script/index.ts

@@ -9,6 +9,7 @@ import { sendMsg2Runtime } from "@api/chrome/runtime"
 import { initLocale } from "@i18n"
 import processLimit from "./limit"
 import printInfo from "./printer"
+import TimelineCollector from './timeline'
 import NormalTracker from "./tracker/normal"
 import RunTimeTracker from "./tracker/run-time"
 
@@ -60,6 +61,8 @@ async function main() {
     normalTracker.init()
     const runTimeTracker = new RunTimeTracker(url)
     runTimeTracker.init()
+    const timelineCollector = new TimelineCollector()
+    timelineCollector.init()
 
     // Execute only one time for each dom
     if (getOrSetFlag()) return

src/content-script/timeline.ts

@@ -0,0 +1,109 @@
+import { sendMsg2Runtime } from '@api/chrome/runtime'
+
+type Session = {
+    startTime: number
+    domain: string
+}
+
+export default class TimelineCollector {
+    private currentSession: Session | null = null;
+
+    private isActive: boolean = false;
+
+    /**
+     * Bind page visibility and focus events
+     */
+    init(): void {
+        document.addEventListener('visibilitychange', () => {
+            if (document.hidden) {
+                this.pauseTracking()
+            } else {
+                this.startTracking()
+            }
+        })
+
+        window.addEventListener('focus', () => this.startTracking())
+        window.addEventListener('blur', () => this.pauseTracking())
+        window.addEventListener('beforeunload', () => this.endCurrentSession())
+
+        if (document.readyState === 'complete') {
+            this.startTracking()
+        } else {
+            window.addEventListener('load', () => this.startTracking())
+        }
+    }
+
+    /**
+     * Start tracking current page
+     */
+    public startTracking(): void {
+        if (this.isActive) return
+
+        this.isActive = true
+        const now: number = Date.now()
+        const domain: string = this.getCurrentDomain()
+
+        this.currentSession = {
+            startTime: now,
+            domain: domain,
+        }
+    }
+
+    /**
+     * Pause tracking
+     */
+    public pauseTracking(): void {
+        if (!this.isActive || !this.currentSession) return
+
+        this.endCurrentSession()
+    }
+
+    /**
+     * End current session and generate event
+     */
+    private endCurrentSession(): void {
+        if (!this.currentSession) return
+
+        const now: number = Date.now()
+        const duration: number = now - this.currentSession.startTime
+
+        const event = this.createEvent(
+            this.currentSession.startTime,
+            this.currentSession.domain,
+            duration,
+        )
+        sendMsg2Runtime('cs.timelineEv', event)
+
+        this.currentSession = null
+        this.isActive = false
+    }
+
+    /**
+     * Create event object
+     */
+    private createEvent(startTime: number, domain: string, duration: number): timer.timeline.Event {
+        return {
+            startTime: startTime,
+            domain: domain,
+            duration: duration,
+        }
+    }
+
+    /**
+     * Get current domain
+     */
+    private getCurrentDomain(): string {
+        try {
+            return new URL(window.location.href).hostname
+        } catch (error) {
+            return window.location.hostname || 'unknown'
+        }
+    }
+
+    /**
+     * Destroy collector and remove event listeners
+     */
+    public destroy(): void {
+        this.endCurrentSession()
+    }
+}

src/database/timeline-database.ts

@@ -0,0 +1,31 @@
+import BaseDatabase from './common/base-database'
+import { REMAIN_WORD_PREFIX } from './common/constant'
+
+const DB_KEY = REMAIN_WORD_PREFIX + 'TL'
+
+type Item = {
+    // start
+    s: number
+    // duration
+    d: number
+    // host
+    h: string
+}
+
+type TimelineData = {
+    date: []
+}
+
+class TimelineDatabase extends BaseDatabase {
+
+    async save(ev: timer.timeline.Event): Promise<void> {
+
+    }
+
+    async importData(_: any): Promise<void> {
+        // do nothing
+    }
+}
+const timelineDatabase = new TimelineDatabase()
+
+export default timelineDatabase

types/timer/mq.d.ts

@@ -36,6 +36,8 @@ declare namespace timer.mq {
         | "cs.idleChange"
         // @since 3.2.0
         | "cs.getRunSites"
+        // @since 3.6.1
+        | "cs.timelineEv"
 
     type ResCode = "success" | "fail" | "ignore"
 

types/timer/timeline.d.ts

@@ -0,0 +1,7 @@
+declare namespace timer.timeline {
+    type Event = {
+        startTime: number
+        domain: string
+        duration: number
+    }
+}