feat: Support open preview page through Code Lens and context menu (#338)

Author: jdneo

package.json

@@ -225,6 +225,10 @@
                 {
                     "command": "leetcode.showSolution",
                     "group": "leetcode@3"
+                },
+                {
+                    "command": "leetcode.previewProblem",
+                    "group": "leetcode@4"
                 }
             ]
         },

src/codelens/CustomCodeLensProvider.ts

@@ -32,6 +32,11 @@ export class CustomCodeLensProvider implements vscode.CodeLensProvider {
                 command: "leetcode.showSolution",
                 arguments: [document.uri],
             }),
+            new vscode.CodeLens(range, {
+                title: "Preview",
+                command: "leetcode.previewProblem",
+                arguments: [document.uri],
+            }),
         ];
     }
 }

src/commands/show.ts

@@ -5,20 +5,45 @@ import * as fse from "fs-extra";
 import * as path from "path";
 import * as unescapeJS from "unescape-js";
 import * as vscode from "vscode";
+import { explorerNodeManager } from "../explorer/explorerNodeManager";
 import { LeetCodeNode } from "../explorer/LeetCodeNode";
 import { leetCodeChannel } from "../leetCodeChannel";
 import { leetCodeExecutor } from "../leetCodeExecutor";
 import { leetCodeManager } from "../leetCodeManager";
 import { IProblem, IQuickItemEx, languages, ProblemState } from "../shared";
+import { getNodeIdFromFile } from "../utils/problemUtils";
 import { DialogOptions, DialogType, openSettingsEditor, promptForOpenOutputChannel, promptForSignIn, promptHintMessage } from "../utils/uiUtils";
 import { selectWorkspaceFolder } from "../utils/workspaceUtils";
 import * as wsl from "../utils/wslUtils";
 import { leetCodePreviewProvider } from "../webview/leetCodePreviewProvider";
 import { leetCodeSolutionProvider } from "../webview/leetCodeSolutionProvider";
 import * as list from "./list";
 
-export async function previewProblem(node: IProblem, isSideMode: boolean = false): Promise<void> {
-    const descString: string = await leetCodeExecutor.getDescription(node);
+export async function previewProblem(input: IProblem | vscode.Uri, isSideMode: boolean = false): Promise<void> {
+    let node: LeetCodeNode;
+    if (input instanceof LeetCodeNode) {
+        node = input;
+    } else if (input instanceof vscode.Uri) {
+        const activeFilePath: string = input.fsPath;
+        const id: string = await getNodeIdFromFile(activeFilePath);
+        if (!id) {
+            vscode.window.showErrorMessage(`Failed to resolve the problem id from file: ${activeFilePath}.`);
+            return;
+        }
+        const cachedNode: LeetCodeNode | undefined = explorerNodeManager.getNodeById(id);
+        if (!cachedNode) {
+            vscode.window.showErrorMessage(`Failed to resolve the problem with id: ${id}.`);
+            return;
+        }
+        node = cachedNode;
+        // Move the preview page aside if it's triggered from Code Lens
+        isSideMode = true;
+    } else {
+        vscode.window.showErrorMessage("Invalid input to fetch the preview data.");
+        return;
+    }
+
+    const descString: string = await leetCodeExecutor.getDescription(node.id);
     leetCodePreviewProvider.show(descString, node, isSideMode);
 }
 
@@ -54,7 +79,7 @@ export async function showSolution(input: LeetCodeNode | vscode.Uri): Promise<vo
     } else if (input instanceof vscode.Uri) {
         problemInput = `"${input.fsPath}"`;
     } else {
-        vscode.window.showErrorMessage("Invalid input to fetch the solution data");
+        vscode.window.showErrorMessage("Invalid input to fetch the solution data.");
         return;
     }
 

src/leetCodeExecutor.ts

@@ -112,8 +112,8 @@ class LeetCodeExecutor implements Disposable {
         return solution;
     }
 
-    public async getDescription(problemNode: IProblem): Promise<string> {
-        return await this.executeCommandWithProgressEx("Fetching problem description...", this.nodeExecutable, [await this.getLeetCodeBinaryPath(), "show", problemNode.id, "-x"]);
+    public async getDescription(problemNodeId: string): Promise<string> {
+        return await this.executeCommandWithProgressEx("Fetching problem description...", this.nodeExecutable, [await this.getLeetCodeBinaryPath(), "show", problemNodeId, "-x"]);
     }
 
     public async listSessions(): Promise<string> {

src/utils/problemUtils.ts

@@ -1,7 +1,9 @@
 // Copyright (c) jdneo. All rights reserved.
 // Licensed under the MIT license.
 
+import * as fse from "fs-extra";
 import * as _ from "lodash";
+import * as path from "path";
 import { IProblem, langExt } from "../shared";
 
 export function genFileExt(language: string): string {
@@ -17,3 +19,18 @@ export function genFileName(node: IProblem, language: string): string {
     const ext: string = genFileExt(language);
     return `${node.id}.${slug}.${ext}`;
 }
+
+export async function getNodeIdFromFile(fsPath: string): Promise<string> {
+    const fileContent: string = await fse.readFile(fsPath, "utf8");
+    let id: string = "";
+    const matchResults: RegExpMatchArray | null = fileContent.match(/@lc.+id=(.+?) /);
+    if (matchResults && matchResults.length === 2) {
+        id = matchResults[1];
+    }
+    // Try to get id from file name if getting from comments failed
+    if (!id) {
+        id = path.basename(fsPath).split(".")[0];
+    }
+
+    return id;
+}