Add Generate All and Clean All commands to go deep in folders. (#135)

DavidSouther · May 27, 2024 · 3071b17 · 3071b17
1 parent 3c761ea
commit 3071b17
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 12 deletions.
diff --git a/core/src/content/content.ts b/core/src/content/content.ts
@@ -295,6 +295,7 @@ export async function loadContent(
   meta: ContentMeta = {},
   depth: number = Number.MAX_SAFE_INTEGER
 ): Promise<Record<string, Content>> {
+  if (depth < 0) throw new Error("Depth in loadContent cannot be negative");
   if (depth < 1) return {};
   LOGGER.debug(`Loading content from ${fs.cwd()}`);
   [system, meta] = await loadAillyRc(fs, system, meta);

diff --git a/extension/package.json b/extension/package.json
@@ -24,10 +24,20 @@
         "command": "ailly.generate",
         "title": "Ailly: Run"
       },
+      {
+        "command": "ailly.generate.all",
+        "title": "Ailly: Run All",
+        "enablement": "explorerViewletFocus"
+      },
       {
         "command": "ailly.clean",
         "title": "Ailly: Clean"
       },
+      {
+        "command": "ailly.clean.all",
+        "title": "Ailly: Clean",
+        "enablement": "explorerViewletFocus"
+      },
       {
         "command": "ailly.edit",
         "title": "Ailly: Edit"
@@ -39,9 +49,17 @@
           "command": "ailly.generate",
           "group": "3_generate"
         },
+        {
+          "command": "ailly.generate.all",
+          "group": "3_generate"
+        },
         {
           "command": "ailly.clean",
           "group": "3_generate"
+        },
+        {
+          "command": "ailly.clean.all",
+          "group": "3_generate"
         }
       ]
     },

diff --git a/extension/src/extension.ts b/extension/src/extension.ts
@@ -27,11 +27,12 @@ export function activate(context: vscode.ExtensionContext) {
   });
 
   registerGenerateCommand(context, {
-    name: "edit",
+    name: "generate.all",
     manager: statusManager,
-    gerund: "editing",
-    pastpart: "edited",
-    edit: true,
+    gerund: "running all in",
+    pastpart: "ran all in",
+    infinitive: "run all in",
+    deep: true,
   });
 
   registerGenerateCommand(context, {
@@ -42,6 +43,23 @@ export function activate(context: vscode.ExtensionContext) {
     clean: true,
   });
 
+  registerGenerateCommand(context, {
+    name: "clean.all",
+    manager: statusManager,
+    gerund: "cleaning all in",
+    pastpart: "cleaned all in",
+    clean: true,
+    deep: true,
+  });
+
+  registerGenerateCommand(context, {
+    name: "edit",
+    manager: statusManager,
+    gerund: "editing",
+    pastpart: "edited",
+    edit: true,
+  });
+
   context.subscriptions.push(
     vscode.commands.registerCommand("ailly.set-engine", async () => {
       vscode.window.showInformationMessage(`Ailly setting engine`);
@@ -65,6 +83,7 @@ export function registerGenerateCommand(
     manager,
     edit = false,
     clean = false,
+    deep = false,
     gerund,
     pastpart,
     infinitive = name,
@@ -73,6 +92,7 @@ export function registerGenerateCommand(
     manager: StatusManager;
     clean?: boolean;
     edit?: boolean;
+    deep?: boolean;
     gerund: string;
     pastpart: string;
     infinitive?: string;

diff --git a/extension/src/generate.ts b/extension/src/generate.ts
@@ -31,10 +31,12 @@ export async function generate(
     extensionEdit,
     manager,
     clean = false,
+    depth = 1,
   }: {
     extensionEdit?: ExtensionEdit;
     manager: StatusManager;
     clean?: boolean;
+    depth?: number;
   }
 ) {
   resetLogger();
@@ -66,6 +68,7 @@ export async function generate(
     path,
     extensionEdit,
     settings,
+    depth,
   });
 
   if (content.length === 0) {
@@ -149,13 +152,15 @@ async function loadContentParts({
   path,
   extensionEdit,
   settings,
+  depth = 1,
 }: {
   fs: FileSystem;
   path: string;
-  extensionEdit?: ExtensionEdit;
+  extensionEdit: ExtensionEdit | undefined;
   settings: PipelineSettings;
+  depth: number;
 }): Promise<[Content[], Record<string, Content>]> {
-  const context = await loadContent(fs, [], settings, 1);
+  const context = await loadContent(fs, [], settings, depth);
   const content: Content[] = Object.values(context).filter((c) =>
     c.path.startsWith(path)
   );
@@ -191,9 +196,3 @@ async function loadContentParts({
 
   return [content, context];
 }
-
-export const TEST_ONLY = {
-  loadContentParts,
-  executeEdit,
-  executeStreaming,
-};