Allow ws host

main.go

@@ -228,6 +228,7 @@ func main() {
 	adderss := flag.String("b", "127.0.0.1:8288", "Bind address")
 	flag.StringVar(&docDir, "d", "documents", "Document directory")
 	distDir := flag.String("w", "dist", "Web dist directory")
+	allowAddr := flag.String("a", "", "Allow other host to connect to websocket")
 	flag.Parse()
 	var err error
 	e = echo.New()
@@ -292,8 +293,12 @@ func main() {
 	e.GET("/*", func(c echo.Context) error {
 		return c.File(filepath.Join(*distDir, "index.html"))
 	})
-
 	go hub.run()
+	upgrader.CheckOrigin = func(r *http.Request) bool {
+		fmt.Printf("Host: %s\n", r.Host)
+		return r.Host == *adderss || r.Host == *allowAddr
+	}
+
 	e.Logger.Fatal(e.Start(*adderss))
 
 }

ws.go

@@ -24,9 +24,6 @@ const (
 )
 
 func init() {
-	//upgrader.CheckOrigin = func(r *http.Request) bool {
-	//	return true
-	//}
 	hub = new(wsHub)
 	hub.clients = make(map[*wsCli]bool)
 	hub.register = make(chan *wsCli)