Show custom diagnostics as TSTL<code> in CLI

ark120202 · ark120202 · commit f78e0d04690c · 2019-04-30T02:47:27.000+05:00
diff --git a/src/tstl.ts b/src/tstl.ts
@@ -6,7 +6,14 @@ import * as CommandLineParser from "./CommandLineParser";
 import * as cliDiagnostics from "./diagnostics";
 
 function createDiagnosticReporter(pretty: boolean): ts.DiagnosticReporter {
-    return (ts as any).createDiagnosticReporter(ts.sys, pretty);
+    const reporter: ts.DiagnosticReporter = (ts as any).createDiagnosticReporter(ts.sys, pretty);
+    return diagnostic => {
+        if (diagnostic.source === "typescript-to-lua") {
+            diagnostic = { ...diagnostic, code: ("TL" + diagnostic.code) as any };
+        }
+
+        reporter(diagnostic);
+    };
 }
 
 function createWatchStatusReporter(options?: ts.CompilerOptions): ts.WatchStatusReporter {
