获取多个题解

Author: ccagml

CHANGELOG.md

@@ -1,3 +1,9 @@
+# version 3.1.3 等待自测
+
+- 重构代码,减少耦合
+- 中文选择 5 个题解,可能一个看不懂
+- 没搬的砖太多了,搬不完了,重置
+
 # version 3.1.2
 
 - 重构 filebutton 代码, 减少代码耦合

src/childCall/childCallModule.ts

@@ -13,7 +13,7 @@ import * as os from "os";
 import * as path from "path";
 import { ExtensionContext, ProgressLocation, Progress } from "vscode";
 import { ConfigurationChangeEvent, Disposable, MessageItem, window, workspace } from "vscode";
-import { DialogOptions, OutPutType, Endpoint, leetcodeHasInited } from "../model/ConstDefind";
+import { DialogOptions, OutPutType, Endpoint, leetcodeHasInited, IQuickItemEx } from "../model/ConstDefind";
 import { getLeetCodeEndpoint, getNodePath } from "../utils/ConfigUtils";
 import { openUrl, ShowMessage } from "../utils/OutputUtils";
 import * as systemUtils from "../utils/SystemUtils";
@@ -175,13 +175,83 @@ class ExecuteService implements Disposable {
     if (!needTranslation) {
       cmd.push("-T");
     }
+    let solution;
     if (cn_help) {
       cmd.push("-f");
+      solution = await this.callWithMsg(
+        "正在获取中文题解~~~",
+        this.nodeExecutable,
+        cmd,
+        undefined,
+        this.tryCnMulSolution,
+        {}
+      );
+    } else {
+      solution = await this.callWithMsg("正在获取题解~~~", this.nodeExecutable, cmd);
     }
-    const solution: string = await this.callWithMsg("正在获取题解~~~", this.nodeExecutable, cmd);
+
     return solution;
   }
 
+  public async tryCnMulSolution(_, child_process, resolve, reject) {
+    child_process.stdout?.on("data", async (data: string | Buffer) => {
+      data = data.toString();
+      BABA.getProxy(BabaStr.LogOutputProxy).get_log().append(data);
+      let successMatch: any;
+      try {
+        successMatch = JSON.parse(data);
+      } catch (e) {
+        successMatch = {};
+      }
+
+      if (successMatch.oper == "requireOper") {
+        let cookie = successMatch.cookie;
+        let arg = successMatch.arg;
+        if (arg.oper == "need_select") {
+          let canSelect = arg.canSelect || [];
+          const picks: Array<IQuickItemEx<string>> = [];
+
+          canSelect.forEach((element) => {
+            if (element?.node?.slug) {
+              picks.push({
+                label: `${element?.node.title}`,
+                description: `作者:${element?.node?.author?.username}`,
+                detail: "",
+                value: element?.node?.slug,
+              });
+            }
+          });
+
+          const choice: IQuickItemEx<string> | undefined = await window.showQuickPick(picks, {
+            ignoreFocusOut: true,
+          });
+          if (!choice) {
+            child_process.stdin?.end();
+            return reject(new Error(successMatch.msg));
+          }
+
+          let select_result = {
+            c: cookie,
+            slug: choice.value,
+          };
+          child_process.stdin?.write(JSON.stringify(select_result));
+        }
+        return;
+      }
+      if (successMatch.code == 100) {
+        child_process.stdin?.end();
+        return resolve(data);
+      } else if (successMatch.code < 0) {
+        child_process.stdin?.end();
+        return reject(new Error(successMatch.msg));
+      }
+    });
+    child_process.stderr?.on("data", (data: string | Buffer) => {
+      BABA.getProxy(BabaStr.LogOutputProxy).get_log().append(data.toString());
+    });
+    child_process.on("error", reject);
+  }
+
   public async getUserContest(needTranslation: boolean, username: string): Promise<string> {
     // solution don't support translation
     const cmd: string[] = [await this.getLeetCodeBinaryPath(), "query", "-b", username];

src/rpc/actionChain/chainNode/leetcode.cn.ts

@@ -299,7 +299,7 @@ function getSolutionArticlesSlugList(question_slug: string, lang: string, cb) {
   opts.json = true;
   opts.body = {
     operationName: "questionSolutionArticles",
-    variables: { questionSlug: question_slug, first: 1, skip: 0, orderBy: "DEFAULT", tagSlugs: [lang] },
+    variables: { questionSlug: question_slug, first: 5, skip: 0, orderBy: "DEFAULT", tagSlugs: [lang] },
     query: [
       "query questionSolutionArticles($questionSlug: String!, $skip: Int, $first: Int, $orderBy: SolutionArticleOrderBy, $userInput: String, $tagSlugs: [String!]) {",
       "questionSolutionArticles(questionSlug: $questionSlug, skip: $skip, first: $first, orderBy: $orderBy, userInput: $userInput, tagSlugs: $tagSlugs) {",
@@ -317,6 +317,10 @@ function getSolutionArticlesSlugList(question_slug: string, lang: string, cb) {
       "fragment solutionArticle on SolutionArticleNode {",
       "      uuid",
       "      slug",
+      "      title",
+      "      author {",
+      "       username",
+      "      }",
       "  byLeetcode",
       "  __typename",
       "}",
@@ -333,7 +337,17 @@ function getSolutionArticlesSlugList(question_slug: string, lang: string, cb) {
       }
     });
 
-    cb(e, temp_result);
+    if (edges.length > 0) {
+      cnSelectSolution(edges, cb);
+    } else {
+      cb(e, temp_result);
+    }
+  });
+}
+
+function cnSelectSolution(edges, cb) {
+  reply.remote_post({ canSelect: edges, oper: "need_select" }, function (_, back_arg) {
+    cb(undefined, back_arg.slug);
   });
 }
 

src/rpc/utils/ReplyUtils.ts

@@ -10,13 +10,19 @@
 let _ = require("underscore");
 
 class Reply {
+  init_recv_flag = false;
   output = _.bind(console.log, console);
   level: any;
   levels = new Map([
     ["INFO", { value: 2 }],
     ["WARN", { value: 3 }],
     ["ERROR", { value: 4 }],
   ]);
+
+  operCookie = 0;
+
+  operWaitMap: Map<number, any> = new Map<number, any>();
+
   setLevel(name: any) {
     this.level = this.levels.get(name) || this.levels.get("INFO");
   }
@@ -57,6 +63,39 @@ class Reply {
     let s = args.map((x) => x.toString()).join(" ");
     this.output(s);
   }
+
+  initRecv() {
+    // 监听父进程的输入
+    this.init_recv_flag = true;
+    process.stdin.on("data", (data) => {
+      reply.recvCallback(data.toString());
+    });
+  }
+
+  recvCallback(data) {
+    let data_ob = JSON.parse(data);
+    let c = data_ob.c;
+    let need_call = this.operWaitMap.get(c);
+    if (need_call) {
+      need_call.callback(need_call.arg, data_ob);
+    }
+  }
+
+  getOperCookie() {
+    this.operCookie = this.operCookie + 1;
+    return this.operCookie;
+  }
+
+  remote_post(oper_data, cb) {
+    if (!this.init_recv_flag) {
+      this.initRecv();
+    }
+    let c = this.getOperCookie();
+    this.operWaitMap.set(c, { arg: oper_data, callback: cb });
+
+    let msg = { oper: "requireOper", cookie: c, arg: oper_data };
+    this.output(JSON.stringify(msg));
+  }
 }
 
 export const reply: Reply = new Reply();