Change api for registering components

3 files changed, 7 insertions, 51 deletions

diff --git a/packages/gitbook-core/src/reducers/file.js b/packages/gitbook-core/src/reducers/file.js
deleted file mode 100644
index 028afc6..0000000
--- a/packages/gitbook-core/src/reducers/file.js
+++ /dev/null
@@ -1,16 +0,0 @@
-const { Record } = require('immutable');
-
-const DEFAULTS = {
-    type:  '',
-    mtime: '',
-    path:  ''
-};
-
-class FileState extends Record(DEFAULTS) {
-    static create(state) {
-        return state instanceof FileState ?
-            state : new FileState(state);
-    }
-}
-
-module.exports = FileState;
diff --git a/packages/gitbook-core/src/reducers/readme.js b/packages/gitbook-core/src/reducers/readme.js
index 8e99f3d..d88f9ec 100644
--- a/packages/gitbook-core/src/reducers/readme.js
+++ b/packages/gitbook-core/src/reducers/readme.js
@@ -1,12 +1,12 @@
 const { Record } = require('immutable');
-const FileState = require('./file');
+const File = require('../models/file');
 
 class ReadmeState extends Record({
-    file: new FileState()
+    file: new File()
 }) {
     constructor(state = {}) {
         super({
-            file: new FileState(state.file)
+            file: new File(state.file)
         });
     }
 
diff --git a/packages/gitbook-core/src/reducers/summary.js b/packages/gitbook-core/src/reducers/summary.js
index 0179ecc..60568ef 100644
--- a/packages/gitbook-core/src/reducers/summary.js
+++ b/packages/gitbook-core/src/reducers/summary.js
@@ -1,45 +1,17 @@
 const { Record, List } = require('immutable');
-const FileState = require('./file');
 
-class SummaryArticle extends Record({
-    title:    '',
-    depth:    0,
-    path:     '',
-    ref:      '',
-    level:    '',
-    articles: List()
-}) {
-    constructor(state) {
-        super({
-            ...state,
-            articles: (new List(state.articles))
-                .map(article => new SummaryArticle(article))
-        });
-    }
-}
-
-class SummaryPart extends Record({
-    title:    '',
-    articles: List()
-}) {
-    constructor(state) {
-        super({
-            ...state,
-            articles: (new List(state.articles))
-                .map(article => new SummaryArticle(article))
-        });
-    }
-}
+const File = require('../models/File');
+const SummaryPart = require('../models/SummaryPart');
 
 
 class SummaryState extends Record({
-    file: new FileState(),
+    file: new File(),
     parts: List()
 }) {
     constructor(state = {}) {
         super({
             ...state,
-            file:  new FileState(state.file),
+            file:  new File(state.file),
             parts: (new List(state.parts))
                 .map(article => new SummaryPart(article))
         });