116 lines
2.8 KiB
TypeScript
116 lines
2.8 KiB
TypeScript
/**
|
||
* @author linhuiw
|
||
* @desc 翻译方法
|
||
* @TODO: index 文件需要添加 mock
|
||
*/
|
||
require('ts-node').register({
|
||
compilerOptions: {
|
||
module: 'commonjs'
|
||
}
|
||
});
|
||
import * as path from 'path';
|
||
import * as fs from 'fs';
|
||
import * as _ from 'lodash';
|
||
import { traverse, getProjectConfig, getLangDir, translateText } from './utils';
|
||
import { baiduTranslateTexts, googleTranslateTexts } from './translate';
|
||
|
||
const CONFIG = getProjectConfig();
|
||
|
||
/**
|
||
* 获取中文文案
|
||
*/
|
||
function getSourceText() {
|
||
const srcLangDir = getLangDir(CONFIG.srcLang);
|
||
const srcFile = path.resolve(srcLangDir, 'index.ts');
|
||
const { default: texts } = require(srcFile);
|
||
|
||
return texts;
|
||
}
|
||
/**
|
||
* 获取对应语言文案
|
||
* @param dstLang
|
||
*/
|
||
function getDistText(dstLang) {
|
||
const distLangDir = getLangDir(dstLang);
|
||
const distFile = path.resolve(distLangDir, 'index.ts');
|
||
let distTexts = {};
|
||
if (fs.existsSync(distFile)) {
|
||
distTexts = require(distFile).default;
|
||
}
|
||
|
||
return distTexts;
|
||
}
|
||
|
||
/**
|
||
* 获取所有未翻译的文案
|
||
* @param 目标语种
|
||
*/
|
||
function getAllUntranslatedTexts(toLang) {
|
||
const texts = getSourceText();
|
||
const distTexts = getDistText(toLang);
|
||
const untranslatedTexts = {};
|
||
/** 遍历文案 */
|
||
traverse(texts, (text, path) => {
|
||
const distText = _.get(distTexts, path);
|
||
if (text === distText || !distText) {
|
||
untranslatedTexts[path] = text;
|
||
}
|
||
});
|
||
return untranslatedTexts;
|
||
}
|
||
|
||
/**
|
||
* Mock 对应语言
|
||
* @param dstLang
|
||
*/
|
||
async function mockCurrentLang(dstLang: string, origin: string) {
|
||
const untranslatedTexts = getAllUntranslatedTexts(dstLang);
|
||
let mocks = {};
|
||
if (origin === 'Google') {
|
||
mocks = await googleTranslateTexts(untranslatedTexts, dstLang);
|
||
} else {
|
||
mocks = await baiduTranslateTexts(untranslatedTexts, dstLang);
|
||
}
|
||
|
||
/** 所有任务执行完毕后，写入mock文件 */
|
||
return writeMockFile(dstLang, mocks);
|
||
}
|
||
/**
|
||
* 写入 Mock 文件
|
||
* @param dstLang
|
||
* @param mocks
|
||
*/
|
||
function writeMockFile(dstLang, mocks) {
|
||
const fileContent = 'export default ' + JSON.stringify(mocks, null, 2);
|
||
const filePath = path.resolve(getLangDir(dstLang), 'mock.ts');
|
||
return new Promise((resolve, reject) => {
|
||
fs.writeFile(filePath, fileContent, err => {
|
||
if (err) {
|
||
reject(err);
|
||
} else {
|
||
resolve();
|
||
}
|
||
});
|
||
});
|
||
}
|
||
/**
|
||
* Mock 语言的未翻译的文案
|
||
* @param lang
|
||
*/
|
||
async function mockLangs(origin: string) {
|
||
const langs = CONFIG.distLangs;
|
||
if (origin === 'Google') {
|
||
const mockPromise = langs.map(lang => {
|
||
return mockCurrentLang(lang, origin);
|
||
});
|
||
return Promise.all(mockPromise);
|
||
} else {
|
||
for (var i = 0; i < langs.length; i++) {
|
||
await mockCurrentLang(langs[i], origin);
|
||
}
|
||
return Promise.resolve();
|
||
}
|
||
}
|
||
|
||
export { mockLangs, getAllUntranslatedTexts };
|