|
1 | 1 | import{getWorkspaces}from"coder/site/src/api/api" |
2 | | -import{WorkspaceAgent}from"coder/site/src/api/typesGenerated" |
| 2 | +import{Workspace,WorkspaceAgent}from"coder/site/src/api/typesGenerated" |
| 3 | +importEventSourcefrom"eventsource" |
3 | 4 | import*aspathfrom"path" |
4 | 5 | import*asvscodefrom"vscode" |
5 | | -import{extractAgents}from"./api-helper" |
| 6 | +import{AgentMetadataEvent,AgentMetadataEventSchemaArray,extractAgents}from"./api-helper" |
| 7 | +import{Storage}from"./storage" |
6 | 8 |
|
7 | 9 | exportenumWorkspaceQuery{ |
8 | 10 | Mine="owner:me", |
9 | 11 | All="", |
10 | 12 | } |
11 | 13 |
|
12 | | -exportclassWorkspaceProviderimplementsvscode.TreeDataProvider<WorkspaceTreeItem>{ |
13 | | -constructor(privatereadonlygetWorkspacesQuery:WorkspaceQuery){} |
| 14 | +exportclassWorkspaceProviderimplementsvscode.TreeDataProvider<vscode.TreeItem>{ |
| 15 | +privateworkspaces:WorkspaceTreeItem[]=[] |
| 16 | +privateagentMetadata:Record<WorkspaceAgent["id"],AgentMetadataEvent[]>={} |
14 | 17 |
|
15 | | -private_onDidChangeTreeData:vscode.EventEmitter<WorkspaceTreeItem|undefined|null|void>= |
16 | | -newvscode.EventEmitter<WorkspaceTreeItem|undefined|null|void>() |
17 | | -readonlyonDidChangeTreeData:vscode.Event<WorkspaceTreeItem|undefined|null|void>= |
| 18 | +constructor(privatereadonlygetWorkspacesQuery:WorkspaceQuery,privatereadonlystorage:Storage){ |
| 19 | +getWorkspaces({q:this.getWorkspacesQuery}) |
| 20 | +.then((workspaces)=>{ |
| 21 | +constworkspacesTreeItem:WorkspaceTreeItem[]=[] |
| 22 | +workspaces.workspaces.forEach((workspace)=>{ |
| 23 | +constshowMetadata=this.getWorkspacesQuery===WorkspaceQuery.Mine |
| 24 | +if(showMetadata){ |
| 25 | +constagents=extractAgents(workspace) |
| 26 | +agents.forEach((agent)=>this.monitorMetadata(agent.id))// monitor metadata for all agents |
| 27 | +} |
| 28 | +consttreeItem=newWorkspaceTreeItem( |
| 29 | +workspace, |
| 30 | +this.getWorkspacesQuery===WorkspaceQuery.All, |
| 31 | +showMetadata, |
| 32 | +) |
| 33 | +workspacesTreeItem.push(treeItem) |
| 34 | +}) |
| 35 | +returnworkspacesTreeItem |
| 36 | +}) |
| 37 | +.then((workspaces)=>{ |
| 38 | +this.workspaces=workspaces |
| 39 | +this.refresh() |
| 40 | +}) |
| 41 | +} |
| 42 | + |
| 43 | +private_onDidChangeTreeData:vscode.EventEmitter<vscode.TreeItem|undefined|null|void>= |
| 44 | +newvscode.EventEmitter<vscode.TreeItem|undefined|null|void>() |
| 45 | +readonlyonDidChangeTreeData:vscode.Event<vscode.TreeItem|undefined|null|void>= |
18 | 46 | this._onDidChangeTreeData.event |
19 | 47 |
|
20 | | -refresh():void{ |
21 | | -this._onDidChangeTreeData.fire() |
| 48 | +refresh(item:vscode.TreeItem|undefined|null|void):void{ |
| 49 | +this._onDidChangeTreeData.fire(item) |
22 | 50 | } |
23 | 51 |
|
24 | | -getTreeItem(element:WorkspaceTreeItem):vscode.TreeItem{ |
| 52 | +asyncgetTreeItem(element:vscode.TreeItem):Promise<vscode.TreeItem>{ |
25 | 53 | returnelement |
26 | 54 | } |
27 | 55 |
|
28 | | -getChildren(element?:WorkspaceTreeItem):Thenable<WorkspaceTreeItem[]>{ |
| 56 | +getChildren(element?:vscode.TreeItem):Thenable<vscode.TreeItem[]>{ |
29 | 57 | if(element){ |
30 | | -if(element.agents.length>0){ |
31 | | -returnPromise.resolve( |
32 | | -element.agents.map((agent)=>{ |
33 | | -constlabel=agent.name |
34 | | -constdetail=`Status:${agent.status}` |
35 | | -returnnewWorkspaceTreeItem(label,detail,"","",agent.name,agent.expanded_directory,[],"coderAgent") |
36 | | -}), |
37 | | -) |
| 58 | +if(elementinstanceofWorkspaceTreeItem){ |
| 59 | +constagents=extractAgents(element.workspace) |
| 60 | +constagentTreeItems=agents.map((agent)=>newAgentTreeItem(agent,element.watchMetadata)) |
| 61 | +returnPromise.resolve(agentTreeItems) |
| 62 | +}elseif(elementinstanceofAgentTreeItem){ |
| 63 | +constsavedMetadata=this.agentMetadata[element.agent.id]||[] |
| 64 | +returnPromise.resolve(savedMetadata.map((metadata)=>newAgentMetadataTreeItem(metadata))) |
38 | 65 | } |
| 66 | + |
39 | 67 | returnPromise.resolve([]) |
40 | 68 | } |
41 | | -returngetWorkspaces({q:this.getWorkspacesQuery}).then((workspaces)=>{ |
42 | | -returnworkspaces.workspaces.map((workspace)=>{ |
43 | | -conststatus= |
44 | | -workspace.latest_build.status.substring(0,1).toUpperCase()+workspace.latest_build.status.substring(1) |
45 | | - |
46 | | -constlabel= |
47 | | -this.getWorkspacesQuery===WorkspaceQuery.All |
48 | | - ?`${workspace.owner_name} /${workspace.name}` |
49 | | - :workspace.name |
50 | | -constdetail=`Template:${workspace.template_display_name||workspace.template_name} • Status:${status}` |
51 | | -constagents=extractAgents(workspace) |
52 | | -returnnewWorkspaceTreeItem( |
53 | | -label, |
54 | | -detail, |
55 | | -workspace.owner_name, |
56 | | -workspace.name, |
57 | | -undefined, |
58 | | -agents[0]?.expanded_directory, |
59 | | -agents, |
60 | | -agents.length>1 ?"coderWorkspaceMultipleAgents" :"coderWorkspaceSingleAgent", |
61 | | -) |
62 | | -}) |
| 69 | +returnPromise.resolve(this.workspaces) |
| 70 | +} |
| 71 | + |
| 72 | +asyncmonitorMetadata(agentId:WorkspaceAgent["id"]):Promise<void>{ |
| 73 | +constagentMetadataURL=newURL(`${this.storage.getURL()}/api/v2/workspaceagents/${agentId}/watch-metadata`) |
| 74 | +constagentMetadataEventSource=newEventSource(agentMetadataURL.toString(),{ |
| 75 | +headers:{ |
| 76 | +"Coder-Session-Token":awaitthis.storage.getSessionToken(), |
| 77 | +}, |
| 78 | +}) |
| 79 | + |
| 80 | +agentMetadataEventSource.addEventListener("data",(event)=>{ |
| 81 | +try{ |
| 82 | +constdataEvent=JSON.parse(event.data) |
| 83 | +constagentMetadata=AgentMetadataEventSchemaArray.parse(dataEvent) |
| 84 | + |
| 85 | +if(agentMetadata.length===0){ |
| 86 | +agentMetadataEventSource.close() |
| 87 | +} |
| 88 | + |
| 89 | +constsavedMetadata=this.agentMetadata[agentId] |
| 90 | +if(JSON.stringify(savedMetadata)!==JSON.stringify(agentMetadata)){ |
| 91 | +this.agentMetadata[agentId]=agentMetadata// overwrite existing metadata |
| 92 | +this.refresh() |
| 93 | +} |
| 94 | +}catch(error){ |
| 95 | +agentMetadataEventSource.close() |
| 96 | +} |
63 | 97 | }) |
64 | 98 | } |
65 | 99 | } |
66 | 100 |
|
67 | 101 | typeCoderTreeItemType="coderWorkspaceSingleAgent"|"coderWorkspaceMultipleAgents"|"coderAgent" |
68 | 102 |
|
69 | | -exportclassWorkspaceTreeItemextendsvscode.TreeItem{ |
| 103 | +classAgentMetadataTreeItemextendsvscode.TreeItem{ |
| 104 | +constructor(metadataEvent:AgentMetadataEvent){ |
| 105 | +constlabel= |
| 106 | +metadataEvent.description.display_name.trim()+": "+metadataEvent.result.value.replace(/\n/g,"").trim() |
| 107 | + |
| 108 | +super(label,vscode.TreeItemCollapsibleState.None) |
| 109 | +this.tooltip="Collected at "+metadataEvent.result.collected_at |
| 110 | +this.contextValue="coderAgentMetadata" |
| 111 | +} |
| 112 | +} |
| 113 | + |
| 114 | +exportclassOpenableTreeItemextendsvscode.TreeItem{ |
70 | 115 | constructor( |
71 | | -publicreadonlylabel:string, |
72 | | -publicreadonlytooltip:string, |
| 116 | +label:string, |
| 117 | +tooltip:string, |
| 118 | +collapsibleState:vscode.TreeItemCollapsibleState, |
| 119 | + |
73 | 120 | publicreadonlyworkspaceOwner:string, |
74 | 121 | publicreadonlyworkspaceName:string, |
75 | 122 | publicreadonlyworkspaceAgent:string|undefined, |
76 | 123 | publicreadonlyworkspaceFolderPath:string|undefined, |
77 | | -publicreadonlyagents:WorkspaceAgent[], |
| 124 | + |
78 | 125 | contextValue:CoderTreeItemType, |
79 | 126 | ){ |
80 | | -super( |
81 | | -label, |
82 | | -contextValue==="coderWorkspaceMultipleAgents" |
83 | | - ?vscode.TreeItemCollapsibleState.Collapsed |
84 | | - :vscode.TreeItemCollapsibleState.None, |
85 | | -) |
| 127 | +super(label,collapsibleState) |
86 | 128 | this.contextValue=contextValue |
| 129 | +this.tooltip=tooltip |
87 | 130 | } |
88 | 131 |
|
89 | 132 | iconPath={ |
90 | 133 | light:path.join(__filename,"..","..","media","logo.svg"), |
91 | 134 | dark:path.join(__filename,"..","..","media","logo.svg"), |
92 | 135 | } |
93 | 136 | } |
| 137 | + |
| 138 | +classAgentTreeItemextendsOpenableTreeItem{ |
| 139 | +constructor(publicreadonlyagent:WorkspaceAgent,watchMetadata=false){ |
| 140 | +constlabel=agent.name |
| 141 | +constdetail=`Status:${agent.status}` |
| 142 | +super( |
| 143 | +label, |
| 144 | +detail, |
| 145 | +watchMetadata ?vscode.TreeItemCollapsibleState.Collapsed :vscode.TreeItemCollapsibleState.None, |
| 146 | +"", |
| 147 | +"", |
| 148 | +agent.name, |
| 149 | +agent.expanded_directory, |
| 150 | +"coderAgent", |
| 151 | +) |
| 152 | +} |
| 153 | +} |
| 154 | + |
| 155 | +exportclassWorkspaceTreeItemextendsOpenableTreeItem{ |
| 156 | +constructor( |
| 157 | +publicreadonlyworkspace:Workspace, |
| 158 | +publicreadonlyshowOwner:boolean, |
| 159 | +publicreadonlywatchMetadata=false, |
| 160 | +){ |
| 161 | +conststatus= |
| 162 | +workspace.latest_build.status.substring(0,1).toUpperCase()+workspace.latest_build.status.substring(1) |
| 163 | + |
| 164 | +constlabel=showOwner ?`${workspace.owner_name} /${workspace.name}` :workspace.name |
| 165 | +constdetail=`Template:${workspace.template_display_name||workspace.template_name} • Status:${status}` |
| 166 | +constagents=extractAgents(workspace) |
| 167 | +super( |
| 168 | +label, |
| 169 | +detail, |
| 170 | +showOwner ?vscode.TreeItemCollapsibleState.Collapsed :vscode.TreeItemCollapsibleState.Expanded, |
| 171 | +workspace.owner_name, |
| 172 | +workspace.name, |
| 173 | +undefined, |
| 174 | +agents[0]?.expanded_directory, |
| 175 | +"coderWorkspaceMultipleAgents", |
| 176 | +) |
| 177 | +} |
| 178 | +} |