[优化] 新增 GeoIP 测试功能，更新相关 API 路由和前端组件

atsf_server/utils/geoip/geoip.go

@@ -16,6 +16,7 @@ import (
 var CurrentProvider GeoIPService
 var geoCache *providerCache
 var providerMutex sync.RWMutex
+var providerFactory = newProvider
 
 const (
 	ProviderDisabled = "disabled"
@@ -111,7 +112,7 @@ func GetRegionUnicodeEmoji(isoCode string) string {
 
 func InitGeoIP() {
 	providerName := normalizeProvider(common.GeoIPProvider)
-	nextProvider, err := newProvider(providerName)
+	nextProvider, err := providerFactory(providerName)
 	if err != nil {
 		slog.Error("initialize GeoIP provider failed", "provider", providerName, "error", err)
 		nextProvider = &EmptyProvider{}
@@ -142,6 +143,24 @@ func GetGeoInfo(ip net.IP) (*GeoInfo, error) {
 	return info, err
 }
 
+func LookupGeoInfoWithProvider(providerName string, ip net.IP) (*GeoInfo, error) {
+	if ip == nil {
+		return nil, fmt.Errorf("IP address cannot be nil")
+	}
+
+	provider, err := providerFactory(normalizeProvider(providerName))
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		if closeErr := provider.Close(); closeErr != nil {
+			slog.Warn("close temporary GeoIP provider failed", "provider", provider.Name(), "error", closeErr)
+		}
+	}()
+
+	return provider.GetGeoInfo(ip)
+}
+
 func UpdateDatabase() error {
 	err := getProvider().UpdateDatabase()
 	if err == nil {
@@ -210,3 +229,15 @@ func getProvider() GeoIPService {
 func float64Pointer(value float64) *float64 {
 	return &value
 }
+
+func ProviderFactoryForTest() func(string) (GeoIPService, error) {
+	return providerFactory
+}
+
+func SetProviderFactoryForTest(factory func(string) (GeoIPService, error)) {
+	if factory == nil {
+		providerFactory = newProvider
+		return
+	}
+	providerFactory = factory
+}

atsf_server/utils/geoip/geoip_test.go

@@ -79,3 +79,21 @@ func TestIsValidProvider(t *testing.T) {
 		}
 	}
 }
+
+func TestLookupGeoInfoWithProviderUsesTemporaryProvider(t *testing.T) {
+	previousFactory := providerFactory
+	providerFactory = func(provider string) (GeoIPService, error) {
+		return &fakeProvider{}, nil
+	}
+	defer func() {
+		providerFactory = previousFactory
+	}()
+
+	info, err := LookupGeoInfoWithProvider("ipinfo", net.ParseIP("8.8.8.8"))
+	if err != nil {
+		t.Fatalf("expected lookup to succeed, got %v", err)
+	}
+	if info == nil || info.ISOCode != "CN" || info.Name != "China" {
+		t.Fatalf("unexpected geo info: %#v", info)
+	}
+}