dblang/src/defaultHandler.ts

import { Attribute, DB, Table } from "./db"
import { Aggregation, Modifier } from "./dbStructure"
import { QueryBuilder, selectQuery } from "./query"
import { allModifierInput, primaryData, serializeReturn } from "./types"

export class Handler {
    /*syncDB(db : DB){

    }*/

    querys = {
        select: (handler: Handler, q: selectQuery): QueryBuilder => {
            const builder = new QueryBuilder();
            builder.addCode("select ");
            builder.append(joinArg(", ")(handler, q.attr));
            builder.addCode(` from ${q.from == null ? 'DUAL' : q.from.serialize(handler)}`);
            if (q.whereD) {
                builder.addCode(" where ");
                builder.append(q.whereD.serialize(handler));
            }
            if (q.groupByD.length > 0) {
                builder.addCode(" group by ");
                builder.append(joinArg(",")(handler, q.groupByD));
                if (q.havingD) {
                    builder.addCode(" having ");
                    builder.append(q.havingD.serialize(handler));
                }
            }

            if (q.havingD) {
                builder.addCode(" limit ");
                builder.addInjection(q.limitD);
            }
            return builder;
        },
        create: (handler: Handler, table: Table): QueryBuilder => {
            const builder = new QueryBuilder();
            builder.addCode(`create table if not exists ${table.toString(handler)}(`);
            let keys = Object.keys(table.dbLangTableAttributes);
            for (let i = 0; i < keys.length; i++) {
                const a = table.dbLangTableAttributes[keys[i]];
                builder.addCode(a.serialize(handler) + " ");
                builder.append(a.serializeSettings(handler));
                if (i + 1 < keys.length) builder.addCode(", ");
            }
            builder.addCode(")");
            return builder;
        },

    }

    constraints = {
        // check constraints
        listPrimaryKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        listForeignKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        listUniqueKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        listChecks: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        // add constraints
        appPrimaryKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        appForeignKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        addUniqueKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        addChecks: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        //should drop all keys to be able to recreate them
        dropPrimaryKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        dropForeignKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        dropUniqueKeys: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
        dropChecks: (handler: Handler, table: Table): QueryBuilder => new QueryBuilder(),
    }

    builders = {
        query: (qb: QueryBuilder): serializeReturn => {
            let args: primaryData[] = [];
            let sql = "";
            for (let i = 0; i < qb.list.length; i++) {
                const [inject, data] = qb.list[i];
                if (inject) {
                    sql += "?";
                    args.push(data);
                } else {
                    sql += data;
                }
            }
            return [sql, args];
        },

        attributeSettings: (handler: Handler, a: Attribute): QueryBuilder => {
            const builder = new QueryBuilder();
            builder.append(a.type.serialize(handler));
            if (a.ops.autoIncrement) {
                builder.addCode(" auto_increment");
            }
            /*if (a.ops.primaryKey) {
                sql += " primary key";
            }*/
            if (a.ops.default != null) {
                if (a.ops.autoIncrement || a.ops.primaryKey || a.ops.notNull)
                    throw new Error(`Can not set default when autoIncrement, primaryKey or notNull ist set on Attribute: ${a.toStringFunc(handler)}`);
                builder.addCode(" default ");
                builder.addInjection(a.ops.default);
            }

            /*if (a.ops.unique != null) {
                if (!a.ops.autoIncrement && !a.ops.primaryKey){
                    sql += " unique";
                }
            }*/
            if (a.ops.notNull != null) {
                if (!a.ops.autoIncrement && !a.ops.primaryKey && a.ops.default == null) {
                    builder.addCode(" not null");
                }
            }

            /*if (a.ops.foreginKey != null) {
                sql += ` foreign key references (${a.ops.foreginKey.link.toStringFunc(handler)})`
            }*/

            return builder;
        },
        escapeID: (key: string): string => {
            if (!key || key === "" || key.includes('\u0000')) throw new Error("Can not escape empty key or with null unicode!");
            if (key.match(/^`.+`$/g)) return key;
            return `\`${key.replace(/`/g, '``')}\``;
        }
    }


    aggregations = {
        count: (handler: Handler, a: Attribute): QueryBuilder => new QueryBuilder([{ data: "count(" + a.toString(handler) + ")" }]),
        sum: (handler: Handler, a: Attribute): QueryBuilder => new QueryBuilder([{ data: "sum(" + a.toString(handler) + ")" }]),
        avg: (handler: Handler, a: Attribute): QueryBuilder => new QueryBuilder([{ data: "avg(" + a.toString(handler) + ")" }]),
        min: (handler: Handler, a: Attribute): QueryBuilder => new QueryBuilder([{ data: "min(" + a.toString(handler) + ")" }]),
        max: (handler: Handler, a: Attribute): QueryBuilder => new QueryBuilder([{ data: "max(" + a.toString(handler) + ")" }]),
    }

    modifiers = {
        and: joinArg("and"),
        or: joinArg("or"),
        le: joinArg("<"),
        leq: joinArg("<="),
        eq: joinArg("="),
        geq: joinArg(">="),
        ge: joinArg(">"),
        plus: joinArg("+"),
        minus: joinArg("-"),
        not: (handler: Handler, a: allModifierInput[]): QueryBuilder => {
            let e = a[0];
            if (e instanceof Attribute) return new QueryBuilder([{ data: "not (" + e.toString(handler) + ")" }])
            if (e instanceof Modifier || e instanceof selectQuery || e instanceof Aggregation) {
                const builder = new QueryBuilder();
                builder.addCode("not (");
                builder.append(e.serialize(handler));
                builder.addCode(")");
                return builder;
            }
            return new QueryBuilder([
                { data: "not(" },
                { inject: true, data: e },
                { data: ")" }
            ])
        }
    }
    datatypes = {
        char: dataTypeSingleNum("char"),
        varchar: dataTypeSingleNum("varchar"),
        binary: dataTypeSingleNum("binary"),
        varbinary: dataTypeSingleNum("varbinary"),

        tinyblob: dataTypeNoArg("tinyblob"),
        blob: dataTypeNoArg("blob"),
        mediumblob: dataTypeNoArg("mediumblob"),
        longblob: dataTypeNoArg("longblob"),
        tinytext: dataTypeNoArg("tinytext"),
        text: dataTypeNoArg("text"),
        mediumtext: dataTypeNoArg("mediumtext"),
        longtext: dataTypeNoArg("longtext"),

        enum: (a: primaryData[]): QueryBuilder => {
            const builder = new QueryBuilder();
            builder.addCode("enum(");
            builder.addInjectionCommaSeperated(a);
            builder.addCode(")");
            return builder;
        },
        set: (a: primaryData[]): QueryBuilder => {
            const builder = new QueryBuilder();
            builder.addCode("set(");
            builder.addInjectionCommaSeperated(a);
            builder.addCode(")");
            return builder;
        },
        bool: dataTypeNoArg("bool"),
        bit: dataTypeNoArg("bit"),
        tinyint: dataTypeNoArg("tinyint"),
        smallint: dataTypeNoArg("smallint"),
        mediumint: dataTypeNoArg("mediumint"),
        int: dataTypeNoArg("int"),
        bigint: dataTypeNoArg("bigint"),

        float: dataTypeDblNum("float"),
        double: dataTypeDblNum("double"),
        decimal: dataTypeDblNum("decimal"),

        date: dataTypeNoArg("data"),
        datetime: dataTypeNoArg("datatime"),
        timestamp: dataTypeNoArg("timestamp"),
        time: dataTypeNoArg("time"),
        year: dataTypeNoArg("year"),
    }

};

function dataTypeNoArg(type: string) {
    return (a: primaryData[]): QueryBuilder => {
        return new QueryBuilder([{
            inject: false,
            data: type
        }]);
    }
}
function dataTypeSingleNum(type: string) {
    return (a: primaryData[]): QueryBuilder => {
        return new QueryBuilder([
            { data: type + "(" },
            { inject: true, data: a[0] },
            { data: ")" }
        ]);
    }
}
function dataTypeDblNum(type: string) {
    return (a: primaryData[]): QueryBuilder => {
        return new QueryBuilder([
            { data: type + "(" },
            { inject: true, data: a[0] },
            { data: ", " },
            { inject: true, data: a[1] },
            { data: ")" }
        ]);
    }
}


function joinArg(type: string) {
    return (handler: Handler, a: (allModifierInput)[]): QueryBuilder => {
        const builder = new QueryBuilder();
        for (let i = 0; i < a.length; i++) {
            const d = a[i];
            if (d instanceof Attribute) builder.addCode(d.toString(handler));
            else if (d instanceof Modifier || d instanceof selectQuery || d instanceof Aggregation) {
                builder.addCode("(");
                builder.append(d.serialize(handler));
                builder.addCode(")");
            } else {
                builder.addInjection(d);
            }
            if (i + 1 < a.length) builder.addCode(" " + type + " ");
        }
        return builder;
    }
}