DebTik/internal/firewall/ipset.go

package firewall

import (
	"os/exec"
	"strings"
)

// SyncAddressLists updates Linux ipsets to match the configured AddressList entries and referenced lists
func SyncAddressLists(entries []AddressListEntry, referencedLists ...string) error {
	// Group addresses by list name
	listMap := make(map[string][]string)
	for _, e := range entries {
		if e.Disabled || e.Name == "" || e.Address == "" {
			continue
		}
		listMap[e.Name] = append(listMap[e.Name], e.Address)
	}

	// Ensure any custom list explicitly referenced in rules exists even if empty
	for _, ref := range referencedLists {
		ref = strings.TrimPrefix(ref, "!")
		if ref != "" && !strings.HasPrefix(ref, "geo:") && !strings.HasPrefix(ref, "geo_") {
			if _, ok := listMap[ref]; !ok {
				listMap[ref] = []string{}
			}
		}
	}

	// Clean up any mk_ sets that no longer exist or are now empty (ignoring mk_geo_ sets managed by GeoIP)
	if out, err := exec.Command("ipset", "list", "-name").Output(); err == nil {
		lines := strings.Split(string(out), "\n")
		for _, line := range lines {
			line = strings.TrimSpace(line)
			if strings.HasPrefix(line, "mk_") && !strings.HasPrefix(line, "mk_geo_") {
				name := strings.TrimPrefix(line, "mk_")
				if _, ok := listMap[name]; !ok {
					_ = exec.Command("ipset", "flush", line).Run()
					_ = exec.Command("ipset", "destroy", line).Run()
				}
			}
		}
	}

	for name, addrs := range listMap {
		setName := "mk_" + strings.ReplaceAll(name, " ", "_")
		// Create set if not exists
		_ = exec.Command("ipset", "create", setName, "hash:net", "-exist").Run()
		_ = exec.Command("ipset", "flush", setName).Run()

		for _, a := range addrs {
			_ = exec.Command("ipset", "add", setName, a, "-exist").Run()
		}
	}

	return nil
}

// GetIPSetName returns the internal ipset name for an address list
func GetIPSetName(listName string) string {
	if listName == "" {
		return ""
	}
	clean := strings.TrimPrefix(listName, "!")
	if strings.HasPrefix(clean, "geo:") {
		code := strings.ToUpper(strings.TrimPrefix(clean, "geo:"))
		return "mk_geo_" + code
	}
	if strings.HasPrefix(clean, "geo_") {
		return "mk_" + clean
	}
	return "mk_" + strings.ReplaceAll(clean, " ", "_")
}

// GetGeoIPSetName returns the internal ipset name for a Geo-Location country or preset
func GetGeoIPSetName(code string) string {
	if code == "" {
		return ""
	}
	clean := strings.TrimPrefix(code, "!")
	clean = strings.TrimPrefix(clean, "geo:")
	clean = strings.TrimPrefix(clean, "geo_")
	return "mk_geo_" + strings.ToUpper(clean)
}