From f3bcb50e3ad0960d810bbec0abe159bf0471704b Mon Sep 17 00:00:00 2001
From: Joona Hoikkala <joohoi@users.noreply.github.com>
Date: Wed, 16 Sep 2020 11:08:09 +0300
Subject: [PATCH] Fixed comma parsing bugs in non-wordlist cli arguments (#298)

---
 main.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index a4ee43a..cc0a4d9 100644
--- a/main.go
+++ b/main.go
@@ -42,7 +42,7 @@ type cliOptions struct {
 	URL                    string
 	outputFormat           string
 	ignoreBody             bool
-	wordlists              multiStringFlag
+	wordlists              wordlistFlag
 	inputcommands          multiStringFlag
 	headers                multiStringFlag
 	cookies                multiStringFlag
@@ -52,12 +52,22 @@ type cliOptions struct {
 }
 
 type multiStringFlag []string
+type wordlistFlag []string
 
 func (m *multiStringFlag) String() string {
 	return ""
 }
 
+func (m *wordlistFlag) String() string {
+	return ""
+}
+
 func (m *multiStringFlag) Set(value string) error {
+	*m = append(*m, value)
+	return nil
+}
+
+func (m *wordlistFlag) Set(value string) error {
 	delimited := strings.Split(value, ",")
 
 	if len(delimited) > 1 {