Calculation/app/service/filing.js

'use strict';

/**
 *
 *
 * @author Mai
 * @date
 * @version
 */
const rootId = '-1';
const filingType = [
    { value: 1, name: '立项文件' },
    { value: 2, name: '招标投标、合同协议文件' },
    { value: 3, name: '勘察、设计文件' },
    { value: 4, name: '征地、拆迁、移民文件' },
    { value: 5, name: '项目管理文件' },
    { value: 6, name: '施工文件' },
    { value: 7, name: '信息系统开发文件' },
    { value: 8, name: '设备文件' },
    { value: 9, name: '监理文件' },
    { value: 10, name: '科研项目文件' },
    { value: 11, name: '生产技术准备、试运行文件' },
    { value: 12, name: '竣工验收文件' },
];
const maxFilingType = 12;

module.exports = app => {
    class Filing extends app.BaseService {

        /**
         * 构造函数
         *
         * @param {Object} ctx - egg全局变量
         * @param {String} tableName - 表名
         * @return {void}
         */
        constructor(ctx) {
            super(ctx);
            this.tableName = 'filing';
        }

        get allFilingType () {
            return filingType.map(x => { return x.value });
        }

        analysisFilingType(filing) {
            const copy = JSON.parse(JSON.stringify(filing));
            const curFilingType = copy.filter(f => {
                return f.is_fixed;
            });
            const checkChildren = function (parent) {
                parent.children = curFilingType.filter(x => { return x.tree_pid === parent.id; });
                if (parent.children.length > 1) parent.children.sort((x, y) => { return x.tree_order - y.tree_order; });
                for (const c of parent.children) {
                    checkChildren(c);
                }
            };
            const topFiling = curFilingType.filter(x => { return x.tree_level === 1; });
            topFiling.sort((x, y) => { return x.tree_order - y.tree_order; });
            for (const tp of topFiling) {
                checkChildren(tp);
            }
            const result = [];
            const getFilingType = function(arr, prefix = '') {
                for (const a of arr) {
                    if (a.children.length) {
                        getFilingType(a.children, prefix ? prefix + '/' + a.name : a.name);
                    } else {
                        result.push({ value: a.filing_type, name: a.name, parentsName: prefix });
                    }
                }
            };
            getFilingType(topFiling);
            return result;
        }

        async getFilingType(spid) {
            const filing = await ctx.service.filing.getValidFiling(ctx.params.id, ctx.subProject.permission.filing_type);
            return this.analysisFilingType(filing);
        }

        async initFiling(spid, templateId, transaction) {
            const count = await this.count({ spid });
            if (count > 0) return;

            const templateFiling = await this.ctx.service.filingTemplate.getAllDataByCondition({
                where: { temp_id: templateId },
                orders: [['tree_level', 'asc']],
            });
            const insertData = [];
            for (const f of templateFiling) {
                f.newId = this.uuid.v4();
                const parent = f.tree_pid !== rootId ? templateFiling.find(x => { return x.id === f.tree_pid; }) : null;
                const newData = {
                    id: f.newId, tree_pid: parent ? parent.newId : rootId, tree_level: f.tree_level, tree_order: f.tree_order,
                    spid, add_user_id: this.ctx.session.sessionUser.accountId,
                    create_uid: this.ctx.session.sessionUser.accountId, is_fixed: f.is_fixed,
                    filing_type: f.filing_type, name: f.name, tips: f.tips, upload_tips: f.upload_tips, file_company: f.file_company,
                };
                insertData.push(newData);
            }
            if (transaction) {
                await transaction.insert(this.tableName, insertData);
            } else {
                await this.db.insert(this.tableName, insertData);
            }
        }

        _buildInitializedFiling(spid, sourceData, operatorUid) {
            const source = (sourceData || []).slice().sort((x, y) => {
                const levelDiff = Number(x.tree_level) - Number(y.tree_level);
                return levelDiff || Number(x.tree_order) - Number(y.tree_order);
            });
            const sourceNodeMap = {};
            const insertData = [];
            for (const node of source) {
                const sourceId = String(node.id || '');
                if (!sourceId || sourceNodeMap[sourceId]) throw '来源项目目录数据错误';
                const sourceParentId = String(node.tree_pid);
                const isRoot = sourceParentId === rootId;
                const parent = isRoot ? null : sourceNodeMap[sourceParentId];
                if (!isRoot && !parent) throw '来源项目目录结构不完整';

                const newNode = {
                    id: this.uuid.v4(),
                    spid,
                    tree_pid: parent ? parent.id : rootId,
                    tree_level: parent ? Number(parent.tree_level) + 1 : 1,
                    tree_order: Number(node.tree_order),
                    name: String(node.name || ''),
                    filing_type: Number(node.filing_type),
                    add_user_id: Number(operatorUid),
                    create_uid: Number(operatorUid),
                    is_fixed: Number(node.is_fixed) === 1 ? 1 : 0,
                    is_deleted: 0,
                    file_count: 0,
                    tips: node.tips || '',
                    upload_tips: node.upload_tips || '',
                    file_company: node.file_company || '',
                    is_rela: Number(node.is_rela) === 1 ? 1 : 0,
                };
                if (!newNode.name || !Number.isInteger(newNode.filing_type) || newNode.filing_type <= 0 ||
                    !Number.isInteger(newNode.tree_order) || newNode.tree_order <= 0) {
                    throw '来源项目目录数据错误';
                }
                sourceNodeMap[sourceId] = newNode;
                insertData.push(newNode);
            }
            return insertData;
        }

        /**
         * 首次初始化项目资料目录。
         *
         * 空白模式创建一个可继续编辑的根目录；模板和项目模式只复制目录字段，
         * 不复制文件与目录授权用户配置。
         *
         * @param {String} spid 目标子项目ID
         * @param {String} initType 初始化方式（blank/template/project）
         * @param {Array} sourceData 来源模板或项目的有效目录
         * @param {Number} operatorUid 初始化用户ID
         * @param {Object|null} templateData 选用的系统模板
         * @param {Object|null} transaction 模板校验与初始化共用的事务
         * @return {Object} 初始化结果
         */
        async initializeDirectory(spid, initType, sourceData, operatorUid, templateData = null, transaction = null) {
            let directorySource = sourceData || [];
            if (initType === 'blank') {
                directorySource = [{
                    id: 'blank-root', tree_pid: rootId, tree_level: 1, tree_order: 1,
                    name: '新建文件夹', filing_type: maxFilingType + 1, is_fixed: 1,
                    tips: '', upload_tips: '', file_company: '', is_rela: 0,
                }];
            } else if (initType !== 'project' && initType !== 'template') {
                throw '初始化方式错误';
            }
            if (initType === 'template' && (!templateData || !templateData.id)) throw '系统模板数据错误';
            if (directorySource.length === 0) throw '来源数据没有可复制的资料目录';
            const insertData = this._buildInitializedFiling(spid, directorySource, operatorUid);

            const conn = transaction || await this.db.beginTransaction();
            try {
                await conn.query('SELECT id FROM ?? WHERE id = ? FOR UPDATE', [
                    this.ctx.service.subProject.tableName, spid,
                ]);
                const activeCount = await conn.count(this.tableName, { spid, is_deleted: 0 });
                if (activeCount > 0) throw '当前项目已经存在资料目录，无需重复初始化';
                await conn.insert(this.tableName, insertData);
                if (initType === 'template') {
                    await conn.update(this.ctx.service.subProject.tableName, {
                        id: spid,
                        filing_template_id: templateData.id,
                        filing_template_name: templateData.name,
                    });
                }
                if (!transaction) await conn.commit();
                return { count: insertData.length };
            } catch (err) {
                if (!transaction) await conn.rollback();
                throw err;
            }
        }

        _filterValidFiling(filing, filingType) {
            const validFiling = filing.filter(x => { return filingType.indexOf(x.filing_type) > -1;});
            const checkParent = function(child) {
                let parent = validFiling.find(x => { return x.id === child.tree_pid; });
                if (!parent) {
                    parent = filing.find(x => { return x.id === child.tree_pid; });
                    validFiling.push(parent);
                }
                if (parent.tree_level > 1) checkParent(parent);
            };
            for (const vf of validFiling) {
                if (vf.tree_level > 1 && vf.is_fixed) checkParent(vf);
            }
            return validFiling;
        }

        async getValidFiling(spid, filingType) {
            if (!filingType || filingType.length === 0) return [];
            const result = await this.getAllDataByCondition({ where: { spid, is_deleted: 0 } });
            if (filingType === 'all') return result;
            return this._filterValidFiling(result, filingType);
        }

        async getPosterityData(id){
            const result = [];
            let cur = await this.getAllDataByCondition({ where: { tree_pid: id } });
            let iLevel = 1;
            while (cur.length > 0 && iLevel < 6) {
                result.push(...cur);
                cur = await this.getAllDataByCondition({ where: { tree_pid: cur.map(x => { return x.id })} });
                iLevel += 1;
            }
            return result;
        }

        _checkFixed(data) {
            if (data.is_fixed) throw '固定分类，不可编辑';
        }

        async getNewName(spid, name = '新增文件类别') {
            const data = await this.db.query(`SELECT * FROM ${this.tableName} WHERE spid = '${spid}' AND name LIKE '${name}%'`);
            if (data.length === 0) return name;

            const _ = this._;
            const names = data.map(x => { return _.toInteger(x.name.replace(name, '')) });
            const filterNames = names.filter(x => { return x > 0 });
            const max = filterNames.reduce((pre, cur) => { return Math.max(pre, cur); }, 0);
            return max >= 0 ? name + (max + 1) : name;
        }
        async getNewFilingType(spid) {
            const max = await this.db.queryOne(`SELECT filing_type FROM ${this.tableName} WHERE spid = '${spid}' ORDER BY filing_type DESC`);
            return max && max.filing_type ? max.filing_type + 1 : maxFilingType + 1;
        }

        async add(data) {
            const parent = await this.getDataById(data.tree_pid);
            // 允许管理员添加顶层
            // if (!parent) throw '添加数据结构错误';
            if (parent && parent.file_count > 0) throw `分类【${parent.name}】下存在文件，不可添加子分类`;

            const sibling = await this.getAllDataByCondition({ where: { spid: this.ctx.subProject.id, tree_pid: parent ? parent.id : rootId, is_deleted: 0 }, orders: [['tree_order', 'asc']]});
            const preChild = data.tree_pre_id ? sibling.find(x => { return x.id === data.tree_pre_id; }) : null;
            const filing_type = parent ? parent.filing_type : await this.getNewFilingType(this.ctx.subProject.id);

            const conn = await this.db.beginTransaction();
            try {
                // 获取当前用户信息
                const sessionUser = this.ctx.session.sessionUser;
                // 获取当前项目信息
                const sessionProject = this.ctx.session.sessionProject;

                const tree_order = preChild ? preChild.tree_order + 1 : (sibling.length > 0 ? sibling[sibling.length - 1].tree_order + 1 : 1);
                const name = data.name === undefined ? await this.getNewName(this.ctx.subProject.id) : data.name;
                const insertData = {
                    id: this.uuid.v4(), spid: this.ctx.subProject.id, add_user_id: sessionUser.accountId,
                    create_uid: sessionUser.accountId,
                    tree_pid: parent ? parent.id : rootId, tree_level: parent ? parent.tree_level + 1 : 1, tree_order,
                    name, filing_type: filing_type, is_fixed: parent ? 0 : 1, file_count: 0, is_deleted: 0
                };
                const operate = await conn.insert(this.tableName, insertData);
                if (operate.affectedRows === 0) throw '新增文件夹失败';

                const updateData = [];
                if (preChild) {
                    sibling.forEach(x => {
                        if (x.tree_order >= tree_order) updateData.push({ id: x.id, tree_order: x.tree_order + 1 });
                    });
                }
                if (updateData.length > 0) await conn.updateRows(this.tableName, updateData);

                await conn.commit();
                return { create: [insertData], update: updateData };
            } catch (error) {
                await conn.rollback();
                throw error;
            }
        }
        async save(data) {
            const filing = await this.getDataById(data.id);
            // this._checkFixed(filing);

            const result = await this.db.update(this.tableName, data);
            if (result.affectedRows > 0) {
                return data;
            } else {
                throw '更新数据失败';
            }
        }
        async del(data) {
            const filing = await this.getDataById(data.id);
            // this._checkFixed(filing);

            const posterity = await this.getPosterityData(data.id);
            const delData = posterity.map(x => {return { id: x.id, is_deleted: 1 }; });
            delData.push({ id: data.id, is_deleted: 1});

            const sibling = await this.getAllDataByCondition({ where: { tree_pid: filing.tree_pid, spid: delData.spid, is_deleted: 0 } });
            const updateData = [];
            sibling.forEach(x => {
                if (x.tree_order > filing.tree_order) updateData.push({ id: x.id, tree_order: x.tree_order - 1});
            });
            const conn = await this.db.beginTransaction();
            try {
                await conn.updateRows(this.tableName, delData);
                if (updateData.length > 0) conn.updateRows(this.tableName, updateData);
                await conn.update(this.ctx.service.file.tableName, { is_deleted: 1}, { where: {filing_id: delData.map(x => { return x.id; })} });
                await this.ctx.service.subProjectFilingPermission.removeFilingNodes(
                    filing.spid, delData.map(x => x.id), conn
                );

                await conn.commit();
                return { delete: delData.map(x => { return x.id }), update: updateData };
            } catch(err) {
                await conn.rollback();
                throw err;
            }
        }
        async move(data) {
            const filing = await this.getDataById(data.id);
            if (!filing || filing.is_deleted) throw '移动的分类不存在，请刷新页面后重试';
            if (this.ctx.subProject && filing.spid !== this.ctx.subProject.id) throw '移动的分类不属于当前项目';
            if (Number(filing.is_fixed)) throw '固定分类不可移动';
            if (data.tree_pid === undefined || data.tree_pid === null || data.tree_pid === '') throw '请选择移动后的目录';
            const treePid = String(data.tree_pid);
            const treeOrder = Number(data.tree_order);
            if (!Number.isInteger(treeOrder) || treeOrder < 0) throw '移动后的目录顺序错误';
            const posterity = await this.getPosterityData(filing.id);
            const filingWithFiles = [filing, ...posterity].find(x => !x.is_deleted && Number(x.file_count) > 0);
            if (filingWithFiles) throw `分类【${filingWithFiles.name}】下存在文件，不可移动目录`;
            if (treePid === String(filing.id) || posterity.find(x => String(x.id) === treePid)) {
                throw '不能将目录移动到自身或其子目录下';
            }
            const parent = treePid === rootId ? null : await this.getDataById(treePid);
            if (treePid !== rootId && (!parent || parent.is_deleted || parent.spid !== filing.spid)) {
                throw '移动后的分类不存在，请刷新页面后重试';
            }
            if (parent && String(filing.tree_pid) !== treePid && Number(parent.file_count) > 0) {
                throw `分类【${parent.name}】下存在文件，不可添加子分类`;
            }
            const sibling = await this.getAllDataByCondition({ where: { spid: filing.spid, tree_pid: treePid, is_deleted: 0 } });
            const updateData = { id: filing.id, tree_order: treeOrder, tree_pid: treePid, tree_level: (parent ? parent.tree_level : 0) + 1 };
            if (data.name !== undefined) {
                const name = String(data.name).trim();
                if (!name) throw '目录名称不能为空';
                if (name.length > 100) throw '目录名称不能超过100个字符';
                updateData.name = name;
            }
            if (data.is_fixed !== undefined) {
                const isFixed = Number(data.is_fixed);
                if (isFixed !== 0 && isFixed !== 1) throw '固定目录状态错误';
                updateData.is_fixed = isFixed;
            }
            const posterityUpdateData = posterity.map(x => {
                return { id: x.id, tree_level: (parent ? parent.tree_level : 0) + 1 - filing.tree_level + x.tree_level };
            });
            const siblingUpdateData = [];
            if (treePid === String(filing.tree_pid)) {
                if (treeOrder < filing.tree_order) {
                    sibling.forEach(x => {
                        if (x.id === filing.id) return;
                        if (x.tree_order < treeOrder) return;
                        if (x.tree_order > filing.tree_order) return;
                        siblingUpdateData.push({ id: x.id, tree_order: x.tree_order + 1 });
                    });
                } else {
                    sibling.forEach(x => {
                        if (x.id === filing.id) return;
                        if (x.tree_order < filing.tree_order) return;
                        if (x.tree_order > treeOrder) return;
                        siblingUpdateData.push({ id: x.id, tree_order: x.tree_order - 1 });
                    });
                }
            } else {
                const orgSibling = await this.getAllDataByCondition({ where: { spid: filing.spid, tree_pid: filing.tree_pid, is_deleted: 0 } });
                orgSibling.forEach(x => {
                    if (x.id === filing.id) return;
                    if (x.tree_order < filing.tree_order) return;
                    siblingUpdateData.push({ id: x.id, tree_order: x.tree_order - 1 });
                });
                sibling.forEach(x => {
                    if (x.id === filing.id) return;
                    if (x.tree_order < treeOrder) return;
                    siblingUpdateData.push({ id: x.id, tree_order: x.tree_order + 1 });
                });
            }

            const conn = await this.db.beginTransaction();
            try {
                await conn.update(this.tableName, updateData);
                if (posterityUpdateData.length > 0) await conn.updateRows(this.tableName, posterityUpdateData);
                if (siblingUpdateData.length > 0) await conn.updateRows(this.tableName, siblingUpdateData);
                await conn.commit();
            } catch (err) {
                await conn.rollback();
                throw err;
            }
            return { update: [updateData, ...posterityUpdateData, ...siblingUpdateData] };
        }

        async editDirectory(data) {
            if (!data || !data.id) throw '请选择需要编辑的目录';
            const filing = await this.getDataById(data.id);
            if (!filing || filing.is_deleted) throw '编辑的分类不存在，请刷新页面后重试';
            if (this.ctx.subProject && filing.spid !== this.ctx.subProject.id) throw '编辑的分类不属于当前项目';
            const name = String(data.name || '').trim();
            if (!name) throw '目录名称不能为空';
            if (name.length > 100) throw '目录名称不能超过100个字符';
            if (data.tree_pid === undefined && data.is_fixed === undefined) {
                if (typeof data.name !== 'string') throw '目录名称格式错误';
                const result = await this.db.update(this.tableName, { name }, {
                    where: { id: filing.id, spid: filing.spid, is_deleted: 0 },
                });
                if (result.affectedRows === 0) throw '更新目录失败';
                return { update: [{ id: filing.id, name }] };
            }
            const isFixed = Number(data.is_fixed);
            if (isFixed !== 0 && isFixed !== 1) throw '固定目录状态错误';
            if (data.tree_pid === undefined || data.tree_pid === null || data.tree_pid === '') throw '请选择目录';
            const treePid = String(data.tree_pid);
            if (treePid === String(filing.tree_pid)) {
                const updateData = { id: filing.id, name, is_fixed: isFixed };
                const result = await this.db.update(this.tableName, updateData);
                if (result.affectedRows === 0) throw '更新目录失败';
                return { update: [updateData] };
            }

            const parent = treePid === rootId ? null : await this.getDataById(treePid);
            if (treePid !== rootId && (!parent || parent.is_deleted || parent.spid !== filing.spid)) {
                throw '移动后的分类不存在，请刷新页面后重试';
            }
            const targetChildren = await this.getAllDataByCondition({
                where: { spid: filing.spid, tree_pid: treePid, is_deleted: 0 },
                orders: [['tree_order', 'asc']],
            });
            const lastChild = targetChildren.length > 0 ? targetChildren[targetChildren.length - 1] : null;
            const treeOrder = lastChild ? Number(lastChild.tree_order) + 1 : 1;
            return await this.move({
                id: filing.id, tree_pid: treePid, tree_order: treeOrder,
                name, is_fixed: isFixed,
            });
        }

        async multiUpdate(spid, data) {
            if (!data || data.length === 0) throw '提交数据格式错误';

            const sourceData = await this.getAllDataByCondition({ where: { spid } });

            const validFields = ['id', 'is_fixed', 'name', 'filing_type', 'tree_order', 'tips', 'upload_tips', 'file_company'];
            const updateData = [];
            for (const d of data) {
                if (!d.id) throw '提交数据格式错误';
                const sd = sourceData.find(x => { return x.id === d.id; });
                if (!sd) throw '提交数据格式错误';

                const nd = {};
                for (const prop in d) {
                    if (validFields.indexOf(prop) < 0) continue;
                    nd[prop] = d[prop];
                }
                updateData.push(nd);
            }
            await this.db.updateRows(this.tableName, updateData);
            return await this.getAllDataByCondition({ where: { spid } });
        }

        async sumFileCount(spid) {
            const result = await this.db.queryOne(`SELECT SUM(file_count) AS file_count FROM ${this.tableName} WHERE spid = '${spid}' and is_deleted = 0`);
            return result.file_count;
        }

        async saveNames(spid, items) {
            if (!Array.isArray(items) || items.length === 0) throw '请选择需要替换名称的分类';
            const ids = new Set();
            const updateData = items.map(item => {
                if (!item || typeof item.id !== 'string' || !item.id || ids.has(item.id)) throw '分类数据格式错误';
                ids.add(item.id);
                if (typeof item.name !== 'string' || typeof item.old_name !== 'string') throw '分类名称格式错误';
                const name = item.name.trim();
                if (!name) throw '目录名称不能为空';
                if (name.length > 100) throw '目录名称不能超过100个字符';
                return { id: item.id, name, old_name: item.old_name };
            });
            const conn = await this.db.beginTransaction();
            try {
                const result = [];
                // 按 ID 更新，保持多批次重叠操作的加锁顺序一致。
                const sortedData = [...updateData].sort((a, b) => a.id.localeCompare(b.id));
                for (const item of sortedData) {
                    const filing = await conn.get(this.tableName, { id: item.id, spid, is_deleted: 0 });
                    if (!filing) throw '部分分类不存在，请刷新页面后重新粘贴';
                    if (Number(filing.is_fixed)) throw '固定分类不可修改名称';
                    if (filing.name !== item.old_name) throw '分类名称已发生变化，请刷新页面后重新粘贴';
                    if (item.name === item.old_name) continue;
                    const operation = await conn.update(this.tableName, { name: item.name }, {
                        where: { id: item.id, spid, is_deleted: 0, is_fixed: 0, name: item.old_name },
                    });
                    if (operation.affectedRows !== 1) throw '分类信息已发生变化，请刷新页面后重新粘贴';
                    result.push({ id: item.id, name: item.name });
                }
                await conn.commit();
                return { update: result };
            } catch (err) {
                await conn.rollback();
                throw err;
            }
        }
        async getDirectoryFilesForUpdate(spid, filingIds, transaction) {
            const sql = 'SELECT id FROM ?? WHERE spid = ? AND filing_id IN (?) AND is_deleted = 0 LIMIT 1 FOR UPDATE';
            const sqlParam = [this.ctx.service.file.tableName, spid, filingIds];
            return await transaction.query(sql, sqlParam);
        }

        async deleteEmptyDirectories(spid, filingIds, updateData, transaction) {
            await transaction.update(this.tableName, { is_deleted: 1 }, {
                where: { spid, id: filingIds, is_deleted: 0 },
            });
            if (updateData.length > 0) await transaction.updateRows(this.tableName, updateData);
            await this.ctx.service.subProjectFilingPermission.removeFilingNodes(spid, filingIds, transaction);
        }
    }

    return Filing;
};