Use orderedMap for language books

2 files changed, 15 insertions, 4 deletions

diff --git a/lib/models/book.js b/lib/models/book.js
index 6daa684..1835032 100644
--- a/lib/models/book.js
+++ b/lib/models/book.js
@@ -162,6 +162,20 @@ Book.prototype.isMultilingual = function() {
 };
 
 /**
+    Add a new language book
+
+    @param {String} language
+    @param {Book} book
+    @return {Book}
+*/
+Book.prototype.addLanguageBook = function(language, book) {
+    var books = this.getBooks();
+    books = books.set(language, book);
+
+    return this.set('books', books);
+};
+
+/**
     Change log level
 
     @param {Book} book
diff --git a/lib/parse/parseBook.js b/lib/parse/parseBook.js
index 3a7d845..c30027c 100644
--- a/lib/parse/parseBook.js
+++ b/lib/parse/parseBook.js
@@ -39,10 +39,7 @@ function parseMultilingualBook(book) {
         .then(parseConfig)
         .then(parseBookContent)
         .then(function(result) {
-            var books = currentBook.getBooks();
-            books = books.push(result);
-
-            return currentBook.set('books', books);
+            return currentBook.addLanguageBook(lang, result);
         });
     }, book);
 }