fix: Fix register command

alibaba · Aug 29, 2018 · 3022645 · 3022645
1 parent 76f1116
commit 3022645
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/components/Register/index.js b/src/components/Register/index.js
@@ -23,11 +23,20 @@ class Register extends BaseComponent {
     const { type } = this;
 
     editor.on(EVENT_BEFORE_ADD_PAGE, ({ className }) => {
+      let host = Editor[className];
       let keys = ['name', 'config', 'extend'];
-      if (type === 'behaviour') keys = ['name', 'behaviour', 'dependences'];
+
+      if (type === 'command') {
+        host = Editor;
+      }
+
+      if (type === 'behaviour') {
+        keys = ['name', 'behaviour', 'dependences'];
+      }
+
       const args = keys.map(key => this.props[key]);
 
-      Editor[className][`register${upperFirst(type)}`](...args);
+      host[`register${upperFirst(type)}`](...args);
     });
   }
 }