Feat: AllowProxyToLocal config

internal/conf/proxy.go

@@ -1,13 +1,15 @@
 package conf
 
 type ProxyConfig struct {
-	MovieProxy bool `yaml:"movie_proxy" env:"PROXY_MOVIE"`
-	LiveProxy  bool `yaml:"live_proxy" env:"PROXY_LIVE"`
+	MovieProxy        bool `yaml:"movie_proxy" env:"PROXY_MOVIE"`
+	LiveProxy         bool `yaml:"live_proxy" env:"PROXY_LIVE"`
+	AllowProxyToLocal bool `yaml:"allow_proxy_to_local" env:"PROXY_ALLOW_TO_LOCAL"`
 }
 
 func DefaultProxyConfig() ProxyConfig {
 	return ProxyConfig{
-		MovieProxy: true,
-		LiveProxy:  true,
+		MovieProxy:        true,
+		LiveProxy:         true,
+		AllowProxyToLocal: false,
 	}
 }

internal/model/movie.go

@@ -67,7 +67,7 @@ func (m *BaseMovie) Validate() error {
 		if err != nil {
 			return err
 		}
-		if utils.IsLocalIP(u.Host) {
+		if !conf.Conf.Proxy.AllowProxyToLocal && utils.IsLocalIP(u.Host) {
 			return errors.New("local ip is not allowed")
 		}
 		switch u.Scheme {
@@ -89,7 +89,7 @@ func (m *BaseMovie) Validate() error {
 		if err != nil {
 			return err
 		}
-		if utils.IsLocalIP(u.Host) {
+		if !conf.Conf.Proxy.AllowProxyToLocal && utils.IsLocalIP(u.Host) {
 			return errors.New("local ip is not allowed")
 		}
 		if u.Scheme != "http" && u.Scheme != "https" {

server/handlers/movie.go

@@ -499,8 +499,12 @@ func ProxyMovie(ctx *gin.Context) {
 }
 
 func proxyURL(ctx *gin.Context, u string, headers map[string]string) error {
-	if l, err := utils.ParseURLIsLocalIP(u); err != nil || l {
-		return err
+	if !conf.Conf.Proxy.AllowProxyToLocal {
+		if l, err := utils.ParseURLIsLocalIP(u); err != nil {
+			return err
+		} else if l {
+			return errors.New("not allow proxy to local")
+		}
 	}
 	r, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
 	if err != nil {