Add Config File

src/config/VafConfigStructure.nim

@@ -0,0 +1,6 @@
+import VafWordlist
+
+type
+  VafConfigStructure* = object
+    DefaultWordlist*: int
+    Wordlists*: seq[VafWordlist]

src/config/VafWordlist.nim

@@ -0,0 +1,5 @@
+
+type
+  VafWordlist* = object
+    Id*: int
+    Path*: string

src/config/importVafConfig.nim

@@ -0,0 +1,24 @@
+import json
+import os
+import strutils
+
+import initVafConfig
+import parseVafConfig
+
+import VafConfigStructure
+
+var homeDir: string = os.getHomeDir()
+
+proc importVafConfig*(): VafConfigStructure = 
+    var configExists: bool = fileExists(homedir & "/.vaf.json")
+    var configContents = readFile(homedir & "/.vaf.json")
+    var jsonData: JsonNode
+
+
+    if configExists:
+      jsonData = parseJson(configContents)
+    # else:
+    #   jsonData = initVafConfig()
+
+    return parseVafConfig(jsonData)
+

src/config/initVafConfig.nim

@@ -0,0 +1,17 @@
+import os
+import strutils
+
+var homeDir: string = os.getHomeDir()
+var defaultJson: string = """{}"""
+
+proc initVafConfig*(): int = 
+    var configExists: bool = fileExists(homedir & "/.vaf.json")
+    var configFile = open(homedir & "/.vaf.json", fmWrite)
+
+    defer: configFile.close()
+
+    if configExists:
+        return 1
+    else:
+        for i in defaultJson.split("\n"):
+            configFile.writeLine(i)

src/config/parseVafConfig.nim

@@ -0,0 +1,36 @@
+import json
+import os
+import strutils
+
+import VafWordlist
+import VafConfigStructure
+
+proc parseWordlist(data: JsonNode): VafWordlist = 
+  var wordlistObject: VafWordlist = VafWordlist()
+
+  if data{"Id"} != nil:
+    wordlistObject.Id = data{"Id"}.getInt()
+
+  if data{"Path"} != nil:
+    wordlistObject.Path = data{"Path"}.getStr()
+
+  return wordlistObject
+
+proc parseVafConfig*(data: JsonNode): VafConfigStructure = 
+  var configObject: VafConfigStructure = VafConfigStructure()
+
+  echo data
+
+  if data{"DefaultWordlist"} == nil:
+    configObject.DefaultWordlist = -1
+  else:
+    configObject.DefaultWordlist = data{"DefaultWordlist"}.getInt()
+
+  if data{"Wordlists"} != nil:
+    for wordlist in data{"Wordlists"}:
+      configObject.Wordlists.add( parseWordlist( wordlist ))
+
+
+  echo configObject
+
+  return configObject

src/config/readVafConfig.nim

@@ -0,0 +1,18 @@
+import json
+import os
+import strutils
+import initVafConfig
+
+var homeDir: string = os.getHomeDir()
+
+proc readVafConfig*(): int = 
+    var configExists: bool = fileExists(homedir & "/.vaf.json")
+    var configContents = readFile(homedir & "/.vaf.json")
+    var jsonNode = parseJson(configContents)
+
+    discard initVafConfig()
+
+    if configExists:
+        return 0
+    else:
+        echo jsonNode

src/utils/VafHttpClient.nim

@@ -1,6 +1,5 @@
 import httpclient
 import VafResponse
-import json
 
 var client = newHttpClient()
 

src/vaf.nim

@@ -8,6 +8,7 @@ import os
 import argparse
 import utils/VafFuzzResult
 import utils/VafColors
+import config/initVafConfig
 
 let p = newParser("vaf - very advanced fuzzer"):
   option("-u", "--url", help="choose url, replace area to fuzz with []")
@@ -23,6 +24,9 @@ let p = newParser("vaf - very advanced fuzzer"):
   flag("-pu", "--printurl", help="prints the url that has been requested")
 
 try:
+
+    # discard initVafConfig()
+
     var parsedArgs = p.parse(commandLineParams())
 
     var url: string = parsedArgs.url