OpenFlare/internal/apps/openflare/geoip/runtime.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package geoip

import (
	"context"
	"io/fs"
	"os"
	"path/filepath"
	"strings"
	"sync"

	"github.com/Rain-kl/Wavelet/internal/apps/agent/geoipdata"
	"github.com/Rain-kl/Wavelet/internal/model"
	pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
	"github.com/Rain-kl/Wavelet/pkg/logger"
)

const (
	serverMMDBRelativePath = "data/GeoLite2-Country.mmdb"
	serverMMDBDirPerm      = 0o750
	serverMMDBFilePerm     = 0o644
)

var (
	runtimeOnce       sync.Once
	runtimeInitErr    error
	currentProviderMu sync.RWMutex
	currentProvider   string
)

// EnsureRuntimeProvider loads OpenFlare options once and configures pkg/geoip.
func EnsureRuntimeProvider(ctx context.Context) error {
	runtimeOnce.Do(func() {
		if err := model.InitOptionMap(ctx); err != nil {
			runtimeInitErr = err
			return
		}
		runtimeInitErr = applyProviderFromModel()
	})
	return runtimeInitErr
}

// RefreshRuntimeProvider reapplies GeoIPProvider after option updates.
func RefreshRuntimeProvider(ctx context.Context) error {
	if err := model.InitOptionMap(ctx); err != nil {
		return err
	}
	return applyProviderFromModel()
}

func applyProviderFromModel() error {
	model.OptionMapRWMutex.RLock()
	provider := strings.TrimSpace(model.GeoIPProvider)
	model.OptionMapRWMutex.RUnlock()
	return ApplyProvider(provider)
}

// ApplyProvider switches the process-wide GeoIP backend.
func ApplyProvider(provider string) error {
	normalized := strings.TrimSpace(strings.ToLower(provider))
	if normalized == "" {
		normalized = pkggeoip.ProviderDisabled
	}

	currentProviderMu.Lock()
	if currentProvider == normalized {
		currentProviderMu.Unlock()
		return nil
	}
	currentProvider = normalized
	currentProviderMu.Unlock()

	if normalized == pkggeoip.ProviderMaxMind {
		path, err := ensureServerMMDB()
		if err != nil {
			logger.WarnF(context.Background(), "[GeoIP] seed MaxMind database failed: %v", err)
		}
		if path != "" {
			pkggeoip.GeoIPFilePath = path
		}
	}

	pkggeoip.InitGeoIP(normalized)
	return nil
}

func ensureServerMMDB() (string, error) {
	path, err := filepath.Abs(serverMMDBRelativePath)
	if err != nil {
		return "", err
	}
	if _, err := os.Stat(path); err == nil {
		return path, nil
	}
	if !os.IsNotExist(err) {
		return "", err
	}

	data, err := fs.ReadFile(geoipdata.FS, geoipdata.DefaultMMDBName)
	if err != nil {
		return "", err
	}
	if err := os.MkdirAll(filepath.Dir(path), serverMMDBDirPerm); err != nil {
		return "", err
	}
	if err := os.WriteFile(path, data, serverMMDBFilePerm); err != nil { //nolint:gosec // world-readable mmdb
		return "", err
	}
	return path, nil
}

// ResetRuntimeForTest clears lazy-init state for unit tests.
func ResetRuntimeForTest() {
	runtimeOnce = sync.Once{}
	runtimeInitErr = nil
	currentProviderMu.Lock()
	currentProvider = ""
	currentProviderMu.Unlock()
}