From 5bb759236d1882e842f64af5b25964f656331c50 Mon Sep 17 00:00:00 2001
From: "sheche@microsoft.com" <sheche@microsoft.com>
Date: Tue, 7 May 2019 19:55:32 +0800
Subject: [PATCH] feat: Add 'All' category into the explorer

---
 src/explorer/LeetCodeTreeDataProvider.ts | 2 ++
 src/explorer/explorerNodeManager.ts      | 8 ++++++++
 src/shared.ts                            | 1 +
 3 files changed, 11 insertions(+)

diff --git a/src/explorer/LeetCodeTreeDataProvider.ts b/src/explorer/LeetCodeTreeDataProvider.ts
index c04346d6..a4339614 100644
--- a/src/explorer/LeetCodeTreeDataProvider.ts
+++ b/src/explorer/LeetCodeTreeDataProvider.ts
@@ -60,6 +60,8 @@ export class LeetCodeTreeDataProvider implements vscode.TreeDataProvider<LeetCod
             return explorerNodeManager.getRootNodes();
         } else {
             switch (element.id) { // First-level
+                case Category.All:
+                    return explorerNodeManager.getAllNodes();
                 case Category.Favorite:
                     return explorerNodeManager.getFavoriteNodes();
                 case Category.Difficulty:
diff --git a/src/explorer/explorerNodeManager.ts b/src/explorer/explorerNodeManager.ts
index dcf1cc09..e58dc8ea 100644
--- a/src/explorer/explorerNodeManager.ts
+++ b/src/explorer/explorerNodeManager.ts
@@ -27,6 +27,10 @@ class ExplorerNodeManager implements Disposable {
 
     public getRootNodes(): LeetCodeNode[] {
         return [
+            new LeetCodeNode(Object.assign({}, defaultProblem, {
+                id: Category.All,
+                name: Category.All,
+            }), false),
             new LeetCodeNode(Object.assign({}, defaultProblem, {
                 id: Category.Difficulty,
                 name: Category.Difficulty,
@@ -46,6 +50,10 @@ class ExplorerNodeManager implements Disposable {
         ];
     }
 
+    public getAllNodes(): LeetCodeNode[] {
+        return Array.from(this.explorerNodeMap.values());
+    }
+
     public getAllDifficultyNodes(): LeetCodeNode[] {
         const res: LeetCodeNode[] = [];
         res.push(
diff --git a/src/shared.ts b/src/shared.ts
index f543cd47..78518454 100644
--- a/src/shared.ts
+++ b/src/shared.ts
@@ -86,6 +86,7 @@ export const defaultProblem: IProblem = {
 };
 
 export enum Category {
+    All = "All",
     Difficulty = "Difficulty",
     Tag = "Tag",
     Company = "Company",