@sqb/connect/orm/util/extract-keyvalues.js

Multi-dialect database connection framework written with TypeScript

import { EntityMetadata } from '../model/entity-metadata.js';
import { isColumnField } from './orm.helper.js';
export function extractKeyValues(entityDef, valueOrInstance, keepOther) {
    const primaryIndex = EntityMetadata.getPrimaryIndex(entityDef);
    if (!primaryIndex)
        throw new Error(`No primary fields defined for "${entityDef.name}" entity`);
    const validateCol = k => {
        const col = EntityMetadata.getField(entityDef, k);
        if (!col)
            throw new Error(`Unknown column (${k}) defined as primary key in entity "${entityDef.name}"`);
        if (!isColumnField(col)) {
            throw new Error(`Column (${k}) defined as primary key in entity "${entityDef.name}" is not a data column`);
        }
    };
    // if entity's primary key has more than one key field
    if (primaryIndex.columns.length > 1) {
        if (typeof valueOrInstance !== 'object') {
            throw new Error(`"${entityDef.name}" entity` +
                ` has more than one primary key field and you must provide all values with an key/value pair`);
        }
        const valueKeys = Object.keys(valueOrInstance);
        const valueKeysUpper = valueKeys.map(x => x.toUpperCase());
        const out = {};
        for (const k of primaryIndex.columns) {
            const i = valueKeysUpper.indexOf(k.toUpperCase());
            if (i < 0)
                throw new Error(`Value of key field "${entityDef.name}.${k}" required to perform this operation`);
            validateCol(k);
            out[k] = valueOrInstance[valueKeys[i]];
        }
        if (keepOther) {
            for (let i = 0; i < valueKeys.length; i++) {
                if (primaryIndex.columns.find(x => x.toUpperCase() === valueKeysUpper[i]))
                    continue;
                out[valueKeys[i]] = valueOrInstance[valueKeys[i]];
            }
        }
        return out;
    }
    const primaryColumnName = primaryIndex.columns[0];
    validateCol(primaryColumnName);
    if (typeof valueOrInstance === 'object') {
        const valueKeys = Object.keys(valueOrInstance);
        const valueKeysUpper = valueKeys.map(x => x.toUpperCase());
        const k = valueKeysUpper.indexOf(primaryColumnName.toUpperCase());
        if (k < 0) {
            throw new Error(`Value of key field "${entityDef.name}.${primaryColumnName}" required to perform this operation`);
        }
        const out = { [primaryColumnName]: valueOrInstance[valueKeys[k]] };
        if (keepOther) {
            for (let i = 0; i < valueKeys.length; i++) {
                if (primaryIndex.columns.find(x => x.toUpperCase() === valueKeysUpper[i]))
                    continue;
                out[valueKeys[i]] = valueOrInstance[valueKeys[i]];
            }
        }
        return out;
    }
    return { [primaryColumnName]: valueOrInstance };
}