Persist and restore find relevant files checked state during workspace duplication

- when duplicating workspace, copy frf context as well

└── apps
    └── editor
        └── src
            ├── commands
            │   └── duplicate-workspace-command.ts
            ├── constants
            │   └── state-keys.ts
            └── context
                ├── context-initialization.ts
                ├── helpers
                │   ├── get-all-workspace-files.ts
                │   └── get-selected-files.ts
                ├── providers
                │   ├── open-editors
                │   │   └── open-editors-provider.ts
                │   ├── selected-files
                │   │   └── selected-files-provider.ts
                │   └── workspace
                │       ├── modules
                │       │   ├── index.ts
                │       │   └── token-calculator.ts
                │       └── workspace-provider.ts
                ├── shared-context-state.ts
                └── utils
                    └── shrink-file
                        ├── index.ts
                        ├── languages
                        │   ├── c-style.ts
                        │   ├── css.ts
                        │   ├── html.ts
                        │   ├── jsx.ts
                        │   ├── python.ts
                        │   ├── ruby.ts
                        │   └── sql.ts
                        ├── shrink-file.spec.ts
                        ├── shrink-file.ts
                        └── test-cases
                            ├── c-style
                            │   ├── input.txt
                            │   └── output.txt
                            ├── css
                            │   ├── input.txt
                            │   └── output.txt
                            ├── html
                            │   ├── input.txt
                            │   └── output.txt
                            ├── jsx
                            │   ├── input.txt
                            │   └── output.txt
                            ├── python
                            │   ├── input.txt
                            │   └── output.txt
                            ├── ruby
                            │   ├── input.txt
                            │   └── output.txt
                            ├── scss
                            │   ├── input.txt
                            │   └── output.txt
                            └── sql
                                ├── input.txt
                                └── output.txt

Author: robertpiosik

apps/editor/src/commands/duplicate-workspace-command.ts

@@ -16,9 +16,6 @@ export const duplicate_workspace_command = (
     async () => {
       const export_state = workspace_provider.get_export_state()
 
-      const checked_files = export_state.regular.checked_files
-      const checked_files_timestamps = export_state.regular.checked_timestamps
-
       const workspace_root_folders =
         vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath) ??
         []
@@ -43,8 +40,10 @@ export const duplicate_workspace_command = (
           RANGES_STATE_KEY
         )
       const context_to_save: DuplicateWorkspaceContext = {
-        checked_files,
-        checked_files_timestamps,
+        checked_files: export_state.regular.checked_files,
+        checked_files_timestamps: export_state.regular.checked_timestamps,
+        frf_checked_files: export_state.frf.checked_files,
+        frf_checked_files_timestamps: export_state.frf.checked_timestamps,
         workspace_root_folders,
         timestamp: Date.now(),
         open_editors,

apps/editor/src/constants/state-keys.ts

@@ -10,6 +10,9 @@ export const LAST_RANGES_SAVE_LOCATION_STATE_KEY = 'last-ranges-save-location'
 export const LAST_APPLY_CONTEXT_OPTION_STATE_KEY = 'last-apply-context-option'
 export const CONTEXT_CHECKED_PATHS_STATE_KEY = 'context-checked-paths'
 export const CONTEXT_CHECKED_TIMESTAMPS_STATE_KEY = 'context-checked-timestamps'
+export const CONTEXT_FRF_CHECKED_PATHS_STATE_KEY = 'context-frf-checked-paths'
+export const CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY =
+  'context-frf-checked-timestamps'
 export const LAST_APPLY_CONTEXT_MERGE_REPLACE_OPTION_STATE_KEY =
   'last-apply-context-merge-replace-option'
 export const LAST_FIND_RELEVANT_FILES_MERGE_REPLACE_OPTION_STATE_KEY =
@@ -116,6 +119,8 @@ export const LAST_COPY_PATHS_FORMAT_STATE_KEY = 'last-copy-paths-format'
 export type DuplicateWorkspaceContext = {
   checked_files: string[]
   checked_files_timestamps: Record<string, number>
+  frf_checked_files?: string[]
+  frf_checked_files_timestamps?: Record<string, number>
   timestamp: number
   workspace_root_folders: string[]
   open_editors?: { path: string; view_column?: number }[]

apps/editor/src/context/context-initialization.ts

@@ -9,6 +9,8 @@ import { EventEmitter } from 'events'
 import {
   CONTEXT_CHECKED_PATHS_STATE_KEY,
   CONTEXT_CHECKED_TIMESTAMPS_STATE_KEY,
+  CONTEXT_FRF_CHECKED_PATHS_STATE_KEY,
+  CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY,
   DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
   RANGES_STATE_KEY,
   type DuplicateWorkspaceContext
@@ -62,6 +64,20 @@ const restore_duplicated_workspace_context = async (
           duplicated_context.checked_files_timestamps
         )
 
+        if (duplicated_context.frf_checked_files) {
+          await extension_context.workspaceState.update(
+            CONTEXT_FRF_CHECKED_PATHS_STATE_KEY,
+            duplicated_context.frf_checked_files
+          )
+        }
+
+        if (duplicated_context.frf_checked_files_timestamps) {
+          await extension_context.workspaceState.update(
+            CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY,
+            duplicated_context.frf_checked_files_timestamps
+          )
+        }
+
         if (duplicated_context.ranges) {
           await extension_context.workspaceState.update(
             RANGES_STATE_KEY,

apps/editor/src/context/providers/workspace/workspace-provider.ts

@@ -5,6 +5,8 @@ import ignore, { Ignore } from 'ignore'
 import {
   CONTEXT_CHECKED_PATHS_STATE_KEY,
   CONTEXT_CHECKED_TIMESTAMPS_STATE_KEY,
+  CONTEXT_FRF_CHECKED_PATHS_STATE_KEY,
+  CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY,
   RANGES_STATE_KEY
 } from '@/constants/state-keys'
 import { IGNORED_LOCK_FILES } from '@/constants/ignored-lock-files'
@@ -225,9 +227,20 @@ export class WorkspaceProvider
   }
 
   private async _save_checked_files_state(): Promise<void> {
-    if (this._is_frf_mode) return
-
     const checked_paths = this.get_all_checked_paths()
+
+    if (this._is_frf_mode) {
+      await this._extension_context.workspaceState.update(
+        CONTEXT_FRF_CHECKED_PATHS_STATE_KEY,
+        checked_paths
+      )
+      await this._extension_context.workspaceState.update(
+        CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY,
+        Object.fromEntries(this._checked_timestamps)
+      )
+      return
+    }
+
     await this._extension_context.workspaceState.update(
       CONTEXT_CHECKED_PATHS_STATE_KEY,
       checked_paths
@@ -247,6 +260,13 @@ export class WorkspaceProvider
         Record<string, number>
       >(CONTEXT_CHECKED_TIMESTAMPS_STATE_KEY)
 
+      const frf_paths = this._extension_context.workspaceState.get<string[]>(
+        CONTEXT_FRF_CHECKED_PATHS_STATE_KEY
+      )
+      const frf_times = this._extension_context.workspaceState.get<
+        Record<string, number>
+      >(CONTEXT_FRF_CHECKED_TIMESTAMPS_STATE_KEY)
+
       await this.gitignore_initialization
       const prev_mode = this._is_frf_mode
 
@@ -258,6 +278,14 @@ export class WorkspaceProvider
         )
       }
 
+      this._is_frf_mode = true
+      if (frf_paths && frf_paths.length > 0) {
+        await this.set_checked_files(
+          frf_paths,
+          frf_times ? new Map(Object.entries(frf_times)) : undefined
+        )
+      }
+
       this._is_frf_mode = prev_mode
     }
     load()