ddl-manager/dist/parser/UnknownExpressionElementParser.ts

store postgres procedures and triggers in files

import { AbstractNode, ColumnReference as ColumnLink } from "psql-lang";
import { ColumnReferenceParser } from "./ColumnReferenceParser";
import {
    UnknownExpressionElement,
    IColumnsMap
} from "../ast";
import { flatMap } from "lodash";

export class UnknownExpressionElementParser {

    private columnReferenceParser = new ColumnReferenceParser();

    parse(syntax: AbstractNode<any>) {
        const columnsMap = this.buildColumnsMap(syntax);
        return new UnknownExpressionElement(syntax, columnsMap);
    }

    buildColumnsMap(
        syntax: AbstractNode<any> | AbstractNode<any>[]
    ) {
        const syntaxes = (
            Array.isArray(syntax) ? 
                syntax : [syntax]
        );
        const columnsMap: IColumnsMap = {};

        const columnLinks = flatMap(syntaxes, scanColumnLinks);

        columnLinks.forEach(columnLink => {
            const columnReference = this.columnReferenceParser.parse(columnLink);
            columnsMap[ columnLink.toString() ] = columnReference;
        });

        return columnsMap
    }
}

function scanColumnLinks(syntax: AbstractNode<any>) {
    const columnLinks = syntax.filterChildrenByInstance(ColumnLink);

    if ( syntax instanceof ColumnLink ) {
        columnLinks.push(syntax);
    }

    return columnLinks;
}