1 files changed, 21 insertions, 9 deletions
diff --git a/lib/models/output.js b/lib/models/output.js
index 3e8dab8..10df67b 100644
--- a/lib/models/output.js
+++ b/lib/models/output.js
@@ -4,10 +4,21 @@ var Book = require('./book');
 
 var Output = Immutable.Record({
     book:       Book(),
+
+    // Map of plugins to use (String -> Plugin)
     plugins:    Immutable.OrderedMap(),
+
+    // Map pages to generation (String -> Page)
     pages:      Immutable.OrderedMap(),
+
+    // List assets (String)
     assets:     Immutable.List(),
-    options:    Immutable.Map()
+
+    // Option for the generation
+    options:    Immutable.Map(),
+
+    // Internal state for the generation
+    state:      Immutable.Map()
 });
 
 Output.prototype.getBook = function() {
@@ -30,6 +41,10 @@ Output.prototype.getAssets = function() {
     return this.get('assets');
 };
 
+Output.prototype.getState = function() {
+    return this.get('state');
+};
+
 /**
     Return logegr for this output (same as book)
 
@@ -40,17 +55,14 @@ Output.prototype.getLogger = function() {
 };
 
 /**
-    Create an Output instance from a book and a set of options
+    Update state of output
 
-    @param {Book} book
-    @param {Object} options
+    @param {Output} output
+    @param {Map} newState
     @return {Output}
 */
-Output.createForBook = function(book, options) {
-    return new Output({
-        book: book,
-        options: options
-    });
+Output.updateState = function(output, newState) {
+    return output.set('state', newState);
 };
 
 module.exports = Output;