Fix flag parsing

main.go

@@ -12,6 +12,7 @@ import (
 	"time"
 )
 
+
 const lifetime time.Duration = 24 * time.Hour
 
 var devices struct {
@@ -32,6 +33,14 @@ func main() {
     publicFolder := flag.String("public", "./public/", "Folder with the public files")
     httpPort := flag.String("http-port", "8180", "Port for the HTTP server")
 
+	// Parse the command-line flags
+	flag.Parse()
+
+	// Check if the pubic folder exists
+	if _, err := os.Stat(*publicFolder); os.IsNotExist(err) {
+		log.Fatalf("Publich folder does not exist")
+	}
+
     devices.d = make([]Device, 0)
 
 	http.HandleFunc("/favicon.ico", func(w http.ResponseWriter, r *http.Request) {})
@@ -42,6 +51,8 @@ func main() {
 	go cleanup()
 
 	fmt.Println("Listen on port", httpPort)
+    fmt.Println("Using public folder", publicFolder)
+
 	// Note: use TLS
     log.Fatal(http.ListenAndServe(":" + *httpPort, nil))
 }