schemaToTypeScript() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/introspect-pg.ts lines 309–636

export const schemaToTypeScript = (schema: PgSchemaInternal, casing: Casing) => {
	// collectFKs
	Object.values(schema.tables).forEach((table) => {
		Object.values(table.foreignKeys).forEach((fk) => {
			const relation = `${fk.tableFrom}-${fk.tableTo}`;
			relations.add(relation);
		});
	});

	const schemas = Object.fromEntries(
		Object.entries(schema.schemas).map((it) => {
			return [it[0], withCasing(it[1], casing)];
		}),
	);

	const enumTypes = Object.values(schema.enums).reduce((acc, cur) => {
		acc.add(`${cur.schema}.${cur.name}`);
		return acc;
	}, new Set<string>());

	const imports = Object.values(schema.tables).reduce(
		(res, it) => {
			const idxImports = Object.values(it.indexes).map((idx) => (idx.isUnique ? 'uniqueIndex' : 'index'));
			const fkImpots = Object.values(it.foreignKeys).map((it) => 'foreignKey');
			if (Object.values(it.foreignKeys).some((it) => isCyclic(it) && !isSelf(it))) {
				res.pg.push('type AnyPgColumn');
			}
			const pkImports = Object.values(it.compositePrimaryKeys).map((it) => 'primaryKey');
			const uniqueImports = Object.values(it.uniqueConstraints).map((it) => 'unique');

			const checkImports = Object.values(it.checkConstraints).map(
				(it) => 'check',
			);

			const policiesImports = Object.values(it.policies).map(
				(it) => 'pgPolicy',
			);

			if (it.schema && it.schema !== 'public' && it.schema !== '') {
				res.pg.push('pgSchema');
			}

			res.pg.push(...idxImports);
			res.pg.push(...fkImpots);
			res.pg.push(...pkImports);
			res.pg.push(...uniqueImports);
			res.pg.push(...policiesImports);
			res.pg.push(...checkImports);

			const columnImports = Object.values(it.columns)
				.map((col) => {
					let patched: string = (importsPatch[col.type] || col.type).replace('[]', '');
					patched = patched === 'double precision' ? 'doublePrecision' : patched;
					patched = patched.startsWith('varchar(') ? 'varchar' : patched;
					patched = patched.startsWith('char(') ? 'char' : patched;
					patched = patched.startsWith('numeric(') ? 'numeric' : patched;
					patched = patched.startsWith('time(') ? 'time' : patched;
					patched = patched.startsWith('timestamp(') ? 'timestamp' : patched;
					patched = patched.startsWith('vector(') ? 'vector' : patched;
					patched = patched.startsWith('geometry(') ? 'geometry' : patched;
					return patched;
				})
				.filter((type) => {
					return pgImportsList.has(type);
				});

			res.pg.push(...columnImports);
			return res;
		},
		{ pg: [] as string[] },
	);

	Object.values(schema.views).forEach((it) => {
		if (it.schema && it.schema !== 'public' && it.schema !== '') {
			imports.pg.push('pgSchema');
		} else if (it.schema === 'public') {
			it.materialized ? imports.pg.push('pgMaterializedView') : imports.pg.push('pgView');
		}

		Object.values(it.columns).forEach(() => {
			const columnImports = Object.values(it.columns)