Add refresh and collapse all support (#26)

Author: MicroFish91

package.json

@@ -222,6 +222,47 @@
         "command": "portfolioInstruments.undeprecate",
         "title": "%portfolioInstruments.undeprecate%",
         "category": "Portfolio Instruments"
+      },
+      {
+        "command": "portfolioInstruments.refresh",
+        "title": "%portfolioInstruments.refresh%",
+        "category": "Portfolio Instruments",
+        "icon": "$(refresh)"
+      },
+      {
+        "command": "portfolioInstruments.refreshSnapshots",
+        "title": "%portfolioInstruments.refreshSnapshots%",
+        "category": "Portfolio Instruments",
+        "icon": "$(refresh)"
+      },
+      {
+        "command": "portfolioInstruments.refreshAccounts",
+        "title": "%portfolioInstruments.refreshAccounts%",
+        "category": "Portfolio Instruments",
+        "icon": "$(refresh)"
+      },
+      {
+        "command": "portfolioInstruments.refreshHoldings",
+        "title": "%portfolioInstruments.refreshHoldings%",
+        "category": "Portfolio Instruments",
+        "icon": "$(refresh)"
+      },
+      {
+        "command": "portfolioInstruments.refreshBenchmarks",
+        "title": "%portfolioInstruments.refreshBenchmarks%",
+        "category": "Portfolio Instruments",
+        "icon": "$(refresh)"
+      },
+      {
+        "command": "portfolioInstruments.refreshSnapshot",
+        "title": "%portfolioInstruments.refreshSnapshot%",
+        "category": "Portfolio Instruments"
+      },
+      {
+        "command": "portfolioInstruments.collapseAll",
+        "title": "%portfolioInstruments.collapseAll%",
+        "category": "Portfolio Instruments",
+        "icon": "$(collapse-all)"
       }
     ],
     "configuration": [
@@ -388,6 +429,26 @@
         {
           "command": "portfolioInstruments.undeprecate",
           "when": "never"
+        },
+        {
+          "command": "portfolioInstruments.refreshSnapshots",
+          "when": "never"
+        },
+        {
+          "command": "portfolioInstruments.refreshAccounts",
+          "when": "never"
+        },
+        {
+          "command": "portfolioInstruments.refreshHoldings",
+          "when": "never"
+        },
+        {
+          "command": "portfolioInstruments.refreshBenchmarks",
+          "when": "never"
+        },
+        {
+          "command": "portfolioInstruments.refreshSnapshot",
+          "when": "never"
         }
       ],
       "view/item/context": [
@@ -506,6 +567,11 @@
           "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /snapshotItem/i",
           "group": "1@1"
         },
+        {
+          "command": "portfolioInstruments.refreshSnapshot",
+          "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /snapshotItem/i",
+          "group": "z@0"
+        },
         {
           "command": "portfolioInstruments.updateMaturationStart",
           "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /snapshotMaturationStartItem/i",
@@ -580,18 +646,48 @@
           "command": "portfolioInstruments.undeprecate",
           "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /benchmarkDeprecatedItem/i",
           "group": "1@1"
+        },
+        {
+          "command": "portfolioInstruments.refreshSnapshots",
+          "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /snapshotsItem/i",
+          "group": "z@0"
+        },
+        {
+          "command": "portfolioInstruments.refreshAccounts",
+          "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /accountsItem/i",
+          "group": "z@0"
+        },
+        {
+          "command": "portfolioInstruments.refreshHoldings",
+          "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /holdingsItem/i",
+          "group": "z@0"
+        },
+        {
+          "command": "portfolioInstruments.refreshBenchmarks",
+          "when": "view =~ /portfolioInstruments.main/i && viewItem =~ /benchmarksItem/i",
+          "group": "z@0"
         }
       ],
       "view/title": [
         {
           "command": "portfolioInstruments.login",
           "when": "view == portfolioInstruments.main",
-          "group": "navigation"
+          "group": "navigation@1"
         },
         {
           "command": "portfolioInstruments.logoutAll",
           "when": "view == portfolioInstruments.main",
-          "group": "navigation"
+          "group": "navigation@2"
+        },
+        {
+          "command": "portfolioInstruments.collapseAll",
+          "when": "view == portfolioInstruments.main",
+          "group": "navigation@3"
+        },
+        {
+          "command": "portfolioInstruments.refresh",
+          "when": "view == portfolioInstruments.main",
+          "group": "navigation@4"
         }
       ]
     }

package.nls.json

@@ -40,5 +40,12 @@
     "portfolioInstruments.updateRebalanceSettings": "Update Rebalance Threshold...",
     "portfolioInstruments.plotSnapshots": "Plot Snapshots...",
     "portfolioInstruments.viewProperties": "View Properties",
-    "portfolioInstruments.undeprecate": "Restore..."
+    "portfolioInstruments.undeprecate": "Restore...",
+    "portfolioInstruments.refresh": "Refresh",
+    "portfolioInstruments.refreshSnapshots": "Refresh",
+    "portfolioInstruments.refreshAccounts": "Refresh",
+    "portfolioInstruments.refreshHoldings": "Refresh",
+    "portfolioInstruments.refreshBenchmarks": "Refresh",
+    "portfolioInstruments.refreshSnapshot": "Refresh",
+    "portfolioInstruments.collapseAll": "Collapse All"
 }

src/commands/refresh/collapseAll.ts

@@ -0,0 +1,6 @@
+import { commands } from "vscode";
+import { CommandContext } from "../registerCommand";
+
+export async function collapseAll(_context: CommandContext): Promise<void> {
+    await commands.executeCommand('workbench.actions.treeView.portfolioInstruments.main.collapseAll');
+}

src/commands/refresh/refresh.ts

@@ -0,0 +1,11 @@
+import { ext } from "../../extensionVariables";
+import { CommandContext } from "../registerCommand";
+import { PiExtTreeItem } from "../../tree/PiExtTreeDataProvider";
+
+export async function refresh(_context: CommandContext): Promise<void> {
+    ext.portfolioInstrumentsTdp.refresh();
+}
+
+export async function refreshNode(_context: CommandContext, node: PiExtTreeItem): Promise<void> {
+    ext.portfolioInstrumentsTdp.refresh(node);
+}

src/commands/registerCommands.ts

@@ -36,6 +36,8 @@ import { updateSnapshotValue } from "./snapshotValues/updateSnapshotValue/update
 import { viewProperties } from "./viewProperties/viewProperties";
 import { toggleShowDeprecatedResources } from "./toggleShowDeprecatedResources/toggleShowDeprecatedResources";
 import { undeprecate } from "./undeprecate/undeprecate";
+import { refresh, refreshNode } from "./refresh/refresh";
+import { collapseAll } from "./refresh/collapseAll";
 
 export function registerCommands() {
     // Auth
@@ -92,4 +94,13 @@ export function registerCommands() {
     registerCommand('portfolioInstruments.setApiEndpoint', setApiEndpoint);
     registerCommand('portfolioInstruments.toggleShowDeprecatedResources', toggleShowDeprecatedResources);
     registerCommand('portfolioInstruments.undeprecate', undeprecate);
+
+    // Refresh
+    registerCommand('portfolioInstruments.refresh', refresh);
+    registerCommand('portfolioInstruments.refreshSnapshots', refreshNode);
+    registerCommand('portfolioInstruments.refreshAccounts', refreshNode);
+    registerCommand('portfolioInstruments.refreshHoldings', refreshNode);
+    registerCommand('portfolioInstruments.refreshBenchmarks', refreshNode);
+    registerCommand('portfolioInstruments.refreshSnapshot', refreshNode);
+    registerCommand('portfolioInstruments.collapseAll', collapseAll);
 }