i18n-automation/index.js

i18n-automation

#!/usr/bin/env node
const path = require('path');
const fs = require('fs');
const { TASK, helpMessage, configFileName } = require('./const');
const { argumentErrorHandler, initConfigFileErrorHandler, configFileNotFoundErrorHandler, configParseErrorHandler } = require('./errorHandler');
const { Extract } = require('./TaskController/extract');
const { Generate } = require('./TaskController/generate');

// 获取用户输入的路径，返回进程的当前工作目录
const workDir = process.cwd();
const argv = process.argv.slice(2);

/**
 * 获取用户输入的任务和参数
 * process.argv返回一个数组，第一个元素是 process.execPath。第二个元素是正被执行的 JavaScript 文件的路径。 其余的元素是任何额外的命令行参数
 * [process.execPath, 文件路径, 参数1 参数2 参数3....]
 */
if (argv.length === 0) argv.push('help');

const initConfig = (workDir) => {
  try {
    const configTp = fs.readFileSync(path.join(__dirname, `./templates/${configFileName}`));
    const targetConfigPath = path.join(workDir, configFileName);
    fs.writeFileSync(targetConfigPath, configTp);
  } catch {
    initConfigFileErrorHandler();
  }
}

// 解析第一个参数
const argv1Handler = argv1 => {
  switch (argv1) {
    case 'init':
      initConfig(workDir);
      process.exit(0);
    case 'help':
      console.log(helpMessage);
      process.exit(0);
    case 'task':
      break;
    default:
      argumentErrorHandler();
  }
}

// 解析第二个参数
const argv2Handler = (argv2) => {
  let taskQueue = [];
  switch(argv2) {
    case '-e':
      taskQueue = [TASK.EXTRACT];
      break;
    case '-g':
      taskQueue = [TASK.GENERATE];
      break;
    case '-eg':
      taskQueue = [TASK.EXTRACT, TASK.GENERATE];
      break;
    default:
      argumentErrorHandler();
  }
  return taskQueue;
}

// 读取配置文件
const readJsonConfig = () => {
  const configFilePath = path.join(workDir, configFileName);
  if (!fs.existsSync(configFilePath)) configFileNotFoundErrorHandler();
  try {
    const jsonObj = fs.readFileSync(configFilePath);
    return JSON.parse(jsonObj);
  } catch(e) {
    configParseErrorHandler(e);
  }
};

// 任务分配器：分配到对应的taskController
const taskStart = (taskType, config) => {
  const { extract, generate } = config;
  
  switch (taskType) {
    case TASK.EXTRACT:
      return Extract(extract); // 提取
    case TASK.GENERATE:
      return Generate(generate); // 生成
  }
}

const tasksLauncher = () => {
  const taskQueue = argv2Handler(argv[1]);
  if (!taskQueue.length) return;
  const config = readJsonConfig(workDir);
  taskQueue.forEach(i => taskStart(i, config));
}

argv1Handler(argv[0]);

// 任务处理
tasksLauncher();