remove svg

example/plugins/header/header.js

@@ -177,6 +177,10 @@ class Header {
    * @public
    */
   save(toolsContent) {
+    /**
+     * @todo sanitize data
+     */
+
     return {
       text: toolsContent.innerHTML,
       level: this.currentLevel
@@ -192,10 +196,6 @@ class Header {
     let text = this._element.innerHTML,
       tag = this._element.tagName;
 
-    /**
-     * @todo sanitize data
-     */
-
     this._data.text = text;
     this._data.level = tag;
 
@@ -256,28 +256,28 @@ class Header {
    * @return {HTMLHeadingElement}
    */
   getTag() {
-    /**
-     * Create element for current Block's level
-     */
-    let tag;
-
     /**
      * Create default element
      */
-    tag = document.createElement(this.levels[0].tag);
+    let tagName = this.levels[0].tag;
 
     /**
-     * Try to reassign to passed level element
+     * Find tag for current level
      */
     for (let index in this.levels) {
       const level = this.levels[index];
 
       if (level.number === this._data.level) {
-        tag = document.createElement(level.tag);
+        tagName = level.tag;
         break;
       }
     }
 
+    /**
+     * Create element for current Block's level
+     */
+    let tag = document.createElement(tagName);
+
     /**
      * Add text to block
      */
@@ -301,7 +301,7 @@ class Header {
    * @return {number}
    */
   get currentLevel() {
-    let level = this.levels.find( level => level.tag === this._element.tagName);
+    let level = this.levels.find( level => level.number === this._data.level);
 
     return level.number;
   }

example/plugins/text/text.js

@@ -107,11 +107,13 @@ class Text {
    * @public
    */
   save(toolsContent) {
-    let toolData = {
+    /**
+     * @todo sanitize data
+     */
+
+    return {
       text: toolsContent.innerHTML
     };
-
-    return toolData;
   }
 
   /**
@@ -122,10 +124,6 @@ class Text {
   get data() {
     let text = this._element.innerHTML;
 
-    /**
-     * @todo sanitize data
-     */
-
     this._data.text = text;
 
     return this._data;