diff --git a/cmd/dirlist.go b/cmd/dirlist.go index 0808b28..7ef5191 100644 --- a/cmd/dirlist.go +++ b/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") + } } diff --git a/config.go b/config.go index b22c808..45172fc 100644 --- a/config.go +++ b/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 { diff --git a/main.go b/main.go index 52251d8..ca4e73e 100644 --- a/main.go +++ b/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) } } }