diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/models/book.js | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/lib/models/book.js b/lib/models/book.js index 2a97f78..0a4a41a 100644 --- a/lib/models/book.js +++ b/lib/models/book.js @@ -176,14 +176,18 @@ Book.prototype.isLanguageBook = function() { }; /** - Return a languages book + Returns the book for a language. + Returns itself if matches the language itself. - @param {String} language - @return {Book} + @param {String} lang Language ID + @return {Book | Null} Null if no such language */ -Book.prototype.getLanguageBook = function(language) { - var books = this.getBooks(); - return books.get(language); +Book.prototype.getLanguageBook = function(lang) { + if (this.getLanguage() === lang) { + return this; + } else { + return this.getBooks().get(lang) || null; + } }; /** |