klz-cables.com/.pnpm-store/v10/files/31/735efc2e65f49b489e8b57774654c38116049f744be863343856b65b1158367e71863babbafceb0f359e9931b740e862260df67e9fe8a648dc59bcd18d48ca

import { bit, boolean, foreignKey, halfvec, index, integer, jsonb, numeric, serial, sparsevec, text, timestamp, uniqueIndex, uuid, varchar, vector } from 'drizzle-orm/pg-core';
import { geometryColumn } from './geometryColumn.js';
const rawColumnBuilderMap = {
    boolean,
    geometry: geometryColumn,
    integer,
    jsonb,
    numeric,
    serial,
    text,
    uuid,
    varchar
};
export const buildDrizzleTable = ({ adapter, rawTable })=>{
    const columns = {};
    for (const [key, column] of Object.entries(rawTable.columns)){
        switch(column.type){
            case 'bit':
                {
                    const builder = bit(column.name, {
                        dimensions: column.dimensions
                    });
                    columns[key] = builder;
                    break;
                }
            case 'enum':
                if ('locale' in column) {
                    columns[key] = adapter.enums.enum__locales(column.name);
                } else {
                    adapter.enums[column.enumName] = adapter.pgSchema.enum(column.enumName, column.options);
                    columns[key] = adapter.enums[column.enumName](column.name);
                }
                break;
            case 'halfvec':
                {
                    const builder = halfvec(column.name, {
                        dimensions: column.dimensions
                    });
                    columns[key] = builder;
                    break;
                }
            case 'numeric':
                {
                    columns[key] = numeric(column.name, {
                        mode: 'number'
                    });
                    break;
                }
            case 'sparsevec':
                {
                    const builder = sparsevec(column.name, {
                        dimensions: column.dimensions
                    });
                    columns[key] = builder;
                    break;
                }
            case 'timestamp':
                {
                    let builder = timestamp(column.name, {
                        mode: column.mode,
                        precision: column.precision,
                        withTimezone: column.withTimezone
                    });
                    if (column.defaultNow) {
                        builder = builder.defaultNow();
                    }
                    columns[key] = builder;
                    break;
                }
            case 'uuid':
                {
                    let builder = uuid(column.name);
                    if (column.defaultRandom) {
                        builder = builder.defaultRandom();
                    }
                    columns[key] = builder;
                    break;
                }
            case 'vector':
                {
                    const builder = vector(column.name, {
                        dimensions: column.dimensions
                    });
                    columns[key] = builder;
                    break;
                }
            default:
                columns[key] = rawColumnBuilderMap[column.type](column.name);
                break;
        }
        if (column.reference) {
            columns[key].references(()=>adapter.tables[column.reference.table][column.reference.name], {
                onDelete: column.reference.onDelete
            });
        }
        if (column.primaryKey) {
            columns[key].primaryKey();
        }
        if (column.notNull) {
            columns[key].notNull();
        }
        if (typeof column.default !== 'undefined') {
            let sanitizedDefault = column.default;
            if (column.type === 'geometry' && Array.isArray(column.default)) {
                sanitizedDefault = `SRID=4326;POINT(${column.default[0]} ${column.default[1]})`;
            }
            columns[key].default(sanitizedDefault);
        }
        if (column.type === 'geometry') {
            if (!adapter.extensions.postgis) {
                adapter.extensions.postgis = true;
            }
        }
    }
    const extraConfig = (cols)=>{
        const config = {};
        if (rawTable.indexes) {
            for (const [key, rawIndex] of Object.entries(rawTable.indexes)){
                let fn = index;
                if (rawIndex.unique) {
                    fn = uniqueIndex;
                }
                if (Array.isArray(rawIndex.on)) {
                    if (rawIndex.on.length) {
                        config[key] = fn(rawIndex.name).on(...rawIndex.on.map((colName)=>cols[colName]));
                    }
                } else {
                    config[key] = fn(rawIndex.name).on(cols[rawIndex.on]);
                }
            }
        }
        if (rawTable.foreignKeys) {
            for (const [key, rawForeignKey] of Object.entries(rawTable.foreignKeys)){
                let builder = foreignKey({
                    name: rawForeignKey.name,
                    columns: rawForeignKey.columns.map((colName)=>cols[colName]),
                    foreignColumns: rawForeignKey.foreignColumns.map((column)=>adapter.tables[column.table][column.name])
                });
                if (rawForeignKey.onDelete) {
                    builder = builder.onDelete(rawForeignKey.onDelete);
                }
                if (rawForeignKey.onUpdate) {
                    builder = builder.onDelete(rawForeignKey.onUpdate);
                }
                config[key] = builder;
            }
        }
        return config;
    };
    adapter.tables[rawTable.name] = adapter.pgSchema.table(rawTable.name, columns, extraConfig);
};

//# sourceMappingURL=buildDrizzleTable.js.map