sizuku 0.6.4 → 0.6.5

package/dist/index.mjs

@@ -432,6 +432,66 @@ function extractRelationsFromSchema(code) {
 	});
 }
 //#endregion
+//#region src/helper/parse-relation.ts
+const RELATIONSHIPS = {
+	"zero-one": "|o",
+	one: "||",
+	"zero-many": "}o",
+	many: "}|"
+};
+function isRelationshipType(type) {
+	return type === "zero-one" || type === "one" || type === "zero-many" || type === "many";
+}
+const CARDINALITY = "(?:zero-one|one|zero-many|many)";
+const RELATION_TYPE = `${CARDINALITY}-to-${CARDINALITY}(?:-optional)?`;
+const RELATION_RE = new RegExp(String.raw`^@relation\s+(\w+)\.(\w+)\s+(\w+)\.(\w+)\s+(${RELATION_TYPE})$`);
+function parseRelation(line) {
+	const match = line.trim().replace(/^\/\/\/\s*/, "").match(RELATION_RE);
+	if (!match) return null;
+	const [, fromModel, fromField, toModel, toField, type] = match;
+	return {
+		fromModel,
+		fromField,
+		toModel,
+		toField,
+		type
+	};
+}
+function makeRelationLine(input) {
+	const parts = input.split("-to-");
+	if (parts.length !== 2) return {
+		ok: false,
+		error: `Invalid input format: ${input}`
+	};
+	const [to, optionalFlag] = parts[1].includes("-optional") ? [parts[1].replace("-optional", ""), "optional"] : [parts[1], ""];
+	const from = parts[0];
+	if (!isRelationshipType(from)) return {
+		ok: false,
+		error: `Invalid relationship: ${from}`
+	};
+	if (!isRelationshipType(to)) return {
+		ok: false,
+		error: `Invalid relationship: ${to}`
+	};
+	return {
+		ok: true,
+		value: `${RELATIONSHIPS[from]}${optionalFlag === "optional" ? ".." : "--"}${RELATIONSHIPS[to]}`
+	};
+}
+function extractRelationsFromAnnotations(code) {
+	return code.map(parseRelation).filter((r) => r !== null).flatMap((r) => {
+		const result = makeRelationLine(r.type);
+		if (!result.ok) return [];
+		return [{
+			fromModel: r.fromModel,
+			fromField: r.fromField,
+			toModel: r.toModel,
+			toField: r.toField,
+			symbol: result.value
+		}];
+	});
+}
+//#endregion
 //#region src/generator/arktype.ts
 function undeclared(objectType) {
 	if (objectType === "strict") return "\"+\":\"reject\",";
@@ -642,8 +702,19 @@ async function emit(code, dir, output) {
 //#region src/generator/mermaid-er.ts
 function mermaidER(code) {
 	const tables = parseTableInfo(code);
-	const relations = extractRelationsFromSchema(code);
-	const relationLines = [...new Set(relations.map((r) => `    ${r.fromModel} ||--${r.isRequired ? "}|" : "}o"} ${r.toModel} : "(${r.fromField}) - (${r.toField})"`))];
+	const inferred = extractRelationsFromSchema(code);
+	const annotated = extractRelationsFromAnnotations(code);
+	const key = (r) => `${r.fromModel}.${r.fromField}->${r.toModel}.${r.toField}`;
+	const merged = /* @__PURE__ */ new Map();
+	for (const r of inferred) merged.set(key(r), {
+		fromModel: r.fromModel,
+		fromField: r.fromField,
+		toModel: r.toModel,
+		toField: r.toField,
+		symbol: r.isRequired ? "||--}|" : "||--}o"
+	});
+	for (const r of annotated) merged.set(key(r), r);
+	const relationLines = [...merged.values()].map((r) => `    ${r.fromModel} ${r.symbol} ${r.toModel} : "(${r.fromField}) - (${r.toField})"`);
 	const tableDefinitions = tables.flatMap((table) => [
 		`    ${table.name} {`,
 		...table.fields.map((field) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sizuku",
-  "version": "0.6.4",
+  "version": "0.6.5",
   "description": "Sizuku is a tool that generates validation schemas for Zod and Valibot, as well as ER diagrams and DBML, from Drizzle schemas annotated with comments.",
   "keywords": [
     "dbml",