test

hugo-encrypt.go

@@ -64,21 +64,17 @@ func encryptPage(path string) {
 }
 
 func main() {
-	var dir = 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 {
+	var Usage = func() {
+	    fmt.Fprintf(os.Stderr, "Usage of %s:\n", os.Args[0])
+
+        flag.PrintDefaults()
+	}
+	sitePath := os.Args[1]
+	if sitePath == nil {
 	    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 {
+	err := filepath.Walk(sitePath, func(path string, f os.FileInfo, err error) error {
 		if f == nil {
 			return err
 		}