From 7892259fe64a39a3a2cc3caedeb7ca8c3396dd56 Mon Sep 17 00:00:00 2001
From: Soreine <nicolas@gitbook.com>
Date: Wed, 11 May 2016 17:54:26 +0200
Subject: [PATCH] Extend Book.getLanguageBook

---
 lib/models/book.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/lib/models/book.js b/lib/models/book.js
index 2a97f78e1..0a4a41a23 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;
+    }
 };
 
 /**