Implement the ability to clear all checkpoints and their associated files via a new quick pick option

Author: robertpiosik

packages/vscode/src/commands/checkpoints-command/actions/clear-all-checkpoints.ts

@@ -0,0 +1,23 @@
+import * as vscode from 'vscode'
+import { CHECKPOINTS_STATE_KEY } from '../../../constants/state-keys'
+import { get_checkpoint_path } from '../utils'
+import { get_checkpoints } from './get-checkpoints'
+
+export const clear_all_checkpoints = async (
+  context: vscode.ExtensionContext
+) => {
+  const checkpoints = await get_checkpoints(context)
+
+  for (const checkpoint of checkpoints) {
+    try {
+      const checkpoint_path = get_checkpoint_path(checkpoint.timestamp)
+      await vscode.workspace.fs.delete(vscode.Uri.file(checkpoint_path), {
+        recursive: true
+      })
+    } catch (error) {
+      console.warn(`Could not delete checkpoint file: ${error}`)
+    }
+  }
+
+  await context.workspaceState.update(CHECKPOINTS_STATE_KEY, [])
+}

packages/vscode/src/commands/checkpoints-command/actions/create-checkpoint.ts

@@ -1,5 +1,8 @@
 import * as vscode from 'vscode'
-import { CHECKPOINTS_STATE_KEY } from '../../../constants/state-keys'
+import {
+  CHECKPOINTS_STATE_KEY,
+  TEMPORARY_CHECKPOINT_STATE_KEY
+} from '../../../constants/state-keys'
 import { WorkspaceProvider } from '../../../context/providers/workspace-provider'
 import type { Checkpoint } from '../types'
 import { get_checkpoints } from './get-checkpoints'
@@ -22,6 +25,25 @@ export const create_checkpoint = async (
   description?: string
 ): Promise<Checkpoint | undefined> => {
   try {
+    const old_temp_checkpoint = context.workspaceState.get<Checkpoint>(
+      TEMPORARY_CHECKPOINT_STATE_KEY
+    )
+    if (old_temp_checkpoint) {
+      try {
+        const checkpoint_path = get_checkpoint_path(
+          old_temp_checkpoint.timestamp
+        )
+        await vscode.workspace.fs.delete(vscode.Uri.file(checkpoint_path), {
+          recursive: true
+        })
+      } catch (error) {
+        console.warn(`Could not delete old temporary checkpoint file: ${error}`)
+      }
+      await context.workspaceState.update(
+        TEMPORARY_CHECKPOINT_STATE_KEY,
+        undefined
+      )
+    }
     const timestamp = Date.now()
     const workspace_folders = vscode.workspace.workspaceFolders!
 

packages/vscode/src/commands/checkpoints-command/actions/index.ts

@@ -1,6 +1,7 @@
 export * from './create-checkpoint'
 export * from './create-temporary-checkpoint'
 export * from './delete-checkpoint'
+export * from './clear-all-checkpoints'
 export * from './get-checkpoints'
 export * from './restore-checkpoint'
 export * from './sync-workspace-from-dir'

packages/vscode/src/commands/checkpoints-command/checkpoints-command.ts

@@ -9,6 +9,7 @@ import relativeTime from 'dayjs/plugin/relativeTime'
 import type { Checkpoint } from './types'
 import {
   create_checkpoint,
+  clear_all_checkpoints,
   delete_checkpoint,
   get_checkpoints,
   restore_checkpoint
@@ -116,6 +117,15 @@ export const checkpoints_command = (
               label: '$(add) New checkpoint',
               alwaysShow: true
             },
+            ...(visible_checkpoints.length > 0
+              ? [
+                  {
+                    id: 'clear-all',
+                    label: '$(trash) Clear all checkpoints',
+                    alwaysShow: true
+                  }
+                ]
+              : []),
             ...(revert_item ? [revert_item] : []),
             ...(visible_checkpoints.length > 0
               ? [
@@ -194,6 +204,25 @@ export const checkpoints_command = (
               TEMPORARY_CHECKPOINT_STATE_KEY,
               undefined
             )
+          } else if (selected.id == 'clear-all') {
+            quick_pick.hide()
+            const confirmation = await vscode.window.showWarningMessage(
+              'Are you sure you want to clear all checkpoints? This action is irreversible.',
+              { modal: true },
+              'Clear All'
+            )
+
+            if (confirmation == 'Clear All') {
+              await clear_all_checkpoints(context)
+              vscode.window.showInformationMessage(
+                'All checkpoints have been cleared.'
+              )
+            }
+            await context.workspaceState.update(
+              TEMPORARY_CHECKPOINT_STATE_KEY,
+              undefined
+            )
+            await show_quick_pick()
           } else if (selected.checkpoint) {
             quick_pick.hide()
             await restore_checkpoint({