Copy readme.json from main language for json output of multilingual

1 files changed, 19 insertions, 2 deletions

diff --git a/lib/output/json/onFinish.js b/lib/output/json/onFinish.js
index 107285a..ff336a2 100644
--- a/lib/output/json/onFinish.js
+++ b/lib/output/json/onFinish.js
@@ -1,4 +1,7 @@
+var path = require('path');
+
 var Promise = require('../../utils/promise');
+var fs = require('../../utils/fs');
 
 /**
     Finish the generation
@@ -7,9 +10,23 @@ var Promise = require('../../utils/promise');
     @return {Output}
 */
 function onFinish(output) {
-    // todo: copy README.json from main language
+    var book = output.getBook();
+    var outputRoot = output.getRoot();
+
+    if (!book.isMultilingual()) {
+        return Promise(output);
+    }
+
+    // Get main language
+    var languages = book.getLanguages();
+    var mainLanguage = languages.getDefaultLanguage();
 
-    return Promise(output);
+    // Copy README.json from it
+    return fs.copy(
+        path.resolve(outputRoot, mainLanguage.getID(), 'README.json'),
+        path.resolve(outputRoot, 'README.json')
+    )
+    .thenResolve(output);
 }
 
 module.exports = onFinish;