From 045c0bddc8323ef902fab708a23d96b542800217 Mon Sep 17 00:00:00 2001
From: alemi <me@alemi.dev>
Date: Sat, 7 Sep 2024 00:41:25 +0200
Subject: [PATCH] fix: expandability, context value in tree

---
 src/tree.ts | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/src/tree.ts b/src/tree.ts
index 63d79e9..97f0e09 100644
--- a/src/tree.ts
+++ b/src/tree.ts
@@ -22,19 +22,20 @@ export class CodempTreeProvider implements vscode.TreeDataProvider<CodempTreeIte
 				case Type.Workspace:
 					if (workspace === null) { return [] };
 					if (element.label == workspace.id()) {
-						return [
-							new CodempTreeItem("Buffers", Type.BufferContainer),
-							new CodempTreeItem("Users", Type.UserContainer)
-						];
+						// return [
+						// 	new CodempTreeItem("Buffers", Type.BufferContainer, true),
+						// 	new CodempTreeItem("Users", Type.UserContainer, true)
+						// ];
+						return workspace.filetree().map((x) => new CodempTreeItem(x, Type.Buffer, false));
 					} else {
 						return [];
 					}
 				case Type.BufferContainer:
 					if (workspace === null) { return [] };
-					return workspace.filetree().map((x) => new CodempTreeItem(x, Type.Buffer));
+					return workspace.filetree().map((x) => new CodempTreeItem(x, Type.Buffer, false));
 				case Type.UserContainer:
 					if (workspace === null) { return [] };
-					return [new CodempTreeItem("TODO", Type.User)]; // TODO keep track of users
+					return [new CodempTreeItem("TODO", Type.User, false)]; // TODO keep track of users
 				case Type.Buffer:
 					return [];
 				case Type.User:
@@ -42,7 +43,7 @@ export class CodempTreeProvider implements vscode.TreeDataProvider<CodempTreeIte
 			}
 		} else {
 			if(client === null)	return [];
-			return workspace_list.map((x) => new CodempTreeItem(x, Type.Workspace));
+			return workspace_list.map((x) => new CodempTreeItem(x, Type.Workspace, true));
 		}
 
 	}
@@ -50,16 +51,19 @@ export class CodempTreeProvider implements vscode.TreeDataProvider<CodempTreeIte
 
 class CodempTreeItem extends vscode.TreeItem {
 	type: Type;
-	constructor(label: string | vscode.TreeItemLabel, type: Type){
-		super(label, vscode.TreeItemCollapsibleState.Expanded);
-		this.type=type;
+	constructor(label: string | vscode.TreeItemLabel, type: Type, expandable: boolean){
+		let state = expandable ? vscode.TreeItemCollapsibleState.Expanded : vscode.TreeItemCollapsibleState.None;
+		console.log(type.toString());
+		super(label, state);
+		this.type = type;
+		this.contextValue = type;
 	}
 }
 
 enum Type {
-	Workspace,
-	BufferContainer,
-	UserContainer,
-	Buffer,
-	User,
+	Workspace = "workspace",
+	BufferContainer = "container_buffer",
+	UserContainer = "container_user",
+	Buffer = "buffer",
+	User = "user",
 }