webpack-i18n-plugin-plus/src/collector/utils.js

webpack国际化插件-升级版，支持script中的中文识别，支持自动翻译

const fs = require('fs');
const path = require('path');
const XLSX = require('xlsx');
const ora = require('ora');
const myOra = ora();
const fileObj = {}
/**
 *
 * @param filePath
 * @param list
 * @returns {*|Array}
 */
module.exports.getFileList = function(filePath, list) {
  list = list || [];
  let files = fs.readdirSync(filePath) || [];

  files.forEach(function(filename) {
    let fileDir = path.join(filePath, filename);
    let stats = fs.statSync(fileDir);

    if (stats.isFile()) {
      list.push(fileDir);
    } else if (stats.isDirectory()) {
      getFileList(fileDir, list);
    }
  });
  return list;
};

/**
 *
 * @param filePath
 */
const readCodeText = function(filePath) {
  let filePathStr = path.resolve(filePath);
  let text = fs.readFileSync(filePathStr, 'utf-8');

  return text;
};
module.exports.readCodeText = readCodeText;
/**
 *
 * @param filePath
 * @param code
 */
module.exports.writeFile = (filePath, code) => {
  if (fileObj[filePath] === JSON.stringify(code)) {
    return false
  }
  fileObj[filePath] = JSON.stringify(code)
  filePath = path.resolve(filePath);
  let dirname = path.dirname(filePath);
  let filePathArr = dirname.split(path.sep);

  /**
   *
   * @param index
   */
  function mkdir(index) {
    let pathArr = filePathArr.slice();
    pathArr.splice(index, filePathArr.length - 1);
    let dirPath = path.normalize(pathArr.join(path.sep));
    if (!fs.existsSync(dirPath) && !/\.[\w\d]+$/.test(dirPath)) {
      fs.mkdirSync(dirPath);
    }
    if (filePathArr.length > 0 && index < filePathArr.length) {
      mkdir(++index);
    }
  }

  mkdir(1);

  fs.writeFileSync(filePath, code, 'utf-8');
};
/**
 *
 * @param filePath
 */
module.exports.deleteFile = (filePath) => {
  if (Array.isArray(filePath)) {
    filePath.forEach((item) => {
      if (fs.existsSync(item)) {
        fs.unlinkSync(item);
      }
    });
  } else {
    if (fs.existsSync(filePath)) {
      fs.unlinkSync(filePath);
    }
  }
};

/**
 *
 * @param version
 * @returns {string|*|XML|void}
 */
module.exports.genPolyfill = function(version) {
  return readCodeText(path.resolve(__dirname, './tplCode/polyfill.js')).replace('${version}', version);
};

/**
 *
 * @param version
 * @returns {string|*|XML|void}
 */
module.exports.genPolyfillTs = function() {
  return readCodeText(path.resolve(__dirname, './tplCode/polyfill.d.ts'));
};

/**
 *
 * @param data
 * @returns {Number|*}
 */
module.exports.genXLSXData = function(data) {
  let ws = XLSX.utils.json_to_sheet(data);
  let wb = XLSX.utils.book_new();
  XLSX.utils.book_append_sheet(wb, ws, 'Sheet');
  return XLSX.write(wb, { type: 'buffer', bookType: 'xlsx' });
};
/**
 *
 * @param data
 * @returns {Number|*}
 */
function deleteFolder(filePath) {
  if (fs.existsSync(filePath)) {
    const files = fs.readdirSync(filePath);
    files.forEach((file) => {
      const nextFilePath = `${filePath}/${file}`;
      const states = fs.statSync(nextFilePath);
      if (states.isDirectory()) {
        //recurse
        deleteFolder(nextFilePath);
      } else {
        //delete file
        fs.unlinkSync(nextFilePath);
      }
    });
    fs.rmdirSync(filePath);
  }
}
module.exports.deleteFolder = deleteFolder;
module.exports.deleteCacheFolder = function() {
  try {
    let cacheFolder = path.resolve(process.cwd(), './node_modules/.cache');
    deleteFolder(cacheFolder);
  } catch (e) {
    console.error(e);
  }
};
/**
 *
 * @param data
 * @returns {Number|*}
 */
let defaultDir = function() {
  return path.resolve(process.cwd(), './i18n');
};
module.exports.defaultDir = defaultDir;
/**
 *
 * @param data
 * @returns {Number|*}
 */
let UN_DO_COUNT = {};
module.exports.setUndoCount = function(key, count) {
  UN_DO_COUNT[key] = count;
};
/**
 *
 * @param data
 * @returns {Number|*}
 */
module.exports.printUndo = function(options) {
  let hasUndo = false;
  Object.keys(UN_DO_COUNT).forEach((key) => {
    if (UN_DO_COUNT[key]) {
      myOra.warn(`${key}语言包剩余 ` + UN_DO_COUNT[key] + ' 条待翻译数据');
      let xlsxPath = path.resolve(options.i18nDir || defaultDir(), './' + key);
      myOra.info(' > 目录：' + xlsxPath + '\n');
      hasUndo = true;
    }
  });
  if (!hasUndo) {
    myOra.succeed('语言包生成完毕！\n');
  }
};
module.exports.translateFormatter = function (i18nMap, formatter) {
  if (typeof formatter === 'function') {
    Object.keys(i18nMap).forEach((key) => {
      i18nMap[key] = formatter(i18nMap[key]) || i18nMap[key]
    })
  }
  return i18nMap
}