support trigger test command in file explorer (#30)

Author: jdneo

package.json

@@ -81,7 +81,7 @@
             },
             {
                 "command": "leetcode.testSolution",
-                "title": "Test",
+                "title": "Test in LeetCode",
                 "category": "LeetCode"
             },
             {
@@ -115,7 +115,7 @@
                 {
                     "command": "leetcode.showProblem",
                     "when": "view == leetCodeExplorer && viewItem == problem",
-                    "group": "1@1"
+                    "group": "leetcode-explorer@1"
                 }
             ],
             "commandPalette": [
@@ -125,9 +125,15 @@
                 }
             ],
             "explorer/context": [
+                {
+                    "command": "leetcode.testSolution",
+                    "when": "explorerResourceIsFolder == false",
+                    "group": "file-explorer@1"
+                },
                 {
                     "command": "leetcode.submitSolution",
-                    "when": "explorerResourceIsFolder == false"
+                    "when": "explorerResourceIsFolder == false",
+                    "group": "file-explorer@2"
                 }
             ]
         },

src/commands/submit.ts

@@ -5,26 +5,17 @@ import { leetCodeManager } from "../leetCodeManager";
 import { leetCodeBinaryPath } from "../shared";
 import { executeCommand } from "../utils/cpUtils";
 import { DialogType, promptForOpenOutputChannel, promptForSignIn, showResultFile } from "../utils/uiUtils";
+import { getActivefilePath } from "../utils/workspaceUtils";
 
 export async function submitSolution(channel: vscode.OutputChannel, uri?: vscode.Uri): Promise<void> {
     if (!leetCodeManager.getUser()) {
         promptForSignIn();
         return;
     }
 
-    let filePath: string;
-    if (uri) {
-        filePath = uri.fsPath;
-    } else {
-        const textEditor: vscode.TextEditor | undefined = vscode.window.activeTextEditor;
-        if (!textEditor) {
-            return;
-        }
-        if (!textEditor.document.save()) {
-            vscode.window.showWarningMessage("Please save the solution file first.");
-            return;
-        }
-        filePath = textEditor.document.uri.fsPath;
+    const filePath: string | undefined = await getActivefilePath(uri);
+    if (!filePath) {
+        return;
     }
 
     try {

src/commands/test.ts

@@ -6,24 +6,18 @@ import { leetCodeManager } from "../leetCodeManager";
 import { IQuickItemEx, leetCodeBinaryPath, UserStatus } from "../shared";
 import { executeCommand } from "../utils/cpUtils";
 import { DialogType, promptForOpenOutputChannel, showFileSelectDialog, showResultFile } from "../utils/uiUtils";
+import { getActivefilePath } from "../utils/workspaceUtils";
 
-export async function testSolution(channel: vscode.OutputChannel): Promise<void> {
+export async function testSolution(channel: vscode.OutputChannel, uri?: vscode.Uri): Promise<void> {
     try {
         if (leetCodeManager.getStatus() === UserStatus.SignedOut) {
             return;
         }
 
-        const activeText: vscode.TextEditor | undefined = vscode.window.activeTextEditor;
-        if (!activeText) {
-            vscode.window.showErrorMessage("Please open a LeetCode solution file first.");
+        const filePath: string | undefined = await getActivefilePath(uri);
+        if (!filePath) {
             return;
         }
-        if (!activeText.document.save()) {
-            vscode.window.showWarningMessage("Please save the solution file first.");
-            return;
-        }
-
-        const filePath = activeText.document.uri.fsPath;
         const picks: Array<IQuickItemEx<string>> = [];
         picks.push(
             {

src/extension.ts

@@ -27,8 +27,8 @@ export async function activate(context: vscode.ExtensionContext) {
         vscode.commands.registerCommand("leetcode.showProblem", (node: LeetCodeNode) => show.showProblem(channel, node)),
         vscode.commands.registerCommand("leetcode.searchProblem", () => show.searchProblem(channel)),
         vscode.commands.registerCommand("leetcode.refreshExplorer", () => leetCodeTreeDataProvider.refresh()),
-        vscode.commands.registerCommand("leetcode.testSolution", () => test.testSolution(channel)),
-        vscode.commands.registerCommand("leetcode.submitSolution", (uri: vscode.Uri) => submit.submitSolution(channel, uri)),
+        vscode.commands.registerCommand("leetcode.testSolution", (uri?: vscode.Uri) => test.testSolution(channel, uri)),
+        vscode.commands.registerCommand("leetcode.submitSolution", (uri?: vscode.Uri) => submit.submitSolution(channel, uri)),
     );
 
     leetCodeManager.on("statusChanged", () => {

src/utils/workspaceUtils.ts

@@ -17,3 +17,21 @@ export async function selectWorkspaceFolder(): Promise<string> {
     }
     return folder ? folder.uri.fsPath : path.join(os.homedir(), ".leetcode");
 }
+
+export async function getActivefilePath(uri?: vscode.Uri): Promise<string | undefined> {
+    let textEditor: vscode.TextEditor | undefined;
+    if (uri) {
+        textEditor = await vscode.window.showTextDocument(uri, { preview: false });
+    } else {
+        textEditor = vscode.window.activeTextEditor;
+    }
+
+    if (!textEditor) {
+        return undefined;
+    }
+    if (textEditor.document.isDirty && !await textEditor.document.save()) {
+        vscode.window.showWarningMessage("Please save the solution file first.");
+        return undefined;
+    }
+    return textEditor.document.uri.fsPath;
+}