Implement file decoration and tree item click handling features

- Added a new `DecorationProvider` class to manage file decorations in the project tree.
- Introduced a command `extension.handleTreeItemClick` to handle double-click events on tree items, allowing for project selection and file marking.
- Updated `package.json` to include the new command in the extension's menu.
- Refactored tree view logic to support the new click handling and decoration features.
- Enhanced project file management by integrating decoration updates on project selection.

Author: BernardXiong

package.json

@@ -98,6 +98,10 @@
                 "command": "extension.openTerminalProject",
                 "title": "Open RT-Thread Terminal",
                 "icon": "$(console)"
+            },
+            {
+                "command": "extension.handleTreeItemClick",
+                "title": "Handle Tree Item Click"
             }
         ],
         "menus": {
@@ -119,11 +123,6 @@
                 }
             ],
             "view/item/context": [
-                {
-                    "command": "extension.switchProject",
-                    "when": "view == projectFilesId && viewItem == project_bsp",
-                    "group": "inline"
-                },
                 {
                     "command": "extension.fastBuildProject",
                     "when": "view == projectFilesId && viewItem == project_bsp",

src/dock.ts

@@ -3,7 +3,7 @@ import * as vscode from 'vscode';
 import os from 'os';
 import fs from 'fs';
 import { getWorkspaceFolder, isRTThreadProject, isRTThreadWorksapce } from './api';
-import { buildGroupsTree, buildProjectTree, buildEmptyProjectTree, ProjectTreeItem, listFolderTreeItem, buildBSPTree } from './project/tree';
+import { buildGroupsTree, buildProjectTree, buildEmptyProjectTree, ProjectTreeItem, listFolderTreeItem, buildBSPTree, setTreeDataChangeEmitter } from './project/tree';
 import { cmds } from './cmds/index';
 
 class CmdTreeDataProvider implements vscode.TreeDataProvider<vscode.TreeItem> {
@@ -188,6 +188,10 @@ class ProjectFilesDataProvider implements vscode.TreeDataProvider<ProjectTreeIte
     private _onDidChangeTreeData: vscode.EventEmitter<ProjectTreeItem | undefined> = new vscode.EventEmitter<ProjectTreeItem | undefined>();
     readonly onDidChangeTreeData: vscode.Event<ProjectTreeItem | undefined> = this._onDidChangeTreeData.event;
 
+    getTreeDataChangeEmitter(): vscode.EventEmitter<ProjectTreeItem | undefined> {
+        return this._onDidChangeTreeData;
+    }
+
     getTreeItem(element: ProjectTreeItem): vscode.TreeItem {
         return element;
     }
@@ -291,6 +295,8 @@ export function initDockView(context: vscode.ExtensionContext) {
         treeDataProvider: _projectFilesDataProvider, showCollapseAll: true
     });
 
+    setTreeDataChangeEmitter(_projectFilesDataProvider.getTreeDataChangeEmitter());
+
     context.subscriptions.push(view);
     vscode.commands.registerCommand('extension.refreshRTThread', () => refreshProjectFilesAndGroups());
 

src/extension.ts

@@ -14,6 +14,7 @@ import { setupVEnv } from './venv';
 import { initAPI } from './api';
 import { openWorkspaceProjectsWebview } from './webviews/project';
 import { initProjectTree } from './project/tree';
+import { DecorationProvider } from './project/fileDecorationProvider';
 
 let _context: vscode.ExtensionContext;
 
@@ -47,6 +48,8 @@ export async function activate(context: vscode.ExtensionContext) {
                 isRTThreadWorksapce = true;
                 vscode.commands.executeCommand('setContext', 'isRTThreadWorksapce', true);
                 context.workspaceState.update('isRTThreadWorksapce', isRTThreadWorksapce);
+
+                new DecorationProvider(context);
             }
         }
         else {

src/project/cmd.ts

@@ -1,5 +1,6 @@
 import * as os from 'os';
 import * as fs from 'fs';
+import * as vscode from 'vscode';
 
 import { getWorkspaceFolder } from '../api';
 import { executeCommand } from '../terminal';
@@ -40,6 +41,9 @@ export function openTerminalProject(arg: any) {
 export function setCurrentProject(arg: any) {
     if (arg) {
         _currentProject = arg.fn;
+
+        let cmd = 'scons -C ' + arg.fn + ' --target=vsc_workspace';
+        executeCommand(cmd);
     }
 
     return;

src/project/fileDecorationProvider.ts

@@ -0,0 +1,61 @@
+import {
+    CancellationToken,
+    Event,
+    EventEmitter,
+    ExtensionContext,
+    FileDecoration,
+    FileDecorationProvider,
+    ThemeColor,
+    Uri,
+    window,
+} from "vscode";
+
+export class DecorationProvider implements FileDecorationProvider {
+    private readonly _onDidChangeFileDecorations: EventEmitter<Uri | Uri[]> =
+        new EventEmitter<Uri | Uri[]>();
+    readonly onDidChangeFileDecorations: Event<Uri | Uri[]> =
+        this._onDidChangeFileDecorations.event;
+    public markedFiles: Set<string> = new Set<string>();
+    private static instance: DecorationProvider;
+
+    constructor(context: ExtensionContext) {
+        // 注册文件装饰提供者
+        context.subscriptions.push(
+            window.registerFileDecorationProvider(this),
+        );
+
+        DecorationProvider.instance = this;
+    }
+
+    static getInstance(): DecorationProvider {
+        return DecorationProvider.instance;
+    }
+
+    provideFileDecoration(uri: Uri, token: CancellationToken): FileDecoration | undefined {
+        if (token.isCancellationRequested) {
+            return;
+        }
+
+        if (this.markedFiles.has(uri.fsPath)) {
+            return {
+                propagate: true,
+                badge: "📌",
+                color: new ThemeColor("terminal.ansiCyan"),
+            };
+        }
+    }
+
+    public async markFile(uri: Uri) {
+        if (!this.markedFiles.has(uri.fsPath)) {
+            this.markedFiles.add(uri.fsPath);
+            this._onDidChangeFileDecorations.fire(uri);
+        }
+    }
+
+    public async unmarkFile(uri: Uri) {
+        if (this.markedFiles.has(uri.fsPath)) {
+            this.markedFiles.delete(uri.fsPath);
+            this._onDidChangeFileDecorations.fire(uri);
+        }
+    }
+}

src/project/tree.ts

@@ -2,6 +2,7 @@ import path from 'path';
 import * as vscode from 'vscode';
 import fs from 'fs';
 import { getWorkspaceFolder, getExtensionPath } from '../api';
+import { DecorationProvider } from './fileDecorationProvider';
 
 /*
  * contexType -> contextValue as following value:
@@ -11,6 +12,12 @@ import { getWorkspaceFolder, getExtensionPath } from '../api';
  * project_bsp
  */
 
+// 全局变量记录当前选中的project_bsp项目
+let currentSelectedBspItem: ProjectTreeItem | null = null;
+
+// 添加树视图刷新事件发射器
+let _onDidChangeTreeData: vscode.EventEmitter<ProjectTreeItem | undefined> | null = null;
+
 export class ProjectTreeItem extends vscode.TreeItem {
     children: ProjectTreeItem[];
     fn: string = '';
@@ -58,12 +65,16 @@ export class ProjectTreeItem extends vscode.TreeItem {
             else if (contextType == 'project_bsp') {
                 this.command = {
                     title: this.name,
-                    command: 'extension.switchProject',
+                    command: 'extension.handleTreeItemClick',
                     tooltip: this.name,
                     arguments: [
                         this
                     ]
                 };
+
+                // with resourceUri, the item will be rendered as a file icon
+                // then the item can be marked with a decoration
+                this.resourceUri = vscode.Uri.file(this.fn);
             }
         }
     }
@@ -333,7 +344,47 @@ export function initProjectTree(context:vscode.ExtensionContext) {
     vscode.commands.registerCommand('extension.openTerminalProject', (arg) => {
         openTerminalProject(arg);
     });
-    vscode.commands.registerCommand('extension.switchProject', (arg) => {
-        setCurrentProject(arg);
+
+    // Add double-clicked
+    let lastClickTime = 0;
+    let lastClickedItem: ProjectTreeItem | null = null;
+    
+    vscode.commands.registerCommand('extension.handleTreeItemClick', (item: ProjectTreeItem) => {
+        const currentTime = Date.now();
+        const doubleClickThreshold = 500; // 500ms内的两次点击被认为是双击
+        
+        if (item.contextType === 'project_bsp') {
+            if (lastClickedItem === item && (currentTime - lastClickTime) < doubleClickThreshold) {
+                if (currentSelectedBspItem && currentSelectedBspItem.fn === item.fn) {
+                    return;
+                }
+
+                // double clicked
+                if (currentSelectedBspItem && currentSelectedBspItem.fn != item.fn) {
+                    DecorationProvider.getInstance().unmarkFile(vscode.Uri.file(currentSelectedBspItem.fn));
+                }
+
+                currentSelectedBspItem = item;
+                DecorationProvider.getInstance().markFile(vscode.Uri.file(item.fn));
+                setCurrentProject(item);
+
+                if (_onDidChangeTreeData) {
+                    _onDidChangeTreeData.fire(undefined);
+                }
+                
+                // reset status
+                lastClickTime = 0;
+                lastClickedItem = null;
+            } else {
+                // one-clicked, just record it.
+                lastClickTime = currentTime;
+                lastClickedItem = item;
+            }
+        }
     });
 }
+
+// 导出函数用于设置树视图刷新事件发射器
+export function setTreeDataChangeEmitter(emitter: vscode.EventEmitter<ProjectTreeItem | undefined>) {
+    _onDidChangeTreeData = emitter;
+}