package qol

import (
	"fmt"
	"net/url"
	"path/filepath"
	"strings"
	"time"
)

func GenerateOutputPaths(outputDir, upstream string, dnssec, keepAlive bool) (csvPath, pcapPath string) {
	proto := DetectProtocol(upstream)
	cleanServer := cleanServerName(upstream)

	// Create date-based subdirectory
	date := time.Now().Format("2006-01-02")
	timestamp := time.Now().Format("150405")

	// Organize hierarchically: resolver/date/filename
	subDir := filepath.Join(outputDir, cleanServer, date)

	// Create simple filename
	base := proto

	// Add flags if enabled
	var flags []string
	if dnssec {
		flags = append(flags, "dnssec")
	}
	if keepAlive {
		flags = append(flags, "persist")
	}

	if len(flags) > 0 {
		base = fmt.Sprintf("%s-%s", base, strings.Join(flags, "-"))
	}

	// Add timestamp
	filename := fmt.Sprintf("%s-%s", base, timestamp)

	return filepath.Join(subDir, filename+".csv"),
		filepath.Join(subDir, filename+".pcap")
}

func cleanServerName(server string) string {
	// Map common servers to readable names
	serverMap := map[string]string{
		"1.1.1.1":               "cloudflare",
		"1.0.0.1":               "cloudflare",
		"cloudflare-dns.com":    "cloudflare",
		"one.one.one.one":       "cloudflare",
		"8.8.8.8":               "google",
		"8.8.4.4":               "google",
		"dns.google":            "google",
		"dns.google.com":        "google",
		"9.9.9.9":               "quad9",
		"149.112.112.112":       "quad9",
		"dns.quad9.net":         "quad9",
		"dns10.quad9.net":       "quad9",
		"208.67.222.222":        "opendns",
		"208.67.220.220":        "opendns",
		"resolver1.opendns.com": "opendns",
		"94.140.14.14":          "adguard",
		"94.140.15.15":          "adguard",
		"dns.adguard.com":       "adguard",
		"dns.adguard-dns.com":   "adguard",
	}

	serverName := ""
	cleanedUrl, err := url.Parse(server)
	if err != nil {
		serverName = server
	} else {
		serverName = cleanedUrl.Hostname()
	}

	// Check if we have a mapping
	if shortName, exists := serverMap[serverName]; exists {
		return shortName
	}

	return serverName
}

func DetectProtocol(upstream string) string {
	if strings.Contains(upstream, "://") {
		u, err := url.Parse(upstream)
		if err == nil && u.Scheme != "" {
			return strings.ToLower(u.Scheme)
		}
	}
	return "do53"
}