Add Code Lens for submitting the answer

jdneo · jdneo · commit 1f48e946eb41 · 2019-01-06T16:21:09.000+08:00
diff --git a/src/codeLensProvider.ts b/src/codeLensProvider.ts
@@ -0,0 +1,28 @@
+// Copyright (c) jdneo. All rights reserved.
+// Licensed under the MIT license.
+
+import * as vscode from "vscode";
+
+class CodeLensProvider implements vscode.CodeLensProvider {
+
+    private validFileNamePattern: RegExp = /\d+\..*\.(.+)/;
+
+    public provideCodeLenses(document: vscode.TextDocument): vscode.ProviderResult<vscode.CodeLens[]> {
+        const fileName: string = document.fileName.trim();
+        const matchResult: RegExpMatchArray | null = fileName.match(this.validFileNamePattern);
+        if (!matchResult) {
+            return undefined;
+        }
+
+        const range: vscode.Range = new vscode.Range(document.lineCount - 1, 0, document.lineCount - 1, 0);
+
+        const lens: vscode.CodeLens = new vscode.CodeLens(range, {
+            title: "🙏 Submit to LeetCode",
+            command: "leetcode.submitSolution",
+        });
+
+        return [lens];
+    }
+}
+
+export const codeLensProvider: CodeLensProvider = new CodeLensProvider();
diff --git a/src/extension.ts b/src/extension.ts
@@ -2,6 +2,7 @@
 // Licensed under the MIT license.
 
 import * as vscode from "vscode";
+import { codeLensProvider } from "./codeLensProvider";
 import * as cache from "./commands/cache";
 import * as plugin from "./commands/plugin";
 import * as session from "./commands/session";
@@ -25,6 +26,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
         leetCodeStatusBarItem,
         leetCodeChannel,
         vscode.window.registerTreeDataProvider("leetCodeExplorer", leetCodeTreeDataProvider),
+        vscode.languages.registerCodeLensProvider({ scheme: "file" }, codeLensProvider),
         vscode.commands.registerCommand("leetcode.deleteCache", () => cache.deleteCache()),
         vscode.commands.registerCommand("leetcode.toogleLeetCodeCn", () => plugin.toogleLeetCodeCn()),
         vscode.commands.registerCommand("leetcode.signin", () => leetCodeManager.signIn()),
