Calculation/app/middleware/budget_check.js

'use strict';

/**
 *
 *
 * @author Mai
 * @date
 * @version
 */

module.exports = options => {
    /**
     * 标段校验 中间件
     * 1. 读取标段数据（包括属性）
     * 2. 检验用户是否可见标段（不校验具体权限）
     *
     * @param {function} next - 中间件继续执行的方法
     * @return {void}
     */
    return function* budgetCheck(next) {
        try {
            if (!this.session.sessionProject.showBudget) {
                throw '该功能已关闭或无法查看';
            }
            // 读取标段数据
            const id = parseInt(this.params.id);
            if (!id) throw '参数错误';
            this.budget = yield this.service.budget.getCurBudget(id);
            if (!this.budget) throw '项目不存在';
            if (this.budget.pid !== this.session.sessionProject.id) throw '您无权查看该项目';

            const subProj = yield this.service.subProject.getDataByCondition({ budget_id: this.budget.id });
            if (subProj) this.budget.name = subProj.name || '';
            if (this.session.sessionUser.is_admin) {
                this.budget.readOnly = false;
            } else {
                const bp = yield this.service.subProjPermission.getBudgetUserPermission(id);
                if (!bp) throw '您无权查看该项目';
                this.budget.readOnly = bp.budget_permission.indexOf(this.service.subProjPermission.PermissionConst.budget.edit.value) < 0;
            }
            yield next;
        } catch (err) {
            this.log(err);
            if (this.helper.isAjax(this.request)) {
                this.ajaxErrorBody(err, '概算投资项目未知错误');
            } else {
                this.postError(err, '概算投资项目未知错误');
                err === '该功能已关闭或无法查看' ? this.redirect('/dashboard') : this.redirect(this.request.headers.referer);
            }
        }
    };
};