drizzle-orm
Version:
Drizzle ORM package for SQL databases
71 lines • 2.29 kB
JavaScript
import { is } from "../entity.js";
import { SQL } from "../sql/sql.js";
import { Subquery } from "../subquery.js";
import { Table } from "../table.js";
import { ViewBaseConfig } from "../view-common.js";
import { CheckBuilder } from "./checks.js";
import { ForeignKeyBuilder } from "./foreign-keys.js";
import { IndexBuilder } from "./indexes.js";
import { PrimaryKeyBuilder } from "./primary-keys.js";
import { SQLiteTable } from "./table.js";
import { UniqueConstraintBuilder } from "./unique-constraint.js";
function getTableConfig(table) {
const columns = Object.values(table[SQLiteTable.Symbol.Columns]);
const indexes = [];
const checks = [];
const primaryKeys = [];
const uniqueConstraints = [];
const foreignKeys = Object.values(table[SQLiteTable.Symbol.InlineForeignKeys]);
const name = table[Table.Symbol.Name];
const extraConfigBuilder = table[SQLiteTable.Symbol.ExtraConfigBuilder];
if (extraConfigBuilder !== void 0) {
const extraConfig = extraConfigBuilder(table[SQLiteTable.Symbol.Columns]);
const extraValues = Array.isArray(extraConfig) ? extraConfig.flat(1) : Object.values(extraConfig);
for (const builder of Object.values(extraValues)) {
if (is(builder, IndexBuilder)) {
indexes.push(builder.build(table));
} else if (is(builder, CheckBuilder)) {
checks.push(builder.build(table));
} else if (is(builder, UniqueConstraintBuilder)) {
uniqueConstraints.push(builder.build(table));
} else if (is(builder, PrimaryKeyBuilder)) {
primaryKeys.push(builder.build(table));
} else if (is(builder, ForeignKeyBuilder)) {
foreignKeys.push(builder.build(table));
}
}
}
return {
columns,
indexes,
foreignKeys,
checks,
primaryKeys,
uniqueConstraints,
name
};
}
function extractUsedTable(table) {
if (is(table, SQLiteTable)) {
return [`${table[Table.Symbol.BaseName]}`];
}
if (is(table, Subquery)) {
return table._.usedTables ?? [];
}
if (is(table, SQL)) {
return table.usedTables ?? [];
}
return [];
}
function getViewConfig(view) {
return {
...view[ViewBaseConfig]
// ...view[SQLiteViewConfig],
};
}
export {
extractUsedTable,
getTableConfig,
getViewConfig
};
//# sourceMappingURL=utils.js.map