Merge pull request #38 from MetabobProject/feat-expiration-for-state

Feat expiration for state

Author: AviGopal

ext-src/extension.ts

@@ -10,6 +10,7 @@ import {
   activateAnalyzeCommand,
 } from './commands';
 import {
+  handleAnalyzeExpiration,
   createOrUpdateUserSession,
   initState,
   AnalyzeDocumentOnSave,
@@ -24,6 +25,8 @@ import {
 import { Analyze } from './state';
 import { Problem } from './types';
 
+let expirationTimer: any = undefined;
+
 export function activate(context: vscode.ExtensionContext): void {
   let previousEditor: vscode.TextEditor | undefined = undefined;
   bootstrapExtensionEventEmitter();
@@ -37,6 +40,15 @@ export function activate(context: vscode.ExtensionContext): void {
   const analyzeDocumentOnSaveConfig = AnalyzeDocumentOnSaveConfig();
 
   try {
+    // handle Analyze State Expiration
+    handleAnalyzeExpiration(context);
+
+    const one_minute = 60_000;
+    const thirty_minutes = one_minute * 30;
+    expirationTimer = setInterval(() => {
+      handleAnalyzeExpiration(context);
+    }, thirty_minutes)
+
     // Create User Session, If already created get the refresh token
     // otherwise, ping server every 60 second to not destroy the token
     // if the user has not done any activity
@@ -460,4 +472,7 @@ export function activate(context: vscode.ExtensionContext): void {
 export function deactivate(): void {
   decorationType.dispose();
   disposeExtensionEventEmitter();
+  if (expirationTimer) {
+    clearInterval(expirationTimer);
+  }
 }

ext-src/helpers/HandleAnalyzeExpiration.ts

@@ -0,0 +1,37 @@
+import vscode from "vscode";
+import { Analyze, AnalyzeState } from "../state";
+
+export const handleAnalyzeExpiration = (context: vscode.ExtensionContext) => {
+    const today = new Date();
+    let results: AnalyzeState = {};
+    const setAnalyzeState = new Analyze(context)
+    const analyzeStateValue = new Analyze(context).get()?.value;
+
+    if (!analyzeStateValue) return undefined;
+
+    // If the expiration field on analyzeState does not exist; discard it.
+    Object.keys(analyzeStateValue).forEach(key => {
+        const problem = analyzeStateValue[key];
+        if (problem.expiration !== undefined) {
+            results[key] = { ...problem }
+        }
+    });
+
+    let nonExpiredResults: AnalyzeState = {};
+
+    // If the problem is expired, discard it as well.
+    Object.keys(results).forEach(key => {
+        const problem = analyzeStateValue[key];
+        if (problem.expiration) {
+            // Ensure both dates are in the same timezone (e.g., UTC)
+            const todayUTC = new Date(today.toISOString());
+            const problemDate = new Date(problem.expiration)
+            if (problemDate >= todayUTC) {
+                nonExpiredResults[key] = { ...problem }
+            }
+        }
+    });
+
+    setAnalyzeState.set(nonExpiredResults);
+    return
+}

ext-src/helpers/HandleDocumentAnalyze.ts

@@ -31,6 +31,9 @@ export const handleDocumentAnalyze = async (
   jobId?: string,
   suppressRateLimitErrors = false,
 ) => {
+  const tomorrow = new Date();
+  tomorrow.setDate(tomorrow.getDate() + 1);
+
   const currentWorkSpaceFolder = Util.getRootFolderName();
   const editor = vscode.window.activeTextEditor;
   if (!editor || editor.document.fileName !== metaDataDocument.filePath) {
@@ -168,6 +171,7 @@ export const handleDocumentAnalyze = async (
         isEndorsed: problem.endorsed,
         isViewed: false,
         fullFilePath: currentWorkSpaceFolder,
+        expiration: tomorrow.toISOString(),
       };
       results[key] = { ...analyzeMetaData };
     });

ext-src/helpers/index.ts

@@ -3,3 +3,4 @@ export * from './CreateOrUpdateUserSession'
 export * from './GenerateDecorations'
 export * from './HandleDocumentAnalyze'
 export * from './InitState'
+export * from "./HandleAnalyzeExpiration"

ext-src/state/Analyze.ts

@@ -15,6 +15,7 @@ export type AnalyseMetaData = {
   isEndorsed?: boolean;
   isViewed?: boolean;
   fullFilePath?: string;
+  expiration?: string;
 };
 
 export type AnalyzeState = {

src/hooks/useUser.tsx

@@ -1,4 +1,5 @@
-import { useContext } from 'react'
-import { AccountSettingContext } from '../context/UserContext'
+import { useContext } from 'react';
+import { AccountSettingContext } from '../context/UserContext';
 
-export const useUser = () => useContext(AccountSettingContext)
+// @ts-ignore
+export const useUser = () => useContext(AccountSettingContext);

src/theme.ts

@@ -29,7 +29,7 @@ const shadows: ThemeOptions['shadows'] = [
   '0px 11px 15px -7px rgba(19, 17, 32, 0.2), 0px 24px 38px 3px rgba(19, 17, 32, 0.14), 0px 9px 46px 8px rgba(19, 17, 32, 0.12)',
 ];
 
-export const registerTheme = () => {
+export const registerTheme = (): ThemeOptions => {
   const vscodeTheme = getComputedStyle(document.documentElement);
   const mainColor = vscodeTheme.getPropertyValue('--vscode-input-foreground');
   const backgroundDefault = vscodeTheme.getPropertyValue('--vscode-input-background');