X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/37e43f4e19ad2bd27b15fe7f0d857218dad39055..5df95741b820e18db893c8ce1fac67b921b7fafc:/src/store/tree-picker/tree-picker.ts

diff --git a/src/store/tree-picker/tree-picker.ts b/src/store/tree-picker/tree-picker.ts
index e19ce3a7a4..a4127fa3bb 100644
--- a/src/store/tree-picker/tree-picker.ts
+++ b/src/store/tree-picker/tree-picker.ts
@@ -3,19 +3,12 @@
 // SPDX-License-Identifier: AGPL-3.0
 
 import { Tree } from "~/models/tree";
-import { TreeItemStatus } from "~/components/tree/tree";
+import { TreeItemStatus } from '~/components/tree/tree';
+export type TreePicker = { [key: string]: Tree<any> };
 
-export type TreePicker = Tree<TreePickerNode>;
+export const getTreePicker = <Value = {}>(id: string) => (state: TreePicker): Tree<Value> | undefined => state[id];
 
-export interface TreePickerNode {
-    id: string;
-    value: any;
-    selected: boolean;
-    collapsed: boolean;
-    status: TreeItemStatus;
-}
-
-export const createTreePickerNode = (data: {id: string, value: any}) => ({
+export const createTreePickerNode = (data: { nodeId: string, value: any }) => ({
     ...data,
     selected: false,
     collapsed: true,