drizzle-orm
Version:
Drizzle ORM package for SQL databases
47 lines • 1.53 kB
JavaScript
import { entityKind } from "../entity.js";
import { Table } from "../table.js";
const InlineForeignKeys = Symbol.for("drizzle:PgInlineForeignKeys");
class PgTable extends Table {
static [entityKind] = "PgTable";
/** @internal */
static Symbol = Object.assign({}, Table.Symbol, {
InlineForeignKeys
});
/**@internal */
[InlineForeignKeys] = [];
/** @internal */
[Table.Symbol.ExtraConfigBuilder] = void 0;
}
function pgTableWithSchema(name, columns, extraConfig, schema, baseName = name) {
const rawTable = new PgTable(name, schema, baseName);
const builtColumns = Object.fromEntries(
Object.entries(columns).map(([name2, colBuilderBase]) => {
const colBuilder = colBuilderBase;
const column = colBuilder.build(rawTable);
rawTable[InlineForeignKeys].push(...colBuilder.buildForeignKeys(column, rawTable));
return [name2, column];
})
);
const table = Object.assign(rawTable, builtColumns);
table[Table.Symbol.Columns] = builtColumns;
if (extraConfig) {
table[PgTable.Symbol.ExtraConfigBuilder] = extraConfig;
}
return table;
}
const pgTable = (name, columns, extraConfig) => {
return pgTableWithSchema(name, columns, extraConfig, void 0);
};
function pgTableCreator(customizeTableName) {
return (name, columns, extraConfig) => {
return pgTableWithSchema(customizeTableName(name), columns, extraConfig, void 0, name);
};
}
export {
InlineForeignKeys,
PgTable,
pgTable,
pgTableCreator,
pgTableWithSchema
};
//# sourceMappingURL=table.js.map