Set up settings in local storage

src/content/add-billability-chart.ts

@@ -3,6 +3,7 @@ import { TimeChimpApi, User } from '../TimeChimpApi';
 import { toIsoDate } from '../date';
 import { endOfWeek, getWeek, startOfWeek, subWeeks } from 'date-fns';
 import { calculateTimeStats } from './stats';
+import { getSettings } from './settings';
 
 const api = new TimeChimpApi();
 
@@ -42,18 +43,17 @@ async function doAddBillabilityChart(date: Date, user: User) {
         api.getCompany(),
     ]);
 
-    // TODO: Only set this when that setting is enabled.
-    const contractHours = user.contractHours;
+    const settings = getSettings();
 
     const stats = calculateTimeStats(
         times,
-        contractHours,
+        settings.relativeToContractHours ? user.contractHours : undefined,
         SHOW_WEEKS,
         ROLLING_AVG_WEEKS,
     );
     createOrUpdateChart(
         stats,
-        !!contractHours,
+        settings.relativeToContractHours,
         company.theme?.mainColor,
         chartContainer,
     );

src/content/settings.ts

@@ -0,0 +1,60 @@
+const STORAGE_KEY = 'tcbc-settings';
+
+export interface Settings {
+    relativeToContractHours: boolean;
+}
+
+let settings: Settings | undefined;
+
+const DEFAULT_SETTINGS: Settings = {
+    relativeToContractHours: false,
+};
+
+export function getSettings(): Settings {
+    // Try to load the settings.
+    if (!settings) {
+        tryLoadSettings();
+    }
+
+    // If the settings are still unset, loading failed.
+    // Set the default settings.
+    if (!settings) {
+        settings = { ...DEFAULT_SETTINGS };
+        saveSettings();
+    }
+
+    return settings;
+}
+
+export function updateSettings(updates: Partial<Settings>) {
+    settings = {
+        ...DEFAULT_SETTINGS,
+        ...settings,
+        ...updates,
+    };
+    saveSettings();
+}
+
+function tryLoadSettings() {
+    try {
+        loadSettings();
+    } catch (e) {
+        console.error(`Failed to load ${STORAGE_KEY}: ${e}`);
+    }
+}
+
+function loadSettings() {
+    const str = localStorage.getItem(STORAGE_KEY);
+    if (!str) {
+        return;
+    }
+
+    const obj = JSON.parse(str);
+    settings = {
+        relativeToContractHours: obj.relativeToContractHours,
+    };
+}
+
+function saveSettings() {
+    localStorage.setItem(STORAGE_KEY, JSON.stringify(settings));
+}