View time in badge

Author: Stigmatoz

src/common/utility.ts

@@ -17,4 +17,14 @@ export function convertStringTimeToSummaryTime(time:string) {
     resultTimeValue += min * 60;
 
     return resultTimeValue;
+}
+
+export function convertSummaryTimeToBadgeString(summaryTime:number):string {
+    const sec = summaryTime;
+    const min = Number((summaryTime / 60).toFixed(0));
+    const hours = Number((summaryTime / (60 * 60)).toFixed(1));
+
+    if (sec < 60) return `${sec}s`;
+    else if (min < 60) return `${min}m`;
+    else return `${hours}h`;
 }

src/compositions/daily-intervals.ts

@@ -7,7 +7,7 @@ export async function closeInterval(domain:string | null): Promise<void>{
     if (domain == null) return;
     const storage = injecStorage();
     const timeIntervalList = await storage.getValue(StorageParams.TIMEINTERVAL_LIST) as TimeInterval[];
-    const item = timeIntervalList.find(x => x.domain === domain && x.day == todayLocalDate());
+    const item = timeIntervalList?.find(x => x.domain === domain && x.day == todayLocalDate());
     item?.closeInterval();
     await storage.saveValue(StorageParams.TIMEINTERVAL_LIST, timeIntervalList);
 }
@@ -17,7 +17,7 @@ export async function addInterval(domain:string | null): Promise<void>{
 
     const storage = injecStorage();
     const timeIntervalList = await storage.getValue(StorageParams.TIMEINTERVAL_LIST) as TimeInterval[];
-    const item = timeIntervalList.find(x => x.domain === domain && x.day == todayLocalDate());
+    const item = timeIntervalList?.find(x => x.domain === domain && x.day == todayLocalDate());
     if (item != undefined) {
         if (item.day == todayLocalDate())
             item.addInterval();

src/compositions/types.ts

@@ -6,6 +6,6 @@ export interface BadgeState {
 
 export enum BadgeColor {
     red = '#fdb8b8',
-    green = '#e7e7e7',
+    green = '#6ec05e',
     none = '#000'
 }

src/tracker.ts

@@ -5,13 +5,14 @@ import { injectTabsRepository } from "./repository/inject-tabs-repository";
 import { isInBlackList } from "./compositions/black-list";
 import { useBadge } from "./compositions/set-badge";
 import { BadgeColor } from "./compositions/types";
-import { INTERVAL_INACTIVITY_DEFAULT, StorageParams,  } from "./storage/storage-params";
+import { INTERVAL_INACTIVITY_DEFAULT, StorageParams, VIEW_TIME_IN_BADGE_DEFAULT,  } from "./storage/storage-params";
 import { injecStorage } from "./storage/inject-storage";
 import { closeInterval } from "./compositions/daily-intervals";
 import { ActiveTab } from "./compositions/activeTab";
 import { isLimitExceeded } from "./compositions/limit-list";
 import { Tab } from "./entity/tab";
 import { useBlockPage } from "./compositions/block-page";
+import { convertSummaryTimeToBadgeString } from "./common/utility";
 
 async function trackTime(){
     const activeTabInstance = ActiveTab.getInstance();
@@ -64,6 +65,25 @@ async function mainTracker(state: Browser.Idle.IdleState, activeTab: Browser.Tab
         }
 
         tab.incSummaryTime();
+
+        const storage = injecStorage();
+        const viewInBadge = (await storage.getValue(
+            StorageParams.VIEW_TIME_IN_BADGE,
+            VIEW_TIME_IN_BADGE_DEFAULT
+        )) as boolean;
+
+        if (viewInBadge)
+            useBadge({
+                tabId: activeTab!.id!,
+                text: convertSummaryTimeToBadgeString(tab.days.at(-1)!.summary),
+                color: BadgeColor.green,
+            });
+        else 
+            useBadge({
+                tabId: activeTab!.id!,
+                text: "",
+                color: BadgeColor.red,
+            });
     }
 }