✨ feat: add runSnippetScript command

Author: wjkang

.eslintrc.js

@@ -37,5 +37,6 @@ module.exports = {
     'no-unused-expressions': 'off',
     'max-len': 'off',
     'class-methods-use-this': 'off',
+		'no-eval': 'off',
   },
 };

package.json

@@ -3,7 +3,7 @@
 	"displayName": "lowcode",
 	"description": "lowcode tool, support ChatGPT",
 	"author": "wjkang <[email protected]>",
-	"version": "1.5.0",
+	"version": "1.6.0",
 	"icon": "asset/icon.png",
 	"publisher": "wjkang",
 	"repository": "https://github.com/lowcoding/lowcode-vscode",
@@ -94,6 +94,10 @@
 			{
 				"command": "lowcode.askChatGPTWithTemplate",
 				"title": "Ask ChatGPT With Template"
+			},
+			{
+				"command": "lowcode.runSnippetScript",
+				"title": "Run Snippet Script"
 			}
 		],
 		"menus": {
@@ -134,6 +138,10 @@
 				{
 					"command": "lowcode.askChatGPTWithTemplate",
 					"group": "lowcode@5"
+				},
+				{
+					"command": "lowcode.runSnippetScript",
+					"group": "lowcode@6"
 				}
 			],
 			"lowcode/explorer/context": [

src/commands/generateCodeV2.ts

@@ -0,0 +1,61 @@
+import * as vscode from 'vscode';
+import * as path from 'path';
+import * as fs from 'fs-extra';
+import { getSnippets } from '../utils/materials';
+import { getEnv, rootPath } from '../utils/vscodeEnv';
+import { getInnerLibs } from '../utils/lib';
+import { getOutputChannel } from '../utils/outputChannel';
+import { createChatCompletionForScript } from '../utils/openai';
+
+const { window } = vscode;
+
+export const registerRunSnippetScript = (context: vscode.ExtensionContext) => {
+  context.subscriptions.push(
+    vscode.commands.registerTextEditorCommand(
+      'lowcode.runSnippetScript',
+      async () => {
+        const templateList = getSnippets().filter(
+          (s) => !s.preview.notShowInCommand,
+        );
+        if (templateList.length === 0) {
+          window.showErrorMessage('请配置模板(代码片段)');
+        }
+        const templateResult = await window.showQuickPick(
+          templateList.map((s) => s.name),
+          { placeHolder: '请选择模板' },
+        );
+        if (!templateResult) {
+          return;
+        }
+        const template = templateList.find((s) => s.name === templateResult);
+        const scriptFile = path.join(template!.path, 'script/index.js');
+        if (fs.existsSync(scriptFile)) {
+          delete eval('require').cache[eval('require').resolve(scriptFile)];
+          const script = eval('require')(scriptFile);
+          if (script.onSelect) {
+            const context = {
+              vscode,
+              workspaceRootPath: rootPath,
+              env: getEnv(),
+              libs: getInnerLibs(),
+              outputChannel: getOutputChannel(),
+              log: getOutputChannel(),
+              createChatCompletion: createChatCompletionForScript,
+              materialPath: template!.path,
+              code: '',
+            };
+            try {
+              await script.onSelect(context);
+            } catch (ex: any) {
+              window.showErrorMessage(ex.toString());
+            }
+          } else {
+            window.showErrorMessage('脚本中未实现 onSelect 方法');
+          }
+        } else {
+          window.showErrorMessage('当前模板中未添加脚本');
+        }
+      },
+    ),
+  );
+};

src/commands/runSnippetScript.ts

@@ -9,6 +9,7 @@ import { init, setLastActiveTextEditorId } from './context';
 import { registerQuickGenerateBlock } from './commands/quickGenerateBlock';
 import { registerChatGPTViewProvider } from './webview';
 import { registerChatGPTCommand } from './commands/chatGPT';
+import { registerRunSnippetScript } from './commands/runSnippetScript';
 
 export function activate(context: vscode.ExtensionContext) {
   vscode.window.onDidChangeActiveTextEditor(
@@ -26,6 +27,8 @@ export function activate(context: vscode.ExtensionContext) {
 
   generateCode(context);
 
+  registerRunSnippetScript(context);
+
   createOrShowWebview(context);
 
   createOrShowAddSnippetWebview(context);

src/extension.ts

@@ -93,6 +93,7 @@ export const genCodeByBlock = async (data: {
         .join(data.path, ...data.createPath)
         .replace(/\\/g, '/'),
       createChatCompletion: createChatCompletionForScript,
+      materialPath: block,
     };
     data.model = {
       ...data.model,
@@ -196,6 +197,7 @@ export const genCodeBySnippet = async (data: {
     outputChannel: getOutputChannel(),
     log: getOutputChannel(),
     createChatCompletion: createChatCompletionForScript,
+    materialPath: snippetPath,
     code: '',
   };
   const extendModel = await hook.beforeCompile(context);