189 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			189 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | var fs = require('fs') | ||
|  | var path = require('path') | ||
|  | var util = require('util') | ||
|  | 
 | ||
|  | function Y18N (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 = {} | ||
|  |   this.writeQueue = [] | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype.__ = function () { | ||
|  |   if (typeof arguments[0] !== 'string') { | ||
|  |     return this._taggedLiteral.apply(this, arguments) | ||
|  |   } | ||
|  |   var args = Array.prototype.slice.call(arguments) | ||
|  |   var str = args.shift() | ||
|  |   var 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([this.directory, this.locale, cb]) | ||
|  |   } else { | ||
|  |     cb() | ||
|  |   } | ||
|  | 
 | ||
|  |   return util.format.apply(util, [this.cache[this.locale][str] || str].concat(args)) | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype._taggedLiteral = function (parts) { | ||
|  |   var args = arguments | ||
|  |   var str = '' | ||
|  |   parts.forEach(function (part, i) { | ||
|  |     var arg = args[i + 1] | ||
|  |     str += part | ||
|  |     if (typeof arg !== 'undefined') { | ||
|  |       str += '%s' | ||
|  |     } | ||
|  |   }) | ||
|  |   return this.__.apply(null, [str].concat([].slice.call(arguments, 1))) | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype._enqueueWrite = function (work) { | ||
|  |   this.writeQueue.push(work) | ||
|  |   if (this.writeQueue.length === 1) this._processWriteQueue() | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype._processWriteQueue = function () { | ||
|  |   var _this = this | ||
|  |   var work = this.writeQueue[0] | ||
|  | 
 | ||
|  |   // destructure the enqueued work.
 | ||
|  |   var directory = work[0] | ||
|  |   var locale = work[1] | ||
|  |   var cb = work[2] | ||
|  | 
 | ||
|  |   var languageFile = this._resolveLocaleFile(directory, locale) | ||
|  |   var serializedLocale = JSON.stringify(this.cache[locale], null, 2) | ||
|  | 
 | ||
|  |   fs.writeFile(languageFile, serializedLocale, 'utf-8', function (err) { | ||
|  |     _this.writeQueue.shift() | ||
|  |     if (_this.writeQueue.length > 0) _this._processWriteQueue() | ||
|  |     cb(err) | ||
|  |   }) | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype._readLocaleFile = function () { | ||
|  |   var localeLookup = {} | ||
|  |   var languageFile = this._resolveLocaleFile(this.directory, this.locale) | ||
|  | 
 | ||
|  |   try { | ||
|  |     localeLookup = JSON.parse(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 | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype._resolveLocaleFile = function (directory, locale) { | ||
|  |   var file = path.resolve(directory, './', locale + '.json') | ||
|  |   if (this.fallbackToLanguage && !this._fileExistsSync(file) && ~locale.lastIndexOf('_')) { | ||
|  |     // attempt fallback to language only
 | ||
|  |     var languageFile = path.resolve(directory, './', locale.split('_')[0] + '.json') | ||
|  |     if (this._fileExistsSync(languageFile)) file = languageFile | ||
|  |   } | ||
|  |   return file | ||
|  | } | ||
|  | 
 | ||
|  | // this only exists because fs.existsSync() "will be deprecated"
 | ||
|  | // see https://nodejs.org/api/fs.html#fs_fs_existssync_path
 | ||
|  | Y18N.prototype._fileExistsSync = function (file) { | ||
|  |   try { | ||
|  |     return fs.statSync(file).isFile() | ||
|  |   } catch (err) { | ||
|  |     return false | ||
|  |   } | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype.__n = function () { | ||
|  |   var args = Array.prototype.slice.call(arguments) | ||
|  |   var singular = args.shift() | ||
|  |   var plural = args.shift() | ||
|  |   var quantity = args.shift() | ||
|  | 
 | ||
|  |   var cb = function () {} // start with noop.
 | ||
|  |   if (typeof args[args.length - 1] === 'function') cb = args.pop() | ||
|  | 
 | ||
|  |   if (!this.cache[this.locale]) this._readLocaleFile() | ||
|  | 
 | ||
|  |   var str = quantity === 1 ? singular : plural | ||
|  |   if (this.cache[this.locale][singular]) { | ||
|  |     str = this.cache[this.locale][singular][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([this.directory, this.locale, cb]) | ||
|  |   } else { | ||
|  |     cb() | ||
|  |   } | ||
|  | 
 | ||
|  |   // if a %d placeholder is provided, add quantity
 | ||
|  |   // to the arguments expanded by util.format.
 | ||
|  |   var values = [str] | ||
|  |   if (~str.indexOf('%d')) values.push(quantity) | ||
|  | 
 | ||
|  |   return util.format.apply(util, values.concat(args)) | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype.setLocale = function (locale) { | ||
|  |   this.locale = locale | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype.getLocale = function () { | ||
|  |   return this.locale | ||
|  | } | ||
|  | 
 | ||
|  | Y18N.prototype.updateLocale = function (obj) { | ||
|  |   if (!this.cache[this.locale]) this._readLocaleFile() | ||
|  | 
 | ||
|  |   for (var key in obj) { | ||
|  |     this.cache[this.locale][key] = obj[key] | ||
|  |   } | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = function (opts) { | ||
|  |   var y18n = new Y18N(opts) | ||
|  | 
 | ||
|  |   // bind all functions to y18n, so that
 | ||
|  |   // they can be used in isolation.
 | ||
|  |   for (var key in y18n) { | ||
|  |     if (typeof y18n[key] === 'function') { | ||
|  |       y18n[key] = y18n[key].bind(y18n) | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   return y18n | ||
|  | } |