fix issue #23

Author: XORbit

cmd/root.go

@@ -45,7 +45,10 @@ func init() {
 
 	rootCmd.Flags().StringP("output", "o", "", "file to export the result (f.json, f.xml, f.txt) / ex: -o result.json")
 
-	rootCmd.Flags().StringToString("regexes", map[string]string{}, "regexes to match in each page / ex: --regexes comments=\"\\<\\!--.*?-->\"")
+	// rootCmd.Flags().StringToString("regexes", map[string]string{}, "regexes to match in each page / ex: --regexes comments=\"\\<\\!--.*?-->\"")
+
+	var regexesFlag shared.RegexFlag
+	rootCmd.Flags().Var(&regexesFlag, "regexes", "regexes to match in each page / ex: --regexes comments=\"\\<\\!--.*?-->\"")
 
 	rootCmd.Flags().IntSliceP("exclude-code", "x", []int{}, "status codes to exclude / ex : -x 404,500")
 

shared/customs.go

@@ -0,0 +1,61 @@
+package shared
+
+import (
+	"fmt"
+	"strings"
+)
+
+type RegexFlag struct {
+	Regexes map[string]string
+}
+
+func (r *RegexFlag) String() string {
+	return fmt.Sprintf("%v", r.Regexes)
+}
+
+func (r *RegexFlag) Set(value string) error {
+	var err error
+	r.Regexes, err = ParseRegexes(value)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (r *RegexFlag) Type() string {
+	return "regex"
+}
+
+func (r *RegexFlag) Value() map[string]string {
+	return r.Regexes
+}
+
+func isBetween(s string, a string, b string, index int) bool {
+	return strings.Count(s[:index], a) == strings.Count(s[index:], b)
+}
+
+func ParseRegexes(stream string) (map[string]string, error) {
+	result := make(map[string]string)
+	var key, value string
+	prev := 0
+	for i, c := range stream {
+		if c == ',' && !isBetween(stream, "{", "}", i) {
+			keyValue := strings.Split(stream[prev:i], "=")
+			if len(keyValue) != 2 {
+				return nil, fmt.Errorf("invalid regexes format")
+			}
+			key = keyValue[0]
+			value = keyValue[1]
+			result[key] = value
+			prev = i + 1
+		}
+	}
+	keyValue := strings.Split(stream[prev:], "=")
+	if len(keyValue) != 2 {
+		return nil, fmt.Errorf("invalid regexes format")
+	}
+	key = keyValue[0]
+	value = keyValue[1]
+	result[key] = value
+	return result, nil
+}

shared/options.go

@@ -171,7 +171,8 @@ func ValidateThenBuildOption(cmd *cobra.Command) (*Options, error) {
 	if err != nil {
 		return nil, err
 	}
-	regexMap, err := cmd.Flags().GetStringToString("regexes")
+	// regexMap, err := cmd.Flags().GetStringToString("regexes")
+	regexMap, err := cmd.Flags().Lookup("regexes").Value.(*RegexFlag).Value(), nil
 	if err != nil {
 		return nil, err
 	}