update

ccagml · ccagml · commit ab84338f1d9f · 2022-11-23T15:35:34.000+08:00
diff --git a/package.json b/package.json
@@ -94,8 +94,43 @@
                 "category": "LeetCode"
             },
             {
-                "command": "lcpr.setBricksType",
-                "title": "设置提交几天后出现在工地",
+                "command": "lcpr.setBricksType0",
+                "title": "设置不出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType1",
+                "title": "设置14天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType2",
+                "title": "设置7天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType3",
+                "title": "设置5天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType4",
+                "title": "设置3天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType5",
+                "title": "设置2天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType6",
+                "title": "设置1天后出现在工地",
+                "category": "LeetCode"
+            },
+            {
+                "command": "lcpr.setBricksType7",
+                "title": "设置1天后出现在工地",
                 "category": "LeetCode"
             },
             {
@@ -106,7 +141,7 @@
             },
             {
                 "command": "lcpr.showSolution",
-                "title": "Show Top Voted Solution",
+                "title": "获取题解",
                 "category": "LeetCode"
             },
             {
@@ -225,7 +260,7 @@
                     "group": "leetcode@3"
                 },
                 {
-                    "command": "lcpr.setBricksType",
+                    "submenu": "lcpr.setBricksType_sub",
                     "when": "view == QuestionExplorer && viewItem =~ /problem*/",
                     "group": "leetcode@4"
                 },
@@ -255,7 +290,7 @@
                     "group": "leetcode@3"
                 },
                 {
-                    "command": "lcpr.setBricksType",
+                    "submenu": "lcpr.setBricksType_sub1",
                     "when": "view == BricksExplorer && viewItem == bricks",
                     "group": "leetcode@4"
                 }
@@ -296,10 +331,10 @@
             ],
             "editor/context": [
                 {
-                    "submenu": "leetcode.editorAction"
+                    "submenu": "lcpr.editorAction"
                 }
             ],
-            "leetcode.editorAction": [
+            "lcpr.editorAction": [
                 {
                     "command": "lcpr.testSolution",
                     "group": "leetcode@1"
@@ -316,12 +351,80 @@
                     "command": "lcpr.previewProblem",
                     "group": "leetcode@4"
                 }
+            ],
+            "lcpr.setBricksType_sub": [
+                {
+                    "command": "lcpr.setBricksType0",
+                    "group": "leetcode@1"
+                },
+                {
+                    "command": "lcpr.setBricksType6",
+                    "group": "leetcode@2"
+                },
+                {
+                    "command": "lcpr.setBricksType5",
+                    "group": "leetcode@3"
+                },
+                {
+                    "command": "lcpr.setBricksType4",
+                    "group": "leetcode@4"
+                },
+                {
+                    "command": "lcpr.setBricksType3",
+                    "group": "leetcode@5"
+                },
+                {
+                    "command": "lcpr.setBricksType2",
+                    "group": "leetcode@6"
+                },
+                {
+                    "command": "lcpr.setBricksType1",
+                    "group": "leetcode@7"
+                }
+            ],
+            "lcpr.setBricksType_sub1": [
+                {
+                    "command": "lcpr.setBricksType0",
+                    "group": "leetcode@1"
+                },
+                {
+                    "command": "lcpr.setBricksType6",
+                    "group": "leetcode@2"
+                },
+                {
+                    "command": "lcpr.setBricksType5",
+                    "group": "leetcode@3"
+                },
+                {
+                    "command": "lcpr.setBricksType4",
+                    "group": "leetcode@4"
+                },
+                {
+                    "command": "lcpr.setBricksType3",
+                    "group": "leetcode@5"
+                },
+                {
+                    "command": "lcpr.setBricksType2",
+                    "group": "leetcode@6"
+                },
+                {
+                    "command": "lcpr.setBricksType1",
+                    "group": "leetcode@7"
+                }
             ]
         },
         "submenus": [
             {
-                "id": "leetcode.editorAction",
+                "id": "lcpr.editorAction",
                 "label": "LCPR菜单"
+            },
+            {
+                "id": "lcpr.setBricksType_sub",
+                "label": "设置砖头类型"
+            },
+            {
+                "id": "lcpr.setBricksType_sub1",
+                "label": "设置砖头类型"
             }
         ],
         "configuration": [
diff --git a/src/controller/BricksViewController.ts b/src/controller/BricksViewController.ts
@@ -55,21 +55,12 @@ class BricksViewController implements Disposable {
         has_qid ? TreeItemCollapsibleState.Collapsed : TreeItemCollapsibleState.None
       )
     );
-
-    // baseNode.push(
-    //   new BricksNode(
-    //     Object.assign({}, defaultProblem, {
-    //       id: BricksNormalId.Today,
-    //       name: BricksNormalId.TodayDesc,
-    //     }),
-    //     false,
-    //     0,
-    //     TreeItemCollapsibleState.Collapsed
-    //   )
-    // );
-
     return baseNode;
   }
+
+  public async setBricksType(node: BricksNode, type) {
+    await bricksDataService.setBricksType(node, type);
+  }
   public dispose(): void {}
 }
 
diff --git a/src/dao/bricksDao.ts b/src/dao/bricksDao.ts
@@ -119,8 +119,10 @@ class BricksDao {
       const value = all_bricks[qid];
       const submit_time = value.submit_time || [];
       const submit_size = submit_time.length;
-      if (submit_size < 1 || submit_time[submit_size - 1] < this.getTimeByType(value.type, today_time)) {
-        all_qid.push(qid);
+      if (value.type > BricksType.TYPE_0) {
+        if (submit_size < 1 || submit_time[submit_size - 1] < this.getTimeByType(value.type, today_time)) {
+          all_qid.push(qid);
+        }
       }
     }
     return all_qid;
@@ -148,6 +150,11 @@ class BricksDao {
     }
     await this.setInfoByQid(qid, temp_data);
   }
+  public async setTypeByQid(qid: string, type) {
+    let temp_data = await this.getInfoByQid(qid);
+    temp_data.type = type;
+    await this.setInfoByQid(qid, temp_data);
+  }
 }
 
 export const bricksDao: BricksDao = new BricksDao();
diff --git a/src/extension.ts b/src/extension.ts
@@ -24,7 +24,7 @@ import { markdownService } from "./service/MarkdownService";
 import { mainContorller } from "./controller/MainController";
 import { loginContorller } from "./controller/LoginController";
 import { getLeetCodeEndpoint } from "./utils/ConfigUtils";
-import { OutPutType } from "./model/Model";
+import { BricksType, OutPutType } from "./model/Model";
 import { bricksDataService } from "./service/BricksDataService";
 import { bricksViewController } from "./controller/BricksViewController";
 
@@ -75,7 +75,28 @@ export async function activate(context: ExtensionContext): Promise<void> {
       commands.registerCommand("lcpr.setDefaultLanguage", () => treeViewController.setDefaultLanguage()),
       commands.registerCommand("lcpr.addFavorite", (node: NodeModel) => treeViewController.addFavorite(node)),
       commands.registerCommand("lcpr.removeFavorite", (node: NodeModel) => treeViewController.removeFavorite(node)),
-      commands.registerCommand("lcpr.problems.sort", () => treeViewController.switchSortingStrategy())
+      commands.registerCommand("lcpr.problems.sort", () => treeViewController.switchSortingStrategy()),
+      commands.registerCommand("lcpr.setBricksType0", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_0)
+      ),
+      commands.registerCommand("lcpr.setBricksType1", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_1)
+      ),
+      commands.registerCommand("lcpr.setBricksType2", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_2)
+      ),
+      commands.registerCommand("lcpr.setBricksType3", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_3)
+      ),
+      commands.registerCommand("lcpr.setBricksType4", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_4)
+      ),
+      commands.registerCommand("lcpr.setBricksType5", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_5)
+      ),
+      commands.registerCommand("lcpr.setBricksType6", (node: NodeModel) =>
+        bricksViewController.setBricksType(node, BricksType.TYPE_6)
+      )
     );
 
     // 设置站点
diff --git a/src/service/BricksDataService.ts b/src/service/BricksDataService.ts
@@ -96,13 +96,20 @@ export class BricksDataService implements TreeDataProvider<BricksNode> {
     }
   }
 
-  public checkSubmit(e: ISubmitEvent) {
+  public async checkSubmit(e: ISubmitEvent) {
     if (e.sub_type == "submit" && e.accepted) {
       let qid: string = e.qid.toString();
       bricksDao.addSubmitTimeByQid(qid);
+      await bricksDataService.refresh();
     }
   }
 
+  public async setBricksType(node: BricksNode, type) {
+    let qid: string = node.id.toString();
+    bricksDao.setTypeByQid(qid, type);
+    await bricksDataService.refresh();
+  }
+
   private parseIconPathFromProblemState(element: BricksNode): string {
     switch (element.state) {
       default:
diff --git a/src/service/ExecuteService.ts b/src/service/ExecuteService.ts
diff --git a/src/service/TreeDataService.ts b/src/service/TreeDataService.ts
