1 | var fs = require('fs')
|
2 | var path = require('path')
|
3 | var util = require('util')
|
4 |
|
5 | function Y18N (opts) {
|
6 |
|
7 | opts = opts || {}
|
8 | this.directory = opts.directory || './locales'
|
9 | this.updateFiles = typeof opts.updateFiles === 'boolean' ? opts.updateFiles : true
|
10 | this.locale = opts.locale || 'en'
|
11 | this.fallbackToLanguage = typeof opts.fallbackToLanguage === 'boolean' ? opts.fallbackToLanguage : true
|
12 |
|
13 |
|
14 | this.cache = {}
|
15 | this.writeQueue = []
|
16 | }
|
17 |
|
18 | Y18N.prototype.__ = function () {
|
19 | if (typeof arguments[0] !== 'string') {
|
20 | return this._taggedLiteral.apply(this, arguments)
|
21 | }
|
22 | var args = Array.prototype.slice.call(arguments)
|
23 | var str = args.shift()
|
24 | var cb = function () {}
|
25 |
|
26 | if (typeof args[args.length - 1] === 'function') cb = args.pop()
|
27 | cb = cb || function () {}
|
28 |
|
29 | if (!this.cache[this.locale]) this._readLocaleFile()
|
30 |
|
31 |
|
32 | if (!this.cache[this.locale][str] && this.updateFiles) {
|
33 | this.cache[this.locale][str] = str
|
34 |
|
35 |
|
36 |
|
37 |
|
38 | this._enqueueWrite([this.directory, this.locale, cb])
|
39 | } else {
|
40 | cb()
|
41 | }
|
42 |
|
43 | return util.format.apply(util, [this.cache[this.locale][str] || str].concat(args))
|
44 | }
|
45 |
|
46 | Y18N.prototype._taggedLiteral = function (parts) {
|
47 | var args = arguments
|
48 | var str = ''
|
49 | parts.forEach(function (part, i) {
|
50 | var arg = args[i + 1]
|
51 | str += part
|
52 | if (typeof arg !== 'undefined') {
|
53 | str += '%s'
|
54 | }
|
55 | })
|
56 | return this.__.apply(null, [str].concat([].slice.call(arguments, 1)))
|
57 | }
|
58 |
|
59 | Y18N.prototype._enqueueWrite = function (work) {
|
60 | this.writeQueue.push(work)
|
61 | if (this.writeQueue.length === 1) this._processWriteQueue()
|
62 | }
|
63 |
|
64 | Y18N.prototype._processWriteQueue = function () {
|
65 | var _this = this
|
66 | var work = this.writeQueue[0]
|
67 |
|
68 |
|
69 | var directory = work[0]
|
70 | var locale = work[1]
|
71 | var cb = work[2]
|
72 |
|
73 | var languageFile = this._resolveLocaleFile(directory, locale)
|
74 | var serializedLocale = JSON.stringify(this.cache[locale], null, 2)
|
75 |
|
76 | fs.writeFile(languageFile, serializedLocale, 'utf-8', function (err) {
|
77 | _this.writeQueue.shift()
|
78 | if (_this.writeQueue.length > 0) _this._processWriteQueue()
|
79 | cb(err)
|
80 | })
|
81 | }
|
82 |
|
83 | Y18N.prototype._readLocaleFile = function () {
|
84 | var localeLookup = {}
|
85 | var languageFile = this._resolveLocaleFile(this.directory, this.locale)
|
86 |
|
87 | try {
|
88 | localeLookup = JSON.parse(fs.readFileSync(languageFile, 'utf-8'))
|
89 | } catch (err) {
|
90 | if (err instanceof SyntaxError) {
|
91 | err.message = 'syntax error in ' + languageFile
|
92 | }
|
93 |
|
94 | if (err.code === 'ENOENT') localeLookup = {}
|
95 | else throw err
|
96 | }
|
97 |
|
98 | this.cache[this.locale] = localeLookup
|
99 | }
|
100 |
|
101 | Y18N.prototype._resolveLocaleFile = function (directory, locale) {
|
102 | var file = path.resolve(directory, './', locale + '.json')
|
103 | if (this.fallbackToLanguage && !this._fileExistsSync(file) && ~locale.lastIndexOf('_')) {
|
104 |
|
105 | var languageFile = path.resolve(directory, './', locale.split('_')[0] + '.json')
|
106 | if (this._fileExistsSync(languageFile)) file = languageFile
|
107 | }
|
108 | return file
|
109 | }
|
110 |
|
111 |
|
112 |
|
113 | Y18N.prototype._fileExistsSync = function (file) {
|
114 | try {
|
115 | return fs.statSync(file).isFile()
|
116 | } catch (err) {
|
117 | return false
|
118 | }
|
119 | }
|
120 |
|
121 | Y18N.prototype.__n = function () {
|
122 | var args = Array.prototype.slice.call(arguments)
|
123 | var singular = args.shift()
|
124 | var plural = args.shift()
|
125 | var quantity = args.shift()
|
126 |
|
127 | var cb = function () {}
|
128 | if (typeof args[args.length - 1] === 'function') cb = args.pop()
|
129 |
|
130 | if (!this.cache[this.locale]) this._readLocaleFile()
|
131 |
|
132 | var str = quantity === 1 ? singular : plural
|
133 | if (this.cache[this.locale][singular]) {
|
134 | str = this.cache[this.locale][singular][quantity === 1 ? 'one' : 'other']
|
135 | }
|
136 |
|
137 |
|
138 | if (!this.cache[this.locale][singular] && this.updateFiles) {
|
139 | this.cache[this.locale][singular] = {
|
140 | one: singular,
|
141 | other: plural
|
142 | }
|
143 |
|
144 |
|
145 |
|
146 |
|
147 | this._enqueueWrite([this.directory, this.locale, cb])
|
148 | } else {
|
149 | cb()
|
150 | }
|
151 |
|
152 |
|
153 |
|
154 | var values = [str]
|
155 | if (~str.indexOf('%d')) values.push(quantity)
|
156 |
|
157 | return util.format.apply(util, values.concat(args))
|
158 | }
|
159 |
|
160 | Y18N.prototype.setLocale = function (locale) {
|
161 | this.locale = locale
|
162 | }
|
163 |
|
164 | Y18N.prototype.getLocale = function () {
|
165 | return this.locale
|
166 | }
|
167 |
|
168 | Y18N.prototype.updateLocale = function (obj) {
|
169 | if (!this.cache[this.locale]) this._readLocaleFile()
|
170 |
|
171 | for (var key in obj) {
|
172 | this.cache[this.locale][key] = obj[key]
|
173 | }
|
174 | }
|
175 |
|
176 | module.exports = function (opts) {
|
177 | var y18n = new Y18N(opts)
|
178 |
|
179 |
|
180 |
|
181 | for (var key in y18n) {
|
182 | if (typeof y18n[key] === 'function') {
|
183 | y18n[key] = y18n[key].bind(y18n)
|
184 | }
|
185 | }
|
186 |
|
187 | return y18n
|
188 | }
|