添加渲染器和反向代理开关

config.go

@@ -81,6 +81,8 @@ func (c *Config) NewPageServerOptions() (*pkg.ServerOptions, error) {
 		MaxCacheSize:        int(cacheSize),
 		HttpClient:          http.DefaultClient,
 		MetaTTL:             time.Minute,
+		EnableRender:        c.Render.Enable,
+		EnableProxy:         c.Proxy.Enable,
 		DefaultErrorHandler: c.ErrorHandler,
 		Cache:               utils.NewCacheMemory(int(cacheMaxSize), int(cacheMaxSize)),
 	}
@@ -135,12 +137,11 @@ type ConfigPage struct {
 }
 
 type ConfigProxy struct {
-	Enable   bool     `yaml:"enable"` // 是否允许反向代理模型
-	DenyList []string `yaml:"deny"`   // 反向代理黑名单
+	Enable bool `yaml:"enable"` // 是否允许反向代理
 }
 
 type ConfigRender struct {
-	Enable bool `yaml:"enable"` // 开启渲染器
+	Enable bool `yaml:"enable"` // 是否开启渲染器
 }
 
 type ConfigCache struct {

pkg/server.go

@@ -42,6 +42,9 @@ type ServerOptions struct {
 
 	MetaTTL time.Duration
 
+	EnableRender bool
+	EnableProxy  bool
+
 	DefaultErrorHandler func(w http.ResponseWriter, r *http.Request, err error)
 }
 
@@ -55,6 +58,8 @@ func DefaultOptions(domain string) ServerOptions {
 		MaxCacheSize:  1024 * 1024 * 10,
 		HttpClient:    http.DefaultClient,
 		MetaTTL:       time.Minute,
+		EnableRender:  true,
+		EnableProxy:   true,
 		DefaultErrorHandler: func(w http.ResponseWriter, r *http.Request, err error) {
 			if errors.Is(err, os.ErrNotExist) {
 				http.Error(w, "page not found.", http.StatusNotFound)
@@ -119,6 +124,7 @@ func (s *Server) Serve(writer http.ResponseWriter, request *http.Request) error
 		return nil
 	}
 
+	if s.options.EnableProxy {
 		for prefix, backend := range meta.Proxy {
 			proxyPath := "/" + meta.Path
 			if strings.HasPrefix(proxyPath, prefix) {
@@ -137,6 +143,7 @@ func (s *Server) Serve(writer http.ResponseWriter, request *http.Request) error
 				return nil
 			}
 		}
+	}
 	// 如果不是反向代理路由则跳过任何配置
 	if request.Method != "GET" {
 		return os.ErrNotExist
@@ -174,7 +181,7 @@ func (s *Server) Serve(writer http.ResponseWriter, request *http.Request) error
 			}
 			writer.Header().Set("Content-Type", mime.TypeByExtension(".html"))
 			writer.WriteHeader(http.StatusNotFound)
-			if render := meta.TryRender(meta.Path, "/404.html"); render != nil {
+			if render := meta.TryRender(meta.Path, "/404.html"); render != nil && s.options.EnableRender {
 				defer result.Close()
 				if err = render.Render(writer, request, result); err != nil {
 					return err
@@ -191,6 +198,9 @@ func (s *Server) Serve(writer http.ResponseWriter, request *http.Request) error
 	}
 	fileName := filepath.Base(meta.Path)
 	render := meta.TryRender(meta.Path)
+	if !s.options.EnableRender {
+		render = nil
+	}
 	defer result.Close()
 	if reader, ok := result.(*utils.CacheContent); ok {
 		writer.Header().Add("X-Cache", "HIT")