better handle user status

Author: sheche@microsoft.com

src/commands/session.ts

@@ -1,13 +1,13 @@
 "use strict";
 
 import * as vscode from "vscode";
+import { leetCodeManager } from "../leetCodeManager";
 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();
+    const signInStatus = leetCodeManager.getUser();
     if (!signInStatus) {
         return [];
     }

src/commands/user.ts

@@ -2,16 +2,15 @@
 
 import * as vscode from "vscode";
 import * as session from "./commands/session";
-import * as user from "./commands/user";
 import { leetcodeChannel } from "./leetCodeChannel";
+import { leetCodeManager } from "./leetCodeManager";
 import { leetCodeStatusBarItem } from "./leetCodeStatusBarItem";
 
 export function activate(context: vscode.ExtensionContext) {
-    const terminal: vscode.Terminal = vscode.window.createTerminal("LeetCode");
+    leetCodeManager.getLoginStatus();
     context.subscriptions.push(
-        terminal,
-        vscode.commands.registerCommand("leetcode.signin", () => user.signIn()),
-        vscode.commands.registerCommand("leetcode.signout", () => user.signOut()),
+        vscode.commands.registerCommand("leetcode.signin", () => leetCodeManager.signIn()),
+        vscode.commands.registerCommand("leetcode.signout", () => leetCodeManager.signOut()),
         vscode.commands.registerCommand("leetcode.selectSessions", () => session.selectSession()),
     );
 }

src/extension.ts

@@ -0,0 +1,119 @@
+"use strict";
+
+import * as cp from "child_process";
+import { EventEmitter } from "events";
+import * as vscode from "vscode";
+import { leetcodeChannel } from "./leetCodeChannel";
+import { UserStatus } from "./shared";
+import { leetCodeBinaryPath } from "./shared";
+import { executeCommand } from "./utils/cpUtils";
+import { DialogType, promptForOpenOutputChannel } from "./utils/uiUtils";
+
+export interface ILeetCodeManager extends EventEmitter {
+    getLoginStatus(): void;
+    getStatus(): UserStatus;
+    getUser(): string | undefined;
+    signIn(): void;
+    signOut(): void;
+}
+
+class LeetCodeManager extends EventEmitter implements ILeetCodeManager {
+    private currentUser: string | undefined;
+    private userStatus: UserStatus;
+
+    constructor() {
+        super();
+        this.currentUser = undefined;
+        this.userStatus = UserStatus.SignedOut;
+    }
+
+    public async getLoginStatus(): Promise<void> {
+        try {
+            const result = await executeCommand("node", [leetCodeBinaryPath, "user"]);
+            this.currentUser = result.slice("You are now login as".length).trim();
+            this.userStatus = UserStatus.SignedIn;
+        } catch (error) {
+            this.currentUser = undefined;
+            this.userStatus = UserStatus.SignedOut;
+        } finally {
+            this.emit("statusChanged");
+        }
+    }
+
+    public async signIn(): Promise<void> {
+        try {
+            const userName: string | undefined = await new Promise(async (resolve: (res: string | undefined) => 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) => {
+                    data = data.toString();
+                    result = result.concat(data);
+                    leetcodeChannel.append(data);
+                });
+
+                childProc.stderr.on("data", (data: string | Buffer) => leetcodeChannel.append(data.toString()));
+
+                childProc.on("error", reject);
+                const name: string | undefined = await vscode.window.showInputBox({
+                    prompt: "Enter user name.",
+                    validateInput: (s: string) => s.trim() ? undefined : "User name must not be empty",
+                });
+                if (!name) {
+                    childProc.kill();
+                    resolve(undefined);
+                }
+                childProc.stdin.write(`${name}\n`);
+                const pwd: string | undefined = await vscode.window.showInputBox({
+                    prompt: "Enter user name.",
+                    password: true,
+                    validateInput: (s: string) => s ? undefined : "Password must not be empty",
+                });
+                if (!pwd) {
+                    childProc.kill();
+                    resolve(undefined);
+                }
+                childProc.stdin.write(`${pwd}\n`);
+                childProc.stdin.end();
+                childProc.on("close", () => {
+                    const match: RegExpMatchArray | null = result.match(/(?:.*) Successfully login as (.*)/i);
+                    if (match && match[1]) {
+                        resolve(match[1]);
+                    } else {
+                        reject(new Error("Failed to sigin in."));
+                    }
+                });
+            });
+            if (userName) {
+                vscode.window.showInformationMessage("Successfully signed in.");
+                this.currentUser = userName;
+                this.userStatus = UserStatus.SignedIn;
+                this.emit("statusChanged");
+            }
+        } catch (error) {
+            promptForOpenOutputChannel("Failed to sign in. Please open the output channel for details", DialogType.error);
+        }
+
+    }
+
+    public async signOut(): Promise<void> {
+        try {
+            await executeCommand("node", [leetCodeBinaryPath, "user", "-L"]);
+            vscode.window.showInformationMessage("Successfully signed out.");
+            this.currentUser = undefined;
+            this.userStatus = UserStatus.SignedOut;
+            this.emit("statusChanged");
+        } catch (error) {
+            promptForOpenOutputChannel("Failed to sign out. Please open the output channel for details", DialogType.error);
+        }
+    }
+
+    public getStatus(): UserStatus {
+        return this.userStatus;
+    }
+
+    public getUser(): string | undefined {
+        return this.currentUser;
+    }
+}
+
+export const leetCodeManager: ILeetCodeManager = new LeetCodeManager();

src/leetCodeManager.ts

@@ -1,9 +1,11 @@
 "use strict";
 
 import * as vscode from "vscode";
+import { leetCodeManager } from "./leetCodeManager";
+import { UserStatus } from "./shared";
 
 export interface ILeetCodeStatusBarItem {
-    updateStatusBar(status: LeetCodeStatus, user?: string): void;
+    updateStatusBar(status: UserStatus, user?: string): void;
     dispose(): void;
 }
 
@@ -13,15 +15,18 @@ class LeetCodeStatusBarItem implements ILeetCodeStatusBarItem {
     constructor() {
         this.statusBarItem = vscode.window.createStatusBarItem();
         this.statusBarItem.command = "leetcode.selectSessions";
+        leetCodeManager.on("statusChanged", () => {
+            leetCodeStatusBarItem.updateStatusBar(leetCodeManager.getStatus(), leetCodeManager.getUser());
+        });
     }
 
-    public updateStatusBar(status: LeetCodeStatus, user?: string): void {
+    public updateStatusBar(status: UserStatus, user?: string): void {
         switch (status) {
-            case LeetCodeStatus.SignedIn:
+            case UserStatus.SignedIn:
                 this.statusBarItem.text = `LeetCode: ${user}`;
                 this.statusBarItem.show();
                 break;
-            case LeetCodeStatus.SignedOut:
+            case UserStatus.SignedOut:
             default:
                 this.statusBarItem.hide();
                 break;
@@ -33,9 +38,4 @@ class LeetCodeStatusBarItem implements ILeetCodeStatusBarItem {
     }
 }
 
-export enum LeetCodeStatus {
-    SignedIn = 1,
-    SignedOut = 2,
-}
-
 export const leetCodeStatusBarItem: ILeetCodeStatusBarItem = new LeetCodeStatusBarItem();

src/leetCodeStatusBarItem.ts

@@ -8,3 +8,8 @@ export const leetCodeBinaryPath: string = path.join(__dirname, "..", "..", "node
 export interface IQuickItemEx<T> extends vscode.QuickPickItem {
     value: T;
 }
+
+export enum UserStatus {
+    SignedIn = 1,
+    SignedOut = 2,
+}