Extend Book.getLanguageBook

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;
+    }
 };
 
 /**