Ensure persistent context state by saving checked files and websites to both workspace and global storage

Author: robertpiosik

packages/vscode/src/context/providers/websites-provider.ts

@@ -162,12 +162,21 @@ export class WebsitesProvider
       CONTEXT_CHECKED_URLS_STATE_KEY,
       checked_urls
     )
+    await this.context.globalState.update(
+      CONTEXT_CHECKED_URLS_STATE_KEY,
+      checked_urls
+    )
   }
 
   public load_checked_websites_state(): void {
-    const checked_urls = this.context.workspaceState.get<string[]>(
+    let checked_urls = this.context.workspaceState.get<string[]>(
       CONTEXT_CHECKED_URLS_STATE_KEY
     )
+    if (!checked_urls) {
+      checked_urls = this.context.globalState.get<string[]>(
+        CONTEXT_CHECKED_URLS_STATE_KEY
+      )
+    }
     if (checked_urls) {
       for (const url of checked_urls) {
         this._checked_websites.set(url, vscode.TreeItemCheckboxState.Checked)

packages/vscode/src/context/providers/workspace-provider.ts

@@ -88,12 +88,21 @@ export class WorkspaceProvider
       CONTEXT_CHECKED_PATHS_STATE_KEY,
       checked_paths
     )
+    await this.context.globalState.update(
+      CONTEXT_CHECKED_PATHS_STATE_KEY,
+      checked_paths
+    )
   }
 
   public load_checked_files_state(): void {
-    const persisted_checked_paths = this.context.workspaceState.get<string[]>(
+    let persisted_checked_paths = this.context.workspaceState.get<string[]>(
       CONTEXT_CHECKED_PATHS_STATE_KEY
     )
+    if (!persisted_checked_paths) {
+      persisted_checked_paths = this.context.globalState.get<string[]>(
+        CONTEXT_CHECKED_PATHS_STATE_KEY
+      )
+    }
     if (persisted_checked_paths && persisted_checked_paths.length > 0) {
       this.set_checked_files(persisted_checked_paths)
     }