bd-training-exam-manager/app/controller/Admin.js

const Controller = require('../core/base_controller');

const fs = require('fs');

const fileName = "Admin";

class AdminController extends Controller {
    // 管理系统新增普通管理员账号信息
    async addAdmin() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const requestMsg = ctx.request.body.requestMsg;
            requestMsg.forEach(element => {
                const salt = this.generateRandomString();
                element.uid = this.generateNumberString('E');
                element.salt = this.encryptRequest(salt);
                element.account_name = this.decryptRequest(element.account_name);
                element.password = this.shaForm('123456', salt);
                element.role = Number(this.decryptRequest(element.role));
                element.status = 0;
                element.created_at = new Date();
                element.updated_at = new Date();
                element.login_status = 0;
            });
            const result = await ctx.service.admin.addAdmin(requestMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                this.setResponse(ctx, 200, '操作成功', "none");
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "addAdmin";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统查看普通管理员账号信息
    async showAdmin() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const uid = JSON.parse(rdmcode).uid;
            const body = ctx.request.body;
            const result = await ctx.service.admin.showAdmin(uid, body);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                let adminMsg = [];
                result.rows.forEach(element => {
                    adminMsg.push(element.dataValues);
                });
                const data = {
                    adminMsg,
                    count: result.count
                };
                this.setResponse(ctx, 200, '操作成功', "deconstruction", data);
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "showAdmin";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统编辑普通管理员账号信息
    async editAdmin() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const uid = JSON.parse(rdmcode).uid;
            const { model, updateMsg, requestMsg } = ctx.request.body;
            const modelType = this.decryptRequest(model);
            const result = await ctx.service.admin.editAdmin(uid, modelType, updateMsg, requestMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                this.setResponse(ctx, 200, '操作成功', "none");
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "editAdmin";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统删除普通管理员账号信息
    async deleteAdmin() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const uid = JSON.parse(rdmcode).uid;
            const { model, password, requestMsg } = ctx.request.body;
            const modelType = this.decryptRequest(model);
            const deCodePassword = this.decryptRequest(password);
            const result = await ctx.service.admin.deleteAdmin(uid, modelType, deCodePassword, requestMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                this.setResponse(ctx, 200, '操作成功', "none");
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "deleteAdmin";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统导出普通管理员账号信息数据
    async downloadAdminData() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const uid = JSON.parse(rdmcode).uid;
            const { form, searchMsg, dataHeaders } = ctx.request.body;
            const result = await ctx.service.admin.downloadAdminData(uid, searchMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result.code) {
                const file = this.downloadData(form, result.data, dataHeaders);
                if (file) {
                    const { filePath, completeName, buf } = file;
                    try {
                        fs.writeFileSync(filePath, Buffer.from(buf));
                        const data = {
                            // 本机服务器地址
                            // 'http://127.0.0.1:8020'
                            // 'http://192.168.31.28:8020'
                            // 公司服务器地址
                            // 'http://123.57.205.46:8020'
                            // 'http://certadmin.hwasmart.com//bdTrainingExamManager'
                            fileUrl: 'http://127.0.0.1:8020' + `/public/${completeName}?time=${new Date().getTime()}`,
                            fileName: completeName,
                        }
                        this.setResponse(ctx, 200, '文件已成功写入', "deconstruction", data);
                        setTimeout(() => {
                            fs.unlinkSync(filePath);
                        }, 600000);
                    } catch (error) {
                        this.setResponse(ctx, 500, '写入文件时发生错误', "none");
                    }
                } else {
                    this.setResponse(ctx, 404, '操作失败', "none");
                }
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "downloadAdminData";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }
}

module.exports = AdminController;