ConstructionOperation/modules/price_info_lib/controllers/index.js

import BaseController from "../../common/base/base_controller";
import CompilationModel from '../../users/models/compilation_model';
const facade = require('../facade/index');
const config = require("../../../config/config.js");

class PriceInfoController extends BaseController {
    async main(req, res) {
        const compilationModel = new CompilationModel();
        const compilationList = await compilationModel.getCompilationList({ _id: 1, name: 1 });
        compilationList.unshift({ _id: 'all', name: '所有' });
        const activeCompilation = compilationList.find(compilation => compilation._id.toString() === req.query.filter);
        if (activeCompilation) {
            activeCompilation.active = 'active';
        } else {
            compilationList[0].active = 'active'
        }
        const filter = req.query.filter ? { compilationID: req.query.filter } : {};
        const libs = await facade.getLibs(filter);
        libs.forEach(lib => {
            compilationList.forEach(compilation => {
                if (compilation._id.toString() === lib.compilationID) {
                    lib.compilationName = compilation.name;
                }
            });
        });
        const listItem = `
            <li class="nav-item">
                <a class="nav-link" href="javascript:void(0);" aria-haspopup="true" aria-expanded="false" data-toggle="modal" data-target="#crawl">导入材料价格信息</a>
            </li>`
        const renderData = {
            title: '材料信息价库',
            userAccount: req.session.managerData.username,
            userID: req.session.managerData.userID,
            libs: libs,
            compilationList: compilationList,
            listItem,
            layout: 'maintain/common/html/layout'
        };
        res.render("maintain/price_info_lib/html/main.html", renderData);
    }

    async editView(req, res) {
        const { libID } = req.query;
        const libs = await facade.getLibs({ ID: libID });
        if (!libs.length) {
            return res.send(404);
        }
        const areaList = await facade.getAreas(libs[0].compilationID);
        const renderData = {
            compilationID: libs[0].compilationID,
            libName: libs[0].name,
            areaList: JSON.stringify(areaList),
            userAccount: req.session.managerData.username,
            userID: req.session.managerData.userID,
            LicenseKey: config.getLicenseKey(process.env.NODE_ENV),
        };
        res.render("maintain/price_info_lib/html/edit.html", renderData);
    }

    async addLib(req, res) {
        try {
            const { name, period, compilationID } = req.body;
            await facade.createLib(name, period, compilationID)
        } catch (err) {
            console.log(err);
        }
        res.redirect(req.headers.referer);
    }

    async renameLib(req, res) {
        try {
            const { libID, name } = JSON.parse(req.body.data);
            await facade.updateLib({ ID: libID }, { name });
            res.json({ error: 0, message: 'rename success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async deleteLib(req, res) {
        try {
            const { libID } = JSON.parse(req.body.data);
            await facade.deleteLib(libID);
            res.json({ error: 0, message: 'delete success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    // 爬取数据
    async crawlData(req, res) {
        try {
            const { from, to, compilationID } = JSON.parse(req.body.data);
            res.setTimeout(1000 * 60 * 60 * 2); // 不设置的话，处理时间过长，会触发默认的响应超时，报错(前端报错，后台还继续在处理)
            await facade.crawlDataByCompilation(compilationID, from, to);
            res.json({ error: 0, message: 'crawl success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async editArea(req, res) {
        try {
            const { updateData } = JSON.parse(req.body.data);
            await facade.updateAres(updateData);
            res.json({ error: 0, message: 'update areas success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async insertArea(req, res) {
        try {
            const { insertData } = JSON.parse(req.body.data);
            await facade.insertAreas(insertData);
            res.json({ error: 0, message: 'update areas success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async deleteArea(req, res) {
        try {
            const { deleteData } = JSON.parse(req.body.data);
            await facade.deleteAreas(deleteData);
            res.json({ error: 0, message: 'update areas success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async getClassData(req, res) {
        try {
            const { libID, areaID } = JSON.parse(req.body.data);
            const data = await facade.getClassData(libID, areaID);
            res.json({ error: 0, message: 'getCLass success', data });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async getPriceData(req, res) {
        try {
            const { classIDList } = JSON.parse(req.body.data);
            const data = await facade.getPriceData(classIDList);
            res.json({ error: 0, message: 'getPriceData success', data });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async editPriceData(req, res) {
        try {
            const { postData } = JSON.parse(req.body.data);
            await facade.editPriceData(postData);
            res.json({ error: 0, message: 'editPrice success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

    async editClassData(req, res) {
        try {
            const { updateData } = JSON.parse(req.body.data);
            await facade.editClassData(updateData);
            res.json({ error: 0, message: 'editClass success' });
        } catch (err) {
            console.log(err);
            res.json({ error: 1, message: err.toString() });
        }
    }

}

module.exports = {
    priceInfoController: new PriceInfoController()
};