feat: add action params for custom tools (#91)

Author: 9aoy

src/index.ts

@@ -221,7 +221,11 @@ type ExtraTool = {
   /**
    * The action to perform when the tool is selected.
    */
-  action?: () => unknown;
+  action?: (context: {
+    templateName: string;
+    distFolder: string;
+    addAgentsMdSearchDirs: (dir: string) => void;
+  }) => unknown;
   /**
    * The custom command to run when the tool is selected.
    */
@@ -374,7 +378,12 @@ export async function create({
       );
       if (matchedTool) {
         if (matchedTool.action) {
-          await matchedTool.action();
+          await matchedTool.action({
+            templateName,
+            distFolder,
+            addAgentsMdSearchDirs: (dir: string) =>
+              agentsMdSearchDirs.push(dir),
+          });
         }
         if (matchedTool.command) {
           runCommand(matchedTool.command, distFolder);

test/custom-tools.test.ts

@@ -35,7 +35,9 @@ test('should run extra tool action', async () => {
       {
         value: 'custom-action',
         label: 'Custom Action',
-        action: () => {
+        action: ({ templateName, distFolder }) => {
+          expect(templateName).toBe('vanilla');
+          expect(distFolder).toBe(projectDir);
           actionCalled = true;
         },
       },