orchid-orm 1.77.0 → 1.77.2

package/dist/migrations/index.mjs

@@ -3,7 +3,6 @@ import { ArrayColumn, DomainColumn, EnumColumn, RawSql, UnknownColumn, VirtualCo
 import path from "node:path";
 import { pathToFileURL } from "node:url";
 import fs from "node:fs/promises";
-import typescript from "typescript";
 export * from "rake-db";
 const viewDataToSql = (viewData, viewName) => {
 	return sqlToRawSql(viewDataToQuerySql(viewData, viewName));
@@ -328,8 +327,9 @@ const compareColumns = async (adapter, domainsMap, ast, currentSchema, compareSq
 		if (!typeCasts.get(dbBaseType)?.has(codeBaseType)) {
 			if (!(dbColumn instanceof EnumColumn) || !(codeColumn instanceof EnumColumn) || !deepCompare(dbColumn.options, codeColumn.options)) {
 				if (verifying) throw new AbortSignal();
+				const tableName = concatSchemaAndName(changeTableData.changeTableAst);
 				if (await promptSelect({
-					message: `Cannot cast type of ${concatSchemaAndName(changeTableData.changeTableAst)}'s column ${key} from ${dbType} to ${codeType}`,
+					message: `Cannot cast type of ${tableName}'s column ${key} from ${dbType} to ${codeType}`,
 					options: [`${colors.yellowBold(`-/+`)} recreate the column, existing data will be ${colors.red("lost")}`, `write migration manually`]
 				})) throw new AbortSignal();
 				dbColumn.data.name = codeColumn.data.name;
@@ -409,7 +409,7 @@ JOIN pg_type AS t ON t.oid = casttarget`);
 		for (const [source, target] of rows) {
 			const set = directTypeCasts.get(source);
 			if (set) set.add(target);
-			else directTypeCasts.set(source, new Set([target]));
+			else directTypeCasts.set(source, /* @__PURE__ */ new Set([target]));
 		}
 		typeCasts = /* @__PURE__ */ new Map();
 		for (const [type, directSet] of directTypeCasts.entries()) {
@@ -540,7 +540,8 @@ const makeComparableDomain = (currentSchema, schemaName, name, column) => {
 		inner = inner.data.item;
 		arrayDims++;
 	}
-	const [typeSchema = "pg_catalog", type] = getSchemaAndTableFromName(currentSchema, getColumnDbType(inner, currentSchema));
+	const fullType = getColumnDbType(inner, currentSchema);
+	const [typeSchema = "pg_catalog", type] = getSchemaAndTableFromName(currentSchema, fullType);
 	return {
 		schemaName,
 		name,
@@ -779,7 +780,7 @@ const processPrimaryKey = (config, changeTableData) => {
 };
 const changePrimaryKey = (config, columnsPrimaryKey, { codeTable, dbTableData: { primaryKey: dbPrimaryKey }, changeTableAst: { shape, add, drop }, changingColumns }) => {
 	const tablePrimaryKey = codeTable.internal.tableData.primaryKey;
-	const primaryKey = [...new Set([...columnsPrimaryKey, ...(config.snakeCase ? tablePrimaryKey?.columns.map((key) => ({
+	const primaryKey = [.../* @__PURE__ */ new Set([...columnsPrimaryKey, ...(config.snakeCase ? tablePrimaryKey?.columns.map((key) => ({
 		key,
 		name: toSnakeCase(key)
 	})) : tablePrimaryKey?.columns.map((key) => ({
@@ -2187,7 +2188,7 @@ const processDefaultPrivileges = (ast, dbStructure, { internal: { roles } }) =>
 				const codeObj = codePrivilege.objectConfigs.find((o) => o.object === objectType);
 				const dbPrivs = dbObj?.privilegeConfigs ?? [];
 				const codePrivs = codeObj?.privilegeConfigs ?? [];
-				const collapsedDbPrivs = collapsePrivilegesToAll({
+				const collapsedDbPrivilege = {
 					owner: dbPrivilege.owner,
 					grantee: dbPrivilege.grantee,
 					schema: dbPrivilege.schema,
@@ -2195,7 +2196,8 @@ const processDefaultPrivileges = (ast, dbStructure, { internal: { roles } }) =>
 						object: objectType,
 						privilegeConfigs: dbPrivs
 					}]
-				}, objectTypeToAllPrivileges).objectConfigs[0]?.privilegeConfigs ?? [];
+				};
+				const collapsedDbPrivs = collapsePrivilegesToAll(collapsedDbPrivilege, objectTypeToAllPrivileges).objectConfigs[0]?.privilegeConfigs ?? [];
 				const toGrant = codePrivs.filter((cp) => !collapsedDbPrivs.some((dp) => privilegeConfigsMatch(cp, dp, objectType, objectTypeToAllPrivileges)));
 				const toRevoke = dbPrivs.filter((dp) => !codePrivs.some((cp) => privilegeConfigsMatch(cp, dp, objectType, objectTypeToAllPrivileges)));
 				if (toGrant.length) {
@@ -2619,9 +2621,7 @@ const dbViewOptionsToAst = (view) => {
 			case "security_barrier":
 				options.securityBarrier = value === "true";
 				break;
-			case "security_invoker":
-				options.securityInvoker = value === "true";
-				break;
+			case "security_invoker": options.securityInvoker = value === "true";
 		}
 	}
 	return options;
@@ -3046,10 +3046,11 @@ const report = (ast, config, currentSchema) => {
 			break;
 		}
 		case "policy": {
-			const summary = formatPolicySummary(a, dbItemName({
+			const table = dbItemName({
 				schema: a.schema,
 				name: a.table
-			}, currentSchema));
+			}, currentSchema);
+			const summary = formatPolicySummary(a, table);
 			const message = a.action === "create" ? `${green("+ create policy")} ${a.name}: ${summary}` : `${red("- drop policy")} ${a.name}: ${summary}`;
 			code.push(message);
 			break;
@@ -3601,13 +3602,11 @@ const addTrailingSemicolon = (code) => {
 function importsToCode(imports) {
 	return Object.entries(imports).map(([from, name]) => `import { ${name} } from '${from}';`).join("\n");
 }
-const { createSourceFile, ScriptTarget, SyntaxKind } = typescript;
 const appCodeGenUpdateDbFile = async (dbPath, tables, extensions, domains, currentSchema) => {
 	const content = await fs.readFile(dbPath, "utf-8");
-	const statements = getTsStatements(content);
-	const importName = getOrchidOrmImportName(statements);
-	if (!importName) throw new Error(`Main file does not contain import of orchid-orm`);
-	const { config, tablesList } = getOrchidOrmArgs(importName, statements);
+	const importInfo = getOrchidOrmImport(content);
+	if (!importInfo) throw new Error(`Main file does not contain import of orchid-orm`);
+	const { config, tablesList } = getOrchidOrmArgs(content, importInfo.importName);
 	const changes = [];
 	let replacedConfig;
 	if (extensions.length || domains.length) {
@@ -3623,32 +3622,18 @@ const appCodeGenUpdateDbFile = async (dbPath, tables, extensions, domains, curre
 		}, ast.baseType.data.name ?? "")},`).join("\n    ")}\n  },`;
 		replacedConfig = code + "\n}";
 	}
-	const tablesChanges = makeTablesListChanges(content, statements, tablesList, tables, dbPath);
+	const tablesChanges = makeTablesListChanges(content, importInfo.lastImportEnd, tablesList, tables, dbPath);
 	if (tablesChanges) addChange(content, changes, tablesChanges.imports.pos, tablesChanges.imports.text);
 	if (replacedConfig) replaceContent(content, changes, config.pos, config.end, replacedConfig);
 	if (tablesChanges) addChange(content, changes, tablesChanges.tablesList.pos, tablesChanges.tablesList.text);
 	return applyChanges(content, changes);
 };
-const getTsStatements = (content) => {
-	return createSourceFile("file.ts", content, ScriptTarget.Latest, true).statements;
-};
-const getOrchidOrmImportName = (statements) => {
-	for (const node of statements) {
-		if (node.kind !== SyntaxKind.ImportDeclaration) continue;
-		const imp = node;
-		if (imp.moduleSpecifier.getText().slice(1, -1) !== "orchid-orm") continue;
-		if (!imp.importClause) continue;
-		const elements = imp.importClause.namedBindings?.elements;
-		if (!elements);
-		for (const element of elements) if (element.name.escapedText === "orchidORM" || element.propertyName && "escapedText" in element.propertyName && element.propertyName.escapedText === "orchidORM") return element.name.escapedText.toString();
-	}
-};
-const makeTablesListChanges = (content, statements, object, tables, dbPath) => {
+const makeTablesListChanges = (content, importPos, object, tables, dbPath) => {
 	const spaces = getTablesListSpaces(content, object);
 	let imports = "";
 	let tablesList = "";
-	const prependComma = object.properties.length && !object.properties.hasTrailingComma;
-	const tablesListNewLine = content.slice(object.properties.end, object.end).includes("\n");
+	const prependComma = object.hasProperties && !object.hasTrailingComma;
+	const tablesListNewLine = content.slice(object.propertiesEnd, object.end).includes("\n");
 	const tablesArr = Object.values(tables);
 	for (let i = 0; i < tablesArr.length; i++) {
 		const { path, className, key } = tablesArr[i];
@@ -3658,15 +3643,13 @@ const makeTablesListChanges = (content, statements, object, tables, dbPath) => {
 		if (i === tablesArr.length - 1 && !tablesListNewLine) tablesList += `\n${spaces}`;
 	}
 	if (!imports.length) return;
-	let importPos = 0;
-	for (const node of statements) if (node.kind === SyntaxKind.ImportDeclaration) importPos = node.end;
 	return {
 		imports: {
 			pos: importPos,
 			text: imports
 		},
 		tablesList: {
-			pos: object.properties.end,
+			pos: object.propertiesEnd,
 			text: tablesList
 		}
 	};
@@ -3675,24 +3658,165 @@ const getTablesListSpaces = (content, object) => {
 	const lines = content.slice(0, object.end).split("\n");
 	return lines[lines.length - 1].match(/^\s+/)?.[0] || "";
 };
-const getOrchidOrmArgs = (importName, statements) => {
-	for (const v of statements) {
-		if (v.kind !== SyntaxKind.VariableStatement) continue;
-		for (const node of v.declarationList.declarations) {
-			const call = node.initializer;
-			if (call?.kind !== SyntaxKind.CallExpression) continue;
-			if (call.expression.getText() !== importName) continue;
-			if (call.arguments.length !== 2) throw new Error("Invalid number of arguments when initializing orchid orm");
-			const object = call.arguments[1];
-			if (object?.kind !== SyntaxKind.ObjectLiteralExpression) throw new Error("Second argument of orchidORM must be an object literal");
-			return {
-				config: call.arguments[0],
-				tablesList: object
-			};
+const getOrchidOrmArgs = (content, importName) => {
+	const call = findCall(content, importName);
+	if (!call) throw new Error("List of tables is not found in main file");
+	const closeParen = findClosingBracket(content, call.openParen);
+	if (closeParen === -1) throw new Error("List of tables is not found in main file");
+	const firstComma = findTopLevelComma(content, call.openParen + 1, closeParen);
+	if (firstComma === -1) throw new Error("Invalid number of arguments when initializing orchid orm");
+	if (findTopLevelComma(content, firstComma + 1, closeParen) !== -1) throw new Error("Invalid number of arguments when initializing orchid orm");
+	const config = makeTrimmedRange(content, call.openParen + 1, firstComma);
+	const tablesList = getObjectRange(content, firstComma + 1, closeParen);
+	if (!tablesList) throw new Error("Second argument of orchidORM must be an object literal");
+	return {
+		config,
+		tablesList
+	};
+};
+const getOrchidOrmImport = (content) => {
+	let lastImportEnd = 0;
+	let importName;
+	const importRegex = /(^|\n)\s*import\s+([\s\S]*?)\s+from\s+['"]([^'"]+)['"]\s*;?/g;
+	let match;
+	while (match = importRegex.exec(content)) {
+		lastImportEnd = importRegex.lastIndex;
+		if (match[3] !== "orchid-orm") continue;
+		const named = match[2].match(/\{([\s\S]*)\}/);
+		if (!named) continue;
+		for (const part of named[1].split(",")) {
+			const item = part.trim();
+			const alias = item.match(/^orchidORM\s+as\s+([A-Za-z_$][\w$]*)$/);
+			if (alias) {
+				importName = alias[1];
+				continue;
+			}
+			if (item === "orchidORM") importName = "orchidORM";
+		}
+	}
+	if (!importName) return;
+	return {
+		importName,
+		lastImportEnd
+	};
+};
+const findCall = (content, importName) => {
+	let index = 0;
+	while (index < content.length) {
+		const found = content.indexOf(importName, index);
+		if (found === -1) return;
+		const before = content[found - 1];
+		const after = content[found + importName.length];
+		if ((!before || !isIdentifierChar(before)) && (!after || !isIdentifierChar(after))) {
+			const openParen = skipWhitespace(content, found + importName.length);
+			if (content[openParen] === "(") return { openParen };
 		}
+		index = found + importName.length;
 	}
 	throw new Error("List of tables is not found in main file");
 };
+const getObjectRange = (content, from, to) => {
+	const pos = skipWhitespace(content, from);
+	if (content[pos] !== "{") return;
+	const end = findClosingBracket(content, pos);
+	if (end === -1 || skipWhitespace(content, end + 1) < to) return;
+	const lastBodyChar = findLastNonWhitespace(content, pos + 1, end);
+	const hasProperties = lastBodyChar !== -1;
+	const hasTrailingComma = hasProperties && content[lastBodyChar] === ",";
+	const propertiesEnd = hasProperties ? lastBodyChar + 1 : pos + 1;
+	return {
+		pos,
+		end: end + 1,
+		propertiesEnd,
+		hasProperties,
+		hasTrailingComma
+	};
+};
+const makeTrimmedRange = (content, from, to) => {
+	return {
+		pos: skipWhitespace(content, from),
+		end: trimWhitespaceEnd(content, from, to)
+	};
+};
+const findTopLevelComma = (content, from, to) => {
+	let depth = 0;
+	for (let i = from; i < to; i++) {
+		const skipped = skipCodeFragment(content, i);
+		if (skipped !== i) {
+			i = skipped - 1;
+			continue;
+		}
+		const char = content[i];
+		if (char === "(" || char === "{" || char === "[") depth++;
+		else if (char === ")" || char === "}" || char === "]") depth--;
+		else if (char === "," && depth === 0) return i;
+	}
+	return -1;
+};
+const findClosingBracket = (content, open) => {
+	const close = getClosingBracket(content[open]);
+	if (!close) return -1;
+	let depth = 1;
+	for (let i = open + 1; i < content.length; i++) {
+		const skipped = skipCodeFragment(content, i);
+		if (skipped !== i) {
+			i = skipped - 1;
+			continue;
+		}
+		const char = content[i];
+		if (char === content[open]) depth++;
+		else if (char === close) {
+			depth--;
+			if (depth === 0) return i;
+		}
+	}
+	return -1;
+};
+const getClosingBracket = (char) => {
+	if (char === "(") return ")";
+	if (char === "{") return "}";
+	if (char === "[") return "]";
+};
+const skipCodeFragment = (content, i) => {
+	const char = content[i];
+	if (char === "'" || char === "\"" || char === "`") return skipQuoted(content, i, char);
+	if (char === "/" && content[i + 1] === "/") {
+		const nextLine = content.indexOf("\n", i + 2);
+		return nextLine === -1 ? content.length : nextLine + 1;
+	}
+	if (char === "/" && content[i + 1] === "*") {
+		const end = content.indexOf("*/", i + 2);
+		return end === -1 ? content.length : end + 2;
+	}
+	return i;
+};
+const skipQuoted = (content, i, quote) => {
+	for (let j = i + 1; j < content.length; j++) {
+		if (content[j] === "\\") {
+			j++;
+			continue;
+		}
+		if (content[j] === quote) return j + 1;
+	}
+	return content.length;
+};
+const skipWhitespace = (content, from) => {
+	let i = from;
+	while (/\s/.test(content[i])) i++;
+	return i;
+};
+const trimWhitespaceEnd = (content, from, to) => {
+	let i = to;
+	while (i > from && /\s/.test(content[i - 1])) i--;
+	return i;
+};
+const findLastNonWhitespace = (content, from, to) => {
+	for (let i = to - 1; i >= from; i--) if (!/\s/.test(content[i])) return i;
+	return -1;
+};
+const isIdentifierChar = (char) => {
+	return /[A-Za-z0-9_$]/.test(char);
+};
 const addChange = (content, changes, at, text, end = at) => {
 	if (changes.length === 0) changes.push([0, at], text, [end, content.length]);
 	else {
@@ -3720,7 +3844,8 @@ const pull = async (adapters, config) => {
 	const [adapter] = adapters;
 	const adapterSchema = adapter.getSchema();
 	const currentSchema = (typeof adapterSchema === "function" ? adapterSchema() : adapterSchema) ?? "public";
-	const asts = await structureToAst(makeStructureToAstCtx(config, currentSchema), adapter, config);
+	const ctx = makeStructureToAstCtx(config, currentSchema);
+	const asts = await structureToAst(ctx, adapter, config);
 	const tableInfos = getTableInfos(asts, config);
 	const exclusiveWriteOptions = { flag: "wx" };
 	const pendingFileWrites = [];
@@ -3746,9 +3871,7 @@ const pull = async (adapters, config) => {
 				version: ast.version
 			});
 			break;
-		case "domain":
-			domains.push(ast);
-			break;
+		case "domain": domains.push(ast);
 	}
 	if (!firstTable && !extensions.length && !domains.length) {
 		await adapter.close();