YangHuCost/web/building_saas/standard_interface/import/base.js

/*
 * @Descripttion: 导入通用代码
 * @Author: vian
 * @Date: 2020-09-09 10:45:54
 */
const INTERFACE_EXPORT_BASE = (() => {

  // xml字符实体
  const XMLEntity = {
    ' ': 'escape{space}',
    ' ': 'escape{simpleSpace}',
    '	': 'escape{tab}',
    '	': 'escape{simpleTab}',
    '
': 'escape{return}',
    '
': 'escape{simpleReturn}',
    '&#000A;': 'escape{newLine}',
    '
': 'escape{simpleNewLine}',
    '<': 'escape{less}',
    '>': 'escape{greater}',
    '&': 'escape{and}',
    '"': 'escape{quot}',
    ''': 'escape{apos}'
  };

  // 避免字符实体进行转义。原文本中含有xml字符实体，转换为其他字符。
  function escapeXMLEntity(str) {
    for (const [key, value] of Object.entries(XMLEntity)) {
      str = str.replace(new RegExp(key, 'g'), value);
    }
    return str;
  }
  // 将文本还原为字符实体
  function restoreXMLEntity(str) {
    for (const [key, value] of Object.entries(XMLEntity)) {
      str = str.replace(new RegExp(value, 'g'), key);
    }
    return str;
  }

  /*
   * 根据字段数组获得所要字段的值 eg: 要获取标段下的单项工程： ['标段', '单项工程'];
   * @param {Object}source 源数据
   *        {Array}fields 字段数组
   * @return {String}
   * @example getValue(source, ['标段', '_文件类型'])
   * */
  function getValue(source, fields) {
    let cur = source;
    for (const field of fields) {
      if (!cur[field]) {
        return '';
      }
      cur = cur[field];
    }
    return cur || '';
  }

  // 获取数据类型
  function _plainType(v) {
    return Object.prototype.toString.call(v).slice(8, -1);
  }

  /*
   * 获取某字段的值，强制返回数组，防止一些错误。如果期待返回数组，可以用此方法。
   * @param {Object}source 数据源
   *        {Array}fields 取的字段
   * @return {Array}
   * @example arrayValue(source, ['标段', '单项工程'])
   * */
  function arrayValue(source, fields) {
    let target = getValue(source, fields);
    if (_plainType(target) === 'Object') {
      target = [target];
    } else if (_plainType(target) !== 'Array') {
      target = []
    }
    return target;
  }

  // 获取费用
  function getFee(fees, fields) {
    if (!Array.isArray(fees) || !fees.length) {
      return '0';
    }
    const feeData = fees.find(fee => fee.fieldName === fields[0]);
    return feeData && feeData[fields[1]] || '0';
  }

  // 合并价格
  function mergeFees(feesA, feesB) {
    if (!feesA) {
      return feesB;
    }
    if (!feesB) {
      return [];
    }
    feesB.forEach(feeB => {
      const sameKindFee = feesA.find(feeA => feeA.fieldName === feeB.fieldName);
      if (sameKindFee) {
        Object.assign(sameKindFee, feeB);
      } else {
        feesA.push(feeB);
      }
    });
    return feesA;
  }

  // 将A对象的一部分属性赋值到B对象上
  function assignAttr(target, source, attrs, isExcepted = false) {
    if (!source || !target) {
      return;
    }
    const sourceAttrs = attrs
      ? isExcepted
        ? Object.keys(source).filter(attr => !attrs.includes(attr))
        : attrs
      : Object.keys(source);
    sourceAttrs.forEach(attr => {
      // 如果是价格，不能简单地覆盖，要合并两个对象的价格
      target[attr] = attr === 'fees' ? mergeFees(target[attr], source[attr]) : source[attr];
    });
  }

  // 获取固定ID
  function getFlag(data) {
    return data.flags && data.flags[0] && data.flags[0].flag || 0;
  }

  // 获取布尔型的数据
  function getBool(v) {
    return v === 'true' ? true : false;
  }

  // 设置成树结构数据
  function setTreeData(data, parent, next) {
    const defalutID = -1;
    data.ID = uuid.v1();
    data.ParentID = parent && parent.ID || defalutID;
    data.NextSiblingID = next && next.ID || defalutID;
  }

  // 递归设置树结构数据，并返回设置好的数据，递归items数组
  function mergeDataRecur(parent, items) {
    const rst = [];
    for (let i = 0; i < items.length; i++) {
      const cur = items[i];
      const next = items[i + 1];
      setTreeData(cur, parent, next);
      rst.push(cur);
      if (cur.items && cur.items.length) {
        rst.push(...mergeDataRecur(cur, cur.items));
      }
    }
    return rst;
  }

  /*
   * 递归获取相关数据，eg：获取组织措施清单下的所有子清单，该子清单们可能由分类及公式措施项各种组合组成。（参考调用处）
   * @param {Object}src(数据源) {Array}fields(二维数组，数组里的成员由需要取的字段组成)
   *         eg: ['组织措施分类'], ['公式计算措施项']，该层数据可能由组织措施分类 或 公式计算措施项组成 (同层不可同时存在)
   *        {Function} 获得源数据后，需要提取的数据方法
   * @return {Array}
   * */
  function getItemsRecur(src, fields, extractFuc) {
    let itemsSrc = [];
    let curField = [''];
    for (const field of fields) {
      itemsSrc = arrayValue(src, field);
      if (itemsSrc.length) {
        curField = field;
        break;
      }
    }
    return itemsSrc.map(itemSrc => {
      const obj = extractFuc(itemSrc, curField);
      obj.items = getItemsRecur(itemSrc, fields, extractFuc);
      return obj;
    });
  }

  // 递归获取相关数据，与上方的方法不同的点在：同层可能出现不同节点,上方方法暂时不取消(防止bug)
  // fields内字段的顺序即决定了提取数据类型的顺序，如fields = [['gruop'], ['item']]，则提取的数据同层中group数据在item数据之前
  function extractItemsRecur(src, fields, extractFuc) {
    const rst = [];
    for (const field of fields) {
      const itemsSrc = arrayValue(src, field);
      if (itemsSrc.length) {
        const items = itemsSrc.map(itemSrc => {
          const obj = extractFuc(itemSrc, field);
          obj.items = extractItemsRecur(itemSrc, fields, extractFuc);
          return obj;
        });
        rst.push(...items);
      }
    }
    return rst;
  }

  const UTIL = Object.freeze({
    escapeXMLEntity,
    restoreXMLEntity,
    getValue,
    arrayValue,
    getFee,
    mergeFees,
    assignAttr,
    setTreeData,
    mergeDataRecur,
    getFlag,
    getBool,
    getItemsRecur,
    extractItemsRecur,
  });


  /*
  * 读取文件转换为utf-8编码的字符串
  * @param {Blob}file
  * @return {Promise}
  * */
  function readAsTextSync(file) {
    return new Promise((resolve, reject) => {
      const fr = new FileReader();
      fr.readAsText(file);    // 默认utf-8，如果出现乱码，得看导入文件是什么编码
      fr.onload = function () {
        resolve(this.result);
      };
      fr.onerror = function () {
        reject('读取文件失败，请重试。');
      }
    });
  }

  /**
   * 
   * @param {Function} entryFunc - 各导入接口提取导入数据方法
   * @param {File} file - 导入的文件
   * @param {String} areaKey - 地区标识，如：'安徽@马鞍山'
   * @param {String} feeRateStandard - 导入接口地区对应的费率标准名称
   * @param {Boolean} escape - 是否需要避免xml中的实体字符转换
   * @return {Promise<Object>}
   */
  async function extractImportData(entryFunc, file, areaKey, feeRateStandard, escape = false) {
    const valuationID = compilationData.ration_valuation[0].id;
    const templateData = await ajaxPost('/pm/api/getImportTemplateData', { user_id: userID, valuationID, feeRateStandard });
    if (!templateData) {
      throw '无法获取有效模板数据。';
    }
    console.log(templateData);
    // 将二进制文件转换成字符串
    let xmlStr = await readAsTextSync(file);
    if (escape) {
      // x2js的str to json的实现方式基于DOMParser，DOMParser会自动将一些实体字符进行转换，比如 “&lt; to <”。如果不想进行自动转换，需要进行处理。
      xmlStr = escapeXMLEntity(xmlStr);
    }
    // 将xml格式良好的字符串转换成对象
    const x2js = new X2JS();
    let xmlObj = x2js.xml_str2json(xmlStr);
    xmlObj = JSON.parse(restoreXMLEntity(JSON.stringify(xmlObj)));
    console.log(xmlObj);
    if (!xmlObj) {
      throw '无有效数据。';
    }
    return await entryFunc(areaKey, xmlObj, templateData);
  }

  return {
    UTIL,
    extractImportData,
  }

})();