support for dirlist size

cmd/dirlist.go

@@ -6,7 +6,15 @@ import (
 	"github.com/charmbracelet/log"
 )
 
-func Dirlist(url string) {
+func Dirlist(size string, url string) {
 	log.Infof("Starting directory scan on %s...", url)
-	fmt.Println(url)
+
+	switch size {
+	case "small":
+		fmt.Println("small")
+	case "medium":
+		fmt.Println("medium")
+	case "large":
+		fmt.Println("large")
+	}
 }

config.go

@@ -10,13 +10,13 @@ import (
 
 type Settings struct {
 	URLs    []string
-	Dirlist bool
+	Dirlist string
 }
 
 func parseURLs() Settings {
 	var url = pflag.StringArrayP("url", "u", []string{}, "URL to check")
 	var file = pflag.StringP("file", "f", "", "File that includes URLs to check")
-	var dirlist = pflag.Bool("dirlist", false, "")
+	var dirlist = pflag.String("dirlist", "none", "Dirlist scan size (small, medium, large)")
 	pflag.Parse()
 
 	if len(*url) > 0 {

main.go

@@ -31,8 +31,8 @@ func main() {
 	settings := parseURLs()
 	for _, url := range settings.URLs {
 		log.Infof("Looking up %s...", url)
-		if settings.Dirlist {
-			cmd.Dirlist(url)
+		if settings.Dirlist != "none" {
+			cmd.Dirlist(settings.Dirlist, url)
 		}
 	}
 }