mirror of
https://github.com/woodchen-ink/proxy-go.git
synced 2025-07-18 00:21:56 +08:00
136 lines
3.8 KiB
Go
136 lines
3.8 KiB
Go
package config
|
||
|
||
import (
|
||
"strings"
|
||
)
|
||
|
||
type Config struct {
|
||
MAP map[string]PathConfig `json:"MAP"` // 路径映射配置
|
||
Compression CompressionConfig `json:"Compression"`
|
||
Security SecurityConfig `json:"Security"` // 安全配置
|
||
}
|
||
|
||
type PathConfig struct {
|
||
DefaultTarget string `json:"DefaultTarget"` // 默认目标URL
|
||
ExtensionMap []ExtRuleConfig `json:"ExtensionMap"` // 扩展名映射规则
|
||
ExtRules []ExtensionRule `json:"-"` // 内部使用，存储处理后的扩展名规则
|
||
RedirectMode bool `json:"RedirectMode"` // 是否使用302跳转模式
|
||
}
|
||
|
||
// ExtensionRule 表示一个扩展名映射规则（内部使用）
|
||
type ExtensionRule struct {
|
||
Extensions []string // 支持的扩展名列表
|
||
Target string // 目标服务器
|
||
SizeThreshold int64 // 最小阈值
|
||
MaxSize int64 // 最大阈值
|
||
RedirectMode bool // 是否使用302跳转模式
|
||
Domains []string // 支持的域名列表，为空表示匹配所有域名
|
||
}
|
||
|
||
type CompressionConfig struct {
|
||
Gzip CompressorConfig `json:"Gzip"`
|
||
Brotli CompressorConfig `json:"Brotli"`
|
||
}
|
||
|
||
type CompressorConfig struct {
|
||
Enabled bool `json:"Enabled"`
|
||
Level int `json:"Level"`
|
||
}
|
||
|
||
type SecurityConfig struct {
|
||
IPBan IPBanConfig `json:"IPBan"` // IP封禁配置
|
||
}
|
||
|
||
type IPBanConfig struct {
|
||
Enabled bool `json:"Enabled"` // 是否启用IP封禁
|
||
ErrorThreshold int `json:"ErrorThreshold"` // 404错误阈值
|
||
WindowMinutes int `json:"WindowMinutes"` // 统计窗口时间（分钟）
|
||
BanDurationMinutes int `json:"BanDurationMinutes"` // 封禁时长（分钟）
|
||
CleanupIntervalMinutes int `json:"CleanupIntervalMinutes"` // 清理间隔（分钟）
|
||
}
|
||
|
||
// 扩展名映射配置结构
|
||
type ExtRuleConfig struct {
|
||
Extensions string `json:"Extensions"` // 逗号分隔的扩展名
|
||
Target string `json:"Target"` // 目标服务器
|
||
SizeThreshold int64 `json:"SizeThreshold"` // 最小阈值
|
||
MaxSize int64 `json:"MaxSize"` // 最大阈值
|
||
RedirectMode bool `json:"RedirectMode"` // 是否使用302跳转模式
|
||
Domains string `json:"Domains"` // 逗号分隔的域名列表，为空表示匹配所有域名
|
||
}
|
||
|
||
// 处理扩展名映射的方法
|
||
func (p *PathConfig) ProcessExtensionMap() {
|
||
p.ExtRules = nil
|
||
|
||
if p.ExtensionMap == nil {
|
||
return
|
||
}
|
||
|
||
// 处理扩展名规则
|
||
for _, rule := range p.ExtensionMap {
|
||
extRule := ExtensionRule{
|
||
Target: rule.Target,
|
||
SizeThreshold: rule.SizeThreshold,
|
||
MaxSize: rule.MaxSize,
|
||
RedirectMode: rule.RedirectMode,
|
||
}
|
||
|
||
// 处理扩展名列表
|
||
for _, ext := range strings.Split(rule.Extensions, ",") {
|
||
ext = strings.TrimSpace(ext)
|
||
if ext != "" {
|
||
extRule.Extensions = append(extRule.Extensions, ext)
|
||
}
|
||
}
|
||
|
||
// 处理域名列表
|
||
if rule.Domains != "" {
|
||
for _, domain := range strings.Split(rule.Domains, ",") {
|
||
domain = strings.TrimSpace(domain)
|
||
if domain != "" {
|
||
extRule.Domains = append(extRule.Domains, domain)
|
||
}
|
||
}
|
||
}
|
||
|
||
if len(extRule.Extensions) > 0 {
|
||
p.ExtRules = append(p.ExtRules, extRule)
|
||
}
|
||
}
|
||
}
|
||
|
||
// GetProcessedExtTarget 快速获取扩展名对应的目标URL，如果存在返回true
|
||
func (p *PathConfig) GetProcessedExtTarget(ext string) (string, bool) {
|
||
if p.ExtRules == nil {
|
||
return "", false
|
||
}
|
||
|
||
for _, rule := range p.ExtRules {
|
||
for _, e := range rule.Extensions {
|
||
if e == ext {
|
||
return rule.Target, true
|
||
}
|
||
}
|
||
}
|
||
|
||
return "", false
|
||
}
|
||
|
||
// GetProcessedExtRule 获取扩展名对应的完整规则信息，包括RedirectMode
|
||
func (p *PathConfig) GetProcessedExtRule(ext string) (*ExtensionRule, bool) {
|
||
if p.ExtRules == nil {
|
||
return nil, false
|
||
}
|
||
|
||
for _, rule := range p.ExtRules {
|
||
for _, e := range rule.Extensions {
|
||
if e == ext {
|
||
return &rule, true
|
||
}
|
||
}
|
||
}
|
||
|
||
return nil, false
|
||
}
|