service-live/app/service/patent/patentwarning.js

'use strict';
const { CrudService } = require('naf-framework-mongoose/lib/service');
const { BusinessError, ErrorCode } = require('naf-core').Error;
const { ObjectId } = require('mongoose').Types;
const _ = require('lodash');
const assert = require('assert');
const compressing = require('compressing');
const { sep } = require('path');
const path = require('path');
const fs = require('fs');
const encoding = require('encoding'); // 编码转换
const fxp = require('fast-xml-parser'); // xml转换
const mime = require('mime-types'); // 任意类型转换
// 专利运营专利申请预警表
class PatentwarningService extends CrudService {
  constructor(ctx) {
    super(ctx, 'patentwarning');
    this.model = this.ctx.model.Patent.Patentwarning;
    this.import_root_path = _.get(this.ctx.app.config.import, 'root_path');
    this.patentInfo = this.ctx.model.Patent.Patentinfo;
    this.patentApply = this.ctx.model.Patent.Patentapply;
    this.personalModel = this.ctx.model.Personal;
  }
  async import({ uri }) {
    uri = uri.replace('/files', this.import_root_path);
    const fileFullName = path.basename(uri);
    const arr = fileFullName.split('.');
    const ext = _.last(arr);
    const fileName = _.head(arr);
    const uncompressFilePath = `${this.import_root_path}${sep}`;
    if (ext !== 'zip') {
      throw new BusinessError(ErrorCode.FILE_FAULT, '只接收压缩格式为zip的压缩包');
    }
    // 解压
    try {
      await compressing.zip.uncompress(uri, `${uncompressFilePath}`);
    } catch (error) {
      throw new BusinessError(ErrorCode.FILE_FAULT, '解压失败');
    }
    // 找到
    let filePaths = [];
    try {
      filePaths = await this.findFileNameLoop(uncompressFilePath, fileName);
    } catch (error) {
      throw new BusinessError(ErrorCode.SERVICE_FAULT, '解析文件具体位置失败');
    }
    if (filePaths.length <= 0) {
      throw new BusinessError(ErrorCode.DATA_NOT_EXIST, '未找到文件');
    }
    const xmlPath = filePaths.find(f => f.includes('.xml'));
    if (!xmlPath) throw new BusinessError(ErrorCode.DATA_NOT_EXIST, '未找到xml文件');
    let create_number;
    try {
      create_number = await this.dealXml(xmlPath);
    } catch (error) {
      throw new BusinessError(ErrorCode.SERVICE_FAULT, '解析申请号失败');
    }
    if (!create_number) throw new BusinessError(ErrorCode.FILE_FAULT, '未找到申请号');
    let result;
    try {
      result = await this.getPatentFromInfo(create_number);
    } catch (error) {
      throw new BusinessError(ErrorCode.SERVICE_FAULT, '查询专利信息失败');
    }
    // 再找
    if (!result) result = await this.getPatentFromApply(create_number);
    // 找不到就算了,弹回去了
    if (!result) throw new BusinessError(ErrorCode.DATA_NOT_EXIST, '未通过申请号找到指定专利');
    let warningData;
    try {
      warningData = await this.getWarningData(result);
    } catch (error) {
      throw new BusinessError(ErrorCode.SERVICE_FAULT, '组织专利信息警告数据失败');
    }
    const tifPaths = filePaths.filter(f => f.includes('.tif'));
    if (tifPaths && tifPaths.length > 0) this.toUploadTif(warningData, tifPaths);
    await this.dirDelete(`${uncompressFilePath}${fileName}`);
    return 'ok';
  }

  /**
   * 上传图片,保存
   * @param {Array} data 警告数据,未保存
   * @param {Array} files tif图片文件路径数组
   */
  async toUploadTif(data, files) {
    const filePaths = [];
    const savePath = ObjectId();
    for (const file of files) {
      let arr = file.split(sep);
      const fullFileName = _.last(arr);
      arr = fullFileName.split('.');
      // 文件后缀
      const ext = _.last(arr);
      console.log('file');
      console.log(file);
      // 图片文件
      const buffer = await fs.readFileSync(file);
      // 新文件名
      const randomStr = ObjectId();
      // 新文件名带后缀
      const newFileName = `${randomStr}.${ext}`;
      // 上传后部分路径
      const shortPath = `platform${sep}patent${sep}${savePath}${sep}`;
      // 上传全路径
      const path = `${this.import_root_path}${sep}${shortPath}`;
      console.log('path');
      console.log(`${path}`);
      console.log('all');
      console.log(`${path}${newFileName}`);
      // 检查上传全路径是否存在,不存在就创建文件夹
      await this.checkPathAndCreate(path);
      // 另存文件
      console.log(`${path}${newFileName}`);
      await fs.writeFileSync(`${path}${newFileName}`, buffer);
      // 替换为files开头
      let dataPath = path.replace(this.import_root_path, '/files');
      while (dataPath.indexOf('\\') >= 0) {
        dataPath = dataPath.replace('\\', '/');
      }
      // 补全请求用路径的所有路径
      dataPath += newFileName;
      filePaths.push(dataPath);
    }
    for (const i of data) {
      i.file_url = filePaths;
    }
    const res = await this.model.insertMany(data);
    return res;

  }

  /**
   * 检查路径,没有创建
   * @param {Array} filePath 图片路径数组
   */
  async checkPathAndCreate(filePath) {
    const arr = filePath.split(sep);
    // w:第一个是盘符,直接和第二个合并
    // l:前几个应该都是存在的.也可以合并
    let head = _.head(arr);
    if (arr[1]) head = `${head}${sep}${arr[1]}`;
    arr.shift();
    arr[0] = head;
    let curPath;
    for (const p of arr) {
      curPath = `${curPath ? curPath + sep : ''}${p}`;
      if (!fs.existsSync(curPath)) {
        await fs.mkdirSync(curPath);
      }
    }
  }

  /**
   * 组织专利预警数据
   * @param {Object} patent 专利
   */
  async getWarningData(patent) {
    console.log(patent);
    const object = {};
    if (_.get(patent, 'create_number'))object.create_number = _.get(patent, 'create_number');
    if (_.get(patent, 'id'))object.patent_id = _.get(patent, 'id');
    if (_.get(patent, 'name'))object.patent_name = _.get(patent, 'name');
    // 找人的信息
    let userIds = _.get(patent, 'user_id', _.get(patent, 'inventer'));
    if (!userIds) throw new BusinessError(ErrorCode.DATA_NOT_EXIST, '未找到专利的发明人');
    if (!_.isArray(userIds)) userIds = [ userIds ];
    userIds = userIds.map(i => ObjectId(i));
    const users = await this.personalModel.find({ _id: userIds });
    const arr = [];
    for (const user of users) {
      if (!_.get(user, '_id') || !_.get(user, 'name')) continue;
      const { _id, name } = user;
      const obj = _.cloneDeep(object);
      obj.to_id = _id;
      obj.to_name = name;
      arr.push(obj);
    }
    return arr;

  }

  /**
   * 在patentApply表中找指定的专利
   * @param {String} create_number 专利的申请号
   */
  async getPatentFromApply(create_number) {
    const res = await this.patentApply.findOne({ create_number });
    return res;
  }

  /**
   * 在patentInfo表中找指定的专利
   * @param {String} create_number 专利的申请号
   */
  async getPatentFromInfo(create_number) {
    const res = await this.patentInfo.findOne({ create_number });
    return res;
  }

  /**
   * 根据路径找到xml,再找到该xml的申请号
   * @param {String} filePath xml的文件路径
   */
  async dealXml(filePath) {
    const buffer = await fs.readFileSync(filePath);
    let str = encoding.convert(buffer, 'UTF8', 'GBK');
    if (!str) throw new BusinessError(ErrorCode.FILE_FAULT, '读取xml文件失败');
    str = str.toString();
    const json = fxp.parse(str);
    const create_number = _.get(json, 'data-bus.TONGZHISXJ.SHUXINGXX.SHENQINGH');
    return create_number;
  }

  /**
   * 找到深层的tif和xml
   * 因为命名方式都是以文件名,之后tif的文件夹也在深层fileName.套娃且文件名都是fileName
   * @param {String} uri temp路径
   * @param {String} fileName 文件名
   */
  async findFileNameLoop(uri, fileName) {
    const filePath = `${uri}${sep}${fileName}`;
    if (!fs.existsSync(filePath)) {
      return [];
    }
    const files = fs.readdirSync(filePath);
    let result = [];
    for (const file of files) {
      console.log(_.trimEnd(filePath));
      console.log(file);
      const curPath = `${_.trimEnd(filePath)}${sep}${file}`;
      if (fs.statSync(curPath).isDirectory()) {
        const res = await this.findFileNameLoop(curPath, file);
        result.push(...res);
      } else {
        result.push(curPath);
      }
    }
    result = result.filter(f => f.includes('.tif') || f.includes('.xml'));
    return result;
  }

  /**
   * 删除路径下的所有内容
   * @param {String} filePath 路径
   */
  async dirDelete(filePath) {
    if (fs.existsSync(filePath)) {
      const files = fs.readdirSync(filePath);
      for (const file of files) {
        const curPath = `${filePath}${sep}${file}`;
        if (fs.statSync(curPath).isDirectory()) { // recurse
          this.dirDelete(curPath);
        } else { // delete file
          fs.unlinkSync(curPath);
        }
      }
      fs.rmdirSync(filePath);
    }
  }
}

module.exports = PatentwarningService;