tethys.backend/providers/query_builder_provider.ts

import { ApplicationService } from '@adonisjs/core/types';
// import { Database, DatabaseQueryBuilder } from '@adonisjs/lucid/database';
import { ModelQueryBuilder } from '@adonisjs/lucid/orm';
// import db from '@adonisjs/lucid/services/db';
// import { LucidModel, LucidRow } from '@adonisjs/lucid/types/model';
import { ChainableContract, ExcutableQueryBuilderContract } from '@adonisjs/lucid/types/querybuilder';

// import { ModelQueryBuilderContract } from '@adonisjs/lucid/types/model';
/*
|--------------------------------------------------------------------------
| Provider
|--------------------------------------------------------------------------
|
| Your application is not ready when this file is loaded by the framework.
| Hence, the top level imports relying on the IoC container will not work.
| You must import them inside the life-cycle methods defined inside
| the provider class.
|
| @example:
|
| public async ready () {
|   const Database = this.app.container.resolveBinding('Adonis/Lucid/Database')
|   const Event = this.app.container.resolveBinding('Adonis/Core/Event')
|   Event.on('db:query', Database.prettyPrint)
| }
|
*/
declare module '@adonisjs/lucid/types/model' {
    // interface ModelQueryBuilderContract<Model extends LucidModel, Result = InstanceType<Model>> {
    export interface ModelQueryBuilderContract<Model extends LucidModel, Result = InstanceType<Model>> extends ChainableContract, ExcutableQueryBuilderContract<Result[]> {
        // macro typescript definitions here
        // whereTrue(columnName: string): this;
        // whereFalse(columnName: string): this;
        // any(): Promise<boolean>;
        // selectCount(): Promise<BigInt>;
        // selectIds(primaryKey?: string): Promise<number[]>;
        // selectId(primaryKey?: string): Promise<number | undefined>;
        // selectIdOrFail(primaryKey?: string): Promise<number>;
        // pluck(valueColumn: string, id?: string): Promise<{ [key: string]: any }>;
        pluck(valueColumn: string, id?: string): Promise<{ [key: string]: any }>;
    }

    export interface LucidRow {
        [key: string]: any;
    }

    
}
declare module '@adonisjs/lucid/orm' {
    interface ModelQueryBuilder {
        pluck(valueColumn: string, id?: string): Promise<{ [key: string]: any }>;
    }
    
    //  class ModelQueryBuilder implements ModelQueryBuilderContract<LucidModel, LucidRow> {
    //     public pluck(valueColumn: string, id?: string): Promise<{ [key: string]: any }>;
    // }
}



export default class QueryBuilderProvider {
    constructor(protected app: ApplicationService) {}

    public register() {
        // Register your own bindings
        // const ModelQueryBuilder = this.app.container.bind('@adonisjs/lucid/orm/ModelQueryBuilder');
    
        // ModelQueryBuilder.macro('whereTrue', function (columnName: string) {
        //     return this.where(columnName, true);
        // });

        // ModelQueryBuilder.macro('whereFalse', function (columnName: string) {
        //     return this.where(columnName, false);
        // });
    }

    public async boot() {
        // All bindings are ready, feel free to use them      
        // const db = await this.app.container.make('lucid.db');     

        ModelQueryBuilder.macro('pluck', async function (this: ModelQueryBuilder, valueColumn: string, id?: string) {
            // let rolesPluck = {};
            let rolesPluck: { [key: number]: any } = {};
            const result = await this.exec();
            result.forEach((user: { [key: string]: any }, index: number) => {
                let idc: number;
                if (!id) {
                    idc = index;
                } else {
                    idc = user[id];
                }
                const value: any = user[valueColumn];
                rolesPluck[idc] = value;
            });
            return rolesPluck;
        });

        // });

        //     validator.rule('foo', () => {})
    }

    public async ready() {
        // App is ready
    }

    public async shutdown() {
        // Cleanup, since app is going down
    }
}