特殊页面支持请求变量

internal/nodes/http_request_page.go

@@ -2,9 +2,9 @@ package nodes
 
 import (
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
+	"github.com/TeaOSLab/EdgeNode/internal/utils"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/logs"
-	"io"
 	"net/http"
 	"os"
 	"regexp"
@@ -48,7 +48,9 @@ func (this *HTTPRequest) doPage(status int) (shouldStop bool) {
 					this.writer.WriteHeader(status)
 				}
 				buf := bytePool1k.Get()
-				_, err = io.CopyBuffer(this.writer, fp, buf)
+				_, err = utils.CopyWithFilter(this.writer, fp, buf, func(p []byte) []byte {
+					return []byte(this.Format(string(p)))
+				})
 				bytePool1k.Put(buf)
 				if err != nil {
 					if !this.canIgnore(err) {

internal/nodes/http_request_shutdown.go

@@ -2,9 +2,9 @@ package nodes
 
 import (
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
+	"github.com/TeaOSLab/EdgeNode/internal/utils"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/logs"
-	"io"
 	"net/http"
 	"os"
 )
@@ -64,7 +64,9 @@ func (this *HTTPRequest) doShutdown() {
 		this.writer.WriteHeader(http.StatusOK)
 	}
 	buf := bytePool1k.Get()
-	_, err = io.CopyBuffer(this.writer, fp, buf)
+	_, err = utils.CopyWithFilter(this.writer, fp, buf, func(p []byte) []byte {
+		return []byte(this.Format(string(p)))
+	})
 	bytePool1k.Put(buf)
 	if err != nil {
 		if !this.canIgnore(err) {

internal/utils/reader_utils.go

@@ -0,0 +1,25 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package utils
+
+import "io"
+
+func CopyWithFilter(writer io.Writer, reader io.Reader, buf []byte, filter func(p []byte) []byte) (written int64, err error) {
+	for {
+		n, err := reader.Read(buf)
+		if n > 0 {
+			n2, err := writer.Write(filter(buf[:n]))
+			written += int64(n2)
+			if err != nil {
+				return written, err
+			}
+		}
+		if err != nil {
+			if err == io.EOF {
+				break
+			}
+			return written, err
+		}
+	}
+	return written, nil
+}