drizzle-docs-generator 0.1.0 → 0.2.0

package/README.ja.md

@@ -5,6 +5,14 @@
 
 Drizzle ORM スキーマから DBML を生成する CLI。JSDoc コメントを Note 句として出力できる。
 
+**✨ 機能:**
+
+- 📁 **ディレクトリインポート対応**: ディレクトリ内のすべてのスキーマファイルを自動インポート
+- 🔄 **拡張子不要**: 拡張子なしのインポートに対応 (例: `import { users } from './users'`)
+- 📝 **JSDoc コメント**: 自動的に DBML の Note 句に変換
+- 🔗 **リレーション対応**: `relations()` または `defineRelations()` から参照を生成
+- 👀 **Watch モード**: ファイル変更時に自動再生成
+
 [English README](./README.md)
 
 ## インストール
@@ -18,9 +26,12 @@ pnpm add -g drizzle-docs-generator
 ## 使い方
 
 ```bash
-# 基本
+# 基本 - 単一ファイル
 drizzle-docs generate ./src/db/schema.ts -d postgresql
 
+# ディレクトリ - ディレクトリ内のすべてのスキーマファイルをインポート
+drizzle-docs generate ./src/db/schema/ -d postgresql
+
 # ファイル出力
 drizzle-docs generate ./src/db/schema.ts -d postgresql -o schema.dbml
 
@@ -83,6 +94,17 @@ const dbml = pgGenerate({
 
 - Node.js >= 24
 - Drizzle ORM v1 beta (1.0.0-beta.10+)
+- ES Modules (ESM): プロジェクトで ESM を使用していること (`package.json` に `"type": "module"`)
+
+## 仕組み
+
+このツールは [tsx](https://github.com/privatenumber/tsx) を使用してスキーマファイルを読み込むため:
+
+✅ **拡張子なしのインポートが動作**: `import { users } from './users'`
+✅ **TypeScript ファイルを直接読み込み**: コンパイル不要
+✅ **ディレクトリインポート**: ディレクトリ内のすべてのスキーマファイルを自動読み込み
+
+スキーマファイルでは、ファイル拡張子を気にせず標準的な TypeScript のモジュール解決を使用できます。
 
 ## License
 

package/README.md

@@ -5,6 +5,14 @@
 
 CLI tool to generate DBML from Drizzle ORM schemas. Extracts JSDoc comments and outputs them as Note clauses.
 
+**✨ Features:**
+
+- 📁 **Directory Import Support**: Import all schema files from a directory
+- 🔄 **No File Extension Required**: Works with extensionless imports (e.g., `import { users } from './users'`)
+- 📝 **JSDoc Comments**: Automatically extracts and converts to DBML Notes
+- 🔗 **Relations Support**: Generate refs from `relations()` or `defineRelations()`
+- 👀 **Watch Mode**: Auto-regenerate on file changes
+
 [日本語版READMEはこちら](./README.ja.md)
 
 ## Install
@@ -18,9 +26,12 @@ pnpm add -g drizzle-docs-generator
 ## Usage
 
 ```bash
-# Basic
+# Basic - single file
 drizzle-docs generate ./src/db/schema.ts -d postgresql
 
+# Directory - import all schema files from directory
+drizzle-docs generate ./src/db/schema/ -d postgresql
+
 # Output to file
 drizzle-docs generate ./src/db/schema.ts -d postgresql -o schema.dbml
 
@@ -83,6 +94,17 @@ const dbml = pgGenerate({
 
 - Node.js >= 24
 - Drizzle ORM v1 beta (1.0.0-beta.10+)
+- ES Modules (ESM): Your project must use ESM (`"type": "module"` in package.json)
+
+## How It Works
+
+This tool uses [tsx](https://github.com/privatenumber/tsx) to load your schema files, which means:
+
+✅ **Extensionless imports work**: `import { users } from './users'`
+✅ **TypeScript files are loaded directly**: No need to compile first
+✅ **Directory imports**: Load all schema files from a directory automatically
+
+Your schema files can use standard TypeScript module resolution without worrying about file extensions.
 
 ## License
 

package/dist/cli/index.js

@@ -1,69 +1,110 @@
 #!/usr/bin/env node
-import { Command as m } from "commander";
-import { readFileSync as u, existsSync as f, watch as d } from "node:fs";
-import { join as g, resolve as i } from "node:path";
-import { pathToFileURL as p } from "node:url";
-import { pgGenerate as h } from "../generator/pg.js";
-import { mysqlGenerate as w } from "../generator/mysql.js";
-import { sqliteGenerate as y } from "../generator/sqlite.js";
+import { Command as y } from "commander";
+import { readFileSync as w, watch as $, existsSync as D, lstatSync as E, readdirSync as q } from "node:fs";
+import { join as m, resolve as f } from "node:path";
+import { pathToFileURL as u } from "node:url";
+import { pgGenerate as b } from "../generator/pg.js";
+import { mysqlGenerate as S } from "../generator/mysql.js";
+import { sqliteGenerate as z } from "../generator/sqlite.js";
 import "drizzle-orm";
 import "drizzle-orm/pg-core";
 import "drizzle-orm/mysql-core";
 import "drizzle-orm/sqlite-core";
 import "typescript";
-const c = new m(), q = g(import.meta.dirname, "../../package.json"), s = JSON.parse(u(q, "utf-8"));
-c.name("drizzle-docs").description(s.description).version(s.version);
-function D(o) {
-  switch (o) {
+import { register as M } from "tsx/esm/api";
+const j = M(), i = new y(), v = m(import.meta.dirname, "../../package.json"), l = JSON.parse(w(v, "utf-8"));
+i.name("drizzle-docs").description(l.description).version(l.version);
+function h(t) {
+  switch (t) {
     case "mysql":
-      return w;
+      return S;
     case "sqlite":
-      return y;
+      return z;
     default:
-      return h;
+      return b;
   }
 }
-async function l(o, e) {
-  const r = p(o).href, t = e.watch ? `?t=${Date.now()}` : "", n = await import(r + t);
-  return D(e.dialect)({
+async function x(t, e) {
+  const o = u(t).href, r = e.watch ? `?t=${Date.now()}` : "", n = await import(o + r);
+  return h(e.dialect)({
     schema: n,
     out: e.output,
     relational: e.relational,
-    source: o
+    source: t
   });
 }
-async function $(o, e) {
-  const r = i(process.cwd(), o);
-  f(r) || (console.error(`Error: Schema file not found: ${r}`), process.exit(1));
+function d(t) {
+  const e = [];
   try {
-    const t = await l(r, e);
-    !e.output && t ? console.log(t) : e.output && console.log(`DBML generated: ${e.output}`);
-  } catch (t) {
-    t instanceof Error ? console.error(`Error generating DBML: ${t.message}`) : console.error("Error generating DBML:", t), process.exit(1);
+    const o = q(t, { withFileTypes: !0 });
+    for (const r of o) {
+      const n = m(t, r.name);
+      r.isDirectory() ? e.push(...d(n)) : r.isFile() && /\.(ts|js|mts|mjs|cts|cjs)$/.test(r.name) && e.push(n);
+    }
+  } catch (o) {
+    o instanceof Error && console.error(`Error reading directory ${t}: ${o.message}`);
   }
+  return e;
 }
-function b(o, e) {
-  const r = i(process.cwd(), o);
-  console.log(`Watching for changes: ${r}`);
-  let t = null;
-  d(r, async (n) => {
-    n === "change" && (t && clearTimeout(t), t = setTimeout(async () => {
+function B(t) {
+  const e = f(process.cwd(), t);
+  if (D(e) || (console.error(`Error: Schema path not found: ${e}`), process.exit(1)), E(e).isDirectory()) {
+    const r = d(e);
+    return r.length === 0 && (console.error(`Error: No schema files found in directory: ${e}`), process.exit(1)), r;
+  }
+  return [e];
+}
+async function F(t, e) {
+  const o = B(t);
+  try {
+    const r = {};
+    for (const a of o) {
+      const g = u(a).href, p = e.watch ? `?t=${Date.now()}` : "";
+      try {
+        const c = await import(g + p);
+        Object.assign(r, c);
+      } catch (c) {
+        throw c instanceof Error && (console.error(`Error importing ${a}: ${c.message}`), console.error(`
+Possible causes:`), console.error("- Syntax error in the schema file"), console.error("- Missing dependencies (make sure drizzle-orm is installed)"), console.error("- Circular dependencies between schema files")), c;
+      }
+    }
+    const s = h(e.dialect)({
+      schema: r,
+      out: e.output,
+      relational: e.relational,
+      source: o[0]
+      // Use first file for source path
+    });
+    !e.output && s ? console.log(s) : e.output && console.log(`DBML generated: ${e.output}`);
+  } catch (r) {
+    r instanceof Error ? console.error(`Error generating DBML: ${r.message}`) : console.error("Error generating DBML:", r), process.exit(1);
+  }
+}
+function G(t, e) {
+  const o = f(process.cwd(), t);
+  console.log(`Watching for changes: ${o}`);
+  let r = null;
+  $(o, async (n) => {
+    n === "change" && (r && clearTimeout(r), r = setTimeout(async () => {
       console.log(`
 File changed, regenerating...`);
       try {
-        const a = await l(r, e);
-        !e.output && a ? console.log(a) : e.output && console.log(`DBML regenerated: ${e.output}`);
-      } catch (a) {
-        a instanceof Error ? console.error(`Error: ${a.message}`) : console.error("Error:", a);
+        const s = await x(o, e);
+        !e.output && s ? console.log(s) : e.output && console.log(`DBML regenerated: ${e.output}`);
+      } catch (s) {
+        s instanceof Error ? console.error(`Error: ${s.message}`) : console.error("Error:", s);
       }
     }, 100));
   });
 }
-c.command("generate").description("Generate DBML from Drizzle schema files").argument("<schema>", "Path to Drizzle schema file").option("-o, --output <file>", "Output file path").option("-d, --dialect <dialect>", "Database dialect (postgresql, mysql, sqlite)", "postgresql").option("-r, --relational", "Use relations() definitions instead of foreign keys").option("-w, --watch", "Watch for file changes and regenerate").action(async (o, e) => {
-  const r = ["postgresql", "mysql", "sqlite"];
-  r.includes(e.dialect) || (console.error(
-    `Error: Invalid dialect "${e.dialect}". Valid options: ${r.join(", ")}`
-  ), process.exit(1)), await $(o, e), e.watch && b(o, e);
+i.command("generate").description("Generate DBML from Drizzle schema files").argument("<schema>", "Path to Drizzle schema file or directory").option("-o, --output <file>", "Output file path").option("-d, --dialect <dialect>", "Database dialect (postgresql, mysql, sqlite)", "postgresql").option("-r, --relational", "Use relations() definitions instead of foreign keys").option("-w, --watch", "Watch for file changes and regenerate").action(async (t, e) => {
+  const o = ["postgresql", "mysql", "sqlite"];
+  o.includes(e.dialect) || (console.error(
+    `Error: Invalid dialect "${e.dialect}". Valid options: ${o.join(", ")}`
+  ), process.exit(1)), await F(t, e), e.watch && G(t, e);
+});
+i.parse();
+process.on("exit", () => {
+  j();
 });
-c.parse();
 //# sourceMappingURL=index.js.map

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\n/**\n * CLI for drizzle-docs-generator\n *\n * Generates DBML files from Drizzle ORM schema definitions.\n */\n\nimport { Command } from \"commander\";\nimport { existsSync, readFileSync, watch } from \"node:fs\";\nimport { join, resolve } from \"node:path\";\nimport { pathToFileURL } from \"node:url\";\nimport { pgGenerate, mysqlGenerate, sqliteGenerate } from \"../generator/index\";\n\nconst program = new Command();\n\n// Use import.meta.dirname (Node 20.11+) to resolve package.json\n// This works correctly even after bundling\nconst packageJsonPath = join(import.meta.dirname, \"../../package.json\");\nconst packageJson = JSON.parse(readFileSync(packageJsonPath, \"utf-8\")) as {\n  version: string;\n  description: string;\n};\n\nprogram.name(\"drizzle-docs\").description(packageJson.description).version(packageJson.version);\n\ntype Dialect = \"postgresql\" | \"mysql\" | \"sqlite\";\n\ninterface GenerateCommandOptions {\n  output?: string;\n  dialect: Dialect;\n  relational?: boolean;\n  watch?: boolean;\n}\n\n/**\n * Get the generate function based on dialect\n */\nfunction getGenerator(dialect: Dialect) {\n  switch (dialect) {\n    case \"mysql\":\n      return mysqlGenerate;\n    case \"sqlite\":\n      return sqliteGenerate;\n    case \"postgresql\":\n    default:\n      return pgGenerate;\n  }\n}\n\n/**\n * Generate DBML from a schema file\n */\nasync function generateDbml(\n  schemaPath: string,\n  options: GenerateCommandOptions,\n): Promise<string | undefined> {\n  // Use file URL for dynamic import (required for ESM)\n  const schemaUrl = pathToFileURL(schemaPath).href;\n\n  // Dynamic import with cache busting for watch mode\n  const cacheBuster = options.watch ? `?t=${Date.now()}` : \"\";\n  const schemaModule = (await import(schemaUrl + cacheBuster)) as Record<string, unknown>;\n\n  const generate = getGenerator(options.dialect);\n\n  const dbml = generate({\n    schema: schemaModule,\n    out: options.output,\n    relational: options.relational,\n    source: schemaPath,\n  });\n\n  return dbml;\n}\n\n/**\n * Run the generate command\n */\nasync function runGenerate(schema: string, options: GenerateCommandOptions): Promise<void> {\n  const schemaPath = resolve(process.cwd(), schema);\n\n  // Check if file exists\n  if (!existsSync(schemaPath)) {\n    console.error(`Error: Schema file not found: ${schemaPath}`);\n    process.exit(1);\n  }\n\n  try {\n    const dbml = await generateDbml(schemaPath, options);\n\n    if (!options.output && dbml) {\n      console.log(dbml);\n    } else if (options.output) {\n      console.log(`DBML generated: ${options.output}`);\n    }\n  } catch (error) {\n    if (error instanceof Error) {\n      console.error(`Error generating DBML: ${error.message}`);\n    } else {\n      console.error(\"Error generating DBML:\", error);\n    }\n    process.exit(1);\n  }\n}\n\n/**\n * Watch mode: regenerate DBML on file changes\n */\nfunction watchSchema(schema: string, options: GenerateCommandOptions): void {\n  const schemaPath = resolve(process.cwd(), schema);\n\n  console.log(`Watching for changes: ${schemaPath}`);\n\n  let debounceTimer: NodeJS.Timeout | null = null;\n\n  watch(schemaPath, async (eventType) => {\n    if (eventType === \"change\") {\n      // Debounce to avoid multiple triggers\n      if (debounceTimer) {\n        clearTimeout(debounceTimer);\n      }\n\n      debounceTimer = setTimeout(async () => {\n        console.log(\"\\nFile changed, regenerating...\");\n        try {\n          const dbml = await generateDbml(schemaPath, options);\n\n          if (!options.output && dbml) {\n            console.log(dbml);\n          } else if (options.output) {\n            console.log(`DBML regenerated: ${options.output}`);\n          }\n        } catch (error) {\n          if (error instanceof Error) {\n            console.error(`Error: ${error.message}`);\n          } else {\n            console.error(\"Error:\", error);\n          }\n        }\n      }, 100);\n    }\n  });\n}\n\nprogram\n  .command(\"generate\")\n  .description(\"Generate DBML from Drizzle schema files\")\n  .argument(\"<schema>\", \"Path to Drizzle schema file\")\n  .option(\"-o, --output <file>\", \"Output file path\")\n  .option(\"-d, --dialect <dialect>\", \"Database dialect (postgresql, mysql, sqlite)\", \"postgresql\")\n  .option(\"-r, --relational\", \"Use relations() definitions instead of foreign keys\")\n  .option(\"-w, --watch\", \"Watch for file changes and regenerate\")\n  .action(async (schema: string, options: GenerateCommandOptions) => {\n    // Validate dialect\n    const validDialects: Dialect[] = [\"postgresql\", \"mysql\", \"sqlite\"];\n    if (!validDialects.includes(options.dialect)) {\n      console.error(\n        `Error: Invalid dialect \"${options.dialect}\". Valid options: ${validDialects.join(\", \")}`,\n      );\n      process.exit(1);\n    }\n\n    // Initial generation\n    await runGenerate(schema, options);\n\n    // Start watch mode if requested\n    if (options.watch) {\n      watchSchema(schema, options);\n    }\n  });\n\nprogram.parse();\n"],"names":["program","Command","packageJsonPath","join","packageJson","readFileSync","getGenerator","dialect","mysqlGenerate","sqliteGenerate","pgGenerate","generateDbml","schemaPath","options","schemaUrl","pathToFileURL","cacheBuster","schemaModule","runGenerate","schema","resolve","existsSync","dbml","error","watchSchema","debounceTimer","watch","eventType","validDialects"],"mappings":";;;;;;;;;;;;;AAcA,MAAMA,IAAU,IAAIC,EAAA,GAIdC,IAAkBC,EAAK,YAAY,SAAS,oBAAoB,GAChEC,IAAc,KAAK,MAAMC,EAAaH,GAAiB,OAAO,CAAC;AAKrEF,EAAQ,KAAK,cAAc,EAAE,YAAYI,EAAY,WAAW,EAAE,QAAQA,EAAY,OAAO;AAc7F,SAASE,EAAaC,GAAkB;AACtC,UAAQA,GAAA;AAAA,IACN,KAAK;AACH,aAAOC;AAAA,IACT,KAAK;AACH,aAAOC;AAAA,IAET;AACE,aAAOC;AAAA,EAAA;AAEb;AAKA,eAAeC,EACbC,GACAC,GAC6B;AAE7B,QAAMC,IAAYC,EAAcH,CAAU,EAAE,MAGtCI,IAAcH,EAAQ,QAAQ,MAAM,KAAK,KAAK,KAAK,IACnDI,IAAgB,MAAM,OAAOH,IAAYE;AAW/C,SATiBV,EAAaO,EAAQ,OAAO,EAEvB;AAAA,IACpB,QAAQI;AAAA,IACR,KAAKJ,EAAQ;AAAA,IACb,YAAYA,EAAQ;AAAA,IACpB,QAAQD;AAAA,EAAA,CACT;AAGH;AAKA,eAAeM,EAAYC,GAAgBN,GAAgD;AACzF,QAAMD,IAAaQ,EAAQ,QAAQ,IAAA,GAAOD,CAAM;AAGhD,EAAKE,EAAWT,CAAU,MACxB,QAAQ,MAAM,iCAAiCA,CAAU,EAAE,GAC3D,QAAQ,KAAK,CAAC;AAGhB,MAAI;AACF,UAAMU,IAAO,MAAMX,EAAaC,GAAYC,CAAO;AAEnD,IAAI,CAACA,EAAQ,UAAUS,IACrB,QAAQ,IAAIA,CAAI,IACPT,EAAQ,UACjB,QAAQ,IAAI,mBAAmBA,EAAQ,MAAM,EAAE;AAAA,EAEnD,SAASU,GAAO;AACd,IAAIA,aAAiB,QACnB,QAAQ,MAAM,0BAA0BA,EAAM,OAAO,EAAE,IAEvD,QAAQ,MAAM,0BAA0BA,CAAK,GAE/C,QAAQ,KAAK,CAAC;AAAA,EAChB;AACF;AAKA,SAASC,EAAYL,GAAgBN,GAAuC;AAC1E,QAAMD,IAAaQ,EAAQ,QAAQ,IAAA,GAAOD,CAAM;AAEhD,UAAQ,IAAI,yBAAyBP,CAAU,EAAE;AAEjD,MAAIa,IAAuC;AAE3C,EAAAC,EAAMd,GAAY,OAAOe,MAAc;AACrC,IAAIA,MAAc,aAEZF,KACF,aAAaA,CAAa,GAG5BA,IAAgB,WAAW,YAAY;AACrC,cAAQ,IAAI;AAAA,8BAAiC;AAC7C,UAAI;AACF,cAAMH,IAAO,MAAMX,EAAaC,GAAYC,CAAO;AAEnD,QAAI,CAACA,EAAQ,UAAUS,IACrB,QAAQ,IAAIA,CAAI,IACPT,EAAQ,UACjB,QAAQ,IAAI,qBAAqBA,EAAQ,MAAM,EAAE;AAAA,MAErD,SAASU,GAAO;AACd,QAAIA,aAAiB,QACnB,QAAQ,MAAM,UAAUA,EAAM,OAAO,EAAE,IAEvC,QAAQ,MAAM,UAAUA,CAAK;AAAA,MAEjC;AAAA,IACF,GAAG,GAAG;AAAA,EAEV,CAAC;AACH;AAEAvB,EACG,QAAQ,UAAU,EAClB,YAAY,yCAAyC,EACrD,SAAS,YAAY,6BAA6B,EAClD,OAAO,uBAAuB,kBAAkB,EAChD,OAAO,2BAA2B,gDAAgD,YAAY,EAC9F,OAAO,oBAAoB,qDAAqD,EAChF,OAAO,eAAe,uCAAuC,EAC7D,OAAO,OAAOmB,GAAgBN,MAAoC;AAEjE,QAAMe,IAA2B,CAAC,cAAc,SAAS,QAAQ;AACjE,EAAKA,EAAc,SAASf,EAAQ,OAAO,MACzC,QAAQ;AAAA,IACN,2BAA2BA,EAAQ,OAAO,qBAAqBe,EAAc,KAAK,IAAI,CAAC;AAAA,EAAA,GAEzF,QAAQ,KAAK,CAAC,IAIhB,MAAMV,EAAYC,GAAQN,CAAO,GAG7BA,EAAQ,SACVW,EAAYL,GAAQN,CAAO;AAE/B,CAAC;AAEHb,EAAQ,MAAA;"}
+{"version":3,"file":"index.js","sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\n/**\n * CLI for drizzle-docs-generator\n *\n * Generates DBML files from Drizzle ORM schema definitions.\n */\n\nimport { Command } from \"commander\";\nimport { existsSync, lstatSync, readdirSync, readFileSync, watch } from \"node:fs\";\nimport { join, resolve } from \"node:path\";\nimport { pathToFileURL } from \"node:url\";\nimport { pgGenerate, mysqlGenerate, sqliteGenerate } from \"../generator/index\";\nimport { register } from \"tsx/esm/api\";\n\n// Register tsx loader to support TypeScript and extensionless imports\nconst unregister = register();\n\nconst program = new Command();\n\n// Use import.meta.dirname (Node 20.11+) to resolve package.json\n// This works correctly even after bundling\nconst packageJsonPath = join(import.meta.dirname, \"../../package.json\");\nconst packageJson = JSON.parse(readFileSync(packageJsonPath, \"utf-8\")) as {\n  version: string;\n  description: string;\n};\n\nprogram.name(\"drizzle-docs\").description(packageJson.description).version(packageJson.version);\n\ntype Dialect = \"postgresql\" | \"mysql\" | \"sqlite\";\n\ninterface GenerateCommandOptions {\n  output?: string;\n  dialect: Dialect;\n  relational?: boolean;\n  watch?: boolean;\n}\n\n/**\n * Get the generate function based on dialect\n */\nfunction getGenerator(dialect: Dialect) {\n  switch (dialect) {\n    case \"mysql\":\n      return mysqlGenerate;\n    case \"sqlite\":\n      return sqliteGenerate;\n    case \"postgresql\":\n    default:\n      return pgGenerate;\n  }\n}\n\n/**\n * Generate DBML from a schema file\n */\nasync function generateDbml(\n  schemaPath: string,\n  options: GenerateCommandOptions,\n): Promise<string | undefined> {\n  // Use file URL for dynamic import (required for ESM)\n  const schemaUrl = pathToFileURL(schemaPath).href;\n\n  // Dynamic import with cache busting for watch mode\n  const cacheBuster = options.watch ? `?t=${Date.now()}` : \"\";\n  const schemaModule = (await import(schemaUrl + cacheBuster)) as Record<string, unknown>;\n\n  const generate = getGenerator(options.dialect);\n\n  const dbml = generate({\n    schema: schemaModule,\n    out: options.output,\n    relational: options.relational,\n    source: schemaPath,\n  });\n\n  return dbml;\n}\n\n/**\n * Find all TypeScript schema files in a directory\n */\nfunction findSchemaFiles(dirPath: string): string[] {\n  const files: string[] = [];\n\n  try {\n    const entries = readdirSync(dirPath, { withFileTypes: true });\n\n    for (const entry of entries) {\n      const fullPath = join(dirPath, entry.name);\n\n      if (entry.isDirectory()) {\n        // Recursively search subdirectories\n        files.push(...findSchemaFiles(fullPath));\n      } else if (entry.isFile() && /\\.(ts|js|mts|mjs|cts|cjs)$/.test(entry.name)) {\n        files.push(fullPath);\n      }\n    }\n  } catch (error) {\n    if (error instanceof Error) {\n      console.error(`Error reading directory ${dirPath}: ${error.message}`);\n    }\n  }\n\n  return files;\n}\n\n/**\n * Resolve schema path (file or directory)\n */\nfunction resolveSchemaPath(schema: string): string[] {\n  const schemaPath = resolve(process.cwd(), schema);\n\n  // Check if path exists\n  if (!existsSync(schemaPath)) {\n    console.error(`Error: Schema path not found: ${schemaPath}`);\n    process.exit(1);\n  }\n\n  // Check if it's a directory\n  const stats = lstatSync(schemaPath);\n  if (stats.isDirectory()) {\n    const schemaFiles = findSchemaFiles(schemaPath);\n    if (schemaFiles.length === 0) {\n      console.error(`Error: No schema files found in directory: ${schemaPath}`);\n      process.exit(1);\n    }\n    return schemaFiles;\n  }\n\n  // Single file\n  return [schemaPath];\n}\n\n/**\n * Run the generate command\n */\nasync function runGenerate(schema: string, options: GenerateCommandOptions): Promise<void> {\n  const schemaPaths = resolveSchemaPath(schema);\n\n  try {\n    // Merge all schema modules\n    const mergedSchema: Record<string, unknown> = {};\n\n    for (const schemaPath of schemaPaths) {\n      const schemaUrl = pathToFileURL(schemaPath).href;\n      const cacheBuster = options.watch ? `?t=${Date.now()}` : \"\";\n\n      try {\n        const schemaModule = (await import(schemaUrl + cacheBuster)) as Record<string, unknown>;\n        Object.assign(mergedSchema, schemaModule);\n      } catch (error) {\n        if (error instanceof Error) {\n          console.error(`Error importing ${schemaPath}: ${error.message}`);\n          console.error(\"\\nPossible causes:\");\n          console.error(\"- Syntax error in the schema file\");\n          console.error(\"- Missing dependencies (make sure drizzle-orm is installed)\");\n          console.error(\"- Circular dependencies between schema files\");\n        }\n        throw error;\n      }\n    }\n\n    const generate = getGenerator(options.dialect);\n    const dbml = generate({\n      schema: mergedSchema,\n      out: options.output,\n      relational: options.relational,\n      source: schemaPaths[0], // Use first file for source path\n    });\n\n    if (!options.output && dbml) {\n      console.log(dbml);\n    } else if (options.output) {\n      console.log(`DBML generated: ${options.output}`);\n    }\n  } catch (error) {\n    if (error instanceof Error) {\n      console.error(`Error generating DBML: ${error.message}`);\n    } else {\n      console.error(\"Error generating DBML:\", error);\n    }\n    process.exit(1);\n  }\n}\n\n/**\n * Watch mode: regenerate DBML on file changes\n */\nfunction watchSchema(schema: string, options: GenerateCommandOptions): void {\n  const schemaPath = resolve(process.cwd(), schema);\n\n  console.log(`Watching for changes: ${schemaPath}`);\n\n  let debounceTimer: NodeJS.Timeout | null = null;\n\n  watch(schemaPath, async (eventType) => {\n    if (eventType === \"change\") {\n      // Debounce to avoid multiple triggers\n      if (debounceTimer) {\n        clearTimeout(debounceTimer);\n      }\n\n      debounceTimer = setTimeout(async () => {\n        console.log(\"\\nFile changed, regenerating...\");\n        try {\n          const dbml = await generateDbml(schemaPath, options);\n\n          if (!options.output && dbml) {\n            console.log(dbml);\n          } else if (options.output) {\n            console.log(`DBML regenerated: ${options.output}`);\n          }\n        } catch (error) {\n          if (error instanceof Error) {\n            console.error(`Error: ${error.message}`);\n          } else {\n            console.error(\"Error:\", error);\n          }\n        }\n      }, 100);\n    }\n  });\n}\n\nprogram\n  .command(\"generate\")\n  .description(\"Generate DBML from Drizzle schema files\")\n  .argument(\"<schema>\", \"Path to Drizzle schema file or directory\")\n  .option(\"-o, --output <file>\", \"Output file path\")\n  .option(\"-d, --dialect <dialect>\", \"Database dialect (postgresql, mysql, sqlite)\", \"postgresql\")\n  .option(\"-r, --relational\", \"Use relations() definitions instead of foreign keys\")\n  .option(\"-w, --watch\", \"Watch for file changes and regenerate\")\n  .action(async (schema: string, options: GenerateCommandOptions) => {\n    // Validate dialect\n    const validDialects: Dialect[] = [\"postgresql\", \"mysql\", \"sqlite\"];\n    if (!validDialects.includes(options.dialect)) {\n      console.error(\n        `Error: Invalid dialect \"${options.dialect}\". Valid options: ${validDialects.join(\", \")}`,\n      );\n      process.exit(1);\n    }\n\n    // Initial generation\n    await runGenerate(schema, options);\n\n    // Start watch mode if requested\n    if (options.watch) {\n      watchSchema(schema, options);\n    }\n  });\n\nprogram.parse();\n\n// Cleanup: Unregister tsx loader when process exits\nprocess.on(\"exit\", () => {\n  unregister();\n});\n"],"names":["unregister","register","program","Command","packageJsonPath","join","packageJson","readFileSync","getGenerator","dialect","mysqlGenerate","sqliteGenerate","pgGenerate","generateDbml","schemaPath","options","schemaUrl","pathToFileURL","cacheBuster","schemaModule","findSchemaFiles","dirPath","files","entries","readdirSync","entry","fullPath","error","resolveSchemaPath","schema","resolve","existsSync","lstatSync","schemaFiles","runGenerate","schemaPaths","mergedSchema","dbml","watchSchema","debounceTimer","watch","eventType","validDialects"],"mappings":";;;;;;;;;;;;;;AAgBA,MAAMA,IAAaC,EAAA,GAEbC,IAAU,IAAIC,EAAA,GAIdC,IAAkBC,EAAK,YAAY,SAAS,oBAAoB,GAChEC,IAAc,KAAK,MAAMC,EAAaH,GAAiB,OAAO,CAAC;AAKrEF,EAAQ,KAAK,cAAc,EAAE,YAAYI,EAAY,WAAW,EAAE,QAAQA,EAAY,OAAO;AAc7F,SAASE,EAAaC,GAAkB;AACtC,UAAQA,GAAA;AAAA,IACN,KAAK;AACH,aAAOC;AAAA,IACT,KAAK;AACH,aAAOC;AAAA,IAET;AACE,aAAOC;AAAA,EAAA;AAEb;AAKA,eAAeC,EACbC,GACAC,GAC6B;AAE7B,QAAMC,IAAYC,EAAcH,CAAU,EAAE,MAGtCI,IAAcH,EAAQ,QAAQ,MAAM,KAAK,KAAK,KAAK,IACnDI,IAAgB,MAAM,OAAOH,IAAYE;AAW/C,SATiBV,EAAaO,EAAQ,OAAO,EAEvB;AAAA,IACpB,QAAQI;AAAA,IACR,KAAKJ,EAAQ;AAAA,IACb,YAAYA,EAAQ;AAAA,IACpB,QAAQD;AAAA,EAAA,CACT;AAGH;AAKA,SAASM,EAAgBC,GAA2B;AAClD,QAAMC,IAAkB,CAAA;AAExB,MAAI;AACF,UAAMC,IAAUC,EAAYH,GAAS,EAAE,eAAe,IAAM;AAE5D,eAAWI,KAASF,GAAS;AAC3B,YAAMG,IAAWrB,EAAKgB,GAASI,EAAM,IAAI;AAEzC,MAAIA,EAAM,gBAERH,EAAM,KAAK,GAAGF,EAAgBM,CAAQ,CAAC,IAC9BD,EAAM,OAAA,KAAY,6BAA6B,KAAKA,EAAM,IAAI,KACvEH,EAAM,KAAKI,CAAQ;AAAA,IAEvB;AAAA,EACF,SAASC,GAAO;AACd,IAAIA,aAAiB,SACnB,QAAQ,MAAM,2BAA2BN,CAAO,KAAKM,EAAM,OAAO,EAAE;AAAA,EAExE;AAEA,SAAOL;AACT;AAKA,SAASM,EAAkBC,GAA0B;AACnD,QAAMf,IAAagB,EAAQ,QAAQ,IAAA,GAAOD,CAAM;AAUhD,MAPKE,EAAWjB,CAAU,MACxB,QAAQ,MAAM,iCAAiCA,CAAU,EAAE,GAC3D,QAAQ,KAAK,CAAC,IAIFkB,EAAUlB,CAAU,EACxB,eAAe;AACvB,UAAMmB,IAAcb,EAAgBN,CAAU;AAC9C,WAAImB,EAAY,WAAW,MACzB,QAAQ,MAAM,8CAA8CnB,CAAU,EAAE,GACxE,QAAQ,KAAK,CAAC,IAETmB;AAAA,EACT;AAGA,SAAO,CAACnB,CAAU;AACpB;AAKA,eAAeoB,EAAYL,GAAgBd,GAAgD;AACzF,QAAMoB,IAAcP,EAAkBC,CAAM;AAE5C,MAAI;AAEF,UAAMO,IAAwC,CAAA;AAE9C,eAAWtB,KAAcqB,GAAa;AACpC,YAAMnB,IAAYC,EAAcH,CAAU,EAAE,MACtCI,IAAcH,EAAQ,QAAQ,MAAM,KAAK,KAAK,KAAK;AAEzD,UAAI;AACF,cAAMI,IAAgB,MAAM,OAAOH,IAAYE;AAC/C,eAAO,OAAOkB,GAAcjB,CAAY;AAAA,MAC1C,SAASQ,GAAO;AACd,cAAIA,aAAiB,UACnB,QAAQ,MAAM,mBAAmBb,CAAU,KAAKa,EAAM,OAAO,EAAE,GAC/D,QAAQ,MAAM;AAAA,iBAAoB,GAClC,QAAQ,MAAM,mCAAmC,GACjD,QAAQ,MAAM,6DAA6D,GAC3E,QAAQ,MAAM,8CAA8C,IAExDA;AAAA,MACR;AAAA,IACF;AAGA,UAAMU,IADW7B,EAAaO,EAAQ,OAAO,EACvB;AAAA,MACpB,QAAQqB;AAAA,MACR,KAAKrB,EAAQ;AAAA,MACb,YAAYA,EAAQ;AAAA,MACpB,QAAQoB,EAAY,CAAC;AAAA;AAAA,IAAA,CACtB;AAED,IAAI,CAACpB,EAAQ,UAAUsB,IACrB,QAAQ,IAAIA,CAAI,IACPtB,EAAQ,UACjB,QAAQ,IAAI,mBAAmBA,EAAQ,MAAM,EAAE;AAAA,EAEnD,SAASY,GAAO;AACd,IAAIA,aAAiB,QACnB,QAAQ,MAAM,0BAA0BA,EAAM,OAAO,EAAE,IAEvD,QAAQ,MAAM,0BAA0BA,CAAK,GAE/C,QAAQ,KAAK,CAAC;AAAA,EAChB;AACF;AAKA,SAASW,EAAYT,GAAgBd,GAAuC;AAC1E,QAAMD,IAAagB,EAAQ,QAAQ,IAAA,GAAOD,CAAM;AAEhD,UAAQ,IAAI,yBAAyBf,CAAU,EAAE;AAEjD,MAAIyB,IAAuC;AAE3C,EAAAC,EAAM1B,GAAY,OAAO2B,MAAc;AACrC,IAAIA,MAAc,aAEZF,KACF,aAAaA,CAAa,GAG5BA,IAAgB,WAAW,YAAY;AACrC,cAAQ,IAAI;AAAA,8BAAiC;AAC7C,UAAI;AACF,cAAMF,IAAO,MAAMxB,EAAaC,GAAYC,CAAO;AAEnD,QAAI,CAACA,EAAQ,UAAUsB,IACrB,QAAQ,IAAIA,CAAI,IACPtB,EAAQ,UACjB,QAAQ,IAAI,qBAAqBA,EAAQ,MAAM,EAAE;AAAA,MAErD,SAASY,GAAO;AACd,QAAIA,aAAiB,QACnB,QAAQ,MAAM,UAAUA,EAAM,OAAO,EAAE,IAEvC,QAAQ,MAAM,UAAUA,CAAK;AAAA,MAEjC;AAAA,IACF,GAAG,GAAG;AAAA,EAEV,CAAC;AACH;AAEAzB,EACG,QAAQ,UAAU,EAClB,YAAY,yCAAyC,EACrD,SAAS,YAAY,0CAA0C,EAC/D,OAAO,uBAAuB,kBAAkB,EAChD,OAAO,2BAA2B,gDAAgD,YAAY,EAC9F,OAAO,oBAAoB,qDAAqD,EAChF,OAAO,eAAe,uCAAuC,EAC7D,OAAO,OAAO2B,GAAgBd,MAAoC;AAEjE,QAAM2B,IAA2B,CAAC,cAAc,SAAS,QAAQ;AACjE,EAAKA,EAAc,SAAS3B,EAAQ,OAAO,MACzC,QAAQ;AAAA,IACN,2BAA2BA,EAAQ,OAAO,qBAAqB2B,EAAc,KAAK,IAAI,CAAC;AAAA,EAAA,GAEzF,QAAQ,KAAK,CAAC,IAIhB,MAAMR,EAAYL,GAAQd,CAAO,GAG7BA,EAAQ,SACVuB,EAAYT,GAAQd,CAAO;AAE/B,CAAC;AAEHb,EAAQ,MAAA;AAGR,QAAQ,GAAG,QAAQ,MAAM;AACvB,EAAAF,EAAA;AACF,CAAC;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drizzle-docs-generator",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "A CLI tool that generates DBML files from Drizzle ORM schema definitions.",
   "keywords": [
     "cli",
@@ -38,13 +38,14 @@
   },
   "dependencies": {
     "commander": "^14.0.2",
-    "drizzle-orm": "1.0.0-beta.10-42c284e"
+    "drizzle-orm": "1.0.0-beta.10-42c284e",
+    "tsx": "^4.21.0",
+    "typescript": "^5.9.3"
   },
   "devDependencies": {
     "@types/node": "^24.10.7",
     "oxfmt": "^0.23.0",
     "oxlint": "^1.38.0",
-    "typescript": "^5.9.3",
     "vite": "^7.3.1",
     "vite-plugin-dts": "^4.5.4",
     "vitest": "^4.0.16"