Calculation/app/service/sub_project.js

'use strict';

/**
 *
 *
 * @author Mai
 * @date
 * @version
 */
const rootId = '-1';

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

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

        async getSubProject(pid, uid, admin) {
            let result = await this.getAllDataByCondition({ where: { project_id: pid, is_delete: 0 } });
            if (admin) return result;

            const permission = await this.ctx.service.subProjPermission.getUserPermission(pid, uid);
            result = result.filter(x => {
                if (x.is_folder) return true;
                const pb = permission.find(y => { return x.id === y.spid});
                if (!pb) return false;
                x.user_permission = pb;
                return x.user_permission.budget_permission.length > 0 || x.user_permission.file_permission.length > 0 || x.manage_permission.length > 0;
            });
            return result;
        }

        _filterEmptyFolder(data) {
            data.sort((a, b) => { return b.tree_level - a.tree_level});
            const result = [];
            for (const d of data) {
                if (!d.is_folder) result.push(d);
                if (result.find(x => { return x.tree_pid === d.id; })) result.push(d);
            }
            return result;
        }

        async getBudgetProject(pid, uid, admin) {
            const sql = `SELECT sp.*, b.std_id From ${this.tableName} sp LEFT JOIN ${this.ctx.service.budget.tableName} b ON sp.budget_id = b.id  
                WHERE sp.project_id = ? AND sp.is_delete = 0`;
            let result = await this.db.query(sql, [pid]);

            const permission = await this.ctx.service.subProjPermission.getUserPermission(pid, uid);
            result = result.filter(x => {
                if (!x.is_folder && !x.budget_id) return false;
                if (x.is_folder || admin) return true;
                const pb = permission.find(y => { return x.id === y.spid});
                if (!pb) return false;
                x.permission = pb.budget_permission;
                x.manage_permission = pb.manage_permission;
                return x.budget_permission.length > 0;
            });
            console.log(result.length);
            return this._filterEmptyFolder(result);
        }

        async getFileProject(pid, uid, admin) {
            let result = await this.getAllDataByCondition({ where: { project_id: pid, is_delete: 0 } });

            const permission = await this.ctx.service.subProjPermission.getUserPermission(pid, uid);
            result = result.filter(x => {
                if (x.is_folder || admin) return true;
                const pb = permission.find(y => { return x.id === y.spid});
                if (!pb) return false;
                x.permission = pb.file_permission;
                x.manage_permission = pb.manage_permission;
                return x.file_permission.length > 0;
            });
            return this._filterEmptyFolder(result);
        }

        async getLastChild(tree_pid) {
            const result = await this.getAllDataByCondition({ where: { tree_pid }, orders: [['tree_order', 'desc']], limit: 1, offset: 0 });
            return result[0];
        }

        async addFolder(data) {
            const parent = await this.getDataById(data.tree_pid);
            if (parent && !parent.is_folder) throw '添加数据结构错误';
            const lastChild = await this.getLastChild(parent ? parent.id : rootId);

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

                const insertData = {
                    id: this.uuid.v4(), project_id: sessionProject.id, user_id: sessionUser.accountId,
                    tree_pid: data.tree_pid,
                    tree_level: parent ? parent.tree_level + 1 : 1,
                    tree_order: lastChild ? lastChild.tree_order + 1 : 1,
                    name: data.name, is_folder: 1,
                };
                const operate = await conn.insert(this.tableName, insertData);

                if (operate.affectedRows === 0) throw '新增文件夹失败';

                await conn.commit();
                return await this.getSubProject(sessionProject.id, sessionUser.accountId, sessionUser.is_admin);
            } catch (error) {
                await conn.rollback();
                throw error;
            }
        }

        async addSubProject(data) {
            const parent = await this.getDataById(data.tree_pid);
            if (parent && !parent.is_folder) throw '添加数据结构错误';
            const lastChild = await this.getLastChild(parent ? parent.id : rootId);

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

                const insertData = {
                    id: this.uuid.v4(), project_id: sessionProject.id, user_id: sessionUser.accountId,
                    tree_pid: data.tree_pid,
                    tree_level: parent ? parent.tree_level + 1 : 1,
                    tree_order: lastChild ? lastChild.tree_order + 1 : 1,
                    name: data.name, is_folder: 0,
                };
                const operate = await conn.insert(this.tableName, insertData);
                // todo 根据节点新增时的其他操作

                if (operate.affectedRows === 0) throw '新增文件夹失败';

                await conn.commit();
                return await this.getSubProject(sessionProject.id, sessionUser.accountId, sessionUser.is_admin);
            } catch (error) {
                await conn.rollback();
                throw error;
            }
        }

        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;
        }

        async dragTo(data) {
            const dragNode = await this.getDataById(data.drag_id);
            const dropNode = await this.getDataById(data.drop_id);
            if (!dragNode || !dropNode || !dropNode.is_folder) throw '拖拽数据结构错误';
            const lastChild = await this.getLastChild(dropNode.id);
            const posterity = await this.getPosterityData(dragNode.id);

            const conn = await this.db.beginTransaction();
            try {
                const updateData = {
                    id: dragNode.id, tree_pid: dropNode.id, tree_level: dropNode.tree_level + 1,
                    tree_order: lastChild ? lastChild.tree_order + 1 : 1,
                };
                await conn.update(this.tableName, updateData);
                if (dragNode.tree_level !== dropNode.tree_level + 1 && posterity.length > 0) {
                    const posterityUpdateData = posterity.map(x => {
                        return { id: x.id, tree_level: dropNode.tree_level + 1 - dragNode.tree_level + x.tree_level }
                    });
                    await conn.updateRows(this.tableName, posterityUpdateData);
                }
                // 升级原来的后项的order
                await conn.query(`UPDATE ${this.tableName} SET tree_order = tree_order-1 WHERE tree_pid = ? AND tree_order > ?`, [dragNode.tree_pid, dragNode.tree_order]);
                await conn.commit();
            } catch (error) {
                await conn.rollback();
                throw error;
            }

            return await this.getSubProject(this.ctx.session.sessionProject.id, this.ctx.session.sessionUser.accountId, this.ctx.session.sessionUser.is_admin);
        }

        async del(id) {
            const node = await this.getDataById(id);
            if (!node) throw '删除的数据不存在';

            const posterity = await this.getPosterityData(node.id);
            const updateData = [
                { id: node.id, is_delete: 1 },
            ];
            posterity.forEach(x => {
                updateData.push({ id: x.id, is_delete: 1});
            });
            await this.db.updateRows(this.tableName, updateData);

            return await this.getSubProject(this.ctx.session.sessionProject.id, this.ctx.session.sessionUser.accountId, this.ctx.session.sessionUser.is_admin);
        }

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

        async setBudgetStd(data) {
            const subProject = await this.getDataById(data.id);
            const budgetStd = await this.ctx.service.budgetStd.getDataById(data.std_id);
            if (!budgetStd) throw '选择的概算标准不存在，请刷新页面重试';

            const conn = await this.db.beginTransaction();
            try {
                const budget_id = await this.ctx.service.budget.add(conn, {
                    pid: subProject.project_id, user_id: subProject.user_id, rela_tender: subProject.rela_tender
                }, budgetStd);
                const updateData = { id: data.id, std_id: budgetStd.id, std_name: budgetStd.name, budget_id };
                await conn.update(this.tableName, updateData);
                await conn.commit();
                return updateData;
            } catch (error) {
                await conn.rollback();
                throw error;
            }
        }

        async setRelaTender(data) {
            const subProject = await this.getDataById(data.id);
            const conn = await this.db.beginTransaction();
            try {
                await conn.update(this.tableName, data);
                await conn.update(this.ctx.service.budget.tableName, { id: subProject.budget_id, rela_tender: data.rela_tender });
                await conn.commit();
                return data;
            } catch (error) {
                await conn.rollback();
                throw error;
            }
        }
    }

    return SubProject;
};