schemaToTypeScript() — drizzle-orm Function Reference

Source Code

drizzle-kit/src/introspect-sqlite.ts lines 73–246

export const schemaToTypeScript = (
	schema: SQLiteSchemaInternal,
	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 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');
			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',
			);

			res.sqlite.push(...idxImports);
			res.sqlite.push(...fkImpots);
			res.sqlite.push(...pkImports);
			res.sqlite.push(...uniqueImports);
			res.sqlite.push(...checkImports);

			const columnImports = Object.values(it.columns)
				.map((col) => {
					return col.type;
				})
				.filter((type) => {
					return sqliteImportsList.has(type);
				});

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

	Object.values(schema.views).forEach((it) => {
		imports.sqlite.push('sqliteView');

		const columnImports = Object.values(it.columns)
			.map((col) => {
				return col.type;
			})
			.filter((type) => {
				return sqliteImportsList.has(type);
			});

		imports.sqlite.push(...columnImports);
	});

	const tableStatements = Object.values(schema.tables).map((table) => {
		const func = 'sqliteTable';
		let statement = '';
		if (imports.sqlite.includes(withCasing(table.name, casing))) {
			statement = `// Table name is in conflict with ${
				withCasing(
					table.name,
					casing,
				)
			} import.\n// Please change to any other name, that is not in imports list\n`;
		}
		statement += `export const ${withCasing(table.name, casing)} = ${func}("${table.name}", {\n`;
		statement += createTableColumns(
			Object.values(table.columns),
			Object.values(table.foreignKeys),
			casing,
		);
		statement += '}';

		// more than 2 fields or self reference or cyclic
		const filteredFKs = Object.values(table.foreignKeys).filter((it) => {