From f429337cc09246cd78b446308d30e2c4fd3d5136 Mon Sep 17 00:00:00 2001
From: lipengyu <lipengyux@gmail.com>
Date: Fri, 2 Dec 2022 13:04:11 +0800
Subject: [PATCH] refactor server/init.go

---
 Makefile       |  2 +-
 server/init.go | 44 +++++++++++++++++++++++++++-----------------
 2 files changed, 28 insertions(+), 18 deletions(-)

diff --git a/Makefile b/Makefile
index 4d530de..36bd4a3 100644
--- a/Makefile
+++ b/Makefile
@@ -25,7 +25,7 @@ demo5: facemask.wasm serve
 demo6: bgblur.wasm serve
 
 serve:
-	$(BROWSER) 'http://localhost:5000'
+	$(BROWSER) 'http://localhost:6060'
 	go run server/init.go
 
 clean:
diff --git a/server/init.go b/server/init.go
index 4628243..62c0e24 100644
--- a/server/init.go
+++ b/server/init.go
@@ -6,37 +6,47 @@ import (
 	"path/filepath"
 )
 
+var defaultConn = &httpConn{
+	host:        "",
+	port:        "6060",
+	path:        "./",
+	cascadePath: "./cascade/",
+}
+
 // httpConn web server connection parameters
 type httpConn struct {
-	port       string
-	root       string
-	cascadeDir string
+	host        string
+	port        string
+	path        string
+	cascadePath string
 }
 
-func main() {
-	httpConn := &httpConn{
-		port:       "6060",
-		root:       "./",
-		cascadeDir: "./cascade/",
-	}
-	initServer(httpConn)
+func (c *httpConn) addr() string {
+	return c.host + ":" + c.port
 }
 
-// initServer initializes the webserver
-func initServer(c *httpConn) {
+func init() {
 	var err error
-	c.root, err = filepath.Abs(c.root)
+	defaultConn.path, err = filepath.Abs(defaultConn.path)
 	if err != nil {
 		log.Fatalln(err)
 	}
+	defaultConn.cascadePath, err = filepath.Abs(defaultConn.cascadePath)
+	if err != nil {
+		log.Fatalln(err)
+	}
+}
 
-	log.Printf("serving %s on localhost:%s", c.root, c.port)
-	http.Handle("/", http.StripPrefix("/", http.FileServer(http.Dir(c.root))))
-	http.Handle("/cascade/", http.StripPrefix("/cascade/", http.FileServer(http.Dir(c.cascadeDir))))
+func main() {
+	log.Printf("serving %s on %s", defaultConn.path, defaultConn.addr())
+
+	http.Handle("/", http.StripPrefix("/", http.FileServer(http.Dir(defaultConn.path))))
+	http.Handle("/cascade/", http.StripPrefix("/cascade/", http.FileServer(http.Dir(defaultConn.cascadePath))))
 
 	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		log.Print(r.RemoteAddr + " " + r.Method + " " + r.URL.String())
 		http.DefaultServeMux.ServeHTTP(w, r)
 	})
-	log.Fatalln(http.ListenAndServe(":"+c.port, handler))
+
+	log.Fatalln(http.ListenAndServe(defaultConn.addr(), handler))
 }