Add handleSave in commands

artf · artf · commit 44793a56ddfd · 2018-07-25T01:21:34.000+02:00
diff --git a/src/commands.js b/src/commands.js
@@ -14,6 +14,7 @@ export default (editor, opts = {}) => {
     }
   }
 
+  // Add the custom code command
   cmd.add(commandNameCustomCode, {
     keyCustomCode,
 
@@ -76,20 +77,30 @@ export default (editor, opts = {}) => {
      * @return {HTMLElement|String}
      */
     getContentActions() {
-      const { editor, options, target } = this;
+      const { editor } = this;
       const btn = document.createElement('button');
       const pfx = editor.getConfig('stylePrefix');
       btn.innerHTML = opts.buttonLabel;
       btn.className = `${pfx}btn-prim ${pfx}btn-import__custom-code`;
-      btn.onclick = () => {
-        const code = this.getCodeViewer().getContent();
-        target.set(keyCustomCode, code);
-        editor.Modal.close();
-      };
+      btn.onclick = () => this.handleSave();
 
       return btn;
     },
 
+    /**
+     * Handle the main save task
+     */
+    handleSave() {
+      const { editor, target } = this;
+      const code = this.getCodeViewer().getContent();
+      target.set(keyCustomCode, code);
+      editor.Modal.close();
+    },
+
+    /**
+     * Return the code viewer instance
+     * @return {CodeViewer}
+     */
     getCodeViewer() {
       const { editor } = this;
 
diff --git a/src/components.js b/src/components.js
@@ -91,7 +91,7 @@ export default (editor, opts = {}) => {
 
       onActive() {
         const target = this.model;
-        this.em.get('Editor').runCommand(commandNameCustomCode, { target });
+        this.em.get('Commands').run(commandNameCustomCode, { target });
       },
     })
   });
diff --git a/src/index.js b/src/index.js
@@ -37,7 +37,7 @@ export default grapesjs.plugins.add('YOUR-PLUGIN-NAME', (editor, opts = {}) => {
     codeViewOptions: {},
 
     // Label for the default save button
-    buttonLabel: 'Insert your code',
+    buttonLabel: 'Save',
 
     // Object to extend the default custom code command.
     // Check the source to see all available methods
