ddl-manager/dist/ast/AssignVariable.ts

store postgres procedures and triggers in files

import { AbstractAstElement } from "./AbstractAstElement";
import { HardCode } from "./HardCode";
import { SimpleSelect } from "./SimpleSelect";
import { Spaces } from "./Spaces";

interface AssignVariableRow {
    variable: string;
    value: AbstractAstElement;
}

export class AssignVariable extends AbstractAstElement {
    readonly variable!: string;
    readonly value!: HardCode | SimpleSelect;

    constructor(row: AssignVariableRow) {
        super();
        Object.assign(this, row);
    }

    template(spaces: Spaces) {
        const valueSQL = this.value
            .toSQL( spaces )
            .trim();
        return [
            spaces + `${this.variable} = ${ valueSQL };`
        ];
    }
}