Set up warning and error diagnostics better

Mermade · Feb 18, 2020 · 0d5d3ea · 0d5d3ea
1 parent 895fb97
commit 0d5d3ea
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/extension.js b/extension.js
@@ -145,9 +145,14 @@ function validate(lint, resolve) {
             dc.delete(editor.document.uri);
             const diagnostics = [];
             let range; // TODO
-            diagnostics.push(new vscode.Diagnostic(range, ex.message, vscode.DiagnosticSeverity.Error));
+            let error = new vscode.Diagnostic(range, ex.message, vscode.DiagnosticSeverity.Error);
+            error.source = 'openapi-lint';
+            diagnostics.push(error);
             for (let warning of options.warnings||[]) {
-                diagnostics.push(new vscode.Diagnostic(range, warning.message + ' ' + warning.ruleName, vscode.DiagnosticSeverity.Warning));
+                let warn = new vscode.Diagnostic(range, warning.message, vscode.DiagnosticSeverity.Warning);
+                warn.source = 'openapi-lint';
+                warn.code = warning.ruleName
+                diagnostics.push(warn);
             }
             dc.set(editor.document.uri, diagnostics);
 	    });