diff --git a/src/OneExplorer/OneExplorer.ts b/src/OneExplorer/OneExplorer.ts index 082ee109..72761e39 100644 --- a/src/OneExplorer/OneExplorer.ts +++ b/src/OneExplorer/OneExplorer.ts @@ -22,6 +22,7 @@ import * as vscode from "vscode"; import { obtainWorkspaceRoots } from "../Utils/Helpers"; import { Logger } from "../Utils/Logger"; +import { BackendContext } from "../Backend/API"; import { ConfigObj } from "./ConfigObject"; import { ArtifactAttr } from "./ArtifactLocator"; import { OneStorage } from "./OneStorage"; @@ -214,15 +215,10 @@ class NodeFactory { "Config nodes cannot have attributes" ); const ext = path.extname(fpath); - switch (ext) { - case ".edgetpucfg": { - node = new ConfigNode(uri, parent, "one.editor.edgetpucfg"); - break; - } - case ".cfg": - default: { - node = new ConfigNode(uri, parent); - } + if (BackendContext.isRegistered("EdgeTPU") && ext === ".edgetpucfg") { + node = new ConfigNode(uri, parent, "one.editor.edgetpucfg"); + } else { + node = new ConfigNode(uri, parent); } break; } diff --git a/src/OneExplorer/OneStorage.ts b/src/OneExplorer/OneStorage.ts index 1e2b5b8d..e5c1eeb8 100644 --- a/src/OneExplorer/OneStorage.ts +++ b/src/OneExplorer/OneStorage.ts @@ -24,6 +24,7 @@ import { Logger } from "../Utils/Logger"; import { ConfigObj } from "./ConfigObject"; import { Node, NodeType } from "./OneExplorer"; +import { BackendContext } from "../Backend/API"; export { CfgToCfgObjMap as _unit_test_CfgToCfgObjMap }; @@ -208,7 +209,9 @@ export class OneStorage { }; try { - const extList = [".cfg", ".edgetpucfg"]; + const extList = BackendContext.isRegistered("EdgeTPU") + ? [".cfg", ".edgetpucfg"] + : [".cfg"]; return roots .map((root) =>