mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-13 07:40:56 +00:00
feat(task): replace built-in task adaptors with a sandboxed JS plugin system (#7076)
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
var defaultTrustedProxyCIDRs = []string{
|
||||
"127.0.0.0/8",
|
||||
"::1",
|
||||
"10.0.0.0/8",
|
||||
"172.16.0.0/12",
|
||||
"192.168.0.0/16",
|
||||
"fc00::/7",
|
||||
}
|
||||
|
||||
// ResolveTrustedProxies parses TRUSTED_PROXIES without applying it to an
|
||||
// engine. The returned slice can be reused by the outer and plugin engines.
|
||||
func ResolveTrustedProxies(raw string) (trustedProxies []string, usedDefaults bool, err error) {
|
||||
raw = strings.TrimSpace(raw)
|
||||
if raw == "" {
|
||||
return append([]string(nil), defaultTrustedProxyCIDRs...), true, nil
|
||||
}
|
||||
if strings.EqualFold(raw, "none") {
|
||||
return nil, false, nil
|
||||
}
|
||||
|
||||
parts := strings.Split(raw, ",")
|
||||
trustedProxies = make([]string, 0, len(parts))
|
||||
for _, part := range parts {
|
||||
trustedProxy := strings.TrimSpace(part)
|
||||
if trustedProxy == "" {
|
||||
continue
|
||||
}
|
||||
if strings.EqualFold(trustedProxy, "none") {
|
||||
return nil, false, errors.New("TRUSTED_PROXIES=none must be used alone")
|
||||
}
|
||||
trustedProxies = append(trustedProxies, trustedProxy)
|
||||
}
|
||||
if len(trustedProxies) == 0 {
|
||||
return nil, false, errors.New("TRUSTED_PROXIES does not contain an IP address or CIDR")
|
||||
}
|
||||
return trustedProxies, false, nil
|
||||
}
|
||||
|
||||
func ConfigureTrustedProxies(engine *gin.Engine, trustedProxies []string) error {
|
||||
if err := engine.SetTrustedProxies(trustedProxies); err != nil {
|
||||
return fmt.Errorf("invalid TRUSTED_PROXIES: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user