Dennis Rodewyk 5 lat temu
rodzic
commit
931bab7120
1 zmienionych plików z 12 dodań i 4 usunięć
  1. 12 4
      hugo-encrypt.go

+ 12 - 4
hugo-encrypt.go

@@ -64,10 +64,18 @@ func encryptPage(path string) {
 }
 
 func main() {
-	if !(os.Args[1]) {
-		flag.Usage = func() {
-			fmt.Println("Lol")
-		}
+	var servername = flag.String("dir", "localhost:8129", "server:port")
+	
+	flag.Parse()
+	
+	flagset := make(map[string]bool)
+	flag.Visit(func(f *flag.Flag) { flagset[f.Name]=true } )
+	
+	if flagset["dir"] {
+	    fmt.Printf("dir set via flags\n")
+	} else {
+	    fmt.Printf("dir not explicitly set, using default\n")
+		os.Exit(1)
 	}
 	publicPath := os.Args[1]
 	err := filepath.Walk(publicPath, func(path string, f os.FileInfo, err error) error {