UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

53 lines 1.47 kB
import { entityKind } from "../../entity.js"; import { SQLiteColumn, SQLiteColumnBuilder } from "./common.js"; class SQLiteCustomColumnBuilder extends SQLiteColumnBuilder { static [entityKind] = "SQLiteCustomColumnBuilder"; constructor(name, fieldConfig, customTypeParams) { super(name, "custom", "SQLiteCustomColumn"); this.config.fieldConfig = fieldConfig; this.config.customTypeParams = customTypeParams; } /** @internal */ build(table) { return new SQLiteCustomColumn( table, this.config ); } } class SQLiteCustomColumn extends SQLiteColumn { static [entityKind] = "SQLiteCustomColumn"; sqlName; mapTo; mapFrom; constructor(table, config) { super(table, config); this.sqlName = config.customTypeParams.dataType(config.fieldConfig); this.mapTo = config.customTypeParams.toDriver; this.mapFrom = config.customTypeParams.fromDriver; } getSQLType() { return this.sqlName; } mapFromDriverValue(value) { return typeof this.mapFrom === "function" ? this.mapFrom(value) : value; } mapToDriverValue(value) { return typeof this.mapTo === "function" ? this.mapTo(value) : value; } } function customType(customTypeParams) { return (dbName, fieldConfig) => { return new SQLiteCustomColumnBuilder( dbName, fieldConfig, customTypeParams ); }; } export { SQLiteCustomColumn, SQLiteCustomColumnBuilder, customType }; //# sourceMappingURL=custom.js.map