PublicPath arg

hugo-encrypt.go

@@ -63,7 +63,8 @@ func encryptPage(path string) {
 }
 
 func main() {
-	err := filepath.Walk("../data/site/public", func(path string, f os.FileInfo, err error) error {
+	publicPath := os.Args[1]
+	err := filepath.Walk(publicPath, func(path string, f os.FileInfo, err error) error {
 		if f == nil {
 			return err
 		}