快速选择调试需要的参数

Author: ccagml

CHANGELOG.md

@@ -1,3 +1,7 @@
+## version 2.17.1
+
+- 增加调试需要参数快速选择按钮
+
 ## version 2.16.1
 
 - 增加中英文多语言的配置

package.json

@@ -2,7 +2,7 @@
     "name": "vscode-leetcode-problem-rating",
     "displayName": "LeetCode",
     "description": "%main.description%",
-    "version": "2.16.2",
+    "version": "2.17.1",
     "author": "ccagml",
     "publisher": "ccagml",
     "license": "MIT",

src/controller/DebugController.ts

@@ -7,14 +7,16 @@
  * Copyright (c) 2023 ccagml . All rights reserved
  */
 
-import { TextDocument, window } from "vscode";
+import { TextDocument, window, Range } from "vscode";
 import { getTextEditorFilePathByUri } from "../utils/SystemUtils";
 import * as fs from "fs";
 import { fileMeta, ProblemMeta, supportDebugLanguages } from "../utils/problemUtils";
 
 import { debugService } from "../service/DebugService";
 import { debugArgDao } from "../dao/debugArgDao";
 
+import { IQuickItemEx } from "../model/Model";
+
 // 做杂活
 class DebugContorller {
   constructor() {}
@@ -66,6 +68,122 @@ class DebugContorller {
       //
     }
   }
+
+  public async addDebugType(document: TextDocument, addType) {
+    const picks: Array<IQuickItemEx<string>> = [
+      { label: "number", detail: "类型说明:数字", value: "number" },
+      { label: "number[]", detail: "类型说明:数字数组", value: "number[]" },
+      { label: "number[][]", detail: "类型说明:数字二维数组", value: "number[][]" },
+      { label: "string", detail: "类型说明:字符串", value: "string" },
+      { label: "string[]", detail: "类型说明:字符串数组", value: "string[]" },
+      { label: "string[][]", detail: "类型说明:字符串二维数组", value: "string[][]" },
+      { label: "ListNode", detail: "类型说明:链表", value: "ListNode" },
+      { label: "ListNode[]", detail: "类型说明:链表数组", value: "ListNode[]" },
+      { label: "character", detail: "类型说明:字节", value: "character" },
+      { label: "character[]", detail: "类型说明:字节数组", value: "character[]" },
+      { label: "character[][]", detail: "类型说明:字节二维数组", value: "character[][]" },
+      { label: "NestedInteger[]", detail: "类型说明:数组", value: "NestedInteger[]" },
+      { label: "MountainArray", detail: "类型说明:数组", value: "MountainArray" },
+      { label: "TreeNode", detail: "类型说明:树节点", value: "TreeNode" },
+    ];
+
+    const choice: IQuickItemEx<string> | undefined = await window.showQuickPick(picks, {
+      title: "添加调试需要的参数",
+      matchOnDescription: false,
+      matchOnDetail: false,
+      placeHolder: "选择要添加的分类",
+      canPickMany: false,
+    });
+    if (!choice) {
+      return;
+    }
+
+    const content: string = document.getText();
+    const matchResult: RegExpMatchArray | null = content.match(/@lc app=.* id=(.*) lang=(.*)/);
+    if (!matchResult || !matchResult[2]) {
+      return undefined;
+    }
+    // 搜集所有debug
+    let debugFlag: boolean = false;
+    for (let i: number = 0; i < document.lineCount; i++) {
+      const lineContent: string = document.lineAt(i).text;
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-end") >= 0) {
+        debugFlag = false;
+      }
+
+      if (debugFlag) {
+        let equal_index = lineContent.indexOf("=");
+        const last_index = document.lineAt(i).range.end.character;
+        if (addType == "paramTypes" && lineContent.indexOf("paramTypes=") >= 0) {
+          window.activeTextEditor?.edit((edit) => {
+            // 参数是个数组;
+            // edit.replace(new Position(i, equal_index + 1), choice.value);
+            let cur_param_str = lineContent.substring(equal_index + 1);
+            let cur_param_array: any = [];
+            try {
+              cur_param_array = JSON.parse(cur_param_str);
+            } catch (error) {
+              cur_param_array = [];
+            }
+
+            cur_param_array.push(choice.value);
+
+            edit.replace(new Range(i, equal_index + 1, i, last_index), JSON.stringify(cur_param_array));
+          });
+        } else if (addType == "returnType" && lineContent.indexOf("returnType=") >= 0) {
+          window.activeTextEditor?.edit((edit) => {
+            edit.replace(new Range(i, equal_index + 1, i, last_index), choice.value);
+          });
+        }
+      }
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-start") >= 0) {
+        debugFlag = true;
+      }
+    }
+    return;
+  }
+  public async resetDebugType(document: TextDocument, addType) {
+    const content: string = document.getText();
+    const matchResult: RegExpMatchArray | null = content.match(/@lc app=.* id=(.*) lang=(.*)/);
+    if (!matchResult || !matchResult[2]) {
+      return undefined;
+    }
+    // 搜集所有debug
+    let debugFlag: boolean = false;
+    for (let i: number = 0; i < document.lineCount; i++) {
+      const lineContent: string = document.lineAt(i).text;
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-end") >= 0) {
+        debugFlag = false;
+      }
+
+      if (debugFlag) {
+        let equal_index = lineContent.indexOf("=");
+        const last_index = document.lineAt(i).range.end.character;
+        if (addType == "paramTypes" && lineContent.indexOf("paramTypes=") >= 0) {
+          window.activeTextEditor?.edit((edit) => {
+            let cur_param_array: any = [];
+            edit.replace(new Range(i, equal_index + 1, i, last_index), JSON.stringify(cur_param_array));
+          });
+        } else if (addType == "returnType" && lineContent.indexOf("returnType=") >= 0) {
+          window.activeTextEditor?.edit((edit) => {
+            edit.replace(new Range(i, equal_index + 1, i, last_index), "");
+          });
+        }
+      }
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-start") >= 0) {
+        debugFlag = true;
+      }
+    }
+    return;
+  }
 }
 
 export const debugContorller: DebugContorller = new DebugContorller();

src/extension.ts

@@ -141,6 +141,12 @@ export async function activate(context: ExtensionContext): Promise<void> {
       }),
       commands.registerCommand("lcpr.simpleDebug", (document: TextDocument, testCase?) =>
         debugContorller.startDebug(document, testCase)
+      ),
+      commands.registerCommand("lcpr.addDebugType", (document: TextDocument, addType) =>
+        debugContorller.addDebugType(document, addType)
+      ),
+      commands.registerCommand("lcpr.resetDebugType", (document: TextDocument, addType) =>
+        debugContorller.resetDebugType(document, addType)
       )
     );
 

src/service/FileButtonService.ts

@@ -209,6 +209,51 @@ export class FileButtonService implements vscode.CodeLensProvider {
     return lineContent.substring(cut_pos);
   }
 
+  public createDebugButton(codeLensLine, document, lineContent): vscode.CodeLens[] {
+    // const last_index = document.lineAt(codeLensLine).range.end.character;
+
+    const range: vscode.Range = new vscode.Range(codeLensLine + 1, 0, codeLensLine + 1, 0);
+    const temp_result: vscode.CodeLens[] = [];
+
+    // paramTypes= []
+    // returnType=
+
+    if (lineContent.indexOf("paramTypes=") >= 0) {
+      temp_result.push(
+        new vscode.CodeLens(range, {
+          title: "addParam",
+          command: "lcpr.addDebugType",
+          arguments: [document, "paramTypes"],
+        })
+      );
+      temp_result.push(
+        new vscode.CodeLens(range, {
+          title: "resetParam",
+          command: "lcpr.resetDebugType",
+          arguments: [document, "paramTypes"],
+        })
+      );
+    }
+
+    if (lineContent.indexOf("returnType=") >= 0) {
+      temp_result.push(
+        new vscode.CodeLens(range, {
+          title: "addReturn",
+          command: "lcpr.addDebugType",
+          arguments: [document, "returnType"],
+        })
+      );
+      temp_result.push(
+        new vscode.CodeLens(range, {
+          title: "resetReturn",
+          command: "lcpr.resetDebugType",
+          arguments: [document, "returnType"],
+        })
+      );
+    }
+    return temp_result;
+  }
+
   public provideCodeLenses(document: vscode.TextDocument): vscode.ProviderResult<vscode.CodeLens[]> {
     const content: string = document.getText();
     const matchResult: RegExpMatchArray | null = content.match(/@lc app=.* id=(.*) lang=(.*)/);
@@ -225,6 +270,8 @@ export class FileButtonService implements vscode.CodeLensProvider {
     const codeLens: vscode.CodeLens[] = [];
     let caseFlag: boolean = false;
     let curCase = "";
+    // 搜集所有debug
+    let debugFlag: boolean = false;
     for (let i: number = 0; i < document.lineCount; i++) {
       const lineContent: string = document.lineAt(i).text;
       if (lineContent.indexOf("@lc code=end") >= 0) {
@@ -248,6 +295,20 @@ export class FileButtonService implements vscode.CodeLensProvider {
         curCase = "";
         caseFlag = false;
       }
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-end") >= 0) {
+        debugFlag = false;
+      }
+
+      if (debugFlag) {
+        this.createDebugButton(i, document, lineContent).forEach((x) => codeLens.push(x));
+      }
+
+      // 收集所有用例
+      if (lineContent.indexOf("@lcpr-div-debug-arg-start") >= 0) {
+        debugFlag = true;
+      }
     }
 
     return codeLens;