Calculation/app/controller/advance_controller.js

'use strict';
const accountGroup = require('../const/account_group').group;
const auditConst = require('../const/audit').advance;
const sendToWormhole = require('stream-wormhole');
const path = require('path');
const fs = require('fs');
module.exports = app => {
    class advanceController extends app.BaseController {

        /**
         * 开工预付款页面(AJAX) GET
         * @param {Object} ctx 全局上下文
         */
        async index(ctx) {
            const type = auditConst.type.start;
            const advanceList = await ctx.service.advance.getAdvanceList(ctx.tender.id, type);
            const latestOrder = await ctx.service.advance.getLastestAdvance(ctx.tender.id, type, true);
            const advancePayTotal = ctx.tender.info.deal_param.startAdvance;
            const progress = await ctx.service.advance.calcProgress(latestOrder, advancePayTotal);
            const renderData = {
                type,
                advanceList,
                latestOrder,
                auditConst,
                preUrl: `/tender/${ctx.tender.id}/advance/${type}/create`,
                jsFiles: this.app.jsFiles.common.concat(this.app.jsFiles.advance.main),
                advancePayTotal,
                progress,
            };
            await this.layout('advance/index.ejs', renderData, 'advance/modal.ejs');
        }

        /**
         * 材料预付款页面(AJAX) GET
         * @param {Object} ctx 全局上下文
         */
        async materialList(ctx) {
            const type = auditConst.type.material;
            const advanceList = await ctx.service.advance.getAdvanceList(ctx.tender.id, type);
            const latestOrder = await ctx.service.advance.getLastestAdvance(ctx.tender.id, type, true);
            const advancePayTotal = ctx.tender.info.deal_param.materialAdvance;
            const progress = await ctx.service.advance.calcProgress(latestOrder, advancePayTotal);
            const renderData = {
                type,
                advanceList,
                latestOrder,
                auditConst,
                preUrl: `/tender/${ctx.tender.id}/advance/${type}/create`,
                jsFiles: this.app.jsFiles.common.concat(this.app.jsFiles.advance.main),
                advancePayTotal,
                progress,
            };
            await this.layout('advance/index.ejs', renderData, 'advance/modal.ejs');
        }

        /**
         * 获取通用的renderData（用于layout, Menu, subMenu部分）
         * @param {Object} ctx 全局上下文
         * @return {{auditConst, jsFiles, accountGroup?, accountList?, auditors, auditHistory, preUrl}}
         * @private
         */
        async _getDefaultRenderData(ctx) {
            const data = {
                auditConst,
                jsFiles: this.app.jsFiles.common.concat(this.app.jsFiles.advance.info),
                preUrl: `/tender/${ctx.tender.id}/advance/${ctx.advance.id}`,
                whiteList: ctx.app.config.multipart.whitelist,
            };
            // 获取所有项目参与者
            if ((ctx.advance.status === auditConst.status.uncheck || ctx.advance.status === auditConst.status.checkNo) && ctx.session.sessionUser.accountId === ctx.advance.uid) {
                data.accountGroup = accountGroup;
                // 获取所有项目参与者
                const accountList = await ctx.service.projectAccount.getAllDataByCondition({
                    where: { project_id: ctx.session.sessionProject.id, enable: 1 },
                    columns: ['id', 'name', 'company', 'role', 'enable', 'is_admin', 'account_group'],
                });
                data.accountList = accountList;
            }
            // 获取审核人左边列表
            data.auditors = await ctx.service.advanceAudit.getAuditorsWithOwner(ctx.advance.id, ctx.advance.times);
            // data.fileList = await ctx.service.advanceFile.getAdvanceFiles({ vid: ctx.advance.id });
            // 获取审批流程中右边列表
            const auditHistory = [];
            const times = ctx.advance.status === auditConst.status.checkNo ? ctx.advance.times - 1 : ctx.advance.times;
            if (times >= 1) {
                for (let i = 1; i <= times; i++) {
                    auditHistory.push(await ctx.service.advanceAudit.getAuditors(ctx.advance.id, i));
                }
            }
            data.auditHistory = auditHistory;
            data.fileList = await ctx.service.advanceFile.getAdvanceFiles({ vid: ctx.advance.id }) || [];
            return data;
        }

        /**
         * 获取审批界面所需的 原报、审批人数据等
         * @param {Object} ctx 全局上下文
         * @private
         */
        async _getAdvanceAuditViewData(ctx) {
            const times = ctx.advance.status === auditConst.status.checkNo ? ctx.advance.times - 1 : ctx.advance.times;
            ctx.advance.user = await ctx.service.projectAccount.getAccountInfoById(ctx.advance.uid);
            ctx.advance.auditHistory = [];
            if (ctx.advance.times > 1) {
                for (let i = 1; i < ctx.advance.times; i++) {
                    ctx.advance.auditHistory.push(await ctx.service.advanceAudit.getAuditors(ctx.advance.id, i));
                }
            }
            // 获取审批流程中左边列表
            ctx.advance.auditors2 = await ctx.service.advanceAudit.getAuditGroupByList(ctx.advance.id, times);
            if (ctx.advance.status === auditConst.status.uncheck || ctx.advance.status === auditConst.status.checkNo) {
                ctx.advance.auditorList = await ctx.service.advanceAudit.getAuditors(ctx.advance.id, ctx.advance.times);
            }
        }

        /**
         * 预付款详情页(AJAX) GET
         * @param {Object} ctx 全局上下文
         */
        async detail(ctx) {
            const advancePayTotal = ctx.advance.type === 0 ? ctx.tender.info.deal_param.startAdvance : ctx.tender.info.deal_param.materialAdvance;
            try {
                const renderData = await this._getDefaultRenderData(ctx);
                await this._getAdvanceAuditViewData(ctx);
                const { uncheck, checkNo } = auditConst.status;
                const { status } = ctx.advance;
                // 获取上一期预付款记录
                const prevAdvance = await ctx.service.advance.getPreviousRecord(ctx.tender.id, ctx.advance.type);
                renderData.isEdited = status === uncheck || status === checkNo;
                renderData.advance = ctx.advance;
                renderData.advancePayTotal = advancePayTotal;
                renderData.prevAdvance = prevAdvance;
                await this.layout('advance/detail.ejs', renderData, 'advance/modal_audit.ejs');
            } catch (error) {
                this.log(error);
                ctx.redirect('/tender/' + ctx.tender.id + '/advance');
            }
        }

        /**
         * 更新
         * @param {Object} ctx 全局上下文
         */
        async update(ctx) {
            const { id } = ctx.advance;
            const data = JSON.parse(ctx.request.body.data);
            try {
                const result = await ctx.service.advance.updateAdvance(data, id);
                if (result) {
                    ctx.body = { err: 0, msg: '请求成功', data: null };
                } else {
                    ctx.body = { err: 0, msg: '请求失败', data: null };
                }
            } catch (error) {
                this.log(error);
                ctx.body = { code: 1, msg: error.toString() };
            }

        }

        /**
         * 添加新一期
         * @param {Object} ctx 全局上下文
         */
        async create(ctx) {
            const type = parseInt(ctx.params.type);
            try {
                const record = await ctx.service.advance.createRecord(type);
                if (!record) {
                    throw '数据错误';
                }
                ctx.redirect(`/tender/${ctx.tender.id}/advance/${record.id}/detail`);
            } catch (error) {
                this.log(error);
                ctx.redirect(`/tender/${ctx.tender.id}/advance`);
            }
        }

        /**
         * 添加审批人
         * @param {Object} ctx 全局上下文
         */
        async addAudit(ctx) {
            try {
                const data = JSON.parse(ctx.request.body.data);
                const audit_id = data && this.app._.toInteger(data.auditorId);
                if (isNaN(audit_id) || audit_id <= 0) {
                    throw '参数错误';
                }
                // 检查权限等
                if (ctx.advance.uid !== ctx.session.sessionUser.accountId) {
                    throw '您无权添加审核人';
                }
                if (ctx.advance.status === auditConst.status.checking || ctx.advance.status === auditConst.status.checked) {
                    throw '当前不允许添加审核人';
                }
                // 检查审核人是否已存在
                const exist = this.app._.find(ctx.advance.auditors, { audit_id });
                if (exist) {
                    throw '该审核人已存在，请勿重复添加';
                }

                const result = await ctx.service.advanceAudit.addAuditor(ctx.tender.id, ctx.advance.id, audit_id, ctx.advance.times);
                if (!result) {
                    throw '添加审核人失败';
                }

                const audit = await ctx.service.advanceAudit.getAuditor(ctx.advance.id, audit_id, ctx.advance.times);
                ctx.body = { err: 0, msg: '', data: audit };
            } catch (err) {
                this.log(err);
                ctx.body = { err: 1, msg: err.toString(), data: null };
            }
        }

        /**
         * 移除审批人
         * @param {Object} ctx 全局上下文
         */
        async deleteAudit(ctx) {
            try {
                const data = JSON.parse(ctx.request.body.data);
                const id = data && data.auditorId instanceof Number ? data.auditorId : this.app._.toNumber(data.auditorId);
                if (isNaN(id) || id <= 0) {
                    throw '参数错误';
                }

                const result = await ctx.service.advanceAudit.deleteAuditor(ctx.advance.id, id, ctx.advance.times);
                if (!result) {
                    throw '移除审核人失败';
                }

                const auditors = await ctx.service.advanceAudit.getAuditGroupByList(ctx.advance.id, ctx.advance.times);
                ctx.body = { err: 0, msg: '', data: auditors };
            } catch (err) {
                ctx.body = { err: 1, msg: err.toString(), data: null };
            }
        }

        /**
         * 上报审批
         * @param {Object} ctx 全局上下文
         */
        async start(ctx) {
            const data = JSON.parse(ctx.request.body.data);
            try {
                // 检查权限等
                if (!ctx.advance) {
                    throw '数据错误';
                }
                if (ctx.advance.uid !== ctx.session.sessionUser.accountId) {
                    throw '您无权上报该期数据';
                }
                if (ctx.advance.status === auditConst.status.checking || ctx.advance.status === auditConst.status.checked) {
                    throw '该预付款期数据当前无法上报';
                }
                if (data.cur_amount === 0) {
                    throw '本期金额不能为空';
                }
                await ctx.service.advanceAudit.start(ctx.advance.id, ctx.advance.times, data);
                ctx.body = { err: 0, msg: '' };
            } catch (error) {
                this.log(error);
                ctx.session.postError = error.toString();
                ctx.body = { err: 1, msg: error.toString() };
            }
        }

        /**
         * 审批
         * @param {Object} ctx 全局上下文
         */
        async checkAudit(ctx) {
            try {
                if (!ctx.advance || ctx.advance.status !== auditConst.status.checking) {
                    throw '当前材料调差期数据有误';
                }
                if (!ctx.advance.curAuditor || ctx.advance.curAuditor.audit_id !== ctx.session.sessionUser.accountId) {
                    throw '您无权进行该操作';
                }
                const data = {
                    checkType: parseInt(ctx.request.body.checkType),
                    opinion: ctx.request.body.opinion,
                };
                if (!data.checkType || isNaN(data.checkType)) {
                    throw '提交数据错误';
                }
                if (data.checkType === auditConst.status.checkNo) {
                    if (!data.checkType || isNaN(data.checkType)) {
                        throw '提交数据错误';
                    }
                }

                await ctx.service.advanceAudit.check(ctx.advance.id, data, ctx.advance.times);

                ctx.redirect(ctx.request.header.referer);
            } catch (err) {
                this.log(err);
                ctx.session.postError = err.toString();
                ctx.redirect(ctx.request.header.referer);
            }
        }

        _checkAdvanceFileCanModify(ctx) {
            // 检查登录用户，是否可操作
            const accountId = ctx.session.sessionUser.accountId;
            if (!ctx.advance.curAuditor) {
                if (ctx.advance.status === auditConst.status.uncheck || ctx.advance.status === auditConst.status.checkNo && accountId === ctx.advance.uid) {
                    return;
                }
                throw '该预付款期当前您无权操作';
            } else {
                if (ctx.advance.curAuditor.audit_id === accountId) return;
                throw '该预付款期当前您无权操作';
            }
        }

        /**
         * 上传附件
         * @param {*} ctx 上下文
         */
        async upload(ctx) {
            let stream;
            try {
                this._checkAdvanceFileCanModify(ctx);
                const parts = this.ctx.multipart({
                    autoFields: true,
                });
                const files = [];
                const create_time = Date.parse(new Date()) / 1000;
                let idx = 0;
                while ((stream = await parts()) !== undefined) {
                    if (!stream.filename) {
                        // 如果没有传入直接返回
                        return;
                    }
                    const fileInfo = path.parse(stream.filename);
                    const filepath = `public/upload/${this.ctx.tender.id.toString()}/yfk/fujian_${create_time + idx.toString() + fileInfo.ext}`;
                    await ctx.helper.saveStreamFile(stream, path.resolve(this.app.baseDir, 'app', filepath));
                    files.push({ filepath, name: stream.filename, ext: fileInfo.ext });
                    ++idx;
                    stream && (await sendToWormhole(stream));
                }
                const in_time = new Date();
                const payload = files.map(file => {
                    let idx;
                    if (Array.isArray(parts.field.name)) {
                        idx = parts.field.name.findIndex(name => name === file.name);
                    } else {
                        idx = 'isString';
                    }
                    const newFile = {
                        uid: ctx.session.sessionUser.accountId,
                        vid: ctx.advance.id,
                        tid: ctx.tender.id,
                        create_time: in_time,
                        filepath: file.filepath,
                        filesize: ctx.helper.bytesToSize(idx === 'isString' ? parts.field.size : parts.field.size[idx]),
                        filename: file.name,
                        fileext: file.ext,
                    };
                    return newFile;
                });
                // 执行文件信息写入数据库
                await ctx.service.advanceFile.saveFileMsgToDb(payload);
                // 将最新的当前标段的所有文件信息返回
                const data = await ctx.service.advanceFile.getAdvanceFiles({ vid: ctx.advance.id });
                ctx.body = { err: 0, msg: '', data };
            } catch (err) {
                stream && (await sendToWormhole(stream));
                this.log(err);
                ctx.body = { err: 1, msg: err.toString(), data: null };
            }
        }

        /**
         * 删除附件
         * @param {Ojbect} ctx 上下文
         */
        async deleteFile(ctx) {
            try {
                const { id } = JSON.parse(ctx.request.body.data);
                const fileInfo = await ctx.service.advanceFile.getDataById(id);
                if (fileInfo) {
                    // 先删除文件
                    await fs.unlinkSync(path.resolve(this.app.baseDir, './app', fileInfo.filepath));
                    // 再删除数据库
                    await ctx.service.advanceFile.delete(id);
                } else {
                    throw '不存在该文件';
                }
                const data = await ctx.service.advanceFile.getAdvanceFiles({ vid: ctx.advance.id });
                ctx.body = { err: 0, msg: '请求成功', data };
            } catch (err) {
                this.log(err);
                ctx.body = { err: 1, msg: err.toString(), data: null };
            }
        }

        /**
         * 下载附件
         * @param {Object} ctx - egg全局变量
         * @return {void}
         */
        async downloadFile(ctx) {
            const id = ctx.params.fid;
            if (id) {
                try {
                    const fileInfo = await ctx.service.advanceFile.getDataById(id);
                    if (fileInfo !== undefined && fileInfo !== '') {
                        const fileName = path.join(__dirname, '../', fileInfo.filepath);
                        // 解决中文无法下载问题
                        const userAgent = (ctx.request.header['user-agent'] || '').toLowerCase();
                        let disposition = '';
                        if (userAgent.indexOf('msie') >= 0 || userAgent.indexOf('chrome') >= 0) {
                            disposition = 'attachment; filename=' + encodeURIComponent(fileInfo.filename);
                        } else if (userAgent.indexOf('firefox') >= 0) {
                            disposition = 'attachment; filename*="utf8\'\'' + encodeURIComponent(fileInfo.filename) + '"';
                        } else {
                            /* safari等其他非主流浏览器只能自求多福了 */
                            disposition = 'attachment; filename=' + new Buffer(fileInfo.filename).toString('binary');
                        }
                        ctx.response.set({
                            'Content-Type': 'application/octet-stream',
                            'Content-Disposition': disposition,
                            'Content-Length': fileInfo.filesize,
                        });
                        ctx.body = await fs.createReadStream(fileName);
                    } else {
                        throw '不存在该文件';
                    }
                } catch (err) {
                    this.log(err);
                    this.setMessage(err.toString(), this.messageType.ERROR);
                }
            }
        }
    }
    return advanceController;
};