Function getDbName

packages/cli/src/actions/pull/utils.ts:96–117

(decl: AbstractDeclaration | DataField | EnumField, includeSchema: boolean = false)

}

export function getDbName(decl: AbstractDeclaration | DataField | EnumField, includeSchema: boolean = false): string {
    if (!('attributes' in decl)) return decl.name;

    const schemaAttr = decl.attributes.find((a) => a.decl.ref?.name === '@@schema');
    let schema = 'public';
    if (schemaAttr) {
        const schemaAttrValue = schemaAttr.args[0]?.value;
        if (schemaAttrValue?.$type === 'StringLiteral') {
            schema = schemaAttrValue.value;
        }
    }

    const formatName = (name: string) => `${schema && includeSchema ? `${schema}.` : ''}${name}`;

    const nameAttr = decl.attributes.find((a) => a.decl.ref?.name === '@@map' || a.decl.ref?.name === '@map');
    if (!nameAttr) return formatName(decl.name);
    const attrValue = nameAttr.args[0]?.value;

    if (attrValue?.$type !== 'StringLiteral') return formatName(decl.name);

    return formatName(attrValue.value);
}

export function getRelationFkName(decl: DataField): string | undefined {
    const relationAttr = decl?.attributes.find((a) => a.decl.ref?.name === '@relation');