Jelajahi Sumber

The one before works

Dennis Rodewyk 5 tahun lalu
induk
melakukan
08438c3812
1 mengubah file dengan 10 tambahan dan 3 penghapusan
  1. 10 3
      hugo-encrypt.go

+ 10 - 3
hugo-encrypt.go

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