package laniakea

import "fmt"

type MsgContext struct {
	Bot           *Bot
	Msg           *Message
	Update        *Update
	From          *User
	CallbackMsgId int
	FromID        int
	Prefix        string
	Text          string
	Args          []string
}

type AnswerMessage struct {
	MessageID int
	Text      string
	IsMedia   bool
	Keyboard  *InlineKeyboard
	ctx       *MsgContext
}

func (ctx *MsgContext) edit(messageId int, text string, keyboard *InlineKeyboard) *AnswerMessage {
	params := &EditMessageTextP{
		MessageID: messageId,
		ChatID:    ctx.Msg.Chat.ID,
		Text:      text,
		ParseMode: ParseMD,
	}
	if keyboard != nil {
		params.ReplyMarkup = keyboard.Get()
	}
	msg, err := ctx.Bot.EditMessageText(params)
	if err != nil {
		ctx.Bot.logger.Errorln(err)
		return nil
	}
	return &AnswerMessage{
		MessageID: msg.MessageID, ctx: ctx, Text: text, IsMedia: false,
	}
}
func (m *AnswerMessage) Edit(text string) *AnswerMessage {
	return m.ctx.edit(m.MessageID, text, nil)
}
func (ctx *MsgContext) EditCallback(text string, keyboard *InlineKeyboard) *AnswerMessage {
	if ctx.CallbackMsgId == 0 {
		ctx.Bot.logger.Errorln("Can't edit non-callback update message")
		return nil
	}

	return ctx.edit(ctx.CallbackMsgId, text, keyboard)
}
func (ctx *MsgContext) EditCallbackf(format string, keyboard *InlineKeyboard, args ...any) *AnswerMessage {
	return ctx.EditCallback(fmt.Sprintf(format, args...), keyboard)
}

func (ctx *MsgContext) editPhotoText(messageId int, text string, kb *InlineKeyboard) *AnswerMessage {
	params := &EditMessageCaptionP{
		ChatID:    ctx.Msg.Chat.ID,
		MessageID: messageId,
		Caption:   text,
		ParseMode: ParseMD,
	}
	if kb != nil {
		params.ReplyMarkup = kb.Get()
	}
	msg, err := ctx.Bot.EditMessageCaption(params)
	if err != nil {
		ctx.Bot.logger.Errorln(err)
	}
	return &AnswerMessage{
		MessageID: msg.MessageID, ctx: ctx, Text: text, IsMedia: true,
	}
}
func (m *AnswerMessage) EditCaption(text string) *AnswerMessage {
	return m.ctx.editPhotoText(m.MessageID, text, nil)
}
func (m *AnswerMessage) EditCaptionKeyboard(text string, kb *InlineKeyboard) *AnswerMessage {
	return m.ctx.editPhotoText(m.MessageID, text, kb)
}

func (ctx *MsgContext) answer(text string, keyboard *InlineKeyboard) *AnswerMessage {
	params := &SendMessageP{
		ChatID:    ctx.Msg.Chat.ID,
		Text:      text,
		ParseMode: ParseMD,
	}
	if keyboard != nil {
		params.ReplyMarkup = keyboard.Get()
	}

	msg, err := ctx.Bot.SendMessage(params)
	if err != nil {
		ctx.Bot.logger.Errorln(err)
		return nil
	}
	return &AnswerMessage{
		MessageID: msg.MessageID, ctx: ctx, IsMedia: false, Text: text,
	}
}
func (ctx *MsgContext) Answer(text string) *AnswerMessage {
	return ctx.answer(text, nil)
}
func (ctx *MsgContext) Answerf(template string, args ...any) *AnswerMessage {
	return ctx.answer(fmt.Sprintf(template, args...), nil)
}
func (ctx *MsgContext) Keyboard(text string, kb *InlineKeyboard) *AnswerMessage {
	return ctx.answer(text, kb)
}

func (ctx *MsgContext) answerPhoto(photoId, text string, kb *InlineKeyboard) *AnswerMessage {
	params := &SendPhotoP{
		ChatID:    ctx.Msg.Chat.ID,
		Caption:   text,
		Photo:     photoId,
		ParseMode: ParseMD,
	}
	if kb != nil {
		params.ReplyMarkup = kb.Get()
	}
	msg, err := ctx.Bot.SendPhoto(params)
	if err != nil {
		ctx.Bot.logger.Errorln(err)
	}
	return &AnswerMessage{
		MessageID: msg.MessageID, ctx: ctx, Text: text, IsMedia: true,
	}
}
func (ctx *MsgContext) AnswerPhoto(photoId, text string) *AnswerMessage {
	return ctx.answerPhoto(photoId, text, nil)
}
func (ctx *MsgContext) AnswerPhotoKeyboard(photoId, text string, kb *InlineKeyboard) *AnswerMessage {
	return ctx.answerPhoto(photoId, text, kb)
}

func (ctx *MsgContext) delete(messageId int) {
	_, err := ctx.Bot.DeleteMessage(&DeleteMessageP{
		ChatID:    ctx.Msg.Chat.ID,
		MessageID: messageId,
	})
	if err != nil {
		ctx.Bot.logger.Errorln(err)
	}
}
func (m *AnswerMessage) Delete() {
	m.ctx.delete(m.MessageID)
}
func (ctx *MsgContext) CallbackDelete() {
	ctx.delete(ctx.CallbackMsgId)
}

func (ctx *MsgContext) Error(err error) {
	_, sendErr := ctx.Bot.SendMessage(&SendMessageP{
		ChatID: ctx.Msg.Chat.ID,
		Text:   fmt.Sprintf(ctx.Bot.errorTemplate, EscapeMarkdown(err.Error())),
	})
	ctx.Bot.logger.Errorln(err)

	if sendErr != nil {
		ctx.Bot.logger.Errorln(sendErr)
	}
}