package tg

import (
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)

// The type represents keyboard to be emdedded into the messages (inline in Telegram terms).
type Inline struct {
	*Keyboard
}

// Convert the inline keyboard to markup for the tgbotapi.
func (kbd *Inline) ToApi() tgbotapi.InlineKeyboardMarkup {
	rows := [][]tgbotapi.InlineKeyboardButton{}
	for _, row := range kbd.Rows {
		if row == nil {
			continue
		}
		buttons := []tgbotapi.InlineKeyboardButton{}
		for _, button := range row {
			if button == nil {
				continue
			}
			buttons = append(buttons, button.ToTelegramInline())
		}
		rows = append(rows, buttons)
	}

	return tgbotapi.NewInlineKeyboardMarkup(rows...)
}

// The type implements message with an inline keyboard.
type InlineCompo struct {
	*MessageCompo
	*Inline
}

// Implementing the Sendable interface.
func (compo *InlineCompo) SendConfig(
	sid SessionId, bot *Bot,
) (*SendConfig) {
	sendConfig := compo.MessageCompo.SendConfig(sid, bot)
	if len(compo.Inline.Rows) > 0 {
		sendConfig.Message.ReplyMarkup = compo.Inline.ToApi()
	}

	return sendConfig
}

// Update the component on the client side.
func (compo *InlineCompo) Update(c *Context) {
	if compo.Message != nil {
		var edit tgbotapi.Chattable
		markup := compo.Inline.ToApi()
		ln := len(markup.InlineKeyboard)
		if ln == 0 || compo.Inline.Rows == nil {
			edit = tgbotapi.NewEditMessageText(
				c.Session.Id.ToApi(),
				compo.Message.MessageID,
				compo.Text,
			)
		} else {
			edit = tgbotapi.NewEditMessageTextAndMarkup(
				c.Session.Id.ToApi(),
				compo.Message.MessageID,
				compo.Text,
				markup,
			)
		}
		msg, _ := c.Bot.Api.Send(edit)
		compo.Message = &msg
	}
	compo.buttonMap = compo.MakeButtonMap()
}

// Implementing the Filterer interface.
func (compo *InlineCompo) Filter(u *Update) bool {
	if compo == nil || u.CallbackQuery == nil {
		return true
	}

	if u.CallbackQuery.Message.MessageID != 
			compo.Message.MessageID {
		return true
	}

	return false
}

// Implementing the Server interface.
func (compo *InlineCompo) Serve(c *Context) {
	for u := range c.Input() {
		compo.OnOneUpdate(c, u)
	}
}

func (compo *InlineCompo) OnOneUpdate(c *Context, u *Update) {
	var act Action
	btns := compo.ButtonMap()
	cb := tgbotapi.NewCallback(
		u.CallbackQuery.ID,
		u.CallbackQuery.Data,
	)
	data := u.CallbackQuery.Data

	_, err := c.Bot.Api.Request(cb)
	if err != nil {
		return
	}

	btn, ok := btns[data]
	if !ok {
		return
	}
	if btn != nil {
		act = btn.Action
	} else if compo.Action != nil {
		act = compo.Action
	}
	c.WithUpdate(u).Run(act)
}