YangHuOperation/modules/users/models/compilation_model.js

/**
 * 编办管理业务逻辑模型
 *
 * @author CaiAoLin
 * @date 2017/7/28
 * @version
 */
import mongoose from "mongoose";
import BaseModel from "../../common/base/base_model";
import uuidV1 from "uuid/v1";

class CompilationModel extends BaseModel {
  /**
   * 允许的块
   *
   * @var {Array}
   */
  sectionList = [
    "suggestion",
    "feasibility",
    "rough",
    "bill",
    "ration",
    "estimation",
  ];

  /**
   * 构造函数
   *
   * @return {void}
   */
  constructor() {
    let parent = super();
    parent.model = mongoose.model("compilation");
    parent.init();
  }

  /**
   * 获取编办列表
   *
   * @return {Promise}
   */
  async getCompilationList(fields = null) {
    // 筛选字段
    let field =
      fields == null
        ? {
            _id: 1,
            name: 1,
            is_release: 1,
            release_time: 1,
            defaultLocation: 1,
            categoryID: 1,
            description: 1,
            overWriteUrl: 1,
            example: 1,
            edition: 1,
            freeUse: 1,
            "ration_valuation.id": 1,
            "ration_valuation.name": 1,
            "ration_valuation.enable": 1,
            "suggestion_valuation.id": 1,
            "suggestion_valuation.name": 1,
            "suggestion_valuation.enable": 1,
            "feasibility_valuation.id": 1,
            "feasibility_valuation.name": 1,
            "feasibility_valuation.enable": 1,
            "rough_valuation.id": 1,
            "rough_valuation.name": 1,
            "rough_valuation.enable": 1,
            "bill_valuation.id": 1,
            "bill_valuation.name": 1,
            "bill_valuation.enable": 1,
            "estimation_valuation.id": 1,
            "estimation_valuation.name": 1,
            "estimation_valuation.enable": 1,
            "bill_valuation.fileTypes": 1,
            type: 1,
          }
        : fields;
    // 一定要查询序号，然后排序输出
    field.serialNumber = 1;
    let compilationData = await this.findDataByCondition(
      { name: { $ne: "" } },
      field,
      false
    );
    if (compilationData) {
      // 按照序号排序
      compilationData.sort((aa, bb) => {
        return (aa.serialNumber || 99999) - (bb.serialNumber || 99999);
      });
      return compilationData;
    }
    return [];
  }

  // 根据用户权限过滤编办
  async getPermissionCompilationList(req, fields = null) {
    const compilationPermission =
      req.session.managerData.compilationPermission || [];
    console.log(`req.session.managerData`);
    console.log(req.session.managerData.compilationPermission);
    const compilationList = await this.getCompilationList(fields);
    console.log(compilationList);
    return compilationList.filter(
      (item) =>
        compilationPermission.includes(item._id) ||
        compilationPermission.includes(item._id.toString())
    );
  }

  /**
   * 获取编办列表
   *
   * @return {Promise}
   */
  async getList() {
    // 筛选字段
    let field = {
      _id: 1,
      name: 1,
      is_release: 1,
      description: 1,
      categoryID: 1,
    };
    let compilationData = await this.findDataByCondition(
      { name: { $ne: "" }, is_release: true },
      field,
      false
    );

    return compilationData === null ? [] : compilationData;
  }

  /**
   * 根据id获取可用的编办数据
   *
   * @param {String} id
   * @return {Promise}
   */
  async getCompilationById(id) {
    let condition = { _id: id, is_release: true };
    let compilationData = await this.findDataByCondition(condition);
    if (!compilationData || compilationData.bill_valuation === undefined) {
      return compilationData;
    }

    if (compilationData.bill_valuation.length > 0) {
      let enableValuation = [];
      for (let index in compilationData.bill_valuation) {
        if (compilationData.bill_valuation[index].enable) {
          enableValuation.push(compilationData.bill_valuation[index]);
        }
      }
      compilationData.bill_valuation = enableValuation;
    }

    if (compilationData.ration_valuation.length > 0) {
      let enableValuation = [];
      for (let index in compilationData.ration_valuation) {
        if (compilationData.ration_valuation[index].enable) {
          enableValuation.push(compilationData.bill_valuation[index]);
        }
      }
      compilationData.ration_valuation = enableValuation;
    }

    return compilationData;
  }

  /**
   * 设置场景
   *
   * @param {string} scene
   * @return {void}
   */
  setScene(scene = "") {
    switch (scene) {
      // 新增
      case "add":
        this.model.schema.path("name").required(true);
        this.model.schema.path("creator").required(true);
        this.model.schema.path("create_time").required(true);
        break;
    }
  }

  /**
   * 新增编办
   *
   * @param {Object} data
   * @return {Promise}
   */
  async add(data) {
    let result = false;
    if (Object.keys(data).length <= 0) {
      return result;
    }

    this.setScene("add");
    data.create_time = new Date().getTime();
    result = this.db.create(data);
    return result;
  }

  /*
   * 设置版本描述
   *
   * @param {String} compilationId
   * @param {String} description
   * @return {Promise}
   * */
  async setDescription(compilationId, description) {
    return await this.updateById(compilationId, { description: description });
  }
  /*
        设置代码覆盖路径
     */
  async setOverWriteUrl(
    compilationId,
    overWriteUrl,
    priceProp,
    consumeAmtProp
  ) {
    return await this.updateById(compilationId, {
      overWriteUrl: overWriteUrl,
      priceProperties: priceProp,
      consumeAmtProperties: consumeAmtProp,
    });
  }

  /*
   * 设置例题
   * @param {String} compilationId
   * @param {Array} example
   * @return {Promise}
   * */
  async setExample(compilationId, example) {
    let data = [];
    if (example) {
      for (let projId of example) {
        if (projId && projId.trim() !== "") data.push(projId.trim());
      }
    }
    return await this.updateById(compilationId, { example: data });
  }

  /**
   * 新增计价规则
   *
   * @param {String} id
   * @param {String} section
   * @param {Object} data
   * @return {Promise}
   */
  async addValuation(id, section, data) {
    let condition = { _id: id };
    let compilationData = await this.findDataByCondition(condition);
    if (compilationData === null || compilationData.name === undefined) {
      throw "没有找到对应的数据";
    }

    if (data.name === undefined || data.name === "") {
      throw "计价规则名称为空";
    }
    if (this.sectionList.indexOf(section) < 0) {
      console.log("2");
      throw "数据有误";
    }

    let insertData = {};
    data.id = uuidV1();
    insertData[section + "_valuation"] = data;
    console.log(insertData);
    let result = await this.db.addToSet(condition, insertData);
    return result.ok === undefined ? false : data.id;
  }

  /**
   * 保存计价数据
   *
   * @param {String} valuationId
   * @param {Object} data
   * @return {Promise}
   */
  async saveValuation(valuationId, data) {
    data = this._filterValuationData(data);
    let sectionString = data.section + "_valuation";
    let condition = {};
    condition[sectionString + ".id"] = valuationId;

    let updateData = {};
    updateData[sectionString + ".$.name"] = data.valuationName;

    let result = await this.db.update(condition, updateData);

    return result !== null && result.ok === 1;
  }

  /**
   * 更改启用/禁用
   *
   * @param {String} valuationId
   * @param {String} section
   * @param {String} enable
   * @return {Promise}
   */
  async switchEnable(valuationId, section, enable) {
    let sectionString = section + "_valuation";
    let condition = {};
    condition[sectionString + ".id"] = valuationId;

    let updateData = {};
    updateData[sectionString + ".$.enable"] = enable === "true";

    let result = await this.db.update(condition, updateData);

    return result !== null && result.ok === 1;
  }

  /**
   * 设置计价规则适用类型
   *
   * @param {String} valuationId
   * @param {String} section
   * @param {String} setFileTypes
   * @return {Promise}
   */
  async setFileTypes(valuationId, section, fileTypes) {
    let sectionString = section + "_valuation";
    let condition = {};
    condition[sectionString + ".id"] = valuationId;

    let updateData = {};
    updateData[sectionString + ".$.fileTypes"] = fileTypes;

    let result = await this.db.update(condition, updateData);

    return result !== null && result.ok === 1;
  }

  /**
   * 过滤计价数据
   *
   * @param {Object} data
   * @return {Object}
   */
  _filterValuationData(data) {
    if (Object.keys(data).length <= 0) {
      console.log("3");
      throw "数据有误";
    }

    if (
      data.section === undefined ||
      data.section === "" ||
      this.sectionList.indexOf(data.section) < 0
    ) {
      throw "类型错误";
    }

    // 判断名称
    if (data.valuationName === undefined || data.valuationName === "") {
      throw "名称不能为空";
    }

    return data;
  }

  /**
   * 获取计价规则数据
   *
   * @param {String} compilationId
   * @param {String} id
   * @param {String} section
   * @return {Promise|Array}
   */
  async getValuation(compilationId, id, section) {
    if (this.sectionList.indexOf(section) < 0) {
      throw "数据有误";
    }
    let compilationData = await this.findDataByCondition({
      _id: compilationId,
    });
    if (Object.keys(compilationData).length <= 0) {
      throw "编办数据有误";
    }
    let result = {};
    let sectionString = section + "_valuation";

    for (let valuation of compilationData[sectionString]) {
      if (valuation.id.toString() === id) {
        result = valuation;
        break;
      }
    }

    return [result, compilationData[sectionString]];
    /* 数据库获取编办
        let condition = {_id: versionId};
        let childCondition = {};
        childCondition[sectionString] = {$elemMatch: {_id: id}};

        let result = await this.db.findOne(condition, childCondition);

        return result !== null && result.bill_valuation.length > 0 ? result.bill_valuation[0] : {};
        */
  }

  /**
   * 删除计价规则
   *
   * @param {String} compilationId
   * @param {String} valuationId
   * @param {String} section
   * @return {Promise}
   */
  async deleteValuation(compilationId, valuationId, section) {
    let condition = { _id: compilationId };
    let sectionString = section + "_valuation";
    let deleteData = {};
    deleteData[sectionString] = { id: valuationId };

    // 利用pull删除嵌套数据
    let result = await this.db.deleteSet(condition, deleteData);

    return result !== null && result.ok === 1;
  }

  /**
   * 发布编办
   *
   * @param {String} id
   * @param {Number} status
   * @return {Promise}
   */
  async release(id, status) {
    // 如果是发布编办则需要判断配置的内容是否满足发布条件
    if (status) {
      let compilationData = await this.findDataByCondition({ _id: id });
      // 最少需要有一个计价规则存在
      if (
        compilationData.suggestion_valuation.length <= 0 &&
        compilationData.feasibility_valuation.length <= 0 &&
        compilationData.rough_valuation.length <= 0 &&
        compilationData.bill_valuation.length <= 0 &&
        compilationData.ration_valuation.length <= 0
      ) {
        throw "至少需要一个计价规则";
      }

      // 判断是否全部禁止
      let hasEnable = false;
      let valuationList = compilationData.bill_valuation.concat(
        compilationData.ration_valuation
      );
      for (let valuation of valuationList) {
        if (valuation.enable) {
          hasEnable = true;
          break;
        }
      }

      if (!hasEnable) {
        throw "至少需要一个可用的计价规则";
      }
    }

    let updateData = {
      is_release: status === 1,
      release_time: new Date().getTime(),
    };
    return this.updateById(id, updateData);
  }

  /**
   * 根据专业工程获取对应专业工程标准库id
   *
   * @param {String} valuationId
   * @param {String} section
   * @param {Number} engineering
   * @return {Promise}
   */
  async getEngineeringLib(valuationId, section, engineering) {
    let sectionString = section + "_valuation";
    let condition = {};
    condition[sectionString + ".id"] = valuationId;

    let compilationData = await this.findDataByCondition(condition);
    if (compilationData === null) {
      throw "不存在对应编办";
    }
    let valuationData = null;
    for (let valuation of compilationData[sectionString]) {
      if (valuation.id === valuationId) {
        valuationData = valuation;
      }
    }

    if (valuationData === null) {
      throw "不存在对应计价规则";
    }

    // 判断是否已有对应数据
    let engineeringList = valuationData.engineering_list;
    let engineeringLib = null;
    for (let tmpEngineering of engineeringList) {
      if (tmpEngineering.engineering === engineering) {
        engineeringLib = tmpEngineering;
        break;
      }
    }

    return engineeringLib;
  }

  /**
   * 新增专业工程标准库
   *
   * @param {String} valuationId
   * @param {String} section
   * @param {Object} data
   * @return {Promise}
   */
  async addEngineeringLib(valuationId, section, data) {
    let sectionString = section + "_valuation";
    let condition = {};
    condition[sectionString + ".id"] = valuationId;

    let insertData = {};
    insertData[sectionString + ".$.engineering_list"] = data;
    let result = await this.db.addToSet(condition, insertData);

    return result.ok === 1;
  }

  /*
   * 设置CLD办事处信息
   *
   * @param {String} compilationId
   * @param {int} category
   * @return {Promise}
   * */
  async updateCategory(compilationId, category) {
    return await this.updateById(compilationId, { categoryID: category });
  }

  /*
   * 设置工程默认所在地
   *
   * @param {String} compilationId
   * @param {int} location
   * @return {Promise}
   * */
  async updateLocation(compilationId, location) {
    return await this.updateById(compilationId, { defaultLocation: location });
  }

  /*
   * 设置是否提供免费版
   *
   * @param {String} compilationId
   * @param {int} location
   * @return {Promise}
   * */
  async updateFreeUse(compilationId, freeUse) {
    return await this.updateById(compilationId, { freeUse: freeUse });
  }

  /*
   * 设置版本号
   *
   * @param {String} compilationId
   * @param {String} edition
   * @return {Promise}
   * */
  async setEdition(compilationId, edition) {
    return await this.updateById(compilationId, { edition: edition });
  }

  /*
   * 设置序号
   *
   * @param {String} compilationId
   * @param {Number} serialNumber
   * @return {Promise}
   * */
  async setSerialNumber(compilationId, serialNumber) {
    return await this.updateById(compilationId, { serialNumber: serialNumber });
  }
}

export default CompilationModel;