schemaToTypeScript() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/introspect-gel.ts lines 211–520

export const schemaToTypeScript = (schema: GelSchemaInternal, 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].replace('::', ''), 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.gel.push('type AnyGelColumn');
			}
			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) => 'gelPolicy',
			);

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

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

			const columnImports = Object.values(it.columns)
				.map((col) => {
					let patched: string = col.type?.replace('[]', '') ?? '';
					patched = patched.startsWith('time without time zone') ? 'localTime' : patched;

					patched = patched === 'double precision' ? 'doublePrecision' : patched;
					patched = patched.startsWith('edgedbt.bigint_t') ? 'bigintT' : patched;

					patched = patched.startsWith('jsonb') ? 'json' : patched;
					patched = patched.startsWith('edgedbt.timestamptz_t') ? 'timestamptz' : patched;
					patched = patched.startsWith('edgedbt.timestamp_t') ? 'timestamp' : patched;

					patched = patched.startsWith('edgedbt.relative_duration_t') ? 'relDuration' : patched;
					patched = patched.startsWith('bytea') ? 'bytes' : patched;

					patched = patched.startsWith('numeric') ? 'decimal' : patched;

					patched = patched.startsWith('edgedbt.duration_t') ? 'duration' : patched;
					patched = patched.startsWith('edgedbt.date_t') ? 'localDate' : patched;
					patched = patched.startsWith('edgedbt.date_duration_t') ? 'dateDuration' : patched;

					return patched;
				})
				.filter((type) => {
					return gelImportsList.has(type);
				});

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