enable switch session

Author: sheche@microsoft.com

package.json

@@ -12,7 +12,8 @@
     ],
     "activationEvents": [
         "onCommand:leetcode.signin",
-        "onCommand:leetcode.signout"
+        "onCommand:leetcode.signout",
+        "onCommand:leetcode.selectSessions"
     ],
     "main": "./out/src/extension",
     "contributes": {
@@ -26,6 +27,11 @@
                 "command": "leetcode.signout",
                 "title": "Sign out",
                 "category": "LeetCode"
+            },
+            {
+                "command": "leetcode.selectSessions",
+                "title": "Select session",
+                "category": "LeetCode"
             }
         ]
     },

src/commands/session.ts

@@ -0,0 +1,64 @@
+"use strict";
+
+import * as vscode from "vscode";
+import { IQuickItemEx, leetCodeBinaryPath } from "../shared";
+import * as cp from "../utils/cpUtils";
+import { DialogType, promptForOpenOutputChannel } from "../utils/uiUtils";
+import { getSignedInAccount } from "./user";
+
+export async function getSessionList(): Promise<ISession[]> {
+    const signInStatus = await getSignedInAccount();
+    if (!signInStatus) {
+        return [];
+    }
+    const result: string = await cp.executeCommand("node", [leetCodeBinaryPath, "session"]);
+    const lines: string[] = result.split("\n");
+    const sessions: ISession[] = [];
+    const reg: RegExp = /(✔?)\s*(\d+)\s+(.*)\s+(\d+ \(\s*\d+\.\d+ %\))\s+(\d+ \(\s*\d+\.\d+ %\))/;
+    for (const line of lines.map((l: string) => l.trim()).filter(Boolean)) {
+        const match: RegExpMatchArray | null = line.match(reg);
+        if (match && match.length === 6) {
+            sessions.push({
+                active: !!match[1],
+                id: match[2].trim(),
+                name: match[3].trim(),
+                acQuestions: match[4].trim(),
+                acSubmits: match[5].trim(),
+            });
+        }
+    }
+    return sessions;
+}
+
+export async function selectSession(): Promise<void> {
+    const choice: IQuickItemEx<string> | undefined = await vscode.window.showQuickPick(parseSessionsToPicks(getSessionList()));
+    if (!choice || choice.label.indexOf("✔") > -1) {
+        return;
+    }
+    try {
+        await cp.executeCommand("node", [leetCodeBinaryPath, "session", "-e", choice.value]);
+        vscode.window.showInformationMessage(`Successfully switched to session '${choice.label}'.`);
+    } catch (error) {
+        await promptForOpenOutputChannel("Failed to switch session. Please open the output channel for details", DialogType.error);
+    }
+}
+
+async function parseSessionsToPicks(p: Promise<ISession[]>): Promise<Array<IQuickItemEx<string>>> {
+    return new Promise(async (resolve: (res: Array<IQuickItemEx<string>>) => void): Promise<void> => {
+        const picks: Array<IQuickItemEx<string>> = (await p).map((s: ISession) => Object.assign({}, {
+            label: `${s.active ? "✔ " : ""}${s.name}`,
+            description: `ID: ${s.id}`,
+            detail: `AC Questions: ${s.acQuestions}, AC Submits: ${s.acSubmits}`,
+            value: s.id,
+        }));
+        resolve(picks);
+    });
+}
+
+export interface ISession {
+    active: boolean;
+    id: string;
+    name: string;
+    acQuestions: string;
+    acSubmits: string;
+}

src/commands/user.ts

@@ -3,6 +3,7 @@
 import * as cp from "child_process";
 import * as vscode from "vscode";
 import { leetcodeChannel } from "../leetCodeChannel";
+import { LeetCodeStatus, leetCodeStatusBarItem } from "../leetCodeStatusBarItem";
 import { leetCodeBinaryPath } from "../shared";
 import { executeCommand } from "../utils/cpUtils";
 import { DialogOptions, DialogType, promptForOpenOutputChannel } from "../utils/uiUtils";
@@ -18,7 +19,7 @@ export async function getSignedInAccount(): Promise<string | undefined> {
             DialogOptions.no,
         );
         if (choice === DialogOptions.yes) {
-            // sign in
+            await vscode.commands.executeCommand("leetcode.signin");
         }
         return undefined;
     }
@@ -27,7 +28,7 @@ export async function getSignedInAccount(): Promise<string | undefined> {
 export async function signIn(): Promise<void> {
     // work around for interactive login
     try {
-        await new Promise(async (resolve: (res: string) => void, reject: (e: Error) => void): Promise<void> => {
+        const userName: string = await new Promise(async (resolve: (res: string) => void, reject: (e: Error) => void): Promise<void> => {
             let result: string = "";
             const childProc: cp.ChildProcess = cp.spawn("node", [leetCodeBinaryPath, "user", "-l"]);
             childProc.stdout.on("data", (data: string | Buffer) => {
@@ -39,11 +40,12 @@ export async function signIn(): Promise<void> {
             childProc.stderr.on("data", (data: string | Buffer) => leetcodeChannel.append(data.toString()));
 
             childProc.on("error", reject);
-            childProc.on("exit", (code: number) => {
-                if (code !== 0 || result.indexOf("ERROR") > -1) {
-                    reject(new Error("Login failed"));
+            childProc.on("exit", () => {
+                const match: RegExpMatchArray | null = result.match(/(?:.*) Successfully login as (.*)/i);
+                if (match && match[1]) {
+                    resolve(match[1]);
                 } else {
-                    resolve(result);
+                    reject(new Error("Login failed"));
                 }
             });
             const user: string | undefined = await vscode.window.showInputBox({
@@ -68,6 +70,7 @@ export async function signIn(): Promise<void> {
             childProc.stdin.end();
         });
         vscode.window.showInformationMessage("Successfully signed in.");
+        leetCodeStatusBarItem.updateStatusBar(LeetCodeStatus.SignedIn, userName);
     } catch (error) {
         await promptForOpenOutputChannel("Failed to sign in. Please open the output channel for details", DialogType.error);
     }
@@ -77,6 +80,7 @@ export async function signOut(): Promise<void> {
     try {
         await executeCommand("node", [leetCodeBinaryPath, "user", "-L"]);
         vscode.window.showInformationMessage("Successfully signed out.");
+        leetCodeStatusBarItem.updateStatusBar(LeetCodeStatus.SignedOut);
     } catch (error) {
         await promptForOpenOutputChannel("Failed to sign out. Please open the output channel for details", DialogType.error);
     }

src/extension.ts

@@ -1,16 +1,22 @@
 "use strict";
 
 import * as vscode from "vscode";
+import * as session from "./commands/session";
 import * as user from "./commands/user";
+import { leetcodeChannel } from "./leetCodeChannel";
+import { leetCodeStatusBarItem } from "./leetCodeStatusBarItem";
 
 export function activate(context: vscode.ExtensionContext) {
     const terminal: vscode.Terminal = vscode.window.createTerminal("LeetCode");
     context.subscriptions.push(
         terminal,
         vscode.commands.registerCommand("leetcode.signin", () => user.signIn()),
         vscode.commands.registerCommand("leetcode.signout", () => user.signOut()),
+        vscode.commands.registerCommand("leetcode.selectSessions", () => session.selectSession()),
     );
 }
 
-// tslint:disable-next-line:no-empty
-export function deactivate() { }
+export function deactivate() {
+    leetcodeChannel.dispose();
+    leetCodeStatusBarItem.dispose();
+}

src/leetCodeChannel.ts

@@ -6,6 +6,7 @@ export interface ILeetCodeChannel {
     appendLine(message: any, title?: string): void;
     append(message: any): void;
     show(): void;
+    dispose(): void;
 }
 
 class LeetCodeChannel implements ILeetCodeChannel {
@@ -27,6 +28,10 @@ class LeetCodeChannel implements ILeetCodeChannel {
     public show(): void {
         this.channel.show();
     }
+
+    public dispose(): void {
+        this.channel.dispose();
+    }
 }
 
 export const leetcodeChannel: ILeetCodeChannel = new LeetCodeChannel();

src/leetCodeStatusBarItem.ts

@@ -0,0 +1,41 @@
+"use strict";
+
+import * as vscode from "vscode";
+
+export interface ILeetCodeStatusBarItem {
+    updateStatusBar(status: LeetCodeStatus, user?: string): void;
+    dispose(): void;
+}
+
+class LeetCodeStatusBarItem implements ILeetCodeStatusBarItem {
+    private readonly statusBarItem: vscode.StatusBarItem;
+
+    constructor() {
+        this.statusBarItem = vscode.window.createStatusBarItem();
+        this.statusBarItem.command = "leetcode.selectSessions";
+    }
+
+    public updateStatusBar(status: LeetCodeStatus, user?: string): void {
+        switch (status) {
+            case LeetCodeStatus.SignedIn:
+                this.statusBarItem.text = `LeetCode: ${user}`;
+                this.statusBarItem.show();
+                break;
+            case LeetCodeStatus.SignedOut:
+            default:
+                this.statusBarItem.hide();
+                break;
+        }
+    }
+
+    public dispose(): void {
+        this.statusBarItem.dispose();
+    }
+}
+
+export enum LeetCodeStatus {
+    SignedIn = 1,
+    SignedOut = 2,
+}
+
+export const leetCodeStatusBarItem: ILeetCodeStatusBarItem = new LeetCodeStatusBarItem();

src/shared.ts

@@ -1,5 +1,10 @@
 "use strict";
 
 import * as path from "path";
+import * as vscode from "vscode";
 
 export const leetCodeBinaryPath: string = path.join(__dirname, "..", "..", "node_modules", "leetcode-cli", "bin", "leetcode");
+
+export interface IQuickItemEx<T> extends vscode.QuickPickItem {
+    value: T;
+}