123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174 |
- 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
- };
- }
|