run-time.ts

import { onRuntimeMessage, sendMsg2Runtime } from "@api/chrome/runtime"

class RunTimeTracker {
    private start: number = Date.now()
    private url: string
    // Real host, including builtin hosts
    private host: string | undefined

    constructor(url: string) {
        this.url = url
        this.start = Date.now()
    }

    init(): void {
        this.fetchSite()

        onRuntimeMessage<void, void>(async req => {
            if (req.code === 'siteRunChange') {
                this.fetchSite()
                return { code: 'success' }
            }
            return { code: 'ignore' }
        })

        setInterval(() => this.collect(), 1000)
    }

    private fetchSite() {
        sendMsg2Runtime('cs.getRunSites', this.url)
            .then((site: timer.site.SiteKey) => this.host = site?.host)
            // Extension reloaded, so terminate
            .catch(() => this.host = undefined)
    }

    private async collect() {
        const now = Date.now()
        const lastTime = this.start

        try {
            if (this.host) {
                const event: timer.core.Event = {
                    start: lastTime,
                    end: now,
                    url: this.url,
                    ignoreTabCheck: false,
                    host: this.host,
                }
                await sendMsg2Runtime('cs.trackRunTime', event)
            }
            this.start = now
        } catch {
        }
    }
}

export default RunTimeTracker