https-socks5-http/pkg/proxy/proxylist.go

package proxy

import (
	"bufio"
	"net"
	"os"
	"path/filepath"
	"strings"
	"sync"

	"s5/pkg/logger"
)

// ProxyList 直连域名管理器
type ProxyList struct {
	patterns []string     // 域名匹配模式
	mu       sync.RWMutex // 读写锁
}

// NewProxyList 创建新的直连域名管理器
func NewProxyList(filename string) *ProxyList {
	dl := &ProxyList{}
	if err := dl.Load(filename); err != nil {
		log := logger.Get()
		log.Errorf("[PROXY] 加载直连域名列表失败: %v", err)
	}
	return dl
}

// Load 加载直连域名列表
func (dl *ProxyList) Load(filename string) error {
	log := logger.Get()
	log.Debugf("[PROXY] =====================================================")
	log.Debugf("[PROXY] 开始加载代理域名列表...")
	log.Debugf("[PROXY] 配置文件: %s", filename)

	// 获取绝对路径
	absPath, err := filepath.Abs(filename)
	if err != nil {
		log.Errorf("[PROXY] 解析文件路径失败: %v", err)
		log.Debugf("[PROXY] =====================================================")
		return err
	}
	log.Debugf("[PROXY] 绝对路径: %s", absPath)

	// 检查文件是否存在
	if _, err := os.Stat(absPath); os.IsNotExist(err) {
		log.Errorf("[PROXY] 文件不存在: %s", absPath)
		log.Debugf("[PROXY] 将创建默认配置文件")

		// 创建默认配置文件
		if err := dl.createDefaultConfig(absPath); err != nil {
			log.Errorf("[PROXY] 创建默认配置文件失败: %v", err)
			log.Debugf("[PROXY] =====================================================")
			return err
		}
	}

	file, err := os.Open(absPath)
	if err != nil {
		log.Errorf("[PROXY] 打开文件失败: %v", err)
		log.Debugf("[PROXY] =====================================================")
		return err
	}
	defer file.Close()

	var patterns []string
	var skipped, added int
	scanner := bufio.NewScanner(file)
	lineNum := 0

	log.Debugf("[PROXY] 开始解析规则:")
	for scanner.Scan() {
		lineNum++
		line := scanner.Text()
		pattern := strings.TrimSpace(line)

		if pattern == "" {
			log.Debugf("[PROXY]   行 %d: 空行，已跳过", lineNum)
			skipped++
			continue
		}
		if strings.HasPrefix(pattern, "#") {
			log.Debugf("[PROXY]   行 %d: 注释行，已跳过: %s", lineNum, line)
			skipped++
			continue
		}

		// 验证规则格式
		if strings.HasPrefix(pattern, "*.") {
			log.Debugf("[PROXY]   行 %d: 添加通配符规则: %s", lineNum, pattern)
		} else if strings.Contains(pattern, "*") {
			log.Warnf("[PROXY]   行 %d: 不支持的通配符位置，规则可能无法正常工作: %s", lineNum, pattern)
		} else {
			log.Debugf("[PROXY]   行 %d: 添加精确匹配规则: %s", lineNum, pattern)
		}

		patterns = append(patterns, pattern)
		added++
	}

	if err := scanner.Err(); err != nil {
		log.Errorf("[PROXY] 读取文件失败: %v", err)
		log.Debugf("[PROXY] =====================================================")
		return err
	}

	dl.mu.Lock()
	dl.patterns = patterns
	dl.mu.Unlock()

	log.Debugf("[PROXY] 规则统计:")
	log.Debugf("[PROXY]   - 总行数: %d", lineNum)
	log.Debugf("[PROXY]   - 跳过行数: %d", skipped)
	log.Debugf("[PROXY]   - 有效规则: %d", added)
	log.Infof("[PROXY] 代理域名列表加载完成")
	log.Debugf("[PROXY] =====================================================")

	return nil
}

// createDefaultConfig 创建默认配置文件
func (dl *ProxyList) createDefaultConfig(path string) error {
	defaultConfig := `# 代理域名列表
# 每行一个域名，支持通配符 *
# 以 # 开头的行为注释

# 需要代理的域名
*.google.com
*.youtube.com
*.facebook.com
*.twitter.com
*.github.com
*.githubusercontent.com
*.cloudflare.com
*.wikipedia.org
*.medium.com
*.reddit.com

# 国外服务
*.netflix.com
*.amazon.com
*.microsoft.com
*.apple.com
*.dropbox.com
*.telegram.org
*.whatsapp.com
*.instagram.com
*.twitch.tv
*.spotify.com

# 开发相关
*.docker.com
*.npmjs.com
*.pypi.org
*.rubygems.org
*.maven.org`

	return os.WriteFile(path, []byte(defaultConfig), 0644)
}

// Match 检查域名是否匹配直连列表
func (dl *ProxyList) Match(domain string) bool {
	dl.mu.RLock()
	defer dl.mu.RUnlock()

	log := logger.Get()
	log.Debugf("[PROXY] =====================================================")
	log.Debugf("[PROXY] 开始检查域名: %s", domain)

	// 处理特殊情况
	if domain == "" {
		log.Warnf("[PROXY] 域名为空")
		log.Debugf("[PROXY] =====================================================")
		return false
	}

	domain = strings.ToLower(domain)
	log.Debugf("[PROXY] 规则匹配:")
	log.Debugf("[PROXY]   - 原始域名: %s", domain)
	log.Debugf("[PROXY]   - 处理后域名: %s", domain)

	// 检查是否是IP地址
	if ip := net.ParseIP(domain); ip != nil {
		log.Debugf("[PROXY]   - 域名是IP地址: %s", ip)
		log.Debugf("[PROXY] =====================================================")
		return false
	}

	// 遍历所有规则进行匹配
	for i, pattern := range dl.patterns {
		log.Debugf("[PROXY]   检查规则 #%d: %s", i+1, pattern)
		if matchDomain(domain, pattern) {
			log.Debugf("[PROXY]   - 匹配成功: %s -> %s", domain, pattern)
			log.Debugf("[PROXY] =====================================================")
			return true
		}
	}

	log.Debugf("[PROXY]   - 未匹配任何规则")
	log.Debugf("[PROXY] =====================================================")
	return false
}

// matchDomain 检查域名是否匹配模式
func matchDomain(domain, pattern string) bool {
	if pattern[0] == '*' {
		suffix := pattern[1:]
		matched := strings.HasSuffix(domain, suffix)
		if matched {
			return true
		}
	}
	return domain == pattern
}