tla-ts 0.0.1 → 0.0.2

package/src/errors.ts

@@ -1,3 +1,5 @@
+import { EXIT_STATUS, getExitCode, setModuleExitCode } from "./exit-codes";
+
 export class TlaError extends Error {
   constructor(message: string, cause?: unknown) {
     super(message, { cause });
@@ -26,9 +28,29 @@ export class TlaEvaluationError extends TlaError {
   }
 }
 
+// An evaluation error that TLC raises as an EvalException from a standard module's Java code. It
+// keeps the name TlaEvaluationError, and index.ts doesn't export it.
+export class TlaModuleError extends TlaEvaluationError {
+  constructor(message: string, exitCode: number = EXIT_STATUS.error) {
+    super(message);
+    this.name = "TlaEvaluationError";
+    setModuleExitCode(this, exitCode);
+  }
+}
+
 export class TlaUnsupportedError extends TlaError {
   constructor(message: string, cause?: unknown) {
     super(message, cause);
     this.name = "TlaUnsupportedError";
   }
 }
+
+export const getErrorExitCode = (error: TlaError) =>
+  getExitCode(error) ??
+  (error instanceof TlaSyntaxError
+    ? EXIT_STATUS.specParseError
+    : error instanceof TlaConfigError
+      ? EXIT_STATUS.configParseError
+      : error instanceof TlaEvaluationError
+        ? EXIT_STATUS.specEvaluationFailure
+        : EXIT_STATUS.error);