Refactor of the flags

go/tests/rc/rc.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"go/ast"
 	"go/parser"
@@ -16,6 +17,62 @@ const (
 	identation = "  "
 )
 
+type strBoolMap map[string]bool
+
+func (a *strBoolMap) String() string {
+	var res string
+	for k, _ := range *a {
+		res += k
+	}
+	return res
+}
+
+func (a *strBoolMap) Set(str string) error {
+	if *a == nil {
+		*a = make(map[string]bool)
+	}
+	s := strings.Split(str, " ")
+	for _, v := range s {
+		(*a)[v] = true
+	}
+	return nil
+}
+
+type arrFlag struct {
+	active  bool
+	content []string
+}
+
+func (a *arrFlag) String() string {
+	return strings.Join(a.content, " ")
+}
+
+func (a *arrFlag) Set(s string) error {
+	a.active = true
+	a.content = strings.Split(s, " ")
+	return nil
+}
+
+// flag that groups a boolean value and a regular expression
+type regexpFlag struct {
+	active bool
+	reg    *regexp.Regexp
+}
+
+func (r *regexpFlag) String() string {
+	if r.reg != nil {
+		return r.reg.String()
+	}
+	return ""
+}
+
+func (r *regexpFlag) Set(s string) error {
+	re := regexp.MustCompile(s)
+	r.active = true
+	r.reg = re
+	return nil
+}
+
 var (
 	allowedImp map[string]map[string]bool // Map of the allowed imports
 	allowedFun map[string]bool            // Map of the allowed built-in functions
@@ -41,6 +98,13 @@ var (
 	allImports      map[string]bool
 	openImports     []string
 	funcOccurrences map[string]int
+	// Flags
+	noArrays          bool
+	noRelativeImports bool
+	noTheseArrays     strBoolMap
+	casting           bool
+	noFor             bool
+	noLit             regexpFlag
 )
 
 // pkgFunc for all the functions of a given package
@@ -229,6 +293,18 @@ func allowedAmount(occurrences map[string]int, allowedImports []string) {
 	}
 }
 
+func init() {
+	flag.Var(&noTheseArrays, "no-these-arrays", "unallowes the array types passed in the flag")
+	flag.Var(&noLit, "no-lit",
+		`The use of string literals matching the pattern --no-lit="{PATTERN}"`+
+			`passed to the program would not be allowed`,
+	)
+	flag.BoolVar(&noRelativeImports, "no-relative-imports", false, `No disallowes the use of relative imports`)
+	flag.BoolVar(&noFor, "no-for", false, `The "for" instruction is not allowed`)
+	flag.BoolVar(&casting, "cast", false, "allowes casting")
+	flag.BoolVar(&noArrays, "no-arrays", false, "unallowes the array types passed in the flag")
+}
+
 func main() {
 	if len(os.Args) < 2 {
 		fmt.Println("No file or directory")