/**
 * @author linhuiw
 * @desc 文件相关操作
 */
import * as vscode from 'vscode';
import * as fs from 'fs-extra';
import * as _ from 'lodash';
import * as prettier from 'prettier';
import { getLangData } from './getLangData';
import { LANG_PREFIX } from './const';

export function updateLangFiles(
  lang: string,
  text: string,
  validateDuplicate: boolean
) {
  if (!lang.startsWith('I18N.')) {
    return;
  }

  const [, filename, ...restPath] = lang.split('.');
  const fullKey = restPath.join('.');
  const targetFilename = `${LANG_PREFIX}${filename}.ts`;

  if (!fs.existsSync(targetFilename)) {
    fs.writeFileSync(targetFilename, generateNewLangFile(fullKey, text));
    addImportToMainLangFile(filename);
    vscode.window.showInformationMessage(`成功新建语言文件 ${targetFilename}`);
  } else {
    // 清除 require 缓存，解决手动更新语言文件后再自动抽取，导致之前更新失效的问题
    const mainContent = getLangData(targetFilename);
    const obj = mainContent;

    if (Object.keys(obj).length === 0) {
      vscode.window.showWarningMessage(
        `${filename} 解析失败，该文件包含的文案无法自动补全`
      );
    }

    if (validateDuplicate && _.get(obj, fullKey) !== undefined) {
      vscode.window.showErrorMessage(
        `${targetFilename} 中已存在 key 为 \`${fullKey}\` 的翻译，请重新命名变量`
      );
      throw new Error('duplicate');
    }
    // \n 会被自动转义成 \\n，这里转回来
    text = text.replace(/\\n/gm, '\n');
    _.set(obj, fullKey, text);
    fs.writeFileSync(
      targetFilename,
      prettierFile(`export default ${JSON.stringify(obj, null, 2)}`)
    );
  }
}
/**
 * 使用 Prettier 格式化文件
 * @param fileContent
 */
function prettierFile(fileContent) {
  try {
    return prettier.format(fileContent, {
      parser: 'typescript',
      trailingComma: 'all',
      singleQuote: true
    });
  } catch (e) {
    console.error(`代码格式化报错！${e.toString()}\n代码为：${fileContent}`);
    return fileContent;
  }
}

export function generateNewLangFile(key: string, value: string) {
  const obj = _.set({}, key, value);

  return prettierFile(`export default ${JSON.stringify(obj, null, 2)}`);
}

export function addImportToMainLangFile(newFilename: string) {
  let mainContent = fs.readFileSync(`${LANG_PREFIX}index.ts`, 'utf8');
  mainContent = mainContent.replace(
    /^(\s*import.*?;)$/m,
    `$1\nimport ${newFilename} from './${newFilename}';`
  );
  if (/\,\n(}\);)/.test(mainContent)) {
    /** 最后一行包含,号 */
    mainContent = mainContent.replace(/(}\);)/, `  ${newFilename},\n$1`);
  } else {
    /** 最后一行不包含,号 */
    mainContent = mainContent.replace(/\n(}\);)/, `,\n  ${newFilename},\n$1`);
  }
  fs.writeFileSync(`${LANG_PREFIX}index.ts`, mainContent);
}