mikro-orm-markdown 0.1.0 → 0.1.2

package/dist/cli.js

@@ -5,10 +5,10 @@ import * as syncFs from "fs";
 import * as fs from "fs/promises";
 import * as path2 from "path";
 import { fileURLToPath, pathToFileURL } from "url";
-import { Command } from "commander";
+import { Command, Option } from "commander";
 
 // src/docs/jsdoc.ts
-import { Project } from "ts-morph";
+import { Project, ts } from "ts-morph";
 function loadJsDoc(filePaths, onWarn) {
   const entities = /* @__PURE__ */ new Map();
   const props = /* @__PURE__ */ new Map();
@@ -47,17 +47,7 @@ function loadJsDoc(filePaths, onWarn) {
         if (classDocs.length > 0) {
           entities.set(className, parseEntityJsDoc(classDocs));
         }
-        const propMap = /* @__PURE__ */ new Map();
-        for (const prop of cls.getProperties()) {
-          const propDocs = prop.getJsDocs();
-          if (propDocs.length === 0) {
-            continue;
-          }
-          const info = parsePropJsDoc(propDocs);
-          if (info.description !== void 0 || info.atLeastOne) {
-            propMap.set(prop.getName(), info);
-          }
-        }
+        const propMap = collectPropJsDocs(cls);
         if (propMap.size > 0) {
           props.set(className, propMap);
         }
@@ -74,6 +64,28 @@ function formatUnknownError(err) {
 function hasGlobPattern(filePath) {
   return /[*?[\]{}]/.test(filePath);
 }
+function collectPropJsDocs(cls) {
+  const propMap = /* @__PURE__ */ new Map();
+  for (const prop of [...cls.getProperties(), ...cls.getGetAccessors()]) {
+    const info = parsePropJsDoc(prop.getJsDocs());
+    addPropInfo(propMap, prop.getName(), info);
+  }
+  for (const prop of getConstructorParameterProperties(cls)) {
+    const info = parseCompilerPropJsDoc(ts.getJSDocCommentsAndTags(prop.compilerNode));
+    addPropInfo(propMap, prop.getName(), info);
+  }
+  return propMap;
+}
+function getConstructorParameterProperties(cls) {
+  return cls.getConstructors().flatMap(
+    (constructorDeclaration) => constructorDeclaration.getParameters().filter((param) => param.isParameterProperty())
+  );
+}
+function addPropInfo(propMap, propName, info) {
+  if (info.description !== void 0 || info.atLeastOne) {
+    propMap.set(propName, info);
+  }
+}
 function parseEntityJsDoc(jsDocs) {
   const namespaces = [];
   const erdNamespaces = [];
@@ -123,10 +135,39 @@ function parsePropJsDoc(jsDocs) {
   }
   return { ...description !== void 0 && { description }, atLeastOne };
 }
+function parseCompilerPropJsDoc(jsDocs) {
+  let description;
+  let atLeastOne = false;
+  for (const doc of jsDocs) {
+    if (ts.isJSDoc(doc)) {
+      const desc = formatCompilerJsDocComment(doc.comment);
+      if (desc && description === void 0) {
+        description = desc;
+      }
+      for (const tag of doc.tags ?? []) {
+        if (tag.tagName.getText() === "atLeastOne") {
+          atLeastOne = true;
+        }
+      }
+      continue;
+    }
+    if (doc.tagName.getText() === "atLeastOne") {
+      atLeastOne = true;
+    }
+  }
+  return { ...description !== void 0 && { description }, atLeastOne };
+}
+function formatCompilerJsDocComment(comment) {
+  if (comment === void 0) {
+    return void 0;
+  }
+  const trimmed = ts.getTextOfJSDocComment(comment)?.trim();
+  return trimmed === void 0 || trimmed === "" ? void 0 : trimmed;
+}
 
 // src/metadata/load.ts
 import * as path from "path";
-import { EntitySchema, MikroORM } from "@mikro-orm/core";
+import { EntitySchema, MetadataStorage, MikroORM } from "@mikro-orm/core";
 var MetadataLoadError = class extends Error {
   constructor(message, cause) {
     super(message);
@@ -162,6 +203,44 @@ function assertNoEntitySchemaEntities(options) {
 Use decorator-based @Entity() classes instead.`
   );
 }
+function hasDecoratorMarker(target) {
+  const pathSymbol = MetadataStorage.PATH_SYMBOL;
+  if (typeof pathSymbol === "symbol" && pathSymbol in target) {
+    return true;
+  }
+  return "__path" in target;
+}
+function isRenderableMeta(meta) {
+  return !meta.pivotTable && !meta.embeddable;
+}
+function assertDiscoveredEntitiesAreSupported(metas) {
+  const confirmed = [];
+  const unconfirmed = [];
+  for (const meta of metas) {
+    if (!isRenderableMeta(meta)) {
+      continue;
+    }
+    if (EntitySchema.REGISTRY.has(meta.class)) {
+      confirmed.push(meta.className);
+    } else if (!hasDecoratorMarker(meta.class)) {
+      unconfirmed.push(meta.className);
+    }
+  }
+  if (confirmed.length === 0 && unconfirmed.length === 0) {
+    return;
+  }
+  const lines = [];
+  if (confirmed.length > 0) {
+    lines.push(`EntitySchema-defined entities are not currently supported: ${confirmed.join(", ")}.`);
+  }
+  if (unconfirmed.length > 0) {
+    lines.push(
+      `Could not confirm these entities are decorator-based @Entity() classes: ${unconfirmed.join(", ")}. This usually means they are EntitySchema-defined entities (also not currently supported). If you're certain these are valid @Entity() classes, this may be a detection false positive in mikro-orm-markdown \u2014 please open an issue: https://github.com/iamkanguk97/mikro-orm-markdown/issues`
+    );
+  }
+  lines.push("Use decorator-based @Entity() classes instead.");
+  throw new MetadataLoadError(lines.join("\n"));
+}
 async function loadEntityMetadata(options) {
   assertNoEntitySchemaEntities(options);
   let orm;
@@ -188,6 +267,7 @@ async function loadEntityMetadata(options) {
         "No entities were discovered. Check that your config specifies at least one entity path or class."
       );
     }
+    assertDiscoveredEntitiesAreSupported(all);
     const baseDir = orm.config.get("baseDir");
     const sourcePaths = [...new Set(all.filter((m) => m.path).map((m) => path.resolve(baseDir, m.path)))];
     return { metas: all, sourcePaths };
@@ -199,57 +279,8 @@ async function loadEntityMetadata(options) {
 // src/model/build.ts
 import { ReferenceKind as ReferenceKind2 } from "@mikro-orm/core";
 
-// src/render/mermaid.ts
+// src/model/diagram.ts
 import { ReferenceKind } from "@mikro-orm/core";
-
-// src/render/escape.ts
-var MARKDOWN_INLINE_SPECIAL_CHARS = /[\\`|*#]/g;
-var MERMAID_IDENTIFIER_INVALID_CHARS = /[^a-zA-Z0-9_]/g;
-function normalizeInlineText(value) {
-  return value.replace(/\r\n?/g, "\n").replace(/[\n\t]+/g, " ").replace(/\s{2,}/g, " ").trim();
-}
-function splitNormalizedLines(value) {
-  return value.replace(/\r\n?/g, "\n").split("\n");
-}
-function escapeHtmlText(value) {
-  return value.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
-}
-function escapeMarkdownInline(value) {
-  return escapeHtmlText(normalizeInlineText(value)).replace(MARKDOWN_INLINE_SPECIAL_CHARS, "\\$&");
-}
-function escapeMarkdownTableCell(value) {
-  return splitNormalizedLines(value).map((line) => escapeMarkdownInline(line)).join("<br>");
-}
-function escapeMarkdownParagraph(value) {
-  return splitNormalizedLines(value).map((line) => escapeMarkdownInline(line)).join("  \n");
-}
-function renderMarkdownBlockQuote(value) {
-  return splitNormalizedLines(value).map((line) => `> ${escapeMarkdownInline(line)}`).join("\n");
-}
-function renderMarkdownInlineCode(value) {
-  const normalized = normalizeInlineText(value);
-  const backtickRuns = normalized.match(/`+/g) ?? [];
-  const longestRun = Math.max(0, ...backtickRuns.map((run2) => run2.length));
-  const fence = "`".repeat(longestRun + 1);
-  const needsPadding = normalized.startsWith("`") || normalized.endsWith("`");
-  const content = needsPadding ? ` ${normalized} ` : normalized;
-  return `${fence}${content}${fence}`;
-}
-function toMermaidIdentifier(value) {
-  const normalized = normalizeInlineText(value).replace(MERMAID_IDENTIFIER_INVALID_CHARS, "_").replace(/_+/g, "_");
-  const identifier = normalized === "" ? "_" : normalized;
-  return /^[a-zA-Z_]/.test(identifier) ? identifier : `_${identifier}`;
-}
-function escapeMermaidQuotedText(value) {
-  return normalizeInlineText(value).replace(/\\/g, "\\\\").replace(/"/g, '\\"');
-}
-function toMarkdownAnchor(value) {
-  return normalizeInlineText(value).toLowerCase().replace(/[^\p{L}\p{N}\s_-]/gu, "").replace(/\s+/g, "-");
-}
-
-// src/render/mermaid.ts
-var MERMAID_LAYOUTS = ["dagre", "elk", "elk.stress"];
-var MERMAID_THEMES = ["default", "neutral", "dark", "forest", "base"];
 var FORMULA_DUMMY_TABLE = {
   alias: "e0",
   name: "",
@@ -309,6 +340,9 @@ function buildColumns(prop, metaByClass, owningMeta) {
       return [];
     }
     const formulaExpr = prop.formula !== void 0 ? resolveFormulaExpr(prop.formula) : void 0;
+    if (prop.persist === false && formulaExpr === void 0) {
+      return [];
+    }
     let embeddedIn;
     let embeddedPropName;
     if (prop.embedded !== void 0) {
@@ -495,6 +529,9 @@ function buildEdge(fromEntity, prop) {
     };
   }
   if (prop.kind === ReferenceKind.ONE_TO_ONE && prop.owner === true) {
+    if (prop.type === fromEntity) {
+      return null;
+    }
     return {
       fromEntity,
       toEntity: prop.type,
@@ -514,75 +551,6 @@ function buildEdge(fromEntity, prop) {
   }
   return null;
 }
-function normalizeType(type) {
-  const t = type.toLowerCase().trim();
-  if (t === "uuid" || t === "text" || t === "string" || t.startsWith("varchar")) {
-    return "string";
-  }
-  if (t === "timestamptz" || t === "timestamp" || t === "datetime") {
-    return "datetime";
-  }
-  if (t === "integer" || t === "int" || t === "bigint" || t === "smallint") {
-    return "integer";
-  }
-  if (t === "doubletype" || t === "double precision" || t === "double" || t === "float" || t === "decimal") {
-    return "float";
-  }
-  if (t === "boolean" || t === "bool") {
-    return "boolean";
-  }
-  if (t === "jsonb") {
-    return "json";
-  }
-  return type;
-}
-function renderErDiagram(model, mermaid) {
-  const lines = [];
-  if (mermaid?.layout !== void 0 || mermaid?.theme !== void 0) {
-    lines.push("---", "config:");
-    if (mermaid.layout !== void 0) {
-      lines.push(`  layout: ${mermaid.layout}`);
-    }
-    if (mermaid.theme !== void 0) {
-      lines.push(`  theme: ${mermaid.theme}`);
-    }
-    lines.push("---");
-  }
-  lines.push("erDiagram");
-  for (const entity of model.entities) {
-    lines.push(`  ${toMermaidIdentifier(entity.className)} {`);
-    for (const col of entity.columns) {
-      lines.push(`    ${renderColumnLine(col)}`);
-    }
-    lines.push("  }");
-  }
-  for (const rel of model.relations) {
-    lines.push(
-      `  ${toMermaidIdentifier(rel.fromEntity)} ${rel.fromCardinality}--${rel.toCardinality} ${toMermaidIdentifier(rel.toEntity)} : "${escapeMermaidQuotedText(rel.label)}"`
-    );
-  }
-  return lines.join("\n");
-}
-function renderColumnLine(col) {
-  let qualifier = "";
-  if (col.isPrimary) {
-    qualifier = " PK";
-  } else if (col.isUnique) {
-    qualifier = " UK";
-  }
-  let comment;
-  if (col.formula !== void 0) {
-    comment = col.formula ? `formula: ${col.formula}` : "formula";
-  } else if (col.isDiscriminator) {
-    comment = "discriminator";
-  } else if (col.embeddedIn !== void 0) {
-    comment = `[${col.embeddedIn}]`;
-  } else if (col.isSelfReference) {
-    comment = "self-ref";
-  }
-  const commentStr = comment !== void 0 ? ` "${escapeMermaidQuotedText(comment)}"` : "";
-  return `${toMermaidIdentifier(normalizeType(col.type))} ${toMermaidIdentifier(col.fieldName)}${qualifier}${commentStr}`;
-}
 
 // src/model/build.ts
 var FROM_ONE_OR_MORE = "}|";
@@ -605,7 +573,10 @@ function buildDocumentModel(metas, jsDocResult, title, description, onWarn) {
     const columns = model.columns.filter(
       (col) => !(col.isForeignKey && col.referencedEntity !== void 0 && hiddenClasses.has(col.referencedEntity))
     );
-    const visibleModel = columns.length === model.columns.length ? model : { ...model, columns };
+    const visibleModel = removeHiddenEntityReferences(
+      columns.length === model.columns.length ? model : { ...model, columns },
+      hiddenClasses
+    );
     const ownPropDocs = jsDocResult.props.get(model.className) ?? /* @__PURE__ */ new Map();
     const propDocs = withEmbeddedPropDocs(ownPropDocs, visibleModel.columns, jsDocResult.props);
     enrichedByClass.set(model.className, { model: visibleModel, jsDoc, propDocs });
@@ -656,6 +627,14 @@ function buildDocumentModel(metas, jsDocResult, title, description, onWarn) {
   });
   return { title, groups, ...description !== void 0 && { description } };
 }
+function removeHiddenEntityReferences(model, hiddenClasses) {
+  if (model.extendsEntity === void 0 || !hiddenClasses.has(model.extendsEntity)) {
+    return model;
+  }
+  const visibleModel = { ...model };
+  delete visibleModel.extendsEntity;
+  return visibleModel;
+}
 function withEmbeddedPropDocs(ownPropDocs, columns, allPropDocs) {
   const merged = new Map(ownPropDocs);
   for (const col of columns) {
@@ -767,6 +746,144 @@ async function withTsMorphMetadataProvider(options, onWarn) {
   }
 }
 
+// src/render/escape.ts
+var MARKDOWN_INLINE_SPECIAL_CHARS = /[\\`|*#\[\]]/g;
+var MARKDOWN_EMPHASIS_UNDERSCORE = /(?<![a-zA-Z0-9])_|_(?![a-zA-Z0-9])/g;
+var MERMAID_IDENTIFIER_INVALID_CHARS = /[^a-zA-Z0-9_]/g;
+function normalizeInlineText(value) {
+  return value.replace(/\r\n?/g, "\n").replace(/[\n\t]+/g, " ").replace(/\s{2,}/g, " ").trim();
+}
+function splitNormalizedLines(value) {
+  return value.replace(/\r\n?/g, "\n").split("\n");
+}
+function escapeHtmlText(value) {
+  return value.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+}
+function escapeMarkdownInline(value) {
+  return escapeHtmlText(normalizeInlineText(value)).replace(MARKDOWN_INLINE_SPECIAL_CHARS, "\\$&").replace(MARKDOWN_EMPHASIS_UNDERSCORE, "\\_");
+}
+function escapeMarkdownTableCell(value) {
+  return splitNormalizedLines(value).map((line) => escapeMarkdownInline(line)).join("<br>");
+}
+function escapeMarkdownParagraph(value) {
+  return splitNormalizedLines(value).map((line) => escapeMarkdownInline(line)).join("  \n");
+}
+function renderMarkdownBlockQuote(value) {
+  return splitNormalizedLines(value).map((line) => `> ${escapeMarkdownInline(line)}`).join("\n");
+}
+function renderMarkdownInlineCode(value) {
+  const normalized = normalizeInlineText(value);
+  const backtickRuns = normalized.match(/`+/g) ?? [];
+  const longestRun = Math.max(0, ...backtickRuns.map((run2) => run2.length));
+  const fence = "`".repeat(longestRun + 1);
+  const needsPadding = normalized.startsWith("`") || normalized.endsWith("`");
+  const content = needsPadding ? ` ${normalized} ` : normalized;
+  return `${fence}${content}${fence}`;
+}
+function toMermaidIdentifier(value) {
+  const normalized = normalizeInlineText(value).replace(MERMAID_IDENTIFIER_INVALID_CHARS, "_").replace(/_+/g, "_");
+  const identifier = normalized === "" ? "_" : normalized;
+  return /^[a-zA-Z_]/.test(identifier) ? identifier : `_${identifier}`;
+}
+function escapeMermaidQuotedText(value) {
+  return normalizeInlineText(value).replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+}
+function toMarkdownAnchor(value) {
+  return normalizeInlineText(value).toLowerCase().replace(/[^\p{L}\p{N}\s_-]/gu, "").replace(/\s+/g, "-");
+}
+
+// src/render/mermaid.ts
+var MERMAID_LAYOUTS = ["dagre", "elk", "elk.stress"];
+var MERMAID_THEMES = ["default", "neutral", "dark", "forest", "base"];
+var GENERIC_TYPE_BY_BASE_NAME = /* @__PURE__ */ new Map([
+  ["uuid", "string"],
+  ["text", "string"],
+  ["string", "string"],
+  ["varchar", "string"],
+  ["character varying", "string"],
+  ["character", "string"],
+  ["char", "string"],
+  ["tinytext", "string"],
+  ["mediumtext", "string"],
+  ["longtext", "string"],
+  ["timestamptz", "datetime"],
+  ["timestamp", "datetime"],
+  ["datetime", "datetime"],
+  ["integer", "integer"],
+  ["int", "integer"],
+  ["bigint", "integer"],
+  ["smallint", "integer"],
+  ["tinyint", "integer"],
+  ["mediumint", "integer"],
+  ["serial", "integer"],
+  ["bigserial", "integer"],
+  ["doubletype", "float"],
+  ["double precision", "float"],
+  ["double", "float"],
+  ["float", "float"],
+  ["decimal", "float"],
+  ["numeric", "float"],
+  ["real", "float"],
+  ["boolean", "boolean"],
+  ["bool", "boolean"],
+  ["jsonb", "json"]
+]);
+function normalizeType(type) {
+  const t = type.toLowerCase().replace(/\s+/g, " ").trim();
+  if (/^tinyint\s*\(\s*1\s*\)$/.test(t)) {
+    return "boolean";
+  }
+  const baseName = t.replace(/\(.*\)/, "").replace(/\b(un)?signed\b/, "").trim();
+  return GENERIC_TYPE_BY_BASE_NAME.get(baseName) ?? type;
+}
+function renderErDiagram(model, mermaid) {
+  const lines = [];
+  if (mermaid?.layout !== void 0 || mermaid?.theme !== void 0) {
+    lines.push("---", "config:");
+    if (mermaid.layout !== void 0) {
+      lines.push(`  layout: ${mermaid.layout}`);
+    }
+    if (mermaid.theme !== void 0) {
+      lines.push(`  theme: ${mermaid.theme}`);
+    }
+    lines.push("---");
+  }
+  lines.push("erDiagram");
+  for (const entity of model.entities) {
+    lines.push(`  ${toMermaidIdentifier(entity.className)} {`);
+    for (const col of entity.columns) {
+      lines.push(`    ${renderColumnLine(col)}`);
+    }
+    lines.push("  }");
+  }
+  for (const rel of model.relations) {
+    lines.push(
+      `  ${toMermaidIdentifier(rel.fromEntity)} ${rel.fromCardinality}--${rel.toCardinality} ${toMermaidIdentifier(rel.toEntity)} : "${escapeMermaidQuotedText(rel.label)}"`
+    );
+  }
+  return lines.join("\n");
+}
+function renderColumnLine(col) {
+  let qualifier = "";
+  if (col.isPrimary) {
+    qualifier = " PK";
+  } else if (col.isUnique) {
+    qualifier = " UK";
+  }
+  let comment;
+  if (col.formula !== void 0) {
+    comment = col.formula ? `formula: ${col.formula}` : "formula";
+  } else if (col.isDiscriminator) {
+    comment = "discriminator";
+  } else if (col.embeddedIn !== void 0) {
+    comment = `[${col.embeddedIn}]`;
+  } else if (col.isSelfReference) {
+    comment = "self-ref";
+  }
+  const commentStr = comment !== void 0 ? ` "${escapeMermaidQuotedText(comment)}"` : "";
+  return `${toMermaidIdentifier(normalizeType(col.type))} ${toMermaidIdentifier(col.fieldName)}${qualifier}${commentStr}`;
+}
+
 // src/render/markdown.ts
 function renderMarkdown(docModel, mermaid) {
   const sections = [`# ${escapeMarkdownInline(docModel.title)}`];
@@ -790,7 +907,7 @@ function renderBulletSection(header, items, renderItem) {
 }
 function renderTableOfContents(groups, anchors) {
   return renderBulletSection("## Contents", groups, (group) => {
-    const label = escapeMarkdownInline(group.name).replace(/[[\]]/g, "\\$&");
+    const label = escapeMarkdownInline(group.name);
     return `- [${label}](#${anchors.get(group) ?? toMarkdownAnchor(group.name)})`;
   });
 }
@@ -875,7 +992,7 @@ function resolveColumnKey(col) {
     return "PK";
   }
   if (col.isForeignKey) {
-    return fkKey;
+    return col.isUnique ? `${fkKey}, UK` : fkKey;
   }
   if (col.isUnique) {
     return "UK";
@@ -988,6 +1105,12 @@ async function generateMarkdown(options) {
 }
 
 // src/cli.ts
+var activeTsxUnregister;
+async function unregisterActiveTsxLoader() {
+  const unregister = activeTsxUnregister;
+  activeTsxUnregister = void 0;
+  await unregister?.();
+}
 function toConfigImportSpecifier(configPath) {
   return pathToFileURL(path2.resolve(configPath)).href;
 }
@@ -1005,64 +1128,78 @@ function findNearestTsconfig(fromPath) {
     dir = parent;
   }
 }
-async function loadOrmOptions(configPath, tsconfigPath) {
+async function loadOrmOptions(configPath, tsconfigPath, loadOptions = {}) {
   const isTypeScriptConfig = configPath.endsWith(".ts");
-  if (isTypeScriptConfig) {
-    let register;
-    try {
-      ({ register } = await import("tsx/esm/api"));
-    } catch {
-      throw new Error('TypeScript config files require the "tsx" package.\nInstall it with: npm install -D tsx');
-    }
-    let tsconfig;
-    if (tsconfigPath !== void 0) {
-      tsconfig = path2.resolve(tsconfigPath);
-      if (!syncFs.existsSync(tsconfig)) {
-        throw new Error(`--tsconfig file not found: ${tsconfig}`);
-      }
-    } else {
-      tsconfig = findNearestTsconfig(configPath);
-    }
-    register(tsconfig !== void 0 ? { tsconfig } : {});
-  }
-  const configUrl = toConfigImportSpecifier(configPath);
-  let mod;
+  let unregisterTsx;
+  let shouldKeepTsxRegistered = false;
   try {
-    mod = await import(
-      /* @vite-ignore */
-      configUrl
-    );
-  } catch (cause) {
     if (isTypeScriptConfig) {
-      const detail = cause instanceof Error ? cause.message : String(cause);
-      throw new Error(
-        `Failed to load TypeScript config.
+      await unregisterActiveTsxLoader();
+      let register;
+      try {
+        ({ register } = await import("tsx/esm/api"));
+      } catch {
+        throw new Error('TypeScript config files require the "tsx" package.\nInstall it with: npm install -D tsx');
+      }
+      let tsconfig;
+      if (tsconfigPath !== void 0) {
+        tsconfig = path2.resolve(tsconfigPath);
+        if (!syncFs.existsSync(tsconfig)) {
+          throw new Error(`--tsconfig file not found: ${tsconfig}`);
+        }
+      } else {
+        tsconfig = findNearestTsconfig(configPath);
+      }
+      unregisterTsx = register(tsconfig !== void 0 ? { tsconfig } : {});
+    }
+    const configUrl = toConfigImportSpecifier(configPath);
+    let mod;
+    try {
+      mod = await import(
+        /* @vite-ignore */
+        configUrl
+      );
+    } catch (cause) {
+      if (isTypeScriptConfig) {
+        const detail = cause instanceof Error ? cause.message : String(cause);
+        throw new Error(
+          `Failed to load TypeScript config.
 ${detail}
 
 If this looks like a decorator/metadata error, the tsconfig applied to your entity files is likely missing "experimentalDecorators" / "emitDecoratorMetadata".
 Make sure a tsconfig.json with those options sits next to your config file, or pass one explicitly with --tsconfig <path>.`,
-        { cause }
+          { cause }
+        );
+      }
+      throw cause;
+    }
+    if (mod.default === void 0) {
+      throw new Error("Config file must use a default export, e.g. `export default defineConfig({ ... })`.");
+    }
+    const config = mod.default;
+    if (typeof config === "function" || config instanceof Promise) {
+      throw new Error(
+        "Config file default export must be a configuration object, not a function or Promise.\nResolve it first, or use the programmatic API instead (see README)."
       );
     }
-    throw cause;
-  }
-  if (mod.default === void 0) {
-    throw new Error("Config file must use a default export, e.g. `export default defineConfig({ ... })`.");
-  }
-  const config = mod.default;
-  if (typeof config === "function" || config instanceof Promise) {
-    throw new Error(
-      "Config file default export must be a configuration object, not a function or Promise.\nResolve it first, or use the programmatic API instead (see README)."
-    );
-  }
-  if (typeof config !== "object" || config === null || Array.isArray(config)) {
-    throw new Error(
-      "Config file default export must be a configuration object, not a primitive value or array.\nExport a plain MikroORM options object instead."
-    );
+    if (typeof config !== "object" || config === null || Array.isArray(config)) {
+      throw new Error(
+        "Config file default export must be a configuration object, not a primitive value or array.\nExport a plain MikroORM options object instead."
+      );
+    }
+    const options = config;
+    const withPreferTs = isTypeScriptConfig && options.preferTs === void 0 ? { ...options, preferTs: true } : options;
+    shouldKeepTsxRegistered = loadOptions.keepTsxRegistered === true;
+    return withPreferTs;
+  } finally {
+    if (unregisterTsx !== void 0) {
+      if (shouldKeepTsxRegistered) {
+        activeTsxUnregister = unregisterTsx;
+      } else {
+        await unregisterTsx();
+      }
+    }
   }
-  const options = config;
-  const withPreferTs = isTypeScriptConfig && options.preferTs === void 0 ? { ...options, preferTs: true } : options;
-  return withPreferTs;
 }
 function formatErrorChain(err) {
   const lines = [];
@@ -1101,20 +1238,6 @@ ${formatFileSystemError(cause)}`, { cause });
 }
 function parseMermaidOptions(opts) {
   const { mermaidLayout, mermaidTheme } = opts;
-  if (mermaidLayout !== void 0 && !MERMAID_LAYOUTS.includes(mermaidLayout)) {
-    process.stderr.write(
-      `Error: Invalid --mermaid-layout "${mermaidLayout}". Allowed values: ${MERMAID_LAYOUTS.join(", ")}
-`
-    );
-    process.exit(1);
-  }
-  if (mermaidTheme !== void 0 && !MERMAID_THEMES.includes(mermaidTheme)) {
-    process.stderr.write(
-      `Error: Invalid --mermaid-theme "${mermaidTheme}". Allowed values: ${MERMAID_THEMES.join(", ")}
-`
-    );
-    process.exit(1);
-  }
   if (mermaidLayout === void 0 && mermaidTheme === void 0) {
     return void 0;
   }
@@ -1129,7 +1252,7 @@ async function run(opts) {
   const mermaid = parseMermaidOptions(opts);
   let ormOptions;
   try {
-    ormOptions = await loadOrmOptions(configPath, opts.tsconfig);
+    ormOptions = await loadOrmOptions(configPath, opts.tsconfig, { keepTsxRegistered: true });
   } catch (err) {
     process.stderr.write(
       `Error: Cannot load config: ${configPath}
@@ -1150,10 +1273,12 @@ ${err instanceof Error ? err.message : String(err)}
 `)
     });
   } catch (err) {
+    await unregisterActiveTsxLoader();
     process.stderr.write(`Error: ${formatDiscoveryError(err)}
 `);
     process.exit(1);
   }
+  await unregisterActiveTsxLoader();
   try {
     await writeMarkdownFile(outPath, markdown);
   } catch (err) {
@@ -1170,7 +1295,17 @@ var program = new Command().name("mikro-orm-markdown").description("Generate Mer
 ).option(
   "--src <paths...>",
   "Source .ts file globs to read JSDoc from when entities run from compiled .js (comments are stripped at build time)"
-).option("--mermaid-layout <layout>", `Mermaid layout engine injected as frontmatter (${MERMAID_LAYOUTS.join("|")})`).option("--mermaid-theme <theme>", `Mermaid theme injected as frontmatter (${MERMAID_THEMES.join("|")})`).action(run);
+).addOption(
+  new Option(
+    "--mermaid-layout <layout>",
+    `Mermaid layout engine injected as frontmatter (${MERMAID_LAYOUTS.join("|")})`
+  ).choices(MERMAID_LAYOUTS)
+).addOption(
+  new Option(
+    "--mermaid-theme <theme>",
+    `Mermaid theme injected as frontmatter (${MERMAID_THEMES.join("|")})`
+  ).choices(MERMAID_THEMES)
+).action(run);
 function isDirectCliExecution() {
   const entryPoint = process.argv[1];
   if (entryPoint === void 0) {