const path = require('path');
const fs = require('fs');
const nodeGypBuild = require('node-gyp-build');
const bindingPath = nodeGypBuild.path(path.join(__dirname, '..'));
const binding = require(bindingPath);
const getAssetsPath = function (bindingPath) {
const bindingDir = path.dirname(bindingPath);
const prebuildsDir = path.dirname(bindingDir);
if (path.basename(prebuildsDir) === 'prebuilds') {
const sharedAssetsPath = path.join(prebuildsDir, 'assets');
if (fs.existsSync(sharedAssetsPath)) {
return sharedAssetsPath;
}
}
return bindingDir;
};
function requireOptionalPeer(packageName) {
const searchPaths = [
process.cwd(),
path.join(__dirname, '..'),
__dirname,
];
for (const searchPath of searchPaths) {
try {
return require(require.resolve(packageName, { paths: [searchPath] }));
} catch (e) {
}
}
return null;
}
const jiebaInfo = requireOptionalPeer('opencc-jieba');
const assetsPath = getAssetsPath(bindingPath);
const isAbsolutePath = function (filePath) {
return path.isAbsolute(filePath) || /^[A-Za-z]:[\\/]/.test(filePath);
};
const getConfigPath = function (config) {
let configPath = config;
if (!isAbsolutePath(config)) {
configPath = path.join(assetsPath, config);
}
return configPath;
};
function patchDictPaths(dict, baseDir) {
if (!dict) return;
if (dict.type === 'group' && Array.isArray(dict.dicts)) {
for (const d of dict.dicts) {
patchDictPaths(d, baseDir);
}
} else if (dict.file && !path.isAbsolute(dict.file)) {
dict.file = path.join(baseDir, dict.file);
}
}
function patchConfigPaths(config, jieba, mainAssetsDir) {
if (config.segmentation) {
config.segmentation.__plugin_library = jieba.pluginLibrary;
if (config.segmentation.resources) {
const res = config.segmentation.resources;
for (const key of Object.keys(res)) {
if (!path.isAbsolute(res[key])) {
res[key] = path.join(jieba.dataDir, res[key]);
}
}
}
}
if (Array.isArray(config.conversion_chain)) {
for (const step of config.conversion_chain) {
patchDictPaths(step.dict, mainAssetsDir);
}
}
}
function resolveJiebaConfigPath(config, jieba) {
if (!jieba || isAbsolutePath(config)) {
return null;
}
if (typeof jieba.resolveConfigPath === 'function') {
return jieba.resolveConfigPath(config);
}
const candidates = [config];
if (!config.endsWith('.json')) {
candidates.push(config + '.json');
}
for (const candidate of candidates) {
if (candidate.includes('/') || candidate.includes('\\')) {
continue;
}
const configPath = path.join(jieba.dataDir, candidate);
if (fs.existsSync(configPath)) {
return configPath;
}
}
return null;
}
const OpenCC = module.exports = function (config) {
if (!config) {
config = 's2t.json';
}
const jiebaConfigPath = resolveJiebaConfigPath(config, jiebaInfo);
if (jiebaConfigPath) {
const raw = JSON.parse(fs.readFileSync(jiebaConfigPath, 'utf-8'));
patchConfigPaths(raw, jiebaInfo, assetsPath);
this.handler = new binding.Opencc(
JSON.stringify(raw),
jiebaInfo.dataDir + '/'
);
return;
}
config = getConfigPath(config);
this.handler = new binding.Opencc(config);
};
OpenCC.OpenCC = OpenCC;
OpenCC.version = binding.Opencc.version();
OpenCC.generateDict = function (inputFileName, outputFileName,
formatFrom, formatTo) {
return binding.Opencc.generateDict(inputFileName, outputFileName,
formatFrom, formatTo);
}
OpenCC.prototype.convert = function (input, callback) {
return this.handler.convert(input.toString(), callback);
};
OpenCC.prototype.convertSync = function (input) {
return this.handler.convertSync(input.toString());
};
OpenCC.prototype.convertPromise = function (input) {
const self = this;
return new Promise(function (resolve, reject) {
self.handler.convert(input.toString(), function (err, text) {
if (err) reject(err);
else resolve(text);
});
});
};