Merge pull request #10 from lowcoding/v1.6.0

V1.6.0

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",
@@ -54,19 +54,19 @@
 	"contributes": {
 		"commands": [
 			{
-				"command": "yapi-code.generateCodeByWebview",
+				"command": "lowcode.generateCodeByWebview",
 				"title": "可视化生成代码"
 			},
 			{
-				"command": "yapi-code.generateCode",
+				"command": "lowcode.generateCode",
 				"title": "生成代码"
 			},
 			{
 				"command": "lowcode.openDownloadMaterials",
 				"title": "下载物料"
 			},
 			{
-				"command": "yapi-code.addSnippet",
+				"command": "lowcode.addSnippet",
 				"title": "添加代码片段"
 			},
 			{
@@ -94,6 +94,10 @@
 			{
 				"command": "lowcode.askChatGPTWithTemplate",
 				"title": "Ask ChatGPT With Template"
+			},
+			{
+				"command": "lowcode.runSnippetScript",
+				"title": "Run Snippet Script"
 			}
 		],
 		"menus": {
@@ -112,19 +116,19 @@
 			],
 			"lowcode/editor/context": [
 				{
-					"command": "yapi-code.generateCodeByWebview",
+					"command": "lowcode.generateCodeByWebview",
 					"group": "lowcode@1"
 				},
 				{
-					"command": "yapi-code.generateCode",
+					"command": "lowcode.generateCode",
 					"group": "lowcode@2"
 				},
 				{
 					"command": "lowcode.openDownloadMaterials",
 					"group": "lowcode@3"
 				},
 				{
-					"command": "yapi-code.addSnippet",
+					"command": "lowcode.addSnippet",
 					"group": "lowcode@4"
 				},
 				{
@@ -134,6 +138,10 @@
 				{
 					"command": "lowcode.askChatGPTWithTemplate",
 					"group": "lowcode@5"
+				},
+				{
+					"command": "lowcode.runSnippetScript",
+					"group": "lowcode@6"
 				}
 			],
 			"lowcode/explorer/context": [
@@ -142,7 +150,7 @@
 					"group": "lowcode@1"
 				},
 				{
-					"command": "yapi-code.generateCodeByWebview",
+					"command": "lowcode.generateCodeByWebview",
 					"group": "lowcode@2"
 				},
 				{

src/commands/addSnippet.ts

@@ -4,7 +4,7 @@ import { showWebView } from '../webview';
 
 export const createOrShowWebview = (context: vscode.ExtensionContext) => {
   context.subscriptions.push(
-    vscode.commands.registerTextEditorCommand('yapi-code.addSnippet', () => {
+    vscode.commands.registerTextEditorCommand('lowcode.addSnippet', () => {
       const content = getSelectedText() || getClipboardText();
       showWebView({
         key: 'main',

src/commands/createOrShowWebview.ts

@@ -4,21 +4,18 @@ import { showWebView } from '../webview';
 
 export const createOrShowWebview = (context: vscode.ExtensionContext) => {
   context.subscriptions.push(
-    vscode.commands.registerCommand(
-      'yapi-code.generateCodeByWebview',
-      (args) => {
-        const path = formatPath(args?.path);
-        showWebView({
-          key: 'main',
-          viewColumn: vscode.ViewColumn.Two,
-          task: path
-            ? {
-                task: 'updateSelectedFolder',
-                data: { selectedFolder: path },
-              }
-            : undefined,
-        });
-      },
-    ),
+    vscode.commands.registerCommand('lowcode.generateCodeByWebview', (args) => {
+      const path = formatPath(args?.path);
+      showWebView({
+        key: 'main',
+        viewColumn: vscode.ViewColumn.Two,
+        task: path
+          ? {
+              task: 'updateSelectedFolder',
+              data: { selectedFolder: path },
+            }
+          : undefined,
+      });
+    }),
   );
 };

src/commands/generateCode.ts

@@ -10,7 +10,7 @@ const { window } = vscode;
 export const generateCode = (context: vscode.ExtensionContext) => {
   context.subscriptions.push(
     vscode.commands.registerTextEditorCommand(
-      'yapi-code.generateCode',
+      'lowcode.generateCode',
       async () => {
         const rawClipboardText = getClipboardText();
         let clipboardText = rawClipboardText.trim();

src/commands/openSnippet.ts

@@ -8,7 +8,7 @@ import { showWebView } from '../webview';
 export const openSnippet = (context: vscode.ExtensionContext) => {
   context.subscriptions.push(
     vscode.commands.registerTextEditorCommand(
-      'yapi-code.openSnippetByWebview',
+      'lowcode.openSnippetByWebview',
       (textEditor: TextEditor, edit: TextEditorEdit, ...args: any[]) => {
         const name = args[0];
         const template = args[1];

src/commands/registerCompletion.ts

@@ -34,7 +34,7 @@ export const registerCompletion = (context: vscode.ExtensionContext) => {
             // 无法直接通过 ejs 编译，说明模板中需要额外的数据，触发命令打开 webview
             completionItem.insertText = '';
             completionItem.command = {
-              command: 'yapi-code.openSnippetByWebview',
+              command: 'lowcode.openSnippetByWebview',
               title: '',
               arguments: [s.name, s.template],
             };

src/commands/runSnippetScript.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/extension.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);
@@ -42,7 +45,7 @@ export function activate(context: vscode.ExtensionContext) {
     vscode.StatusBarAlignment.Left,
     -1,
   );
-  statusBarItem.command = 'yapi-code.generateCodeByWebview';
+  statusBarItem.command = 'lowcode.generateCodeByWebview';
   statusBarItem.text = '$(octoface) Low Code';
   statusBarItem.tooltip = '可视化生成代码';
   statusBarItem.show();

src/utils/generate.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);