package main

import (
	"encoding/json"
	"fmt"
	"net/http"
	"regexp"
	"strings"

	"github.com/nbd-wtf/go-nostr"
	"github.com/pelletier/go-toml"
)

var kindNames = map[int]string{
	0:     "profile metadata",
	1:     "text note",
	2:     "relay recommendation",
	3:     "contact list",
	4:     "encrypted direct message",
	5:     "event deletion",
	6:     "repost",
	7:     "reaction",
	8:     "badge award",
	40:    "channel creation",
	41:    "channel metadata",
	42:    "channel message",
	43:    "channel hide message",
	44:    "channel mute user",
	1984:  "report",
	9735:  "zap",
	9734:  "zap request",
	10002: "relay list",
	30008: "profile badges",
	30009: "badge definition",
	30078: "app-specific data",
	30023: "article",
}

func generateClientList(code string, event *nostr.Event) []map[string]string {
	if strings.HasPrefix(code, "nevent") || strings.HasPrefix(code, "note") {
		return []map[string]string{
			{"name": "native client", "url": "nostr:" + code},
			{"name": "Snort", "url": "https://Snort.social/e/" + code},
			{"name": "Coracle", "url": "https://coracle.social/" + code},
			{"name": "Satellite", "url": "https://satellite.earth/thread/" + event.ID},
			{"name": "Iris", "url": "https://iris.to/" + code},
			{"name": "Yosup", "url": "https://yosup.app/thread/" + event.ID},
			{"name": "Nostr.band", "url": "https://nostr.band/" + code},
			{"name": "Primal", "url": "https://primal.net/thread/" + event.ID},
			{"name": "Nostribe", "url": "https://www.nostribe.com/post/" + event.ID},
			{"name": "Nostrid", "url": "https://web.nostrid.app/note/" + event.ID},
		}
	} else if strings.HasPrefix(code, "npub") || strings.HasPrefix(code, "nprofile") {
		return []map[string]string{
			{"name": "native client", "url": "nostr:" + code},
			{"name": "Snort", "url": "https://snort.social/p/" + code},
			{"name": "Coracle", "url": "https://coracle.social/" + code},
			{"name": "Satellite", "url": "https://satellite.earth/@" + code},
			{"name": "Iris", "url": "https://iris.to/" + code},
			{"name": "Yosup", "url": "https://yosup.app/profile/" + event.PubKey},
			{"name": "Nostr.band", "url": "https://nostr.band/" + code},
			{"name": "Primal", "url": "https://primal.net/profile/" + event.PubKey},
			{"name": "Nostribe", "url": "https://www.nostribe.com/profile/" + event.PubKey},
			{"name": "Nostrid", "url": "https://web.nostrid.app/account/" + event.PubKey},
		}
	} else if strings.HasPrefix(code, "naddr") {
		return []map[string]string{
			{"name": "native client", "url": "nostr:" + code},
			{"name": "habla", "url": "https://habla.news/a/" + code},
			{"name": "blogstack", "url": "https://blogstack.io/" + code},
		}
	} else {
		return []map[string]string{
			{"name": "native client", "url": "nostr:" + code},
		}
	}
}

func mergeMaps[K comparable, V any](m1 map[K]V, m2 map[K]V) map[K]V {
	for k, v := range m2 {
		m1[k] = v
	}
	return m1
}

func prettyJsonOrRaw(j string) string {
	var parsedContent any
	if err := json.Unmarshal([]byte(j), &parsedContent); err == nil {
		if t, err := toml.Marshal(parsedContent); err == nil && len(t) > 0 {
			return string(t)
		}
	}
	return j
}

func getPreviewStyle(r *http.Request) string {
	ua := strings.ToLower(r.Header.Get("User-Agent"))
	accept := r.Header.Get("Accept")

	switch {
	case strings.Contains(ua, "telegrambot"):
		return "telegram"
	case strings.Contains(ua, "twitterbot"):
		return "twitter"
	case strings.Contains(ua, "mattermost"):
		return "mattermost"
	case strings.Contains(ua, "slack"):
		return "slack"
	case strings.Contains(ua, "discord"):
		return "discord"
	case strings.Contains(ua, "whatsapp"):
		return "whatsapp"
	case strings.Contains(accept, "text/html"):
		return ""
	default:
		return "unknown"
	}
}

func BasicFormatting(input string) string {
	lines := strings.Split(input, "\n")

	var processedLines []string
	for _, line := range lines {
		processedLine := ReplaceURLsWithTags(line)
		processedLines = append(processedLines, processedLine)
	}

	return strings.Join(processedLines, "<br/>")
}

func ReplaceURLsWithTags(line string) string {

	// Match and replace image URLs with <img> tags
	imageExtensions := []string{".jpg", ".jpeg", ".png", ".webp", ".gif"}
	for _, extension := range imageExtensions {
		regexPattern := fmt.Sprintf(`\s*(https?://\S+%s)\s*`, extension)
		regex := regexp.MustCompile(regexPattern)
		matches := regex.FindAllString(line, -1)
		for _, match := range matches {
			imgTag := fmt.Sprintf(`<img src="%s" alt="">`, strings.ReplaceAll(match, "\n", ""))
			line = strings.ReplaceAll(line, match, imgTag)
			return line
		}
	}

	// Match and replace mp4 URLs with <video> tag
	videoExtensions := []string{".mp4", ".ogg", ".webm"}
	for _, extension := range videoExtensions {
		regexPattern := fmt.Sprintf(`\s*(https?://\S+%s)\s*`, extension)
		regex := regexp.MustCompile(regexPattern)
		matches := regex.FindAllString(line, -1)
		for _, match := range matches {
			videoTag := fmt.Sprintf(`<video controls width="100%%"><source src="%s"></video>`, strings.ReplaceAll(match, "\n", ""))
			line = strings.ReplaceAll(line, match, videoTag)
			return line
		}
	}

	// Match and replace npup1, nprofile1, note1, nevent1, etc
	nostrRegexPattern := `\s*nostr:((npub|note|nevent|nprofile)1[a-z0-9]+)\s*`
	nostrRegex := regexp.MustCompile(nostrRegexPattern)
	line = nostrRegex.ReplaceAllStringFunc(line, func(match string) string {
		submatch := nostrRegex.FindStringSubmatch(match)
		if len(submatch) < 2 {
			return match
		}
		capturedGroup := submatch[1]
		first6 := capturedGroup[:6]
		last6 := capturedGroup[len(capturedGroup)-6:]
		replacement := fmt.Sprintf(`<a href="/%s">%s</a>`, capturedGroup, first6+"…"+last6)
		return replacement
	})

	// Match and replace other URLs with <a> tags
	hrefRegexPattern := `\S*(https?://\S+)\S*`
	hrefRegex := regexp.MustCompile(hrefRegexPattern)
	line = hrefRegex.ReplaceAllString(line, `<a href="$1">$1</a>`)
	return line
}