更新 gotemplate 的参数注入

pkg/renders/gotemplate.go

@@ -3,11 +3,11 @@ package renders
 import (
 	"bytes"
 	"io"
-	"net"
 	"net/http"
-	"strings"
 	"text/template"
 
+	"code.d7z.net/d7z-project/gitea-pages/pkg/utils"
+
 	sprig "github.com/go-task/slim-sprig/v3"
 )
 
@@ -24,47 +24,13 @@ func (g GoTemplate) Render(w http.ResponseWriter, r *http.Request, input io.Read
 	}
 	out := &bytes.Buffer{}
 	parse, err := template.New("tmpl").Funcs(sprig.FuncMap()).Option("missingkey=error").Parse(string(dataB))
-	headers := make(map[string]string)
-	for k, vs := range r.Header {
-		headers[k] = strings.Join(vs, ",")
-	}
 	if err != nil {
 		return err
 	}
-	err = parse.Execute(out, map[string]interface{}{
-		"Request": map[string]any{
-			"Headers":    headers,
-			"Request":    r.RequestURI,
-			"RemoteAddr": r.RemoteAddr,
-			"RemoteIP":   GetRemoteIP(r),
-		},
-	})
+	err = parse.Execute(out, utils.NewTemplateInject(r, nil))
 	if err != nil {
 		return err
 	}
 	_, err = out.WriteTo(w)
 	return err
 }
-
-// 注意，相关 ip 获取未做反向代理安全判断，可能导致安全降级
-
-func GetRemoteIP(r *http.Request) string {
-	// 最先取 cloudflare 的头
-	if ip := r.Header.Get("CF-Connecting-IP"); ip != "" {
-		return ip
-	}
-	if forwardedFor := r.Header.Get("X-Forwarded-For"); forwardedFor != "" {
-		ips := strings.Split(forwardedFor, ",")
-		if len(ips) > 0 {
-			return strings.TrimSpace(ips[0])
-		}
-	}
-	if realIP := r.Header.Get("X-Real-IP"); realIP != "" {
-		return realIP
-	}
-	ip, _, err := net.SplitHostPort(r.RemoteAddr)
-	if err != nil {
-		return r.RemoteAddr
-	}
-	return ip
-}