123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
// Copyright 2020-2020 The Mandarine.TS Framework authors. All rights reserved. MIT license.
import { Log } from "../../logger/log.ts";import { ApplicationContext } from "../../main-core/application-context/mandarineApplicationContext.ts";import type { Mandarine } from "../../main-core/Mandarine.ns.ts";import { ReflectUtils } from "../../main-core/utils/reflectUtils.ts";import type { PostgresConnector } from "../connectors/postgreSQLConnector.ts";import { MandarineORMException } from "../core/exceptions/mandarineORMException.ts";import { lexicalProcessor } from "../core/lexicalProcessor.ts";
/** * This class is one of the most important class for MQL * This class resolves the methods of your repository as it works as bridge between your repositories and Mandarine's Engine. */export class PostgresRepositoryProxy<T> implements Mandarine.ORM.RepositoryProxy {
    public readonly SUPPORTED_KEYWORDS = ["and", "or", "isnotnull", "isnull", "isempty", "isnotempty", "startingwith", "endswith", "like", "greaterthan", "lessthan"];
    public entity: Mandarine.ORM.Entity.Table;
    private logger: Log = Log.getLogger("PostgresRepositoryProxy");
    constructor(entity: Mandarine.ORM.Entity.Table) {        this.entity = entity;    }
    public async executeQuery(query: any) {        let dbClient: PostgresConnector = this.getEntityManager().getDatabaseClient();            try{                let queryExecution = await dbClient.query(query);                if(!queryExecution) {                    this.logger.debug("Query execution returned null and operation was aborted");                    return null;                }                let rowsOfObjects = queryExecution.query.result.rowsOfObjects();                if(rowsOfObjects.length == 0) {                    return null;                } else if(rowsOfObjects.length >= 1) {                    return rowsOfObjects;                }            }catch(error){                return undefined;            }    }
    public getEntityManager(): Mandarine.ORM.Entity.EntityManager {        let entityManager: Mandarine.ORM.Entity.EntityManager = ApplicationContext.getInstance().getEntityManager();        return entityManager;    }
    public async save(model: any): Promise<any> {        if(!ReflectUtils.checkClassInitialized(model)) throw new MandarineORMException(MandarineORMException.INSTANCE_IN_SAVE, "PostgresRepositoryProxy");        let entityManager: Mandarine.ORM.Entity.EntityManager = this.getEntityManager();        let columns: Array<Mandarine.ORM.Entity.Column> = this.entity.columns;
        let modelObject: { [name: string] : any } = JSON.parse(JSON.stringify(model));        Object.keys(modelObject).forEach((modelColumn) => {            if(!columns.some(col => col.name?.toLowerCase() == modelColumn.toLowerCase())) delete modelObject[modelColumn];        });
        let dialect = entityManager.getDialectClass();
        const primaryKey = this.entity.primaryKey;
        if(primaryKey == (undefined || null) || (primaryKey.fieldName != undefined && model[primaryKey.fieldName] == (null || undefined))) {            // INSERTION                let insertionValues: Array<any> = new Array<any>();                let queryData = (<Mandarine.ORM.Dialect.Dialect>dialect).insertStatement((<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity), this.entity, modelObject, true);                let query = queryData.query;                let queryColumnValues = queryData.insertionValuesObject;                Object.keys(queryColumnValues).forEach((col, index) => insertionValues.push(`$${index + 1}`));
                query = query.replace("%values%", insertionValues.join(", "));                let saveQuery = await this.executeQuery({                    text: query,                    args: Object.values(queryColumnValues)                });
                if(saveQuery === undefined) {                    return false;                } else {                    return true;                }        } else {            // UPDATE                let updateValues: Array<any> = new Array<any>();                let queryData = (<Mandarine.ORM.Dialect.Dialect>dialect).updateStatement((<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity), this.entity, modelObject);                let query = queryData.query;                let queryColumnValues = queryData.updateValuesObject;                Object.keys(queryColumnValues).forEach((col, index) => updateValues.push(`$${index + 1}`));                query = query.replace("%values%", updateValues.join(", "));
                // Primary key value                query = query.replace("%primaryKeyValue%", `$${updateValues.length + 1}`);                // Query args                let queryArgs = Object.values(queryColumnValues);                queryArgs.push(model[<string> primaryKey.fieldName]);                                    let saveQuery = await this.executeQuery({                    text: query,                    args: queryArgs                });
                if(saveQuery === undefined) {                    return false;                } else {                    return true;                }        }
    }
    public async findAll() {        let entityManager: Mandarine.ORM.Entity.EntityManager = this.getEntityManager();
        let dialect = entityManager.getDialectClass();        let query = (<Mandarine.ORM.Dialect.Dialect>dialect).selectStatement((<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity));        return this.executeQuery(query);    }
    public async countAll() {        let entityManager: Mandarine.ORM.Entity.EntityManager = this.getEntityManager();
        let dialect = entityManager.getDialectClass();        let query = (<Mandarine.ORM.Dialect.Dialect>dialect).selectAllCountStatement((<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity));        return parseFloat((<any>await this.executeQuery(query))[0]?.count || 0);    }
    public async deleteAll() {        let entityManager: Mandarine.ORM.Entity.EntityManager = this.getEntityManager();
        let dialect = entityManager.getDialectClass();        let query = (<Mandarine.ORM.Dialect.Dialect>dialect).deleteStatement((<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity));
        let deleteQuery = await this.executeQuery(query);
        if(deleteQuery === undefined) {            return false;        } else {            return true;        }    }
    public lexicalProcessor(methodName: string, proxyType: Mandarine.ORM.ProxyType): string {        const dialect  = this.getEntityManager().getDialectClass();        return lexicalProcessor(this, methodName, proxyType, (<Mandarine.ORM.Dialect.Dialect>dialect).getTableMetadata(this.entity), this.entity, (<Mandarine.ORM.Dialect.Dialect>dialect));    }
    public async mainProxy(nativeMethodName: string, proxyType: Mandarine.ORM.ProxyType, args: Array<any>): Promise<any> {        let mqlQuery: string = this.lexicalProcessor(nativeMethodName, proxyType);
       let query: any = this.executeQuery({        text: mqlQuery,        args: args        });
        if(query) {            if(proxyType == "countBy") {                return parseFloat((await query)[0]?.count || 0);            } else if(proxyType == "existsBy") {                return ((await query)[0].count) >= 1;            } else {                return query;            }        }    }
    public async manualProxy(query: String, secure: boolean, args: Array<any>) {            if(secure != undefined && secure == true) {                return this.executeQuery({                    text: query,                    args: args                });            } else {                return this.executeQuery(query);            }    }
}