117 lines
3.0 KiB
JavaScript
117 lines
3.0 KiB
JavaScript
/* eslint-disable no-console */
|
|
|
|
var Q = require('q');
|
|
var _ = require('lodash');
|
|
var tmp = require('tmp');
|
|
var path = require('path');
|
|
|
|
var Book = require('../').Book;
|
|
var NodeFS = require('../lib/fs/node');
|
|
var fs = require('../lib/utils/fs');
|
|
|
|
require('./assertions');
|
|
|
|
// Create filesystem instance for testing
|
|
var nodeFS = new NodeFS();
|
|
|
|
function setupFS(files) {
|
|
return Q.nfcall(tmp.dir.bind(tmp)).get(0)
|
|
.then(function(rootFolder) {
|
|
return _.chain(_.pairs(files))
|
|
.sortBy(0)
|
|
.reduce(function(prev, pair) {
|
|
return prev.then(function() {
|
|
var filename = path.resolve(rootFolder, pair[0]);
|
|
var buf = pair[1];
|
|
|
|
if (_.isObject(buf)) buf = JSON.stringify(buf);
|
|
if (_.isString(buf)) buf = new Buffer(buf, 'utf-8');
|
|
|
|
return fs.mkdirp(path.dirname(filename))
|
|
.then(function() {
|
|
return fs.writeFile(filename, buf);
|
|
});
|
|
});
|
|
}, Q())
|
|
.value()
|
|
.then(function() {
|
|
return rootFolder;
|
|
});
|
|
});
|
|
}
|
|
|
|
// Setup a mock book for testing using a map of files
|
|
function setupBook(files, opts) {
|
|
opts = opts || {};
|
|
opts.log = function() { };
|
|
|
|
return setupFS(files)
|
|
.then(function(folder) {
|
|
opts.fs = nodeFS;
|
|
opts.root = folder;
|
|
|
|
return new Book(opts);
|
|
});
|
|
}
|
|
|
|
// Setup a book with default README/SUMMARY
|
|
function setupDefaultBook(files, summary, opts) {
|
|
var summaryContent = '# Summary \n\n' +
|
|
_.map(summary, function(article) {
|
|
return '* [' + article.title +'](' + article.path + ')';
|
|
})
|
|
.join('\n');
|
|
|
|
return setupBook(_.defaults(files || {}, {
|
|
'README.md': 'Hello',
|
|
'SUMMARY.md': summaryContent
|
|
}), opts);
|
|
}
|
|
|
|
// Output a book with a specific generator
|
|
function outputDefaultBook(Output, files, summary, opts) {
|
|
return setupDefaultBook(files, summary, opts)
|
|
.then(function(book) {
|
|
// Parse the book
|
|
return book.parse()
|
|
|
|
// Start generation
|
|
.then(function() {
|
|
var output = new Output(book);
|
|
return output.generate()
|
|
.thenResolve(output);
|
|
});
|
|
});
|
|
}
|
|
|
|
// Output a book with a specific generator
|
|
function outputBook(Output, files, opts) {
|
|
return setupBook(files, opts)
|
|
.then(function(book) {
|
|
// Parse the book
|
|
return book.parse()
|
|
|
|
// Start generation
|
|
.then(function() {
|
|
var output = new Output(book);
|
|
return output.generate()
|
|
.thenResolve(output);
|
|
});
|
|
});
|
|
}
|
|
|
|
// Log an error
|
|
function logError(err) {
|
|
console.log(err.stack || err);
|
|
}
|
|
|
|
module.exports = {
|
|
fs: nodeFS,
|
|
setupFS: setupFS,
|
|
setupBook: setupBook,
|
|
outputBook: outputBook,
|
|
setupDefaultBook: setupDefaultBook,
|
|
outputDefaultBook: outputDefaultBook,
|
|
logError: logError
|
|
};
|