fix: make highlights consistent (#6)

Author: Jujuyeh

package-lock.json

@@ -3,7 +3,7 @@
   "publisher": "sysdig",
   "displayName": "Sysdig Scanner",
   "description": "Sysdig Scanner for Visual Studio Code",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "icon": "img/logo.png",
   "repository": "https://github.com/sysdiglabs/vscode-extension",
   "engines": {

package.json

@@ -4,7 +4,7 @@ import * as fs from 'fs';
 import { Report } from '../types';
 import { outputChannel, vulnTreeDataProvider, policyTreeDataProvider } from '../extension';
 import { checkInternetConnectivity } from '../utils/connectivity';
-import { highlightImage } from '../fileScanners';
+import { clearDecorations, highlightImage } from '../fileScanners';
 import { Z_VERSION_ERROR } from 'zlib';
 
 const VM_SCAN_FILE = 'vm_scan.json';
@@ -167,6 +167,7 @@ function updateVulnerabilities(statusBar: vscode.StatusBarItem, report: Report,
     vulnTreeDataProvider.updateVulnTree(report.result.packages, report.info.resultUrl, report.result.layers, source, range);
     policyTreeDataProvider.addPolicies(report.result.policyEvaluations || []);
     if (source) {
+        clearDecorations(source);
         highlightImage(report, "", source, range);
     }
 }

src/runners/vmScanRunner.ts

@@ -22,6 +22,11 @@ function traverseNode(node: yaml_ast.YAMLNode, path: string[], imageInfos: Image
     for (const mapping of map.mappings) {
       const keyNode = mapping.key;
       const valueNode = mapping.value;
+
+      if (!keyNode || !valueNode) {
+        continue;
+      }
+
       if (keyNode.value === 'image' && valueNode.kind === yaml_ast.Kind.SCALAR) {
         const imageInfo: ImageInfo = {
           image: valueNode.value,