var Q = require("q"); var _ = require("lodash"); var path = require("path"); var fs = require("./fs"); var parse = require("../parse"); var template = require("./template"); var generators = { "site": require("./generator_site"), "json": require("./generator_json") }; var generate = function(options) { var generator = null; var files; options = _.defaults(options || {}, { // Folders to use input: null, output: null, // Output generator generator: "site", // Book title, keyword, description title: null, description: "Book generated using GitBook", // Origin github repository id github: null, githubHost: 'https://github.com/' }); if (!options.github || !options.title || !options.input || !options.output) { return Q.reject(new Error("Need options: github, title, input, output")); } if (!generators[options.generator]) { return Q.reject(new Error("Invalid generator (availables are: "+_.keys(generators).join(", "))); } // Clean output folder return fs.remove(options.output) .then(function() { return fs.mkdirp(options.output); }) // List all files in the repository .then(function() { return fs.list(options.input); }) // Check repository is valid .then(function(_files) { files = _files; if (!_.contains(files, "SUMMARY.md") || !_.contains(files, "README.md")) { return Q.reject(new Error("Invalid gitbook repository, need SUMMARY.md and README.md")); } }) // Get summary .then(function() { return fs.readFile(path.join(options.input, "SUMMARY.md"), "utf-8") .then(function(_summary) { options.summary = parse.summary(_summary); // Parse navigation options.navigation = parse.navigation(options.summary); }); }) // Create the generator .then(function() { generator = new generators[options.generator](options); }) // Copy file and replace markdown file .then(function() { return Q.all( _.chain(files) .map(function(file) { if (!file) return; if (file[file.length -1] == "/") { return Q(generator.transferFolder(file)); } else if (path.extname(file) == ".md" && options.navigation[file] != null) { return fs.readFile(path.join(options.input, file), "utf-8") .then(function(content) { return Q(generator.convertFile(content, file)); }); } else { return Q(generator.transferFile(file)); } }) .value() ); }) // Finish gneration .then(function() { return generator.finish(); }) .fail(function(err) { console.log(err); return Q.reject(err); }) }; module.exports = { generators: generators, folder: generate };