package domains

import (
	v1 "billionmail-core/api/domains/v1"
	"billionmail-core/internal/service/public"
	"context"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/text/gregex"
	"net"
	"strings"
)

// ValidateARecord checks if the given A record is valid
func ValidateARecord(record v1.DNSRecord) bool {
	if strings.ToUpper(record.Type) != "A" && strings.ToUpper(record.Type) != "AAAA" {
		return false
	}

	// Query A records
	ips, err := net.LookupIP(record.Host)
	if err != nil {
		return false
	}

	// Check if any IP matches the expected server IP
	for _, ip := range ips {
		if ip.String() == record.Value {
			return true
		}
	}

	return false
}

// ValidateTXTRecord checks if the given TXT record is valid
func ValidateTXTRecord(record v1.DNSRecord, domain string) bool {
	if strings.ToUpper(record.Type) != "TXT" {
		return false
	}

	if record.Host != "@" {
		domain = strings.TrimSuffix(record.Host, ".") + "." + domain
	}

	cacheKey := "txt_records_" + domain
	txtRecords := make([]string, 0)
	cacheTxtRecords := public.GetCache(cacheKey)

	if cacheTxtRecords != nil {
		txtRecords = cacheTxtRecords.([]string)
		g.Log().Debug(context.Background(), "get txt records from cache", cacheTxtRecords)
	} else {
		// Query TXT records
		var err error
		txtRecords, err = net.LookupTXT(domain)
		if err != nil {
			// g.Log().Error(context.Background(), "query txt records failed", err)
			return false
		}
		g.Log().Debug(context.Background(), "query txt records success", txtRecords)
		// Store TXT records in cache
		public.SetCache(cacheKey, txtRecords, 60)
	}

	vl := strings.ToLower(record.Value)

	switch {
	case strings.HasPrefix(vl, "v=dmarc"):
		// Match DMARC Record
		for _, txt := range txtRecords {
			txtl := strings.ToLower(txt)
			if strings.HasPrefix(txtl, "v=dmarc") && (strings.Contains(vl, "p=quarantine") || strings.Contains(vl, "q=reject")) && strings.Contains(vl, "rua=") {
				return true
			}
		}
	case strings.HasPrefix(vl, "v=dkim"):
		// Match DKIM Record
		break
	case strings.HasPrefix(vl, "v=spf"):
		// Match SPF Record
		addrs := make([]string, 0)

		ms, err := gregex.MatchAllString("[+-]?((?:ip4|ip6|include):[^ ]+)", vl)

		if err != nil {
			return false
		}

		for _, m := range ms {
			if len(m) < 2 {
				continue
			}

			addrs = append(addrs, m[1])
		}

		for _, txt := range txtRecords {
			txtl := strings.TrimSpace(strings.ToLower(txt))
			if strings.HasPrefix(txtl, "v=spf") && (strings.HasSuffix(txtl, "~all") || strings.HasSuffix(txtl, "-all")) {
				spfPassed := true
				for _, addr := range addrs {
					if !strings.Contains(txtl, addr) {
						spfPassed = false
						break
					}
				}

				if spfPassed {
					return true
				}
			}
		}
	}

	// Check if any TXT record matches the expected value
	for _, txt := range txtRecords {
		if txt == record.Value {
			return true
		}
	}

	return false
}

// ValidateMXRecord checks if the given MX record is valid
func ValidateMXRecord(record v1.DNSRecord, domain string, aRecordHosts ...string) bool {
	if strings.ToUpper(record.Type) != "MX" {
		return false
	}

	if record.Host != "@" {
		domain = strings.TrimSuffix(record.Host, ".") + "." + domain
	}

	// Query A records for the domain
	domains := append([]string{domain}, aRecordHosts...)

	// Check if any MX record matches the expected value
	aRecordHosts = append([]string{record.Value}, aRecordHosts...)

	for _, d := range domains {
		// Query MX records
		mxRecords, err := net.LookupMX(d)
		if err != nil {
			// g.Log().Error(context.Background(), "query mx record failed", err)
			return false
		}

		g.Log().Debug(context.Background(), "query mx record success", mxRecords)

		for _, mx := range mxRecords {
			mxHost := strings.TrimSuffix(mx.Host, ".")
			for _, aRecordHost := range aRecordHosts {
				if aRecordHost == mxHost {
					return true
				}
			}
		}
	}

	return false
}

// ValidatePTRRecord checks if the given PTR record is valid
func ValidatePTRRecord(record v1.DNSRecord) bool {
	if strings.ToUpper(record.Type) != "PTR" {
		return false
	}

	// Query PTR records
	ptrRecords, err := net.LookupAddr(record.Host)
	if err != nil {
		return false
	}

	g.Log().Debug(context.Background(), "query ptr record success", ptrRecords)

	// Check if any PTR record matches the expected value
	for _, ptr := range ptrRecords {
		if strings.TrimSuffix(ptr, ".") == record.Value {
			return true
		}
	}

	return false
}