You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
174 lines
6.1 KiB
174 lines
6.1 KiB
let shim; |
|
class Y18N { |
|
constructor(opts) { |
|
// configurable options. |
|
opts = opts || {}; |
|
this.directory = opts.directory || './locales'; |
|
this.updateFiles = typeof opts.updateFiles === 'boolean' ? opts.updateFiles : true; |
|
this.locale = opts.locale || 'en'; |
|
this.fallbackToLanguage = typeof opts.fallbackToLanguage === 'boolean' ? opts.fallbackToLanguage : true; |
|
// internal stuff. |
|
this.cache = Object.create(null); |
|
this.writeQueue = []; |
|
} |
|
__(...args) { |
|
if (typeof arguments[0] !== 'string') { |
|
return this._taggedLiteral(arguments[0], ...arguments); |
|
} |
|
const str = args.shift(); |
|
let cb = function () { }; // start with noop. |
|
if (typeof args[args.length - 1] === 'function') |
|
cb = args.pop(); |
|
cb = cb || function () { }; // noop. |
|
if (!this.cache[this.locale]) |
|
this._readLocaleFile(); |
|
// we've observed a new string, update the language file. |
|
if (!this.cache[this.locale][str] && this.updateFiles) { |
|
this.cache[this.locale][str] = str; |
|
// include the current directory and locale, |
|
// since these values could change before the |
|
// write is performed. |
|
this._enqueueWrite({ |
|
directory: this.directory, |
|
locale: this.locale, |
|
cb |
|
}); |
|
} |
|
else { |
|
cb(); |
|
} |
|
return shim.format.apply(shim.format, [this.cache[this.locale][str] || str].concat(args)); |
|
} |
|
__n() { |
|
const args = Array.prototype.slice.call(arguments); |
|
const singular = args.shift(); |
|
const plural = args.shift(); |
|
const quantity = args.shift(); |
|
let cb = function () { }; // start with noop. |
|
if (typeof args[args.length - 1] === 'function') |
|
cb = args.pop(); |
|
if (!this.cache[this.locale]) |
|
this._readLocaleFile(); |
|
let str = quantity === 1 ? singular : plural; |
|
if (this.cache[this.locale][singular]) { |
|
const entry = this.cache[this.locale][singular]; |
|
str = entry[quantity === 1 ? 'one' : 'other']; |
|
} |
|
// we've observed a new string, update the language file. |
|
if (!this.cache[this.locale][singular] && this.updateFiles) { |
|
this.cache[this.locale][singular] = { |
|
one: singular, |
|
other: plural |
|
}; |
|
// include the current directory and locale, |
|
// since these values could change before the |
|
// write is performed. |
|
this._enqueueWrite({ |
|
directory: this.directory, |
|
locale: this.locale, |
|
cb |
|
}); |
|
} |
|
else { |
|
cb(); |
|
} |
|
// if a %d placeholder is provided, add quantity |
|
// to the arguments expanded by util.format. |
|
const values = [str]; |
|
if (~str.indexOf('%d')) |
|
values.push(quantity); |
|
return shim.format.apply(shim.format, values.concat(args)); |
|
} |
|
setLocale(locale) { |
|
this.locale = locale; |
|
} |
|
getLocale() { |
|
return this.locale; |
|
} |
|
updateLocale(obj) { |
|
if (!this.cache[this.locale]) |
|
this._readLocaleFile(); |
|
for (const key in obj) { |
|
if (Object.prototype.hasOwnProperty.call(obj, key)) { |
|
this.cache[this.locale][key] = obj[key]; |
|
} |
|
} |
|
} |
|
_taggedLiteral(parts, ...args) { |
|
let str = ''; |
|
parts.forEach(function (part, i) { |
|
const arg = args[i + 1]; |
|
str += part; |
|
if (typeof arg !== 'undefined') { |
|
str += '%s'; |
|
} |
|
}); |
|
return this.__.apply(this, [str].concat([].slice.call(args, 1))); |
|
} |
|
_enqueueWrite(work) { |
|
this.writeQueue.push(work); |
|
if (this.writeQueue.length === 1) |
|
this._processWriteQueue(); |
|
} |
|
_processWriteQueue() { |
|
const _this = this; |
|
const work = this.writeQueue[0]; |
|
// destructure the enqueued work. |
|
const directory = work.directory; |
|
const locale = work.locale; |
|
const cb = work.cb; |
|
const languageFile = this._resolveLocaleFile(directory, locale); |
|
const serializedLocale = JSON.stringify(this.cache[locale], null, 2); |
|
shim.fs.writeFile(languageFile, serializedLocale, 'utf-8', function (err) { |
|
_this.writeQueue.shift(); |
|
if (_this.writeQueue.length > 0) |
|
_this._processWriteQueue(); |
|
cb(err); |
|
}); |
|
} |
|
_readLocaleFile() { |
|
let localeLookup = {}; |
|
const languageFile = this._resolveLocaleFile(this.directory, this.locale); |
|
try { |
|
// When using a bundler such as webpack, readFileSync may not be defined: |
|
if (shim.fs.readFileSync) { |
|
localeLookup = JSON.parse(shim.fs.readFileSync(languageFile, 'utf-8')); |
|
} |
|
} |
|
catch (err) { |
|
if (err instanceof SyntaxError) { |
|
err.message = 'syntax error in ' + languageFile; |
|
} |
|
if (err.code === 'ENOENT') |
|
localeLookup = {}; |
|
else |
|
throw err; |
|
} |
|
this.cache[this.locale] = localeLookup; |
|
} |
|
_resolveLocaleFile(directory, locale) { |
|
let file = shim.resolve(directory, './', locale + '.json'); |
|
if (this.fallbackToLanguage && !this._fileExistsSync(file) && ~locale.lastIndexOf('_')) { |
|
// attempt fallback to language only |
|
const languageFile = shim.resolve(directory, './', locale.split('_')[0] + '.json'); |
|
if (this._fileExistsSync(languageFile)) |
|
file = languageFile; |
|
} |
|
return file; |
|
} |
|
_fileExistsSync(file) { |
|
return shim.exists(file); |
|
} |
|
} |
|
export function y18n(opts, _shim) { |
|
shim = _shim; |
|
const y18n = new Y18N(opts); |
|
return { |
|
__: y18n.__.bind(y18n), |
|
__n: y18n.__n.bind(y18n), |
|
setLocale: y18n.setLocale.bind(y18n), |
|
getLocale: y18n.getLocale.bind(y18n), |
|
updateLocale: y18n.updateLocale.bind(y18n), |
|
locale: y18n.locale |
|
}; |
|
}
|
|
|