From 84beaf8acc2b27c1e95371a3ef798b5b7ec1b792 Mon Sep 17 00:00:00 2001
From: leowrites <leo@liuwork.ca>
Date: Tue, 10 Sep 2024 09:45:01 -0400
Subject: [PATCH] Refactor and throw errors when the path is a directory

---
 memory-viz/bin/cli.js | 47 +++++++++++++++----------------------------
 1 file changed, 16 insertions(+), 31 deletions(-)

diff --git a/memory-viz/bin/cli.js b/memory-viz/bin/cli.js
index 7997aa2b..e9ba6b2f 100755
--- a/memory-viz/bin/cli.js
+++ b/memory-viz/bin/cli.js
@@ -7,41 +7,33 @@ const { program } = require("commander");
 const { json } = require("node:stream/consumers");
 
 function parseFilePath(input) {
-    if (input) {
-        return pathExists(input, `File`);
-    } else {
-        return undefined;
+    const fullPath = path.resolve(process.cwd(), input);
+    if (fs.existsSync(fullPath, "File")) {
+        return fullPath;
     }
+    console.error(`Error: File ${fullPath} does not exist.`);
+    process.exit(1);
 }
 
 function parseOutputPath(input) {
-    let outputDir, filename;
     if (isPathDirectory(input)) {
-        outputDir = input;
-        filename = "memory-viz.svg";
-    } else {
-        const parsedPath = path.parse(input);
-        outputDir = parsedPath.dir;
-        filename = `${parsedPath.name}.svg`;
+        console.error(`Error: Output path must be a file`);
+        process.exit(1);
+    }
+    const parsedPath = path.parse(input);
+    const outputDir = parsedPath.dir;
+    const filename = `${parsedPath.name}.svg`;
+    if (!fs.existsSync(outputDir)) {
+        console.error(`Error: Output directory does not exist`);
+        process.exit(1);
     }
-    return { outputDir, filename };
+    return path.join(outputDir, filename);
 }
 
 function isPathDirectory(rawOutputPath) {
     return rawOutputPath.slice(-1) == "/";
 }
 
-// helper function for parsing paths
-function pathExists(inputPath, errMsg) {
-    const fullPath = path.resolve(process.cwd(), inputPath);
-    if (!fs.existsSync(fullPath)) {
-        console.error(`Error: ${errMsg} ${fullPath} does not exist.`);
-        process.exit(1);
-    } else {
-        return fullPath;
-    }
-}
-
 function parseRoughjsConfig(input) {
     const pairs = input.split(",");
     const keyValuePairs = pairs.map((pair) => pair.split("="));
@@ -113,14 +105,7 @@ function runMemoryViz(jsonContent) {
 
     try {
         if (options.output) {
-            const { outputDir, filename } = options.output;
-
-            if (!fs.existsSync(outputDir)) {
-                fs.mkdirSync(outputDir, { recursive: true });
-            }
-
-            const outputPath = path.join(outputDir, filename);
-            m.save(outputPath);
+            m.save(options.output);
         } else {
             m.save();
         }