add create tree view wrapper

iterative · Aug 3, 2021 · 1376336 · 1376336
1 parent acffc32
commit 1376336
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 18 deletions.
diff --git a/extension/src/experiments/model/filterBy/tree.ts b/extension/src/experiments/model/filterBy/tree.ts
@@ -6,12 +6,12 @@ import {
   TreeDataProvider,
   TreeItem,
   TreeItemCollapsibleState,
-  Uri,
-  window
+  Uri
 } from 'vscode'
 import { getFilterId } from '.'
 import { Experiments } from '../..'
 import { definedAndNonEmpty, flatten } from '../../../util/array'
+import { createTreeView } from '../../../vscode/tree'
 
 type FilterItem = {
   description: string
@@ -33,14 +33,7 @@ export class ExperimentsFilterByTree
     this.onDidChangeTreeData = experiments.experimentsChanged.event
 
     this.dispose.track(
-      window.createTreeView<string | FilterItem>(
-        'dvc.views.experimentsFilterByTree',
-        {
-          canSelectMany: true,
-          showCollapseAll: true,
-          treeDataProvider: this
-        }
-      )
+      createTreeView<FilterItem>('dvc.views.experimentsFilterByTree', this)
     )
 
     this.experiments = experiments

diff --git a/extension/src/experiments/paramsAndMetrics/tree.ts b/extension/src/experiments/paramsAndMetrics/tree.ts
@@ -7,13 +7,13 @@ import {
   TreeItem,
   TreeItemCollapsibleState,
   TreeView,
-  Uri,
-  window
+  Uri
 } from 'vscode'
 import { Status } from './model'
 import { Experiments } from '..'
 import { Resource, ResourceLocator } from '../../resourceLocator'
 import { definedAndNonEmpty, flatten } from '../../util/array'
+import { createTreeView } from '../../vscode/tree'
 
 type ParamsAndMetricsItem = {
   description: string | undefined
@@ -41,13 +41,9 @@ export class ExperimentsParamsAndMetricsTree
     this.onDidChangeTreeData = experiments.paramsOrMetricsChanged.event
 
     this.view = this.dispose.track(
-      window.createTreeView<string | ParamsAndMetricsItem>(
+      createTreeView<ParamsAndMetricsItem>(
         'dvc.views.experimentsParamsAndMetricsTree',
-        {
-          canSelectMany: true,
-          showCollapseAll: true,
-          treeDataProvider: this
-        }
+        this
       )
     )
 

diff --git a/extension/src/vscode/tree.ts b/extension/src/vscode/tree.ts
@@ -0,0 +1,11 @@
+import { TreeDataProvider, TreeView, window } from 'vscode'
+
+export const createTreeView = <T>(
+  name: string,
+  treeDataProvider: TreeDataProvider<string | T>
+): TreeView<string | T> =>
+  window.createTreeView<string | T>(name, {
+    canSelectMany: true,
+    showCollapseAll: true,
+    treeDataProvider
+  })