@anthonyalayo/drizzle-kit 1.0.1

package/src/serializer/pgImports.ts

@@ -0,0 +1,106 @@
+import { is } from 'drizzle-orm';
+import {
+	AnyPgTable,
+	isPgDomain,
+	isPgEnum,
+	isPgMaterializedView,
+	isPgSequence,
+	isPgView,
+	PgDomain,
+	PgEnum,
+	PgMaterializedView,
+	PgPolicy,
+	PgRole,
+	PgSchema,
+	PgSequence,
+	PgTable,
+	PgView,
+} from 'drizzle-orm/pg-core';
+import { safeRegister } from '../cli/commands/utils';
+
+export const prepareFromExports = (exports: Record<string, unknown>) => {
+	const tables: AnyPgTable[] = [];
+	const domains: PgDomain<any>[] = [];
+	const enums: PgEnum<any>[] = [];
+	const schemas: PgSchema[] = [];
+	const sequences: PgSequence[] = [];
+	const roles: PgRole[] = [];
+	const policies: PgPolicy[] = [];
+	const views: PgView[] = [];
+	const matViews: PgMaterializedView[] = [];
+
+	const i0values = Object.values(exports);
+	i0values.forEach((t) => {
+		// TODO confirm this makes sense
+		if (isPgDomain(t)) {
+			domains.push(t);
+		}
+
+		if (isPgEnum(t)) {
+			enums.push(t);
+			return;
+		}
+		if (is(t, PgTable)) {
+			tables.push(t);
+		}
+
+		if (is(t, PgSchema)) {
+			schemas.push(t);
+		}
+
+		if (isPgView(t)) {
+			views.push(t);
+		}
+
+		if (isPgMaterializedView(t)) {
+			matViews.push(t);
+		}
+
+		if (isPgSequence(t)) {
+			sequences.push(t);
+		}
+
+		if (is(t, PgRole)) {
+			roles.push(t);
+		}
+
+		if (is(t, PgPolicy)) {
+			policies.push(t);
+		}
+	});
+
+	return { tables, domains, enums, schemas, sequences, views, matViews, roles, policies };
+};
+
+export const prepareFromPgImports = async (imports: string[]) => {
+	const tables: AnyPgTable[] = [];
+	const domains: PgDomain<any>[] = [];
+	const enums: PgEnum<any>[] = [];
+	const schemas: PgSchema[] = [];
+	const sequences: PgSequence[] = [];
+	const views: PgView[] = [];
+	const roles: PgRole[] = [];
+	const policies: PgPolicy[] = [];
+	const matViews: PgMaterializedView[] = [];
+
+	const { unregister } = await safeRegister();
+	for (let i = 0; i < imports.length; i++) {
+		const it = imports[i];
+
+		const i0: Record<string, unknown> = require(`${it}`);
+		const prepared = prepareFromExports(i0);
+
+		tables.push(...prepared.tables);
+		domains.push(...prepared.domains);
+		enums.push(...prepared.enums);
+		schemas.push(...prepared.schemas);
+		sequences.push(...prepared.sequences);
+		views.push(...prepared.views);
+		matViews.push(...prepared.matViews);
+		roles.push(...prepared.roles);
+		policies.push(...prepared.policies);
+	}
+	unregister();
+
+	return { tables: Array.from(new Set(tables)), domains, enums, schemas, sequences, views, matViews, roles, policies };
+};