feat: add generate doc

Author: convexwf

package.json

@@ -2,7 +2,7 @@
   "name": "vscode-leetcode",
   "displayName": "LeetCode",
   "description": "Solve LeetCode problems in VS Code",
-  "version": "0.19.1",
+  "version": "0.20.1",
   "author": "convexwf",
   "publisher": "LeetCode",
   "license": "MIT",
@@ -32,6 +32,7 @@
     "onCommand:leetcode.manageSessions",
     "onCommand:leetcode.refreshExplorer",
     "onCommand:leetcode.pickOne",
+    "onCommand:leetcode.generateDocumentation",
     "onCommand:leetcode.showProblem",
     "onCommand:leetcode.showDocumentation",
     "onCommand:leetcode.previewProblem",
@@ -83,6 +84,11 @@
         "title": "Pick One",
         "category": "LeetCode"
       },
+      {
+        "command": "leetcode.generateDocumentation",
+        "title": "Generate Documentation",
+        "category": "LeetCode"
+      },
       {
         "command": "leetcode.showProblem",
         "title": "Show Problem",

src/commands/show.ts

@@ -52,13 +52,91 @@ export async function pickOne(): Promise<void> {
     await showProblemInternal(randomProblem);
 }
 
+export async function generateDocumentation(): Promise<void> {
+    const ids: string | undefined = await vscode.window.showInputBox({
+        prompt: "Enter problem id. e.g. 1, 1-10, 1,2,3, 1,2,3-10",
+        ignoreFocusOut: true,
+        validateInput: (s: string): string | undefined => s && s.trim() ? undefined : "The input must not be empty",
+    });
+    if (!ids) {
+        return;
+    }
+    const idArray: string[] = ids.split(",");
+    const nodeArray: IProblem[] = [];
+    for (const id of idArray) {
+        const _id: string = id.trim();
+        if (_id.search("-") > 0) {
+            const range: string[] = _id.split("-");
+            const start: number = parseInt(range[0], 10);
+            const end: number = parseInt(range[1], 10);
+            if (start > 0 && end > 0 && start <= end) {
+                for (let i: number = start; i <= end; i++) {
+                    console.log(`Show documentation: ${i}`);
+                    const node: IProblem | undefined = explorerNodeManager.getNodeById(i.toString());
+                    if (node) {
+                        nodeArray.push(node);
+                    }
+                }
+            }
+        }
+        else {
+            console.log(`Show documentation: ${_id}`);
+            const node: IProblem | undefined = explorerNodeManager.getNodeById(_id);
+            if (node) {
+                nodeArray.push(node);
+            }
+        }
+    }
+
+    for (const node of nodeArray) {
+        const finalPath: string = await assureDoucumentation(node);
+        if (!finalPath) {
+            continue;
+        }
+        try {
+            await leetCodeExecutor.showDocumentationInternal(node, finalPath);
+        }
+        catch (error) {
+            console.log(error);
+            vscode.window.showErrorMessage(error);
+        }
+    }
+}
+
 export async function showProblem(node?: LeetCodeNode): Promise<void> {
     if (!node) {
         return;
     }
     await showProblemInternal(node);
 }
 
+async function assureDoucumentation(node: IProblem): Promise<string> {
+    const leetCodeConfig: vscode.WorkspaceConfiguration = getWorkspaceConfiguration();
+    const workspaceFolder: string = await selectWorkspaceFolder();
+    if (!workspaceFolder) {
+        return "";
+    }
+
+    const fileFolder: string = leetCodeConfig.get<string>(`filePath.doc.folder`, "").trim();
+    const fileName: string = leetCodeConfig.get<string>(`filePath.doc.filename`, "").trim();
+    if (fileFolder === "" || fileName === "") {
+        await promptForOpenOutputChannel("Please configure the file path first.", DialogType.error);
+        return "";
+    }
+
+    let finalPath: string = path.join(workspaceFolder, fileFolder, fileName);
+    if (finalPath) {
+        finalPath = await resolveRelativePath(finalPath, node, "md");
+        if (!finalPath) {
+            leetCodeChannel.appendLine("Showing problem canceled by user.");
+            return "";
+        }
+    }
+    finalPath = wsl.useWsl() ? await wsl.toWinPath(finalPath) : finalPath;
+    console.log(`Documentation path: ${finalPath}`);
+    return finalPath;
+}
+
 export async function showDocumentation(node?: LeetCodeNode): Promise<void> {
     if (!node) {
         return;

src/commands/submit.ts

@@ -43,10 +43,10 @@ export async function submitSolution(uri?: vscode.Uri): Promise<void> {
 
         // 匹配通过测试用例数、运行时间和内存使用情况的正则表达式
         const regex_cases_passed = /(\d+)\/(\d+) cases passed/;
-        const regex_runtime_percentage = /Your runtime beats (\d+\.\d+) %/;
-        const regex_memory_percentage = /Your memory usage beats (\d+\.\d+) %/;
+        const regex_runtime_percentage = /Your runtime beats  (\d+(.\d+)?) %/;
+        const regex_memory_percentage = /Your memory usage beats  (\d+(.\d+)?) %/;
         const regex_runtime_ms = /(\d+) ms/;
-        const regex_memory_usage = /\((\d+\.\d+) MB\)/;
+        const regex_memory_usage = /\( (\d+(.\d+)?) MB\)/;
 
         // 提取通过测试用例数、运行时间和内存使用情况
         const [cases_passed, total_cases] = regex_cases_passed.exec(message)?.slice(1) ?? [];

src/extension.ts

@@ -59,6 +59,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
             vscode.commands.registerCommand("leetcode.showProblem", (node: LeetCodeNode) => show.showProblem(node)),
             vscode.commands.registerCommand("leetcode.showDocumentation", (node: LeetCodeNode) => show.showDocumentation(node)),
             vscode.commands.registerCommand("leetcode.pickOne", () => show.pickOne()),
+            vscode.commands.registerCommand("leetcode.generateDocumentation", () => show.generateDocumentation()),
             vscode.commands.registerCommand("leetcode.searchProblem", () => show.searchProblem()),
             vscode.commands.registerCommand("leetcode.showSolution", (input: LeetCodeNode | vscode.Uri) => show.showSolution(input)),
             vscode.commands.registerCommand("leetcode.refreshExplorer", () => leetCodeTreeDataProvider.refresh()),

src/leetCodeExecutor.ts

@@ -156,6 +156,11 @@ class LeetCodeExecutor implements Disposable {
 
     public async showDocumentationInternal(problemNode: IProblem, filePath: string): Promise<void> {
 
+        if (await fse.pathExists(filePath)) {
+            await window.showInformationMessage(`File ${filePath} already exists!`);
+            return;
+        }
+
         const descString: string = await this.getDescription(problemNode.id, false);
         const description = this.parseDescription(descString, problemNode);
         const { title, url, category, difficulty, likes, dislikes, body } = description;
@@ -188,13 +193,12 @@ class LeetCodeExecutor implements Disposable {
 
         // console.log("md_content", md_content);
 
-        if (!await fse.pathExists(filePath)) {
-            await fse.createFile(filePath);
-            await fse.writeFile(filePath, md_content);
-            workspace.openTextDocument(filePath).then((document) => {
-                window.showTextDocument(document);
-            });
-        }
+
+        await fse.createFile(filePath);
+        await fse.writeFile(filePath, md_content);
+        workspace.openTextDocument(filePath).then((document) => {
+            window.showTextDocument(document);
+        });
     }
 
     public async showProblem(problemNode: IProblem, language: string, filePath: string, showDescriptionInComment: boolean = false, needTranslation: boolean): Promise<void> {

src/utils/workspaceUtils.ts

@@ -72,17 +72,17 @@ export async function insertSubmitResult(result?: string): Promise<string | unde
 
     // 获取当前光标所在行
     const currentLine = editor.selection.active.line;
-    // 向前遍历找到第一次出现 "// @lc code=start" 的行
+    // 向前遍历找到第一次出现 "@lc code=start" 的行
     let startLine = currentLine;
     while (startLine >= 0) {
         const lineText = editor.document.lineAt(startLine).text;
-        if (lineText.includes('// @lc code=start')) {
+        if (lineText.includes('@lc code=start')) {
             break;
         }
         startLine--;
     }
     if (startLine < 0) {
-        vscode.window.showErrorMessage("Please add '// @lc code=start' in your solution file.");
+        vscode.window.showErrorMessage("Please add '@lc code=start' in your solution file.");
         return;
     }
 
@@ -110,21 +110,21 @@ export async function copyCodeBlock(): Promise<string | undefined> {
     // 获取当前光标所在行
     const currentLine = editor.selection.active.line;
 
-    // 向前遍历找到第一次出现 "// @lc code=start" 的行
+    // 向前遍历找到第一次出现 "@lc code=start" 的行
     let startLine = currentLine;
     while (startLine >= 0) {
         const lineText = editor.document.lineAt(startLine).text;
-        if (lineText.includes('// @lc code=start')) {
+        if (lineText.includes('@lc code=start')) {
             break;
         }
         startLine--;
     }
 
-    // 向后遍历找到第一次出现 "// @lc code=end" 的行
+    // 向后遍历找到第一次出现 "@lc code=end" 的行
     let endLine = currentLine;
     while (endLine < editor.document.lineCount) {
         const lineText = editor.document.lineAt(endLine).text;
-        if (lineText.includes('// @lc code=end')) {
+        if (lineText.includes('@lc code=end')) {
             break;
         }
         endLine++;