package lib

import (
	"fmt"
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
	"log"
	"math/rand"
	"net/url"
	"strconv"
	"time"
)

var (
	bot *tgbotapi.BotAPI
	err error
)
var userState = make(map[string]int)
var linkState = make(map[int]string)

func init() {
	// 7632434434:AAG-Ac9RlLTKHnGtOSUe8_SUD8j_nDWz_AU
	// bot, err = tgbotapi.NewBotAPI("7322066653:AAG3qYWH5jzAW8ZtdFEWg0Fjy0Il5FLOsg8")
	bot, err = tgbotapi.NewBotAPI("7455883553:AAEwoqr6EAUZehhpJUWkhPiu4r3W-fngelE")
	if err != nil {
		log.Panic(err)
	}
	bot.Debug = true
	log.Printf("Authorized on account %s", bot.Self.UserName)
	//db.Test()
}

func number1() int {
	rand.Seed(time.Now().UnixNano())

	// 生成一个0到100之间的随机整数
	randomNumber := rand.Intn(100) // 生成一个 [0, 100) 范围内的随机数
	return randomNumber
}
func number2() int {
	rand.Seed(time.Now().UnixNano())

	// 生成一个0到100之间的随机整数
	randomNumber := rand.Intn(10) // 生成一个 [0, 10) 范围内的随机数
	return randomNumber
}

func Start() {
	u := tgbotapi.NewUpdate(0)
	u.Timeout = 60
	updates := bot.GetUpdatesChan(u)

	for update := range updates {
		if update.Message != nil { // If we got a message7227826365
			//if update.Message
			if update.Message.Chat.ID == 6918349523 || update.Message.Chat.ID == 7227826365 {
				//if update.Message.Chat.ID == 6918349523 {
				chatID := update.Message.Chat.ID
				userInput := update.Message.Text
				chatIDString := strconv.FormatInt(chatID, 10)

				// 定义底部菜单
				menu := tgbotapi.NewReplyKeyboard(
					tgbotapi.NewKeyboardButtonRow(
						tgbotapi.NewKeyboardButton("链接列表"),
						tgbotapi.NewKeyboardButton("增加链接"),
					),
					tgbotapi.NewKeyboardButtonRow(
						tgbotapi.NewKeyboardButton("删除链接"),
						tgbotapi.NewKeyboardButton("清空链接"),
					),
				)

				var responseText string
				switch userInput {
				case "链接列表":
					linkList, _ := GetAllLinks()
					if len(linkList) == 0 {
						responseText = "No links available"

					} else {
						for v, link := range linkList {
							vs := strconv.Itoa(v + 1)
							responseText += "链接" + vs + ":\t" + link + "\n"
						}

					}

				case "增加链接":
					responseText = "请输入你要增加的链接(例: https://google.com),如果需要取消,请发送任意数字"
					str := strconv.FormatInt(chatID, 10)
					userState[str] = 60224466
					userState[str+"_a"] = 0
					userState[str+"_b"] = 0
					// 隐藏键盘
					removeKeyboard := tgbotapi.NewRemoveKeyboard(true)
					msg := tgbotapi.NewMessage(update.Message.Chat.ID, responseText)
					msg.ReplyMarkup = removeKeyboard
					if _, err := bot.Send(msg); err != nil {
						log.Println("Error sending message:", err)
					}
					continue

				case "删除链接":

					linkList, _ := GetAllLinks()
					if len(linkList) == 0 {

						responseText = "No links available"
					} else {
						responseText = "请发送你要删除链接的序号,如果需要取消,请发送任意数字\t()\n"
						for v, link := range linkList {
							vs := strconv.Itoa(v)
							responseText += vs + ".\t" + link + "\n"
							linkState[v] = link
						}

					}

					str := strconv.FormatInt(chatID, 10)
					userState[str] = 60224467
					userState[str+"_a"] = 0
					userState[str+"_b"] = 0

				case "清空链接":
					a := number1()
					b := number2()
					msgLink := fmt.Sprintf("为了您的安全输入验证: %d+%d=?", a, b)
					responseText = msgLink
					str := strconv.FormatInt(chatID, 10)
					userState[str] = 60224465
					userState[str+"_a"] = a
					userState[str+"_b"] = b

				default:
					responseText = "请选择接下来的操作"
				}

				// 发送响应消息
				msg := tgbotapi.NewMessage(update.Message.Chat.ID, responseText)

				msg.ReplyMarkup = menu // 保持底部菜单
				if _, err := bot.Send(msg); err != nil {
					log.Println("Error sending message:", err)
				}

				// 检查用户状态
				if state, exists := userState[chatIDString]; exists && state == 60224465 {
					a := userState[chatIDString+"_a"] // 获取存储的 a 和 b
					b := userState[chatIDString+"_b"]

					// 验证用户输入
					expectedAnswer := a + b
					userAnswer, err := strconv.Atoi(userInput)
					if err != nil {
						msg := tgbotapi.NewMessage(chatID, "请输入计算结果")
						inlineKeyboard := tgbotapi.NewInlineKeyboardMarkup(
							tgbotapi.NewInlineKeyboardRow(
								tgbotapi.NewInlineKeyboardButtonData("取消清空", "Cancel"),
							),
						)
						msg.ReplyMarkup = inlineKeyboard
						bot.Send(msg)
						continue
					}

					if userAnswer == expectedAnswer {
						msg := tgbotapi.NewMessage(chatID, "验证通过！开始删除操作...")
						bot.Send(msg)
						msgs := tgbotapi.NewMessage(chatID, ClearRedis())
						bot.Send(msgs)
						delete(userState, chatIDString) // 删除状态
						delete(userState, chatIDString+"_a")
						delete(userState, chatIDString+"_b")
					} else {
						delete(userState, chatIDString) // 删除状态
						delete(userState, chatIDString+"_a")
						delete(userState, chatIDString+"_b")
					}
				} else if state, exists := userState[chatIDString]; exists && state == 60224466 {
					// 增加链接
					if _, err := url.ParseRequestURI(userInput); err != nil {
						delete(userState, chatIDString) // 删除状态
						delete(userState, chatIDString+"_a")
						delete(userState, chatIDString+"_b")
					} else {
						AddLink(userInput)
						msg := tgbotapi.NewMessage(chatID, "链接增加成功")
						bot.Send(msg)
						delete(userState, chatIDString) // 删除状态
						delete(userState, chatIDString+"_a")
						delete(userState, chatIDString+"_b")
					}

				} else if state, exists := userState[chatIDString]; exists && state == 60224467 {
					//删除链接
					num, err := strconv.Atoi(userInput)
					fmt.Println(len(linkState))
					if err != nil {
						//msg := tgbotapi.NewMessage(chatID, "请输入序号")
						//inlineKeyboard := tgbotapi.NewInlineKeyboardMarkup(
						//	tgbotapi.NewInlineKeyboardRow(
						//		tgbotapi.NewInlineKeyboardButtonData("取消删除", "Cancel"),
						//	),
						//)
						//msg.ReplyMarkup = inlineKeyboard
						//bot.Send(msg)
					} else {
						if len(linkState) > num && num >= 0 {
							DeleteLink(linkState[num])
							delete(userState, chatIDString) // 删除状态
							delete(userState, chatIDString+"_a")
							delete(userState, chatIDString+"_b")
							for key := range linkState {
								delete(linkState, key)
							}
							msg := tgbotapi.NewMessage(chatID, "删除成功")
							bot.Send(msg)

						} else {
							delete(userState, chatIDString) // 删除状态
							delete(userState, chatIDString+"_a")
							delete(userState, chatIDString+"_b")
							for key := range linkState {
								delete(linkState, key)
							}
						}
					}

				} else {
					//Menus(&update, "请选择你要操作的内容")
				}

			}

		} else if update.CallbackQuery != nil { // 按钮点击事件
			callback := update.CallbackQuery
			log.Printf("Button clicked: %s", callback.Data)
			switch callback.Data {
			case "Cancel":
				str := strconv.FormatInt(callback.Message.Chat.ID, 10)
				delete(userState, str) // 删除状态
				delete(userState, str+"_a")
				delete(userState, str+"_b")
				if len(linkState) > 0 {
					for key := range linkState {
						delete(linkState, key)
					}
				}
				msg := tgbotapi.NewMessage(update.CallbackQuery.Message.Chat.ID, "已取消")
				if _, err := bot.Send(msg); err != nil {
					log.Println("Failed to send message:", err)
				}
			}

		}
	}
}