package telegram

import (
	"fmt"
	"regexp"

	tgbotapi "github.com/BiRabittoh/telegram-bot-api/v5"
)

type Replacer struct {
	Regex  *regexp.Regexp
	Format string
}

var parseMode = "markdown"
var linkMessage = "[🔗](%s) Da %s."
var regexFlags = "(?i)(?m)"
var replacers = []Replacer{
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:(?:https?:)?\/\/)?(?:(?:www|m)\.)?(?:(?:youtube(?:-nocookie)?\.com|youtu.be))(?:\/(?:[\w\-]+\?v=|embed\/|live\/|v\/|shorts\/)?)([\w\-]+)`),
		Format: "https://y.outube.duckdns.org/%s",
	},
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:https?:\/\/)(?:www\.)?twitter\.com\/(?:#!\/)?(.*)\/status(?:es)?\/([^\/\?\s]+)`),
		Format: "https://fxtwitter.com/%s/status/%s",
	},
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:https?:\/\/)?(?:www\.)?x\.com\/(?:#!\/)?(.*)\/status(?:es)?\/([^\/\?\s]+)`),
		Format: "https://fixupx.com/%s/status/%s",
	},
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:https?:\/\/)?(?:www\.)?instagram\.com\/(reel|p)\/([A-Za-z0-9_-]{11})[\/\?\w=&]*`),
		Format: "https://ddinstagram.com/%s/%s",
	},
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:https?:\/\/)?(?:(?:www)|(?:vm))?\.?tiktok\.com\/@([\w\d_.]+)\/(?:video)\/(\d+)`),
		Format: "https://www.vxtiktok.com/@%s/video/%s",
	},
	{
		Regex:  regexp.MustCompile(regexFlags + `(?:https?:\/\/)?(?:(?:www)|(?:vm))?\.?tiktok\.com\/([\w]+)\/?`),
		Format: "https://vm.vxtiktok.com/%s/",
	},
}

func parseText(message string) []string {
	links := []string{}

	for _, replacer := range replacers {
		foundMatches := replacer.Regex.FindStringSubmatch(message)

		if len(foundMatches) == 0 {
			continue
		}
		captureGroups := foundMatches[1:]

		var formatArgs []interface{}
		for _, match := range captureGroups {
			if match != "" {
				formatArgs = append(formatArgs, match)
			}
		}

		formatted := fmt.Sprintf(replacer.Format, formatArgs...)
		links = append(links, formatted)
	}
	return links
}

func getUserMention(user tgbotapi.User) string {
	return fmt.Sprintf("[@%s](tg://user?id=%d)", user.UserName, user.ID)
}

func handleLinks(bot *tgbotapi.BotAPI, message *tgbotapi.Message) {
	links := []string{}

	if message.Text != "" {
		textLinks := parseText(message.Text)
		links = append(links, textLinks...)
	}

	if message.Caption != "" {
		captionLinks := parseText(message.Caption)
		links = append(links, captionLinks...)
	}

	if len(links) == 0 {
		return
	}

	user := getUserMention(*message.From)

	for _, link := range links {
		text := fmt.Sprintf(linkMessage, link, user)
		msg := tgbotapi.NewMessage(message.Chat.ID, text)
		msg.MessageThreadID = message.MessageThreadID
		msg.ParseMode = parseMode
		msg.ReplyMarkup = inlineKeyboardFeedback
		bot.Send(msg)
	}

}