@dineug/erd-editor 4.2.0 → 5.0.0

package/dist/utils/generator-code/swift.js

@@ -0,0 +1,222 @@
+import { query } from "../../erd-editor-schema/dist/query/index.js";
+import { ColumnOption, Database } from "../../constants/schema.js";
+import { bHas } from "../bit.js";
+import { orderByNameASC } from "../tableName.js";
+import { getNameCase, splitLines } from "./utils.js";
+import { POSTGRES_OBJECT_ID_TYPES, getColumnType } from "./columnTypes.js";
+//#region src/utils/generator-code/swift.ts
+var SWIFT_HEADER = ["", "import Foundation"];
+var INDENT = "    ";
+var STRUCT_HEAD = "nonisolated struct";
+var CONFORMANCES = "Codable, Hashable, Sendable";
+var SWIFT_KEYWORDS = /* @__PURE__ */ new Set([
+	"Any",
+	"Self",
+	"_",
+	"as",
+	"associatedtype",
+	"break",
+	"case",
+	"catch",
+	"class",
+	"continue",
+	"default",
+	"defer",
+	"deinit",
+	"do",
+	"else",
+	"enum",
+	"extension",
+	"fallthrough",
+	"false",
+	"fileprivate",
+	"for",
+	"func",
+	"guard",
+	"if",
+	"import",
+	"in",
+	"init",
+	"inout",
+	"internal",
+	"is",
+	"let",
+	"nil",
+	"operator",
+	"precedencegroup",
+	"private",
+	"protocol",
+	"public",
+	"repeat",
+	"rethrows",
+	"return",
+	"self",
+	"static",
+	"struct",
+	"subscript",
+	"super",
+	"switch",
+	"throw",
+	"throws",
+	"true",
+	"try",
+	"typealias",
+	"var",
+	"where",
+	"while"
+]);
+var MEMBER_KEYWORDS = /* @__PURE__ */ new Set(["Protocol", "Type"]);
+var TYPE_NAMES = /* @__PURE__ */ new Set([
+	"Bool",
+	"Codable",
+	"CodingKey",
+	"Data",
+	"Date",
+	"Decimal",
+	"Double",
+	"Float",
+	"Hashable",
+	"Int16",
+	"Int32",
+	"Int64",
+	"Int8",
+	"Sendable",
+	"String",
+	"UInt16",
+	"UInt32",
+	"UInt64",
+	"UInt8",
+	"UUID"
+]);
+var CODING_KEYS = "CodingKeys";
+var RESERVED_KEYS = /* @__PURE__ */ new Set([CODING_KEYS, "RawValue"]);
+var SWIFT_SCALARS = {
+	bool: "Bool",
+	i8: "Int8",
+	i16: "Int16",
+	i32: "Int32",
+	i64: "Int64",
+	u8: "UInt8",
+	u16: "UInt16",
+	u32: "UInt32",
+	u64: "UInt64",
+	f32: "Float",
+	f64: "Double",
+	decimal: "Decimal",
+	string: "String",
+	bytes: "Data",
+	uuid: "UUID",
+	json: "String",
+	date: "Date",
+	time: "String",
+	timeTz: "String",
+	dateTime: "Date",
+	dateTimeUtc: "Date",
+	dateTimeOffset: "Date",
+	interval: "String"
+};
+var NUL = String.fromCharCode(0);
+function createCode(state) {
+	const { doc: { tableIds }, collections } = state;
+	const tables = query(collections).collection("tableEntities").selectByIds(tableIds).sort(orderByNameASC);
+	if (tables.length === 0) return "";
+	const buffer = [...SWIFT_HEADER];
+	tables.forEach((table) => {
+		buffer.push("");
+		formatTable(state, {
+			buffer,
+			table
+		});
+	});
+	buffer.push("");
+	return buffer.join("\n");
+}
+function createTableCode(state, table) {
+	const buffer = [...SWIFT_HEADER, ""];
+	formatTable(state, {
+		buffer,
+		table
+	});
+	buffer.push("");
+	return buffer.join("\n");
+}
+function formatTable(state, { buffer, table }) {
+	const { settings: { tableNameCase, columnNameCase, database }, collections } = state;
+	const head = `${STRUCT_HEAD} ${toStructName(getNameCase(table.name, tableNameCase))}: ${CONFORMANCES}`;
+	const columns = query(collections).collection("tableColumnEntities").selectByIds(table.columnIds);
+	formatComment(buffer, "", table.comment);
+	if (columns.length === 0) {
+		buffer.push(`${head} {}`);
+		return;
+	}
+	const properties = columns.map((column) => toProperty(column, columnNameCase, database));
+	buffer.push(`${head} {`);
+	properties.forEach(({ name, type, column }) => {
+		formatComment(buffer, INDENT, column.comment);
+		buffer.push(`${INDENT}var ${name}: ${type}`);
+	});
+	formatCodingKeys(buffer, properties);
+	buffer.push("}");
+}
+function toProperty(column, columnNameCase, database) {
+	const key = toPropertyKey(getNameCase(column.name, columnNameCase));
+	const type = swiftType(column.dataType, database);
+	const isNullable = !bHas(column.options, ColumnOption.primaryKey) && !bHas(column.options, ColumnOption.notNull);
+	return {
+		name: toPropertyName(key),
+		key,
+		type: isNullable ? `${type}?` : type,
+		column
+	};
+}
+function formatCodingKeys(buffer, properties) {
+	if (properties.every(({ key, column }) => key === column.name)) return;
+	buffer.push("");
+	buffer.push(`${INDENT}enum ${CODING_KEYS}: String, CodingKey {`);
+	properties.forEach(({ name, key, column }) => {
+		buffer.push(key === column.name ? `${INDENT}${INDENT}case ${name}` : `${INDENT}${INDENT}case ${name} = ${toStringLiteral(column.name)}`);
+	});
+	buffer.push(`${INDENT}}`);
+}
+function swiftType(dataType, database) {
+	const columnType = getColumnType(dataType, database);
+	let type = swiftScalar(columnType, database);
+	for (let depth = 0; depth < columnType.arrayDepth; depth++) type = `[${type}]`;
+	return type;
+}
+function swiftScalar({ scalar, base, isMoney }, database) {
+	if (database === Database.MariaDB && scalar === "uuid") return "String";
+	if (database === Database.PostgreSQL && (isMoney || POSTGRES_OBJECT_ID_TYPES.has(base))) return "String";
+	return SWIFT_SCALARS[scalar];
+}
+function toStructName(name) {
+	if (TYPE_NAMES.has(name)) return `${name}_`;
+	return SWIFT_KEYWORDS.has(name) ? `\`${name}\`` : name;
+}
+function toPropertyKey(name) {
+	return RESERVED_KEYS.has(name) ? `${name}_` : name;
+}
+function toPropertyName(key) {
+	return SWIFT_KEYWORDS.has(key) || MEMBER_KEYWORDS.has(key) ? `\`${key}\`` : key;
+}
+function toStringLiteral(value) {
+	let literal = "\"";
+	for (const character of value) {
+		const code = character.charCodeAt(0);
+		if (character === "\"" || character === "\\") literal += `\\${character}`;
+		else if (code < 32 || code === 127) literal += escapeCodePoint(character);
+		else literal += character;
+	}
+	return `${literal}"`;
+}
+function formatComment(buffer, indent, comment) {
+	const lines = splitLines(comment.replaceAll(NUL, escapeCodePoint(NUL))).map((line) => line.trimEnd());
+	while (lines[0] === "") lines.shift();
+	while (lines[lines.length - 1] === "") lines.pop();
+	lines.forEach((line) => buffer.push(line === "" ? `${indent}///` : `${indent}/// ${line}`));
+}
+function escapeCodePoint(character) {
+	return `\\u{${character.charCodeAt(0).toString(16).toUpperCase()}}`;
+}
+//#endregion
+export { CONFORMANCES, INDENT, STRUCT_HEAD, SWIFT_HEADER, SWIFT_KEYWORDS, TYPE_NAMES, createCode, createTableCode, formatComment, formatTable, swiftType, toPropertyKey, toPropertyName, toStringLiteral, toStructName };