feat(serverHandler): add middleware interface

Author: marjune163

src/middleware/context.go

@@ -0,0 +1,9 @@
+package middleware
+
+type Context struct {
+	PrefixReqPath string
+	VhostReqPath  string
+	AliasReqPath  string
+	AliasFsPath   string
+	AliasFsRoot   string
+}

src/middleware/main.go

@@ -0,0 +1,5 @@
+package middleware
+
+import "net/http"
+
+type Middleware func(w http.ResponseWriter, r *http.Request, context *Context) (processed bool)

src/param/main.go

@@ -2,6 +2,7 @@ package param
 
 import (
 	"crypto/tls"
+	"mjpclab.dev/ghfs/src/middleware"
 	"mjpclab.dev/ghfs/src/serverError"
 	"mjpclab.dev/ghfs/src/util"
 	"os"
@@ -83,6 +84,8 @@ type Param struct {
 
 	AccessLog string
 	ErrorLog  string
+
+	Middlewares []middleware.Middleware
 }
 
 type Params []*Param

src/serverHandler/aliasHandler.go

@@ -1,6 +1,7 @@
 package serverHandler
 
 import (
+	"mjpclab.dev/ghfs/src/middleware"
 	"mjpclab.dev/ghfs/src/param"
 	"mjpclab.dev/ghfs/src/serverLog"
 	"mjpclab.dev/ghfs/src/tpl"
@@ -106,6 +107,8 @@ type aliasHandler struct {
 	contentVaryV1 string
 	contentVary   string
 
+	middlewares []middleware.Middleware
+
 	fileServer http.Handler
 }
 
@@ -131,7 +134,7 @@ func (h *aliasHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	}
 
 	// data
-	data := h.getResponseData(r)
+	data, fsPath := h.getResponseData(r)
 	h.logErrors(data.errors)
 	file := data.File
 	if file != nil {
@@ -167,7 +170,7 @@ func (h *aliasHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	// regular flows
+	// archive
 	if len(r.URL.RawQuery) >= 3 {
 		switch r.URL.RawQuery[:3] {
 		case "tar":
@@ -182,6 +185,12 @@ func (h *aliasHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
+	// middlewares
+	if h.middleware(w, r, data, fsPath) {
+		return
+	}
+
+	// final process
 	item := data.Item
 	if data.WantJson {
 		h.json(w, r, data)
@@ -275,6 +284,8 @@ func newAliasHandler(
 		pageVary:      ap.pageVary,
 		contentVaryV1: ap.contentVaryV1,
 		contentVary:   ap.contentVary,
+
+		middlewares: p.Middlewares,
 	}
 	return h
 }

src/serverHandler/middleware.go

@@ -0,0 +1,28 @@
+package serverHandler
+
+import (
+	"mjpclab.dev/ghfs/src/middleware"
+	"net/http"
+)
+
+func (h *aliasHandler) middleware(w http.ResponseWriter, r *http.Request, data *responseData, fsPath string) (processed bool) {
+	if len(h.middlewares) == 0 {
+		return false
+	}
+
+	context := &middleware.Context{
+		PrefixReqPath: data.prefixReqPath,
+		VhostReqPath:  data.rawReqPath,
+		AliasReqPath:  data.handlerReqPath,
+		AliasFsPath:   fsPath,
+		AliasFsRoot:   h.root,
+	}
+
+	for i := range h.middlewares {
+		if h.middlewares[i](w, r, context) {
+			return true
+		}
+	}
+
+	return false
+}

src/serverHandler/responseData.go

@@ -291,7 +291,7 @@ func dereferenceSymbolLinks(reqFsPath string, subItems []os.FileInfo) (errs []er
 	return
 }
 
-func (h *aliasHandler) getResponseData(r *http.Request) *responseData {
+func (h *aliasHandler) getResponseData(r *http.Request) (data *responseData, fsPath string) {
 	var errs []error
 
 	prefixReqPath := r.URL.RawPath
@@ -458,5 +458,5 @@ func (h *aliasHandler) getResponseData(r *http.Request) *responseData {
 		Context:       context,
 
 		NeedDirSlashRedirect: needDirSlashRedirect,
-	}
+	}, reqFsPath
 }