import { Response, NextFunction, Application } from "express";
import { Request } from "express-jwt";
import { ReturnHelper } from "../helpers/express/return";
import { OrmHelper } from "../helpers/orm";
import Joi from "joi";
import { ILogObj, Logger } from "tslog";
import { Language } from "../langs/lang";
import { Paging } from "entity";
import CommonHelper from "../helpers/common";
import * as fastcsv from 'fast-csv';
import dayjs from "dayjs";
import { Menu } from "entity";

const log: Logger<ILogObj> = new Logger({ name: '[MenuController]', type: 'pretty' });

export class MenuController {
    static async list(req: Request, res: Response, next: NextFunction): Promise<Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.parameters['filter'] = {
            in: 'query',
            type: 'string'
        }
        #swagger.parameters['limit'] = {
            in: 'query',
            required: true,
            type: 'number'
        }
        #swagger.parameters['page'] = {
            in: 'query',
            required: true,
            type: 'number'
        }
        #swagger.parameters['with_deleted'] = {
            in: 'query',
            required: true,
            type: 'boolean'
        }
        #swagger.parameters['order_field'] = {
            in: 'query',
            required: true,
            type: 'string'
        }
        #swagger.parameters['order_direction'] = {
            in: 'query',
            required: true,
            schema: {
                '@enum': ['ASC', 'DESC']
            }
        }
        #swagger.parameters['token'] = {
            in: 'query',
            required: true,
            type: 'string'
        }
        */

        try {
            const schema = Joi.object().keys({
                filter: Joi.string().allow('').optional().label('Filter'),
                page: Joi.number().required().min(1).label('Page'),
                limit: Joi.number().required().min(1).label('Limit'),
                with_deleted: Joi.bool().required().label('With Deleted'),
                order_field: Joi.string().required().label('Order Field'),
                order_direction: Joi.string().allow('asc', 'desc').required().label('Order Direction'),
                token: Joi.string().required().label('Token'),
            });

            const param: Paging = await schema.validateAsync(req.query);

            const repo = OrmHelper.DB.getRepository(Menu);

            const offset = (param.page - 1) * param.limit

            let whereAttr = '';
            let whereVal: any = {};

            if (param.filter) {
                whereAttr = 'name like :filter_like  ';
                // whereVal['filter'] = param.filter
                whereVal['filter_like'] = '%' + param.filter + '%'
            }

            const res_count = repo.createQueryBuilder()
                .where(whereAttr, whereVal);
            const res_list = repo.createQueryBuilder()
                .where(whereAttr, whereVal)
                .andWhere('id_parent is null')
                .orderBy(param.order_field, param.order_direction)
                .offset(offset)
                .limit(param.limit);

            if (param.with_deleted) {
                res_count.withDeleted();
                res_list.withDeleted();
            }

            const res_list_all = await repo.createQueryBuilder().andWhere('id_parent is not null').getMany();
            let list_data_final = await res_list.getMany();

            let list_child: any = {};
            for (let p of res_list_all) {
                if (!list_child[p.id_parent]) {
                    list_child[p.id_parent] = [];
                }

                list_child[p.id_parent].push(p);
            }

            for (let f of list_data_final) {
                if (list_child[f.id]) {
                    f.children = list_child[f.id];

                    for (let g of f.children) {
                        if (list_child[g.id]) {
                            g.children = list_child[g.id];

                            for (let h of g.children) {
                                if (list_child[h.id]) {
                                    h.children = list_child[h.id];

                                    for (let i of h.children) {
                                        if (list_child[i.id]) {
                                            i.children = list_child[i.id];

                                            for (let j of i.children) {
                                                if (list_child[j.id]) {
                                                    j.children = list_child[j.id];

                                                    for (let k of j.children) {
                                                        if (list_child[k.id]) {
                                                            k.children = list_child[k.id];
                                                        }
                                                    }
                                                }
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }

            const current_page = param.page;
            const total_count_data = await res_count.getCount();
            const list_data = list_data_final;
            const count_data = CommonHelper.countObject(list_data);

            return ReturnHelper.successResponselist(res, 200, Language.lang.success_view, count_data, current_page, total_count_data, list_data);
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 400, 401, Language.lang.failed_view, err.message);
        }
    }

    static async export(req: Request, res: Response, next: NextFunction): Promise<void | Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.parameters['filter'] = {
            in: 'query',
            type: 'string'
        }
        #swagger.parameters['filter'] = {
            in: 'query',
            type: 'string'
        }
        #swagger.parameters['order_field'] = {
            in: 'query',
            required: true,
            type: 'string'
        }
        #swagger.parameters['order_direction'] = {
            in: 'query',
            required: true,
            schema: {
                '@enum': ['ASC', 'DESC']
            }
        }
        #swagger.parameters['token'] = {
            in: 'query',
            required: true,
            type: 'string'
        }
        */

        try {
            const schema = Joi.object().keys({
                filter: Joi.string().allow('').optional().label('Filter'),
                order_field: Joi.string().required().label('Order Field'),
                order_direction: Joi.string().allow('asc', 'desc').required().label('Order Direction'),
                token: Joi.string().required().label('Token'),
            });

            const param: Paging = await schema.validateAsync(req.query);

            const repo = OrmHelper.DB.getRepository(Menu);

            let whereAttr = '';
            let whereVal: any = {};

            if (param.filter) {
                whereAttr = 'module like :filter_like or name like :filter_like ';
                // whereVal['filter'] = param.filter
                whereVal['filter_like'] = '%' + param.filter + '%'
            }

            const filename = "menu.csv";

            res.setHeader('Content-Type', 'text/csv');
            res.setHeader('Content-Disposition', 'attachment; filename=' + filename);

            const csvStream = fastcsv.format({
                headers: true,
                writeHeaders: true,
                transform: (row: Menu): any => ({
                    ...row,
                    created_at: dayjs(row.created_at).format('DD-MM-YYYY HH:MM:ss'),
                })
            });
            csvStream.pipe(res);

            const limit = 50;

            const fetchAndWrite = async (page: any) => {
                const offset = (page - 1) * limit

                const data = await repo.createQueryBuilder()
                    .where(whereAttr, whereVal)
                    .select([
                        'id',
                        'code',
                        'name',
                        'status',
                        'created_at'])
                    .orderBy(param.order_field, param.order_direction)
                    .offset(offset)
                    .limit(limit).getRawMany();

                if (CommonHelper.countObject(data) === 0) {
                    csvStream.end();
                } else {
                    data.forEach((item: any) => csvStream.write(item));

                    if (CommonHelper.countObject(data) == limit) {
                        fetchAndWrite(page + 1);
                    } else {
                        csvStream.end();
                    }
                }
            };

            fetchAndWrite(1);
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 400, 401, Language.lang.failed_view, err.message);
        }
    }

    static async create(req: Request, res: Response, next: NextFunction): Promise<Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.security = [{
            "bearerAuth": []
        }]
        
        #swagger.requestBody = {
            required: true,
            content: {
                "application/json": {
                    schema: {
                         $ref: "#/components/schemas/menu"
                    }  
                }
            }
        }
        */

        try {
            const schema = Joi.object().keys({
                module: Joi.string().max(128).required().label('Module'),
                name: Joi.string().max(128).required().label('Name'),
                link: Joi.string().max(256).required().label('Link'),
                id_parent: Joi.string().uuid().optional().allow('').label('ID Parent'),
                order: Joi.number().min(1).required().label('Order'),
                icon: Joi.string().max(128).optional().allow('').label('Icon'),
                status: Joi.string().required().label('Status'),
            });

            const param: Menu = await schema.validateAsync(req.body);

            const data = new Menu()
            data.module = param.module
            data.name = param.name
            data.link = param.link
            data.order = param.order
            data.icon = param.icon
            data.status = param.status

            if (param.id_parent) {
                data.id_parent = param.id_parent
            }

            await OrmHelper.DB.manager.save(data);

            return ReturnHelper.successResponseAny(res, 200, Language.lang.success_insert, data);
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 500, 401, Language.lang.failed_insert, err.message);
        }
    }

    static async update(req: Request, res: Response, next: NextFunction): Promise<Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.security = [{
            "bearerAuth": []
        }]
         #swagger.parameters['id'] = {
            in: 'path',
            description: 'Menu ID.',
            required: true,
            type: 'string'
        }
    
         #swagger.requestBody = {
            required: true,
            content: {
                "application/json": {
                    schema: {
                         $ref: "#/components/schemas/menu"
                    }  
                }
            }
        }
        */

        try {
            const schema = Joi.object().keys({
                id: Joi.string().uuid().required().label('ID'),
                module: Joi.string().max(128).required().label('Module'),
                name: Joi.string().max(128).required().label('Name'),
                link: Joi.string().max(256).required().label('Link'),
                id_parent: Joi.string().uuid().optional().allow('').label('ID Parent'),
                order: Joi.number().min(1).required().label('Order'),
                icon: Joi.string().max(128).optional().allow('').label('Icon'),
                status: Joi.string().required().label('Status'),
            });

            req.body.id = req.params['id'];

            const param: Menu = await schema.validateAsync(req.body);

            const repo = OrmHelper.DB.getRepository(Menu);

            const data = await repo.findOneBy({ id: param.id });

            if (data != null) {
                data.module = param.module
                data.name = param.name
                data.link = param.link
                data.order = param.order
                data.icon = param.icon
                data.status = param.status

                if (param.id_parent) {
                    data.id_parent = param.id_parent
                }

                await repo.save(data);

                return ReturnHelper.successResponseAny(res, 200, Language.lang.success_update, data);
            } else {
                return ReturnHelper.errorResponse(res, 404, 401, Language.lang.failed_not_found, "");
            }
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 500, 402, Language.lang.failed_update, err.message);
        }
    }

    static async delete(req: Request, res: Response, next: NextFunction): Promise<Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.security = [{
            "bearerAuth": []
        }]
        #swagger.parameters['id'] = {
            in: 'path',
            description: 'Menu ID.',
            required: true,
            type: 'string'
        }
        */
        try {
            const schema = Joi.object().keys({
                id: Joi.string().uuid().required().label('ID')
            });

            const param: Menu = await schema.validateAsync(req.params);

            const repo = OrmHelper.DB.getRepository(Menu);

            const affected = (await repo.softDelete({ id: param.id })).affected;

            if (affected > 0) {
                return ReturnHelper.successResponseAny(res, 200, Language.lang.success_delete, {});
            } else {
                return ReturnHelper.errorResponse(res, 404, 401, Language.lang.failed_not_found, "");
            }
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 500, 402, Language.lang.failed_delete, err.message);
        }
    }

    static async restore(req: Request, res: Response, next: NextFunction): Promise<Response> {
        /*
        #swagger.tags = ['Menu']
        #swagger.security = [{
            "bearerAuth": []
        }]
            
        #swagger.parameters['id'] = {
            in: 'path',
            description: 'Menu ID.',
            required: true,
            type: 'string'
        }
        */
        try {
            const schema = Joi.object().keys({
                id: Joi.string().uuid().required().label('ID')
            });

            const param: Menu = await schema.validateAsync(req.params);

            const repo = OrmHelper.DB.getRepository(Menu);

            const affected = (await repo.restore({ id: param.id })).affected;

            if (affected > 0) {
                return ReturnHelper.successResponseAny(res, 200, Language.lang.success_restore, {});
            } else {
                return ReturnHelper.errorResponse(res, 404, 401, Language.lang.failed_not_found, "");
            }
        } catch (e: unknown) {
            log.error(e);
            const err = e as Error;

            return ReturnHelper.errorResponse(res, 500, 402, Language.lang.failed_restore, err.message);
        }
    }
}