handle end of day notification updates (#464)

* handle end of day notification updates

* add preferences_update ws case

package.json

@@ -1,7 +1,7 @@
 {
   "name": "swdc-vscode",
   "displayName": "Code Time",
-  "version": "2.7.0",
+  "version": "2.7.1",
   "publisher": "softwaredotcom",
   "description": "Code Time is an open source plugin that provides programming metrics right in Visual Studio Code.",
   "author": {
@@ -141,14 +141,14 @@
   },
   "dependencies": {
     "@swdotcom/editor-flow": "1.1.1",
-    "@types/uuid": "^8.3.4",
-    "@types/ws": "^8.5.3",
-    "axios": "^1.3.2",
-    "date-fns": "^2.29.3",
+    "@types/uuid": "^9.0.4",
+    "@types/ws": "^8.5.6",
+    "axios": "^1.5.1",
+    "date-fns": "^2.30.0",
     "node-cache": "^5.1.2",
-    "open": "^8.4.0",
+    "open": "^9.1.0",
     "swdc-tracker": "1.5.5",
-    "uuid": "^9.0.0",
-    "ws": "^8.11.0"
+    "uuid": "^9.0.1",
+    "ws": "^8.14.2"
   }
 }

src/extension.ts

@@ -33,6 +33,7 @@ import {initializeFlowModeState} from './managers/FlowManager';
 import { ExtensionManager } from './managers/ExtensionManager';
 import { LocalStorageManager } from './managers/LocalStorageManager';
 import { setSessionStorageManager } from './managers/FileManager';
+import { setEndOfDayNotification } from './notifications/endOfDay';
 
 let TELEMETRY_ON = true;
 let currentColorKind: number | undefined = undefined;
@@ -140,6 +141,11 @@ export async function intializePlugin() {
     // INIT session summary sync manager
     SyncManager.getInstance();
   }, 3000);
+
+  setTimeout(() => {
+    // Set the end of the day notification trigger if it's enabled
+    setEndOfDayNotification();
+  }, 5000);
 }
 
 export function getCurrentColorKind() {

src/managers/ConfigManager.ts

@@ -2,6 +2,7 @@ import {commands, ViewColumn, WebviewPanel, window} from 'vscode';
 import {getUser} from '../DataController';
 import {isResponseOk, appGet, appPut} from '../http/HttpClient';
 import {getConnectionErrorHtml} from '../local/404';
+import { setEndOfDayNotification } from '../notifications/endOfDay';
 
 let currentPanel: WebviewPanel | undefined = undefined;
 
@@ -61,6 +62,8 @@ export async function getEditSettingsHtml(): Promise<string> {
 export async function updateSettings(path: string, jsonData: any, reloadSettings: false) {
   await appPut(path, jsonData);
   await getUser();
+  // update the end of the day notification trigger
+  setEndOfDayNotification();
   // update the sidebar
   commands.executeCommand('codetime.refreshCodeTimeView');
 

src/notifications/endOfDay.ts

@@ -23,9 +23,7 @@ export const setEndOfDayNotification = async () => {
   const workHours: any = JSON.parse(cachedUser.profile.work_hours)
 
   // If the end of day notification setting is turned on (if undefined or null, will default to true)
-  if (preferences.notifications?.endOfDayNotification !== false) {
-    const jwt = getItem('jwt');
-
+  if (preferences.notifications?.endOfDayNotification) {
     const d = new Date();
     const day = format(d, 'EEE').toLowerCase();
     let msUntilEndOfTheDay = 0;

src/websockets.ts

@@ -6,6 +6,7 @@ import { handleIntegrationConnectionSocketEvent } from './message_handlers/integ
 import { handleCurrentDayStatsUpdate } from './message_handlers/current_day_stats_update';
 import { handleFlowStateMessage } from './message_handlers/flow_state';
 import { userDeletedCompletionHandler } from './DataController';
+import { setEndOfDayNotification } from './notifications/endOfDay';
 
 const WebSocket = require('ws');
 
@@ -238,6 +239,9 @@ const handleIncomingMessage = (data: any) => {
       case 'account_deleted':
         userDeletedCompletionHandler();
         break;
+      case 'preferences_update':
+        setEndOfDayNotification();
+        break;
     }
   } catch (e) {
     if (data) {