Convert scripts to TS

Author: mahmoud-elmorabea

package.json

@@ -6,6 +6,7 @@
   "scripts": {
     "build": "tsc",
     "buildJson": "npx tsx scripts/generate-json-definitions.ts",
+    "buildDiagrams": "npx tsx scripts/generate-diagrams.ts",
     "validateGeneration": "npx tsx scripts/validate-json-generated.ts",
     "test": "jest"
   },

scripts/generate-diagrams.ts

@@ -0,0 +1,74 @@
+import fs from "fs";
+import path from "path";
+import { fileURLToPath } from "url";
+import { walkDir, isJsonFile, getMermaidOutputPath } from "./utils";
+import logger from "./log-styling";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const INPUT_DIR = path.resolve(__dirname, "../generated/json");
+const OUTPUT_DIR = path.resolve(__dirname, "../generated/mermaid");
+
+type Event = {
+  id: string;
+  label: string;
+  next?: string;
+  success?: string;
+  error?: string;
+};
+
+function toMermaid(events: Event[]): string {
+  const lines: string[] = ["graph TD"];
+
+  for (const event of events) {
+    const { id, label, next, success, error } = event;
+
+    lines.push(`${id}["${label}"]`);
+    if (next) {
+      lines.push(`${id} --> ${next}`);
+    }
+    if (success) {
+      lines.push(`${id} -->|Success| ${success}`);
+    }
+    if (error) {
+      lines.push(`${id} -->|Error| ${error}`);
+    }
+  }
+
+  return lines.join("\n");
+}
+
+function generateDiagram(filePath: string): void {
+  const relativePath = path.relative(INPUT_DIR, filePath);
+  logger.info(`Generating diagram for: ${relativePath}`);
+
+  const outputFile = getMermaidOutputPath(filePath, INPUT_DIR, OUTPUT_DIR);
+  fs.mkdirSync(path.dirname(outputFile), { recursive: true });
+
+  const events: Event[] = JSON.parse(fs.readFileSync(filePath, "utf-8"));
+  const output = toMermaid(events);
+
+  fs.writeFileSync(outputFile, output);
+}
+
+if (!fs.existsSync(INPUT_DIR)) {
+  logger.warn(`Generated directory is missing: ${INPUT_DIR}`);
+  process.exit(0);
+}
+
+logger.heading("Generating Mermaid diagrams from JSON files...");
+logger.blank();
+
+walkDir(INPUT_DIR, (filePath: string) => {
+  if (isJsonFile(filePath)) {
+    generateDiagram(filePath);
+  } else {
+    logger.warn(
+      `Skipping unsupported file: ${path.relative(INPUT_DIR, filePath)}`
+    );
+  }
+});
+
+logger.blank();
+logger.success("Mermaid diagrams generated successfully!");

scripts/generate-json-definitions.ts

@@ -6,7 +6,7 @@ import {
   walkDir,
   isJsonnetFile,
   isLibSonnetFile,
-  getOutputPath,
+  getJsonOutputPath,
 } from "./utils";
 import logger from "./log-styling";
 
@@ -18,7 +18,7 @@ const OUTPUT_DIR = path.resolve(__dirname, "../generated/json");
 
 function buildJsonnetFile(inputPath: string): void {
   const relativePath = path.relative(INPUT_DIR, inputPath);
-  const outputFile = getOutputPath(inputPath, INPUT_DIR, OUTPUT_DIR);
+  const outputFile = getJsonOutputPath(inputPath, INPUT_DIR, OUTPUT_DIR);
 
   fs.mkdirSync(path.dirname(outputFile), { recursive: true });
 

scripts/utils.ts

@@ -29,14 +29,33 @@ export function isLibSonnetFile(file: string) {
   return file.endsWith(".libsonnet");
 }
 
+/**
+ * Determines if a file is a valid JSON file
+ */
+export function isJsonFile(file: string) {
+  return file.endsWith(".json");
+}
+
 /**
  * Converts a .jsonnet input path to a .json output path.
  */
-export function getOutputPath(
+export function getJsonOutputPath(
   inputPath: string,
   inputRoot: string,
   outputRoot: string
 ) {
   const relativePath = path.relative(inputRoot, inputPath);
   return path.join(outputRoot, relativePath.replace(/\.jsonnet$/, ".json"));
 }
+
+/**
+ * Converts a .json input path to a .mmd output path.
+ */
+export function getMermaidOutputPath(
+  inputPath: string,
+  inputRoot: string,
+  outputRoot: string
+) {
+  const relativePath = path.relative(inputRoot, inputPath);
+  return path.join(outputRoot, relativePath.replace(/\.json$/, ".mmd"));
+}

scripts/validate-json-generated.ts

@@ -2,7 +2,7 @@ import fs from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
 import logger from "./log-styling";
-import { walkDir, isJsonnetFile, getOutputPath } from "./utils";
+import { walkDir, isJsonnetFile, getJsonOutputPath } from "./utils";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
@@ -20,7 +20,7 @@ const missingFiles: string[] = [];
 walkDir(INPUT_DIR, (inputPath: string) => {
   if (isJsonnetFile(inputPath)) {
     const relativePath = path.relative(INPUT_DIR, inputPath);
-    const expectedOutputPath = getOutputPath(inputPath, INPUT_DIR, OUTPUT_DIR);
+    const expectedOutputPath = getJsonOutputPath(inputPath, INPUT_DIR, OUTPUT_DIR);
 
     if (!fs.existsSync(expectedOutputPath)) {
       missingFiles.push(relativePath);