feat: re-add cookie-based login method

Author: tomoyachen

src/leetCodeManager.ts

@@ -6,14 +6,14 @@ import { EventEmitter } from "events";
 import * as vscode from "vscode";
 import { leetCodeChannel } from "./leetCodeChannel";
 import { leetCodeExecutor } from "./leetCodeExecutor";
-import { Endpoint, loginArgsMapping, urls, urlsCn, UserStatus } from "./shared";
+import { Endpoint, IQuickItemEx, loginArgsMapping, urls, urlsCn, UserStatus } from "./shared";
 import { createEnvOption } from "./utils/cpUtils";
 import { DialogType, openUrl, promptForOpenOutputChannel } from "./utils/uiUtils";
 import * as wsl from "./utils/wslUtils";
 import { getLeetCodeEndpoint } from "./commands/plugin";
 import { globalState } from "./globalState";
 import { queryUserData } from "./request/query-user-data";
-import { parseQuery, sleep } from "./utils/toolUtils";
+import { parseQuery } from "./utils/toolUtils";
 
 class LeetCodeManager extends EventEmitter {
     private currentUser: string | undefined;
@@ -68,8 +68,117 @@ class LeetCodeManager extends EventEmitter {
         }
     }
 
+    public async signInWithCli(loginMethod: string): Promise<void> {
+        const commandArg: string | undefined = loginArgsMapping.get(loginMethod)
+        if (!commandArg) {
+            throw new Error(`The login method "${loginMethod}" is not supported.`)
+        }
+        const isByCookie: boolean = loginMethod === 'Cookie'
+        const inMessage: string = isByCookie ? 'sign in by cookie' : 'sign in'
+
+        try {
+            const userName: string | undefined = await new Promise(
+                async (resolve: (res: string | undefined) => void, reject: (e: Error) => void): Promise<void> => {
+                    const leetCodeBinaryPath: string = await leetCodeExecutor.getLeetCodeBinaryPath()
+
+                    const childProc: cp.ChildProcess = wsl.useWsl()
+                        ? cp.spawn('wsl', [leetCodeExecutor.node, leetCodeBinaryPath, 'user', commandArg], { shell: true })
+                        : cp.spawn(leetCodeExecutor.node, [leetCodeBinaryPath, 'user', commandArg], {
+                              shell: true,
+                              env: createEnvOption(),
+                          })
+
+                    childProc.stdout?.on('data', async (data: string | Buffer) => {
+                        data = data.toString()
+                        leetCodeChannel.append(data)
+                        if (data.includes('twoFactorCode')) {
+                            const twoFactor: string | undefined = await vscode.window.showInputBox({
+                                prompt: 'Enter two-factor code.',
+                                ignoreFocusOut: true,
+                                validateInput: (s: string): string | undefined => (s && s.trim() ? undefined : 'The input must not be empty'),
+                            })
+                            if (!twoFactor) {
+                                childProc.kill()
+                                return resolve(undefined)
+                            }
+                            childProc.stdin?.write(`${twoFactor}\n`)
+                        }
+                        const successMatch: RegExpMatchArray | null = data.match(this.successRegex)
+                        if (successMatch && successMatch[1]) {
+                            childProc.stdin?.end()
+                            return resolve(successMatch[1])
+                        } else if (data.match(this.failRegex)) {
+                            childProc.stdin?.end()
+                            return reject(new Error('Faile to login'))
+                        }
+                    })
+
+                    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 username or E-mail.',
+                        ignoreFocusOut: true,
+                        validateInput: (s: string): string | undefined => (s && s.trim() ? undefined : 'The input must not be empty'),
+                    })
+                    if (!name) {
+                        childProc.kill()
+                        return resolve(undefined)
+                    }
+                    childProc.stdin?.write(`${name}\n`)
+                    const pwd: string | undefined = await vscode.window.showInputBox({
+                        prompt: isByCookie ? 'Enter cookie' : 'Enter password.',
+                        password: true,
+                        ignoreFocusOut: true,
+                        validateInput: (s: string): string | undefined =>
+                            s ? undefined : isByCookie ? 'Cookie must not be empty' : 'Password must not be empty',
+                    })
+                    if (!pwd) {
+                        childProc.kill()
+                        return resolve(undefined)
+                    }
+                    childProc.stdin?.write(`${pwd}\n`)
+                }
+            )
+            if (userName) {
+                vscode.window.showInformationMessage(`Successfully ${inMessage}.`)
+                this.currentUser = userName
+                this.userStatus = UserStatus.SignedIn
+                this.emit('statusChanged')
+            }
+        } catch (error) {
+            promptForOpenOutputChannel(`Failed to ${inMessage}. Please open the output channel for details`, DialogType.error)
+        }
+    }
+
     public async signIn(): Promise<void> {
-        openUrl(this.getAuthLoginUrl());
+        const picks: Array<IQuickItemEx<string>> = []
+        picks.push(
+            {
+                label: 'Web Authorization',
+                detail: 'Open browser to authorize login on the website',
+                value: 'WebAuth',
+                description: '[Recommended]'
+            },
+            {
+                label: 'LeetCode Cookie',
+                detail: 'Use LeetCode cookie copied from browser to login',
+                value: 'Cookie',
+            }
+        )
+
+        const choice: IQuickItemEx<string> | undefined = await vscode.window.showQuickPick(picks)
+        if (!choice) {
+            return
+        }
+        const loginMethod: string = choice.value
+
+        if (loginMethod === 'WebAuth') {
+            openUrl(this.getAuthLoginUrl())
+            return
+        }
+
+        this.signInWithCli(loginMethod)
     }
 
     public async signOut(): Promise<void> {
@@ -136,15 +245,16 @@ class LeetCodeManager extends EventEmitter {
                 } else if (data.match(this.failRegex)) {
                     childProc.stdin?.end();
                     return reject(new Error("Faile to login"));
+                } else if (data.match(/login: /)){
+                    childProc.stdin?.write(`${name}\n`);
+                } else if (data.match(/cookie: /)){
+                    childProc.stdin?.write(`${cookie}\n`);
                 }
             });
 
             childProc.stderr?.on("data", (data: string | Buffer) => leetCodeChannel.append(data.toString()));
 
             childProc.on("error", reject);
-            childProc.stdin?.write(`${name}\n`);
-            await sleep(800);
-            childProc.stdin?.write(`${cookie}\n`);
         });
     }
 }