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