package bot

import (
	"time"

	"github.com/celestix/gotgproto/dispatcher"
	"github.com/celestix/gotgproto/ext"
	"github.com/duke-git/lancet/v2/slice"
	"github.com/gotd/contrib/middleware/floodwait"
	"github.com/gotd/contrib/middleware/ratelimit"
	"github.com/gotd/td/telegram"
	"github.com/krau/SaveAny-Bot/common"
	"github.com/krau/SaveAny-Bot/config"
	"golang.org/x/time/rate"
)

func FloodWaitMiddleware() []telegram.Middleware {
	waiter := floodwait.NewSimpleWaiter().WithMaxRetries(uint(config.Cfg.Telegram.FloodRetry))
	ratelimiter := ratelimit.New(rate.Every(time.Millisecond*100), 5)
	return []telegram.Middleware{
		waiter,
		ratelimiter,
	}
}

const noPermissionText string = `
您不在白名单中, 无法使用此 Bot.
您可以部署自己的实例: https://github.com/krau/SaveAny-Bot
`

func checkPermission(ctx *ext.Context, update *ext.Update) error {
	userID := update.GetUserChat().GetID()
	if !slice.Contain(config.Cfg.GetUsersID(), userID) {
		if config.Cfg.AsPublicCopyMediaBot {
			tryCopyMedia(ctx, update)
			return dispatcher.EndGroups
		}
		ctx.Reply(update, ext.ReplyTextString(noPermissionText), nil)
		return dispatcher.EndGroups
	}
	return dispatcher.ContinueGroups
}

func tryCopyMedia(ctx *ext.Context, update *ext.Update) {
	if !config.Cfg.AsPublicCopyMediaBot {
		return
	}
	if update.EffectiveMessage == nil || update.EffectiveMessage.Message == nil {
		return
	}
	msg := update.EffectiveMessage.Message
	if link := linkRegex.FindString(update.EffectiveMessage.Text); link != "" {
		linkChatID, messageID, err := parseLink(ctx, link)
		if err != nil {
			return
		}
		fileMessage, err := GetTGMessage(ctx, linkChatID, messageID)
		if err != nil {
			return
		}
		if fileMessage == nil || fileMessage.Media == nil {
			return
		}
		msg = fileMessage
	}
	if _, ok := msg.GetMedia(); !ok || msg.Media == nil {
		ctx.Reply(update, ext.ReplyTextString("消息中没有文件或媒体"), nil)
		return
	}
	common.Log.Tracef("Got copy media request from %d", update.EffectiveChat().GetID())
	if _, err := copyMediaToChat(ctx, msg, update.EffectiveChat().GetID()); err != nil {
		common.Log.Errorf("Failed to copy media: %v", err)
	}
}