refactor and add middleware

admin/admin.go

@@ -18,9 +18,14 @@ func about(w http.ResponseWriter, r *http.Request) {
 
 }
 
+func setConfig(w http.ResponseWriter, r *http.Request) {
+
+}
+
 var AdminRoutes = []Route{
 	// Admin Routes
 	{"GET", "/about", about},
+	{"Post", "/config", setConfig},
 }
 
 var AdminServerMux *server.RestMux
@@ -31,4 +36,5 @@ func init() {
 	for _, route := range AdminRoutes {
 		AdminServerMux.HandleFunc(route.Method, route.Path, route.Handle)
 	}
+	AdminServerMux.Use(server.BasicAuth)
 }

gohttp.go

@@ -12,13 +12,12 @@ import (
 	"git.pyer.club/kingecg/gologger"
 )
 
-var conf model.GoHttpdConfig
-
 type GoHttp struct {
 	logger gologger.Logger
 }
 
 func (g *GoHttp) Start() {
+	conf := model.GetConfig()
 	g.logger = gologger.GetLogger("Server")
 	g.logger.Info("start gohttpd")
 	// if g.conf != nil {
@@ -78,8 +77,8 @@ func LoadConfig(configPath string) {
 
 	// read content from cpath
 	content, _ := os.ReadFile(cpath)
-	json.Unmarshal(content, &conf)
-	gologger.Configure(conf.Logging)
+	json.Unmarshal(content, &model.Config)
+	gologger.Configure(model.Config.Logging)
 	logger := gologger.GetLogger("Server")
 	logger.Info("Load config success")
 }

model/model.go

@@ -16,10 +16,13 @@ type PathRewrite struct {
 }
 
 type HttpServerConfig struct {
+	Name       string `json:"name"`
 	ServerName string `json:"server"`
 	Port       int    `json:"port"`
 	Host       string `json:"host"`
 	Paths      []HttpPath
+	Username   string `json:"username"`
+	Password   string `json:"password"`
 }
 
 type GoHttpdConfig struct {
@@ -32,3 +35,28 @@ var DefaultAdminConfig HttpServerConfig = HttpServerConfig{
 	ServerName: "admin",
 	Port:       8080,
 }
+
+var Config GoHttpdConfig = GoHttpdConfig{}
+
+func GetConfig() *GoHttpdConfig {
+	return &Config
+}
+
+func SetServerConfig(c *HttpServerConfig) {
+	for i, s := range Config.Servers {
+		if s.Name == c.Name {
+			Config.Servers[i] = c
+			return
+		}
+	}
+	Config.Servers = append(Config.Servers, c)
+}
+
+func GetServerConfig(name string) *HttpServerConfig {
+	for _, s := range Config.Servers {
+		if s.Name == name {
+			return s
+		}
+	}
+	return nil
+}

server/middleware.go

@@ -0,0 +1,32 @@
+package server
+
+import (
+	"net/http"
+
+	"git.pyer.club/kingecg/gohttpd/model"
+)
+
+type Middleware func(w http.ResponseWriter, r *http.Request, next func())
+
+func BasicAuth(w http.ResponseWriter, r *http.Request, next func()) {
+	config := model.GetConfig()
+
+	if config.Admin.Username == "" || config.Admin.Password == "" {
+		next()
+		return
+	}
+	user, pass, ok := r.BasicAuth()
+	if ok && user == config.Admin.Username && pass == config.Admin.Password {
+		next()
+	} else {
+		w.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
+		http.Error(w, "Unauthorized.", http.StatusUnauthorized)
+	}
+}
+
+func Parse(w http.ResponseWriter, r *http.Request, next func()) {
+	if r.Method == "POST" || r.Method == "PUT" {
+		r.ParseForm()
+	}
+	next()
+}

server/server.go

@@ -9,12 +9,27 @@ import (
 
 // 可以嵌套的Rest http server mux
 type RestMux struct {
-	Path      string
-	imux      *http.ServeMux
-	rmuxPaths []string
+	Path        string
+	imux        *http.ServeMux
+	rmuxPaths   []string
+	middlewares []Middleware
 }
 
+func (mux *RestMux) Use(m Middleware) {
+	mux.middlewares = append(mux.middlewares, m)
+}
 func (mux *RestMux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	canContinue := false
+	if len(mux.middlewares) > 0 {
+		for _, m := range mux.middlewares {
+			canContinue = false
+			m(w, r, func() { canContinue = true })
+			if !canContinue {
+				return
+			}
+		}
+	}
+
 	_, has := lo.Find(mux.rmuxPaths, func(s string) bool {
 		return strings.HasPrefix(r.URL.Path, s)
 	})