@@ -12,6 +12,7 @@ import { ExplainNode, ExplainTree } from "./explain/nodes";
1212import { DoveResultsView , ExplainTreeItem } from "./explain/doveResultsView" ;
1313import { DoveNodeView , PropertyNode } from "./explain/doveNodeView" ;
1414import { ResultSetPanelProvider , SqlParameter } from "./resultSetPanelProvider" ;
15+ import { DoveTreeDecorationProvider } from "./explain/doveTreeDecorationProvider" ;
1516import { generateSqlForAdvisedIndexes } from "./explain/advice" ;
1617import { updateStatusBar } from "../jobManager/statusBar" ;
1718import { DbCache } from "../../language/providers/logic/cache" ;
@@ -57,6 +58,7 @@ let doveResultsView = new DoveResultsView();
5758let doveResultsTreeView : TreeView < ExplainTreeItem > = doveResultsView . getTreeView ( ) ;
5859let doveNodeView = new DoveNodeView ( ) ;
5960let doveNodeTreeView : TreeView < PropertyNode > = doveNodeView . getTreeView ( ) ;
61+ let doveTreeDecorationProvider = new DoveTreeDecorationProvider ( ) ; // Self-registers as a tree decoration providor
6062
6163export function initialise ( context : vscode . ExtensionContext ) {
6264 setCancelButtonVisibility ( false ) ;
@@ -408,6 +410,9 @@ async function runHandler(options?: StatementInfo) {
408410
409411 explainTree = new ExplainTree ( explained . vedata ) ;
410412 const topLevel = explainTree . get ( ) ;
413+ const rootNode = doveResultsView . setRootNode ( topLevel ) ;
414+ doveNodeView . setNode ( rootNode . explainNode ) ;
415+ doveTreeDecorationProvider . updateTreeItems ( rootNode ) ;
411416
412417 const graph = new CytoscapeGraph ( ) ;
413418
0 commit comments