Calculation/app/lib/stage_im.js

'use strict';

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

const Ledger = require('./ledger');
const imType = require('../const/tender').imType;

class StageIm {
    constructor (ctx) {
        const self = this;
        this.ctx = ctx;
        this._ = this.ctx.helper._;
        // mainData
        this.billsTree = new Ledger.billsTree(this.ctx, {
            id: 'ledger_id',
            pid: 'ledger_pid',
            order: 'order',
            level: 'level',
            rootId: -1,
            keys: ['id', 'tender_id', 'ledger_id'],
            stageId: 'id',
            calcFields: ['deal_tp', 'total_price', 'contract_tp', 'qc_tp', 'gather_tp'],
            calc: function (node) {
                if (node.children && node.children.length === 0) {
                    node.gather_qty = self.ctx.helper.add(node.contract_qty, node.qc_qty);
                }
                node.gather_tp = self.ctx.helper.add(node.contract_tp, node.qc_tp);
            }
        });
        this.pos = new Ledger.pos({
            id: 'id', ledgerId: 'lid',
            updateFields: ['contract_qty', 'qc_qty', 'postil'],
            calc: function (p) {
                p.gather_qty = self.ctx.helper.add(p.contract_qty, p.qc_qty);
            }
        });
        // relaData
        this.change = null;
        this.details = null;
        // result
        this.ImData = [];
        this.ImBillsData = [];
        //
        this.imFields = ['uuid', 'doc_code', 'peg', 'bw', 'xm', 'drawing_code', 'calc_memo', 'calc_img'];
        this.splitChar = '-';
    }

    // 加载数据
    async _loadMainData () {
        const billsData = await this.ctx.service.ledger.getData(this.ctx.tender.id);
        if (this.ctx.stage.readOnly) {
            const curStage = await this.ctx.service.stageBills.getAuditorStageData(this.ctx.tender.id,
                this.ctx.stage.id, this.ctx.stage.curTimes, this.ctx.stage.curOrder);
            this.ctx.helper.assignRelaData(billsData, [
                {data: curStage, fields: ['contract_qty', 'contract_tp', 'qc_qty', 'qc_tp'], prefix: '', relaId: 'lid'}
            ]);
        } else {
            const curStage = await this.ctx.service.stageBills.getLastestStageData(this.ctx.tender.id, this.ctx.stage.id);
            this.ctx.helper.assignRelaData(billsData, [
                {data: curStage, fields: ['contract_qty', 'contract_tp', 'qc_qty', 'qc_tp'], prefix: '', relaId: 'lid'}
            ]);
        }
        this.billsTree.loadDatas(billsData);
        this.billsTree.calculateAll();

        const posData = await this.ctx.service.pos.getAllDataByCondition({ where: {tid: this.ctx.tender.id }});
        if (this.ctx.stage.readOnly) {
            const curPosStage = await this.ctx.service.stagePos.getAuditorStageData2(this.ctx.tender.id,
                this.ctx.stage.id, this.ctx.stage.curTimes, this.ctx.stage.curOrder);
            this.ctx.helper.assignRelaData(posData, [
                {data: curPosStage, fields: ['contract_qty', 'qc_qty'], prefix: '', relaId: 'pid'}
            ]);
        } else {
            const curPosStage = await this.ctx.service.stagePos.getLastestStageData2(this.ctx.tender.id, this.ctx.stage.id);
            this.ctx.helper.assignRelaData(posData, [
                {data: curPosStage, fields: ['contract_qty', 'qc_qty'], prefix: '', relaId: 'pid'}
            ]);
        }
        this.pos.loadDatas(posData);
        this.pos.calculateAll();
    }
    async _loadRelaData () {
        if (this.ctx.stage.readOnly) {
            this.details = await this.ctx.service.stageDetail.getAuditorStageData(this.ctx.tender.id, this.ctx.stage.id, this.ctx.stage.curTimes, this.ctx.stage.curOrder);
        } else {
            this.details = await this.ctx.service.stageDetail.getLastestStageData(this.ctx.tender.id, this.ctx.stage.id);
        }
        if (this.ctx.stage.readOnly) {
            this.changes = await this.ctx.service.stageChange.getAuditorAllStageData(this.ctx.tender.id, this.ctx.stage.id, this.ctx.stage.curTimes, this.ctx.stage.curOrder);
        } else {
            this.changes = await this.ctx.service.stageChange.getLastestAllStageData(this.ctx.tender.id, this.ctx.stage.id);
        }
    }

    // 检查汇总节点
    async _checkGather() {
        const gatherNodes = this.ctx.stage.im_gather_node ? this.ctx.stage.im_gather_node.split(',') : [];
        for (const node of this.billsTree.datas) {
            node.check = gatherNodes.indexOf(node.id) !== -1;
        }
    }

    // 辅助取值方法
    _getNumberFormat(num, length) {
        let s = '0000000000';
        s = s + num;
        return s.substr(s.length - length);
    }
    _getNodeByLevel(node, level) {
        let cur = node;
        while (cur && cur.level > level) {
            cur = this.billsTree.getParent(cur);
        }
        return cur;
    }
    _checkPeg(text) {
        const pegReg = /[a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?/;
        return pegReg.test(text);
    }
    _getPegStr(text) {
        const pegReg1 = /[a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?[~～—][a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?/;
        const result1 = text.match(pegReg1);
        if (result1) {
            return result1[0];
        } else {
            const pegReg2 = /[a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?-[a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?/;
            const result2 = text.match(pegReg2);
            if (result2) {
                return result2[0];
            } else {
                const pegReg3 = /[a-zA-Z]?[kK][0-9]+[+＋][0-9]{3}([.][0-9]+)?/;
                const result3 = text.match(pegReg3);
                return result3 ? result3[0] : '';
            }
        }
    }
    _getPegNode (node) {
        if (node) {
            if (this._checkPeg(node.name)) {
                return node;
            } else {
                const parent = this.billsTree.getParent(node);
                return parent ? this._getPegNode(parent) : null;
            }
        }
    }
    _getDrawingCode(node) {
        if (!node) {
            return '';
        } else if (node.drawing_code) {
            return node.drawing_code;
        } else {
            const parent = this.billsTree.getParent(node);
            return parent ? this._getDrawingCode(parent) : '';
        }
    }
    _getZlNormalBw(node, peg) {
        if (peg) {
            const subPeg1 = this._getNodeByLevel(node, peg.level + 1);
            const subPeg2 = this._getNodeByLevel(node, peg.level + 2);
            let result = peg.name;
            if (subPeg1 && subPeg1.id !== peg.id && subPeg1.id !== node.id) {
                result = result + '-' + subPeg1.name;
                if (subPeg2 && subPeg2.id !== subPeg1.id && subPeg2.id !== node.id) {
                    result = result + '-' + subPeg2.name;
                }
            }
            return result;
        } else {
            if (node.level === 2 || node.level === 3) {
                return node.name;
            } else if (node.level >= 4) {
                let parent = this.billsTree.getParent(node), result = parent.name;
                while (parent.level > 3 && parent) {
                    parent = this._getNodeByLevel(node, parent.level - 1);
                    result = parent.name + '-' + result;
                }
                return result;
            } else {
                return '';
            }
        }
    }
    _getZlGatherBw(node, peg) {
        if (peg) {
            const subPeg1 = this._getNodeByLevel(node, peg.level + 1);
            let result = peg.name;
            if (subPeg1 && subPeg1.id !== peg.id) {
                result = result + '-' + subPeg1.name;
            }
            return result;
        } else {
            if (node.level < 3) {
                return node.name;
            } else {
                let parent = node, result = parent.name;
                while (parent.level > 3 && parent) {
                    parent = this._getNodeByLevel(node, parent.level - 1);
                    result = parent.name + '-' + result;
                }
                return result;
            }
        }
    }

    _checkCustomDetail(im) {
        const self = this;
        const cd = this._.find(this.details, function (d) {
            return im.lid === d.lid &&
                (!im.code || im.code === d.code) &&
                (!im.name || im.name === d.name) &&
                (!im.unit || im.unit === d.unit) &&
                self.ctx.helper.checkZero(self.ctx.helper.sub(im.unit_price, d.unit_price)) &&
                (!im.pid || im.pid === d.pid);
        });
        if (cd) {
            this._.assignInWith(im, cd, function (oV, sV, key) {
                return self.imFields.indexOf(key) > -1 && sV !== undefined && sV !== null ? sV : oV;
            });
        }
    }

    _getCalcMemo(im) {
        if (im.calc_memo !== undefined && im.calc_memo !== null && im.calc_memo !== '') return;

        if (im.leafXmjs && im.leafXmjs.length > 0) {
            const memo = ['本期计量：' + im.jl + ' ' + im.unit];
            for (const lx of im.leafXmjs) {
                for (const p of lx.pos) {
                    memo.push(p.name + '：' + p.jl + ' ' + im.unit);
                }
            }
            im.calc_memo = memo.join('\n');
        } else if (im.check) {
            const memo = [];
            for (const [i, b] of im.gclBills.entries()) {
                if (b.pos && b.pos.length > 0) {
                    memo.push('清单' + (i+1) + b.b_code + ' ' + b.name);
                    for (const p of b.pos) {
                        memo.push(p.name + '：' + p.jl + ' ' + b.unit);
                    }
                } else {
                    memo.push('清单' + (i+1) + b.b_code + ' ' + b.name + '：' + b.jl + ' ' + b.unit);
                }
            }
            im.calc_memo = memo.join('\n');
        } else {
            im.calc_memo =  '';
        }
    }
    _getChangeInfo(im) {
        if (im.changes && im.changes.length > 0) {
            const code = this._.uniq(this._.map(im.changes, 'c_code'));
            if (!im.bgl_code || im.bgl_code === '') im.bgl_code = code.join(';');
            const new_code = this._.uniq(this._.map(im.changes, 'c_new_code'));
            if (!im.bgl_drawing_code || im.bgl_drawing_code === '') im.bgl_drawing_code = new_code.join(';');
        }
    }

    _generateTzPosData(node, gclBills) {
        if (!gclBills.pos) {
            gclBills.pos = [];
        }
        const posRange = this.pos.getLedgerPos(node.id);
        if (!posRange) { return }
        for (const p of posRange) {
            if (!p.gather_qty || this.ctx.helper.checkZero(p.gather_qty)) { continue; }
            let lp = this._.find(gclBills.pos, {name: p.name});
            if (!lp) {
                lp = {name: p.name};
                gclBills.pos.push(lp);
            }
            lp.jl = this.ctx.helper.add(lp.jl, p.gather_qty);
            lp.contract_jl = this.ctx.helper.add(lp.contract_jl, p.contract_qty);
            lp.qc_jl = this.ctx.helper.add(lp.qc_jl, p.qc_qty);
        }
    }
    _generateTzGclBillsData(node, im) {
        if (!im.gclBills) {
            im.gclBills = [];
        }
        const posterity = this.billsTree.getPosterity(node);
        for (const p of posterity) {
            if ((!p.b_code || p.b_code === '') || (p.children && p.children.length > 0)) {
                continue;
            }
            if ((!p.contract_tp || p.contract_tp === 0) && (!p.qc_tp || p.qc_tp === 0)) {
                continue;
            }
            let b = this._.find(im.gclBills, {bid: p.id});
            if (!b) {
                b = {imid: im.id, bid: p.id, b_code: p.b_code, name: p.name, unit: p.unit};
                im.gclBills.push(b);
                this.ImBillsData.push(b);
            }
            b.jl = this.ctx.helper.add(b.jl, p.gather_qty);
            b.contract_jl = this.ctx.helper.add(b.contract_jl, p.contract_qty);
            b.qc_jl = this.ctx.helper.add(b.qc_jl, p.qc_qty);
            this._generateTzPosData(p, b);
        }
    }
    _generateTzChangeData(node, im) {
        if (!im.changes) {
            im.changes = [];
        }
        const posterity = this.billsTree.getPosterity(node);
        for (const p of posterity) {
            if (p.children && p.children.length > 0) {
                continue;
            }
            if ((!p.qc_tp || p.qc_tp === 0)) {
                continue;
            }
            const posRange = this.pos.getLedgerPos(p.id);
            if (!posRange) {
                for (const c of this.changes) {
                    if (c.lid === p.id && c.pid == -1 && c.qty && c.qty !== 0) {
                        im.changes.push(c);
                    }
                }
            } else {
                for (const pp of posRange) {
                    if ((!pp.qc_tp || pp.qc_tp === 0)) {
                        continue;
                    }
                    for (const c of this.changes) {
                        if (c.lid === p.id && c.pid === pp.id && c.qty && c.qty !== 0) {
                            im.changes.push(c);
                        }
                    }
                }
            }
        }
    }
    /**
     * 生成 0号台账 中间计量数据
     * @param {Object} node - 生成中间计量表的节点
     */
    _generateTzImData (node) {
        if (node.gather_tp) {
            const peg = this._getPegNode(node);
            const im = {
                id: this.ImData.length + 1,
                lid: node.id, pid: '', code: node.code,
                jl: node.gather_tp, contract_jl: node.contract_tp, qc_jl: node.qc_tp,
                peg: peg ? this._getPegStr(peg.name) : '', drawing_code: this._getDrawingCode(node),
            };
            if (this.ctx.stage.im_gather && node.check) {
                im.bw = this._getZlGatherBw(node, peg);
                im.xm = '';
            } else {
                im.bw = this._getZlNormalBw(node, peg);
                im.xm = node.name;
            }
            this._checkCustomDetail(im);
            im.check = this.ctx.stage.im_gather && node.check;
            this._generateTzGclBillsData(node, im);
            this.ImData.push(im);
            this._generateTzChangeData(node, im);
        }
    }

    _generateZlPosData(node, lx) {
        if (!lx.pos) {
            lx.pos = [];
        }
        const posRange = this.pos.getLedgerPos(node.id);
        if (!posRange) { return }
        for (const p of posRange) {
            if (!p.gather_qty || this.ctx.helper.checkZero(p.gather_qty)) { continue; }
            let lp = this._.find(lx.pos, {name: p.name});
            if (!lp) {
                lp = {name: p.name};
                lx.pos.push(lp);
            }
            lp.jl = this.ctx.helper.add(lp.jl, p.gather_qty);
            lp.contract_jl = this.ctx.helper.add(lp.contract_jl, p.contract_qty);
            lp.qc_jl = this.ctx.helper.add(lp.qc_jl, p.qc_qty);
        }
    }
    _generateZlLeafXmjData(node, im) {
        if (!im.leafXmjs) {
            im.leafXmjs = [];
        }
        const leafXmj = this.billsTree.getLeafXmjParent(node);
        if (!leafXmj) { return }
        let lx = this._.find(im.leafXmjs, {lxid: leafXmj.id});
        if (!lx) {
            lx = {
                lxid: leafXmj.id,
                code: leafXmj.code,
                name: leafXmj.name
            };
            im.leafXmjs.push(lx);
        }
        lx.jl = this.ctx.helper.add(lx.jl, node.gather_qty);
        lx.contract_jl = this.ctx.helper.add(lx.contract_jl, node.contract_qty);
        lx.qc_jl = this.ctx.helper.add(lx.qc_jl, node.qc_qty);
        this._generateZlPosData(node, lx);
    }
    _generateZlChangeData(node, im) {
        if (!im.changes) {
            im.changes = [];
        }
        if ((!node.qc_qty || node.qc_qty === 0)) {
            return;
        }
        const posRange = this.pos.getLedgerPos(node.id);
        if (!posRange) {
            for (const c of this.changes) {
                if (c.lid === node.id && c.pid == -1 && c.qty && c.qty !== 0) {
                    im.changes.push(c);
                }
            }
        } else {
            for (const p of posRange) {
                if ((!p.qc_qty || p.qc_qty === 0)) {
                    continue;
                }
                for (const c of this.changes) {
                    if (c.lid === node.id && c.pid === p.id && c.qty && c.qty !== 0) {
                        im.changes.push(c);
                    }
                }
            }
        }
    }
    /**
     * 生成 总量控制 中间计量数据
     * @param {Object} node - 生成中间计量表的节点
     */
    _generateZlImData (node) {
        const self = this;
        const nodeImData = [], posterity = this.billsTree.getPosterity(node);
        for (const p of posterity) {
            if (p.children && p.children.length > 0 ) { continue; }
            if (!p.b_code || p.b_code === '') { continue }
            if (this.ctx.helper.checkZero(p.contract_qty) && this.ctx.helper.checkZero(p.qc_qty) ) { continue; }
            let im = nodeImData.find(function (d) {
                return d.lid === node.id &&
                    d.code === p.b_code && p.name === d.name && p.unit === d.unit &&
                    this.ctx.helper.checkZero(self.ctx.helper.sub(p.unit_price, d.unit_price));
            });
            if (!im) {
                const peg = this._getPegNode(node);
                im = {
                    id: this.ImData.length + 1,
                    lid: node.id, pid: '', code: p.b_code, name: p.name, unit: p.unit, unit_price: p.unit_price,
                    jl: 0, contract_jl: 0, qc_jl: 0,
                    peg: peg ? this._getPegStr(peg.name) : ''
                };
                if (this.ctx.stage.im_gather && node.check) {
                    im.check = true;
                    im.bw = this._getZlGatherBw(node, peg);
                    im.xm = '';
                    im.drawing_code = this._getDrawingCode(node);
                } else {
                    im.check = false;
                    im.bw = this._getZlNormalBw(node, peg);
                    im.xm = node.name;
                    im.drawing_code = this._getDrawingCode(p);
                }
                nodeImData.push(im);
                this._checkCustomDetail(im);
                this.ImData.push(im);
            }
            if (!this.ctx.stage.im_gather || !node.check) {
                this._generateZlLeafXmjData(p, im, 'gather_qty');
            }
            this._generateZlChangeData(p, im);
            im.jl = this.ctx.helper.add(im.jl, p.gather_qty);
            im.contract_jl = this.ctx.helper.add(im.contract_jl, p.contract_qty);
            im.qc_jl = this.ctx.helper.add(im.qc_jl, p.qc_qty);
        }
    }

    _generateBwImData (node) {
        const posterity = this.billsTree.getPosterity(node);
        for (const p of posterity) {
            if (p.children && p.children.length > 0 ) { continue; }
            if (!p.b_code || p.b_code === '') { continue }
            const peg = this._getPegNode(node);
            const pPos = this.pos.getLedgerPos(p.id);
            const bw = this._getZlNormalBw(node, peg);
            if (pPos && pPos.length > 0) {
                for (const pp of pPos) {
                    if (this.ctx.helper.checkZero(pp.contract_qty) && this.ctx.helper.checkZero(pp.qc_qty)) { continue }
                    const im = {
                        id: this.ImData.length + 1,
                        lid: node.id, code: p.b_code, name: p.name, unit: p.unit, unit_price: p.unit_price, pid: pp.id,
                        jl: pp.gather_qty, contract_jl: pp.contract_qty, qc_jl: pp.qc_qty,
                        bw: bw,
                        peg: this._checkPeg(pp.name) ? this._getPegStr(pp.name) : (peg ? this._getPegStr(peg.name) : ''),
                        xm: pp.name,
                        drawing_code: pp.drawing_code,
                        changes: [],
                    };
                    im.calc_memo = '本期计量：' + im.jl + ' ' + im.unit;
                    this._checkCustomDetail(im);
                    this.ImData.push(im);
                    if (pp.qc_qty && pp.qc_qty !== 0) {
                        for (const c of this.changes) {
                            if (c.lid === p.id && c.pid === pp.id && c.qty && c.qty !== 0) {
                                im.changes.push(c);
                            }
                        }
                    }
                }
            } else {
                if (this.ctx.helper.checkZero(p.contract_qty) && this.ctx.helper.checkZero(p.qc_qty)) { continue }

                const im = {
                    id: this.ImData.length + 1,
                    lid: node.id, code: p.b_code, name: p.name, unit: p.unit, unit_price: p.unit_price, pid: '',
                    jl: p.gather_qty, contract_jl: p.contract_qty, qc_jl: p.qc_qty,
                    bw: bw,
                    peg: peg ? this._getPegStr(peg.name) : '',
                    xm: node.name,
                    drawing_code: this._getDrawingCode(node),
                    changes: [],
                };
                im.calc_memo = '本期计量：' + im.jl + ' ' + im.unit;
                this._checkCustomDetail(im);
                this.ImData.push(im);
                if (p.qc_qty && p.qc_qty !== 0) {
                    for (const c of this.changes) {
                        if (c.lid === p.id && c.pid == -1 && c.qty && c.qty !== 0) {
                            im.changes.push(c);
                        }
                    }
                }
            }
        }
    }

    /**
     * 递归 生成中间计量表
     * @param {Array} nodes
     */
    _recursiveBuildImData (nodes) {
        if (!nodes || nodes.length === 0) { return; }
        for (const node of nodes) {
            if (this.billsTree.isLeafXmj(node) || (this.ctx.stage.im_type !== imType.bw.value && this.ctx.stage.im_gather && node.check)) {
                if (this.ctx.stage.im_type === imType.tz.value) {
                    this._generateTzImData(node);
                } else if (this.ctx.stage.im_type === imType.zl.value) {
                    this._generateZlImData(node);
                } else if (this.ctx.stage.im_type === imType.bw.value) {
                    this._generateBwImData(node);
                }
            } else {
                this._recursiveBuildImData(node.children);
            }
        }
    }

    // 生成中间计量数据
    async buildImData () {
        const self = this;
        // 初始化
        await this._loadMainData();
        await this._loadRelaData();
        if (this.ctx.stage.im_gather) {
            this._checkGather();
        }
        // 生成数据
        this._recursiveBuildImData(this.billsTree.children);
        // 排序
        if (this.ctx.stage.im_type !== imType.tz.value) {
            this.ImData.sort(function (x, y) {
                return self.ctx.helper.compareCode(x.code, y.code);
            });
        }
        // 生成数据(需要缓存，并清理缓存)
        const pre = (this.ctx.stage.im_pre && this.ctx.stage.im_pre !== '') ? this.ctx.stage.im_pre + splitChar : '';
        for (const [i, im] of this.ImData.entries()) {
            im.im_code = pre + this._getNumberFormat(this.ctx.stage.order, 2) + this.splitChar + this._getNumberFormat(i + 1, 3);
            if (im.gclBills) {
                for (const b of im.gclBills) {
                    b.im_code = im.im_code;
                }
            }

            this._getCalcMemo(im);
            delete im.leafXmjs;
            delete im.gclBills;

            this._getChangeInfo(im);
            delete im.changes;
        }
    }

    // 生成中间计量汇总数据
    async buildImGatherData () {
        this._loadMainData();
        if (stage.im_gather) {
            this._checkGather();
        }
    }
}

module.exports = StageIm;