show editor highlights

Author: jrieken

src/calls/editorHighlights.ts

@@ -0,0 +1,54 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import { Call } from './model';
+import { TreeObject } from './provider';
+
+export class EditorHighlights {
+
+    private readonly _decorationType = vscode.window.createTextEditorDecorationType({
+        backgroundColor: new vscode.ThemeColor('editor.findMatchHighlightBackground'),
+        rangeBehavior: vscode.DecorationRangeBehavior.ClosedClosed,
+        overviewRulerLane: vscode.OverviewRulerLane.Center,
+        overviewRulerColor: new vscode.ThemeColor('editor.findMatchHighlightBackground'),
+    });
+
+    constructor(readonly view: vscode.TreeView<TreeObject>) { }
+
+    show() {
+        const { activeTextEditor: editor } = vscode.window;
+        if (!editor) {
+            return;
+        }
+        const [sel] = this.view.selection;
+        if (!(sel instanceof Call)) {
+            return
+        }
+        const call = sel;
+
+        if (call.locations) {
+            const ranges: vscode.Range[] = [];
+            for (const loc of call.locations) {
+                if (loc.uri.toString() === editor.document.uri.toString()) {
+                    ranges.push(loc.range);
+                }
+            }
+            editor.setDecorations(this._decorationType, ranges);
+        }
+    }
+
+    hide() {
+        const { activeTextEditor: editor } = vscode.window;
+        if (editor) {
+            editor.setDecorations(this._decorationType, []);
+        }
+    }
+
+    refresh() {
+        this.hide();
+        this.show();
+    }
+}

src/calls/index.ts

@@ -7,15 +7,21 @@ import * as vscode from 'vscode';
 import { CallsDirection, CallsModel, Call } from './model';
 import { DataProvider } from './provider';
 import { History, HistoryItem } from './history';
+import { EditorHighlights } from './editorHighlights';
 
 export function register(disposables: vscode.Disposable[]) {
 
     const viewId = 'calls-view.tree';
     const history = new History();
     const provider = new DataProvider(history);
-
     const view = vscode.window.createTreeView(viewId, { treeDataProvider: provider });
 
+    // highlight management
+    const highlights = new EditorHighlights(view);
+    vscode.window.onDidChangeActiveTextEditor(() => view.visible && highlights.refresh(), undefined, disposables);
+    view.onDidChangeVisibility(e => e.visible ? highlights.show() : highlights.hide(), undefined, disposables);
+    view.onDidChangeSelection(() => highlights.refresh(), undefined, disposables);
+
     let callsDirection = CallsDirection.Outgoing;
     vscode.commands.executeCommand('setContext', 'calls-view.mode', 'showOutgoing');
 
@@ -72,6 +78,7 @@ export function register(disposables: vscode.Disposable[]) {
 
     const clearCommand = () => {
         updateModel(undefined);
+        highlights.hide();
         if (history.items.length === 0) {
             view.message = `To populate this view, open an editor and run the 'Show Call Hierarchy'-command.`;
         } else {

src/calls/model.ts

@@ -12,7 +12,7 @@ export const enum CallsDirection {
 
 export class Call {
 
-    constructor(readonly item: vscode.CallHierarchyItem, readonly parent: Call | undefined, readonly location: vscode.Location[] | undefined) { }
+    constructor(readonly item: vscode.CallHierarchyItem, readonly parent: Call | undefined, readonly locations: vscode.Location[] | undefined) { }
 }
 
 export class CallsModel {