@zenstackhq/cli 3.1.1 → 3.2.0

package/dist/index.cjs

@@ -28,7 +28,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 ));
 
 // src/index.ts
-var import_language3 = require("@zenstackhq/language");
+var import_language4 = require("@zenstackhq/language");
 var import_colors9 = __toESM(require("colors"), 1);
 var import_commander = require("commander");
 
@@ -289,6 +289,7 @@ __name(run3, "run");
 
 // src/actions/generate.ts
 var import_common_helpers = require("@zenstackhq/common-helpers");
+var import_language3 = require("@zenstackhq/language");
 var import_ast2 = require("@zenstackhq/language/ast");
 var import_utils = require("@zenstackhq/language/utils");
 var import_colors4 = __toESM(require("colors"), 1);
@@ -296,6 +297,7 @@ var import_jiti = require("jiti");
 var import_node_fs6 = __toESM(require("fs"), 1);
 var import_node_path4 = __toESM(require("path"), 1);
 var import_node_url = require("url");
+var import_chokidar = require("chokidar");
 var import_ora = __toESM(require("ora"), 1);
 
 // src/plugins/index.ts
@@ -363,6 +365,81 @@ var typescript_default = plugin2;
 
 // src/actions/generate.ts
 async function run4(options) {
+  const model = await pureGenerate(options, false);
+  if (options.watch) {
+    const logsEnabled = !options.silent;
+    if (logsEnabled) {
+      console.log(import_colors4.default.green(`
+Enabled watch mode!`));
+    }
+    const schemaExtensions = import_language3.ZModelLanguageMetaData.fileExtensions;
+    const getRootModelWatchPaths = /* @__PURE__ */ __name((model2) => new Set(model2.declarations.filter((v) => v.$cstNode?.parent?.element.$type === "Model" && !!v.$cstNode.parent.element.$document?.uri?.fsPath).map((v) => v.$cstNode.parent.element.$document.uri.fsPath)), "getRootModelWatchPaths");
+    const watchedPaths = getRootModelWatchPaths(model);
+    if (logsEnabled) {
+      const logPaths = [
+        ...watchedPaths
+      ].map((at) => `- ${at}`).join("\n");
+      console.log(`Watched file paths:
+${logPaths}`);
+    }
+    const watcher = (0, import_chokidar.watch)([
+      ...watchedPaths
+    ], {
+      alwaysStat: false,
+      ignoreInitial: true,
+      ignorePermissionErrors: true,
+      ignored: /* @__PURE__ */ __name((at) => !schemaExtensions.some((ext) => at.endsWith(ext)), "ignored")
+    });
+    const reGenerateSchema = (0, import_common_helpers.singleDebounce)(async () => {
+      if (logsEnabled) {
+        console.log("Got changes, run generation!");
+      }
+      try {
+        const newModel = await pureGenerate(options, true);
+        const allModelsPaths = getRootModelWatchPaths(newModel);
+        const newModelPaths = [
+          ...allModelsPaths
+        ].filter((at) => !watchedPaths.has(at));
+        const removeModelPaths = [
+          ...watchedPaths
+        ].filter((at) => !allModelsPaths.has(at));
+        if (newModelPaths.length) {
+          if (logsEnabled) {
+            const logPaths = newModelPaths.map((at) => `- ${at}`).join("\n");
+            console.log(`Added file(s) to watch:
+${logPaths}`);
+          }
+          newModelPaths.forEach((at) => watchedPaths.add(at));
+          watcher.add(newModelPaths);
+        }
+        if (removeModelPaths.length) {
+          if (logsEnabled) {
+            const logPaths = removeModelPaths.map((at) => `- ${at}`).join("\n");
+            console.log(`Removed file(s) from watch:
+${logPaths}`);
+          }
+          removeModelPaths.forEach((at) => watchedPaths.delete(at));
+          watcher.unwatch(removeModelPaths);
+        }
+      } catch (e) {
+        console.error(e);
+      }
+    }, 500, true);
+    watcher.on("unlink", (pathAt) => {
+      if (logsEnabled) {
+        console.log(`Removed file from watch: ${pathAt}`);
+      }
+      watchedPaths.delete(pathAt);
+      watcher.unwatch(pathAt);
+      reGenerateSchema();
+    });
+    watcher.on("change", () => {
+      reGenerateSchema();
+    });
+  }
+}
+__name(run4, "run");
+async function pureGenerate(options, fromWatch) {
   const start = Date.now();
   const schemaFile = getSchemaFile(options.schema);
   const model = await loadSchemaDocument(schemaFile);
@@ -371,7 +448,8 @@ async function run4(options) {
   if (!options.silent) {
     console.log(import_colors4.default.green(`Generation completed successfully in ${Date.now() - start}ms.
 `));
-    console.log(`You can now create a ZenStack client with it.
+    if (!fromWatch) {
+      console.log(`You can now create a ZenStack client with it.
 
 \`\`\`ts
 import { ZenStackClient } from '@zenstackhq/orm';
@@ -383,9 +461,11 @@ const client = new ZenStackClient(schema, {
 \`\`\`
 
 Check documentation: https://zenstack.dev/docs/`);
+    }
   }
+  return model;
 }
-__name(run4, "run");
+__name(pureGenerate, "pureGenerate");
 function getOutputPath(options, schemaFile) {
   if (options.output) {
     return options.output;
@@ -1165,13 +1245,13 @@ var seedAction = /* @__PURE__ */ __name(async (options, args) => {
 }, "seedAction");
 function createProgram() {
   const program = new import_commander.Command("zen").alias("zenstack").helpOption("-h, --help", "Show this help message").version(getVersion(), "-v --version", "Show CLI version");
-  const schemaExtensions = import_language3.ZModelLanguageMetaData.fileExtensions.join(", ");
+  const schemaExtensions = import_language4.ZModelLanguageMetaData.fileExtensions.join(", ");
   program.description(`${import_colors9.default.bold.blue("\u03B6")} ZenStack is the modern data layer for TypeScript apps.
 
 Documentation: https://zenstack.dev/docs`).showHelpAfterError().showSuggestionAfterError();
   const schemaOption = new import_commander.Option("--schema <file>", `schema file (with extension ${schemaExtensions}). Defaults to "zenstack/schema.zmodel" unless specified in package.json.`);
   const noVersionCheckOption = new import_commander.Option("--no-version-check", "do not check for new version");
-  program.command("generate").description("Run code generation plugins").addOption(schemaOption).addOption(noVersionCheckOption).addOption(new import_commander.Option("-o, --output <path>", "default output directory for code generation")).addOption(new import_commander.Option("--lite", "also generate a lite version of schema without attributes").default(false)).addOption(new import_commander.Option("--lite-only", "only generate lite version of schema without attributes").default(false)).addOption(new import_commander.Option("--silent", "suppress all output except errors").default(false)).action(generateAction);
+  program.command("generate").description("Run code generation plugins").addOption(schemaOption).addOption(noVersionCheckOption).addOption(new import_commander.Option("-o, --output <path>", "default output directory for code generation")).addOption(new import_commander.Option("-w, --watch", "enable watch mode").default(false)).addOption(new import_commander.Option("--lite", "also generate a lite version of schema without attributes").default(false)).addOption(new import_commander.Option("--lite-only", "only generate lite version of schema without attributes").default(false)).addOption(new import_commander.Option("--silent", "suppress all output except errors").default(false)).action(generateAction);
   const migrateCommand = program.command("migrate").description("Run database schema migration related tasks.");
   const migrationsOption = new import_commander.Option("--migrations <path>", 'path that contains the "migrations" directory');
   migrateCommand.command("dev").addOption(schemaOption).addOption(noVersionCheckOption).addOption(new import_commander.Option("-n, --name <name>", "migration name")).addOption(new import_commander.Option("--create-only", "only create migration, do not apply")).addOption(migrationsOption).description("Create a migration from changes in schema and apply it to the database").action((options) => migrateAction("dev", options));
@@ -1223,6 +1303,9 @@ async function main() {
       exitCode = 1;
     }
   }
+  if (program.args.includes("generate") && (program.args.includes("-w") || program.args.includes("--watch"))) {
+    return;
+  }
   if (telemetry.isTracking) {
     setTimeout(() => {
       process.exit(exitCode);