Add/remove imports

Author: kunaltaitkar

README.md

@@ -2,7 +2,7 @@
 
 ![visitors](https://visitor-badge.laobi.icu/badge?page_id=kunaltaitkar.vue-script-parser)
 
-This package can be used to parse vue script section and it provides an interface to add, update and delete entities of different life cycles hooks of VueJS.
+This package is used to parse vue script section and it provides an interface to add, update and delete entities of different life cycles hooks of VueJS.
 
 
 ## Installation
@@ -79,3 +79,22 @@ export default {
 }
 ```
 
+## Data
+
+### To add new variable in data 
+
+``` 
+let instance = new VueScriptParser(code)
+instace.addData('country','INDIA')
+```
+
+#### Output
+
+```
+
+```
+
+
+
+
+

example/Demo.vue

@@ -11,6 +11,8 @@
     <button @click="downloadScript()">Download Script</button>
     <button @click="addMethod">Add method</button>
     <button @click="removeMethod">Remove method</button>
+    <button @click="addImport">Add import</button>
+    <button @click="removeImport">Remove import</button>
       <p align="center">Input</p>
       <textarea
         cols="91"
@@ -36,7 +38,9 @@ import VueScriptParser from '../index'
 export default {
   data() {
     return {
-      input: `export default {
+      input: `
+      import a from 'a'
+      export default {
         data() {
           return {}
         },
@@ -68,6 +72,7 @@ export default {
         key: '',
         val:''
       }
+      this.updateJSON()
     },
     removeVariable() {
       this.instance.removeData(this.variable.key)
@@ -76,6 +81,7 @@ export default {
         key: '',
         val:''
       }
+      this.updateJSON()
     },
     downloadScript() {
       alert(this.instance.script)
@@ -84,15 +90,31 @@ export default {
       this.instance.addMounted(this.mountedBody)
       this.input = this.instance.script
       this.mountedBody = ''
+      this.updateJSON()
     },
     addMethod() {
       this.instance.addMethod('test', "console.log('this is test method')", "args1,arg2")
       this.instance.addMethod('test2', "console.log('this is test 2 method')")
       this.input = this.instance.script
+      this.updateJSON()
     },
     removeMethod() {
       this.instance.removeMethod('test2')
       this.input = this.instance.script
+      this.updateJSON()
+    },
+    addImport() {
+      this.instance.addImport("import c from 'c'")
+      this.input = this.instance.script
+      this.updateJSON()
+    },
+    removeImport() {
+      this.instance.removeImport("import c from 'c'")
+      this.input = this.instance.script
+      this.updateJSON()
+    },
+    updateJSON() {
+      this.output = JSON.stringify(this.instance, undefined, 2)
     }
   },
 }

index.js

@@ -7,25 +7,54 @@ import { pattern } from './modules/constant'
 class VueScriptParser {
     constructor(vueScript = '') {
 
-        //set original script
+        // set original script
         this.script = vueScript
 
-        //load imports
+        // load imports
+        this.imports = this.importsReader()
 
-
-        //load data
+        // load data
         this.data = this.dataReader()
 
-
-        //load mounted
+        // load mounted
         this.mounted = this.mountedReader()
 
-
-        //load methods
+        // load methods
         this.methods = this.methodsReader()
 
     }
+    // to read imports from given script and parse
+    importsReader () {
+        return this.script.match(pattern.importsRegex) || []
+    }
+    // to add new import statement
+    addImport (statement = '') {
+        if (!statement) {
+            return
+        }
+
+        (this.imports || []).push(statement)
+
+        let importStr = '';
+
+        (this.imports || []).forEach((imp) => {
+            importStr += imp + '\n'
+        })
+
+        this.script = this.script.replace(pattern.importsRegex, importStr)
+        this.imports = this.importsReader() || []
+
+    }
+    // to remove import from given script
+    removeImport (statement) {
+        if (!statement) {
+            return
+        }
 
+        this.script = this.script.replace(statement, '')
+        this.imports = this.importsReader() || []
+    }
+    // to read data section from given script and  parses
     dataReader () {
         let processedData = extractAndUpdateData(this.script, pattern.dataRegex)
 
@@ -47,7 +76,7 @@ class VueScriptParser {
 
         return result
     }
-
+    // to add new variable in data section of given script
     addData (key, value) {
         let processedData = extractAndUpdateData(this.script, pattern.dataRegex)
 
@@ -77,9 +106,9 @@ class VueScriptParser {
             pattern.dataRegex,
             script
         )
-
+        this.data = this.dataReader() || []
     }
-
+    // to remove variable from data section of given script
     removeData (key) {
 
         let processedData = extractAndUpdateData(this.script, pattern.dataRegex)
@@ -110,9 +139,10 @@ class VueScriptParser {
             pattern.dataRegex,
             script
         )
-
+        this.data = this.dataReader() || []
 
     }
+    // to read mounted section body from given script
     mountedReader () {
 
         let newScript = ''
@@ -127,7 +157,7 @@ class VueScriptParser {
 
         return newScript
     }
-
+    // to add/update mounted body in given script
     addMounted (body) {
 
         let updatedVueScript = extractAndUpdateMounted(
@@ -139,8 +169,9 @@ class VueScriptParser {
         if (updatedVueScript) {
             this.script = updatedVueScript
         }
+        this.mounted = this.mountedReader() || ''
     }
-
+    // to read methods section from given script and parse
     methodsReader () {
         let matchMethods = this.script.match(pattern.methodsRegex)
         if (!matchMethods) {
@@ -156,7 +187,7 @@ class VueScriptParser {
         return methodsData || []
 
     }
-
+    // to add new method in given script
     addMethod (name = '', body = '', args = '') {
 
         let newMethod = {
@@ -190,8 +221,9 @@ class VueScriptParser {
             methodsString.substring(lastIndex, methodsString.length)
 
         this.script = this.script.replace(pattern.methodsRegex, updatedMethods)
+        this.methods = this.methodsReader() || []
     }
-
+    // to remove method from given script
     removeMethod (name = '') {
 
         let matchMethods = this.script.match(pattern.methodsRegex)
@@ -241,6 +273,7 @@ class VueScriptParser {
             methodsString.substring(lastIndex, methodsString.length)
 
         this.script = this.script.replace(pattern.methodsRegex, updatedMethodsData)
+        this.methods = this.methodsReader() || []
     }
 }
 

modules/constant.js

@@ -1,6 +1,7 @@
 
 
 const pattern = {
+    importsRegex: /import .*/g,
     dataRegex: /data\s*\(\s*\)\s*\{\s*return\s*{([^]*)}/g,
     mountedRegex: /mounted\s*\(\s*\)\s*{([^]*)}/g,
     methodsRegex: /methods\s*:\s*{([^]*)}/g