Laniakea/keyboard.go

package laniakea

import (
	"fmt"

	"git.scuroneko.dev/scuroneko/extypes"
	"git.scuroneko.dev/scuroneko/laniakea/tgapi"
)

const (
	// ButtonStyleDanger marks a destructive inline keyboard action.
	ButtonStyleDanger tgapi.KeyboardButtonStyle = "danger"
	// ButtonStyleSuccess marks a confirmatory inline keyboard action.
	ButtonStyleSuccess tgapi.KeyboardButtonStyle = "success"
	// ButtonStylePrimary marks a primary inline keyboard action.
	ButtonStylePrimary tgapi.KeyboardButtonStyle = "primary"
)

// InlineKeyboardButtonBuilder is a fluent builder for creating a single inline keyboard button.
//
// Use NewInlineKeyboardButton() to start, then chain methods to configure:
//   - SetIconCustomEmojiID() — adds a custom emoji icon
//   - SetStyle() — sets visual style (danger/success/primary)
//   - SetURL() — makes button open a URL
//   - SetCallbackDataJSON() — attaches structured command + args for bot handling
//
// Call Build to validate and produce the final tgapi.InlineKeyboardButton.
// Builder methods are immutable — each returns a copy.
type InlineKeyboardButtonBuilder struct {
	text    string
	emojiID string
	style   tgapi.KeyboardButtonStyle

	url  string
	data string

	payloadType BotPayloadType
}

// NewInlineKeyboardButton creates a new button builder with the given display text.
// The button will have no URL, no style, and no callback data by default.
func NewInlineKeyboardButton(text string) InlineKeyboardButtonBuilder {
	return InlineKeyboardButtonBuilder{text: text}
}

// SetIconCustomEmojiID sets a custom emoji ID to display as the button's icon.
// This is a Telegram Bot API feature for custom emoji icons.
func (b InlineKeyboardButtonBuilder) SetIconCustomEmojiID(id string) InlineKeyboardButtonBuilder {
	b.emojiID = id
	return b
}

// SetStyle sets the visual style of the button.
// Valid values: ButtonStyleDanger, ButtonStyleSuccess, ButtonStylePrimary.
// If not set, the button uses the default style.
func (b InlineKeyboardButtonBuilder) SetStyle(style tgapi.KeyboardButtonStyle) InlineKeyboardButtonBuilder {
	b.style = style
	return b
}

// SetURL sets a URL that will be opened when the button is pressed.
// It clears callback data because Telegram requires exactly one button action.
func (b InlineKeyboardButtonBuilder) SetURL(url string) InlineKeyboardButtonBuilder {
	b.url = url
	if url != "" {
		b.data = ""
	}
	return b
}

// SetPayloadType sets the encoding used by SetCallbackData.
func (b InlineKeyboardButtonBuilder) SetPayloadType(t BotPayloadType) InlineKeyboardButtonBuilder {
	b.payloadType = t
	return b
}

// SetCallbackDataJSON sets a structured callback payload that will be sent to the bot
// when the button is pressed. The command and arguments are serialized as JSON.
//
// Args are converted to strings using fmt.Sprint. Non-string types (e.g., int, bool)
// are safely serialized, but complex structs may not serialize usefully.
//
// Example: SetCallbackDataJSON("delete_user", 123, "confirm") → {"cmd":"delete_user","args":["123","confirm"]}.
func (b InlineKeyboardButtonBuilder) SetCallbackDataJSON(cmd string, args ...any) InlineKeyboardButtonBuilder {
	b.url = ""
	b.data = NewCallbackData(cmd, args...).ToJSON()
	return b
}

// SetCallbackDataBase64 sets a Base64-encoded structured callback payload.
// Base64 does not bypass Telegram's 64-byte callback-data limit.
func (b InlineKeyboardButtonBuilder) SetCallbackDataBase64(cmd string, args ...any) InlineKeyboardButtonBuilder {
	b.url = ""
	b.data = NewCallbackData(cmd, args...).ToBase64()
	return b
}

// SetCallbackDataCompact sets a structured callback payload encoded as compact text.
func (b InlineKeyboardButtonBuilder) SetCallbackDataCompact(cmd string, args ...any) InlineKeyboardButtonBuilder {
	b.url = ""
	b.data = NewCallbackData(cmd, args...).ToCompact()
	return b
}

// SetCallbackDataCompactBase64 sets a compact callback payload encoded as Base64.
func (b InlineKeyboardButtonBuilder) SetCallbackDataCompactBase64(cmd string, args ...any) InlineKeyboardButtonBuilder {
	b.url = ""
	b.data = NewCallbackData(cmd, args...).ToCompactBase64()
	return b
}

// SetCallbackData sets a structured callback payload using the configured payload type.
// The default payload type is JSON.
func (b InlineKeyboardButtonBuilder) SetCallbackData(cmd string, args ...any) InlineKeyboardButtonBuilder {
	b.url = ""
	switch b.payloadType {
	case BotPayloadJSON:
		b.data = NewCallbackData(cmd, args...).ToJSON()
	case BotPayloadBase64:
		b.data = NewCallbackData(cmd, args...).ToBase64()
	case BotPayloadCompact:
		b.data = NewCallbackData(cmd, args...).ToCompact()
	case BotPayloadCompactBase64:
		b.data = NewCallbackData(cmd, args...).ToCompactBase64()
	default:
		b.data = NewCallbackData(cmd, args...).ToJSON()
	}
	return b
}

func (b InlineKeyboardButtonBuilder) build() tgapi.InlineKeyboardButton {
	return tgapi.InlineKeyboardButton{
		Text:              b.text,
		URL:               b.url,
		Style:             b.style,
		IconCustomEmojiID: b.emojiID,
		CallbackData:      b.data,
	}
}

// Validate checks that the button has exactly one action and valid callback data.
func (b InlineKeyboardButtonBuilder) Validate() error {
	return validateInlineKeyboardButton(b.build())
}

// Build validates and returns the configured inline keyboard button.
func (b InlineKeyboardButtonBuilder) Build() (tgapi.InlineKeyboardButton, error) {
	button := b.build()
	if err := validateInlineKeyboardButton(button); err != nil {
		return tgapi.InlineKeyboardButton{}, err
	}
	return button, nil
}

// InlineKeyboard is a stateful builder for constructing Telegram inline keyboard layouts.
//
// Buttons are added row-by-row. When a row reaches maxRow, it is automatically flushed.
// Call AddLine() to manually end a row, or Get() to finalize and retrieve the markup.
//
// The keyboard is not thread-safe. Build it in a single goroutine.
type InlineKeyboard struct {
	// CurrentLine is the row currently being built.
	CurrentLine extypes.Slice[tgapi.InlineKeyboardButton]
	// Lines contains completed keyboard rows.
	Lines  [][]tgapi.InlineKeyboardButton
	maxRow int // Max buttons per row (e.g., 3 or 4)

	payloadType BotPayloadType // Serialization format for callback data (JSON or Base64)
}

// NewInlineKeyboardJSON creates a new keyboard builder with the specified maximum
// number of buttons per row.
//
// Example: NewInlineKeyboardJSON(3) creates a keyboard with at most 3 buttons per line.
func NewInlineKeyboardJSON(maxRow int) *InlineKeyboard {
	return NewInlineKeyboard(BotPayloadJSON, maxRow)
}

// NewInlineKeyboardBase64 creates a new keyboard builder with the specified maximum
// number of buttons per row, using Base64 encoding for button payloads.
//
// Example: NewInlineKeyboardBase64(3) creates a keyboard with at most 3 buttons per line.
func NewInlineKeyboardBase64(maxRow int) *InlineKeyboard {
	return NewInlineKeyboard(BotPayloadBase64, maxRow)
}

// NewInlineKeyboardCompact creates a keyboard builder using compact callback payloads.
func NewInlineKeyboardCompact(maxRow int) *InlineKeyboard {
	return NewInlineKeyboard(BotPayloadCompact, maxRow)
}

// NewInlineKeyboardCompactBase64 creates a keyboard builder using Base64-encoded compact payloads.
func NewInlineKeyboardCompactBase64(maxRow int) *InlineKeyboard {
	return NewInlineKeyboard(BotPayloadCompactBase64, maxRow)
}

// NewInlineKeyboard creates a new keyboard builder with the specified payload encoding
// type and maximum number of buttons per row.
//
// Use NewInlineKeyboardJSON or NewInlineKeyboardBase64 for the common cases.
func NewInlineKeyboard(payloadType BotPayloadType, maxRow int) *InlineKeyboard {
	return &InlineKeyboard{
		CurrentLine: make(extypes.Slice[tgapi.InlineKeyboardButton], 0),
		Lines:       make([][]tgapi.InlineKeyboardButton, 0),
		maxRow:      maxRow,
		payloadType: payloadType,
	}
}

// SetPayloadType sets the keyboard-local serialization format for callback data added via
// AddCallbackButton and AddCallbackButtonStyle methods.
// It overrides the bot's default payload type for this keyboard only.
func (in *InlineKeyboard) SetPayloadType(t BotPayloadType) *InlineKeyboard {
	in.payloadType = t
	return in
}

// GetPayloadType returns the keyboard-local callback payload encoding type.
func (in *InlineKeyboard) GetPayloadType() BotPayloadType { return in.payloadType }

// SetMaxRow sets the maximum number of buttons appended to a row before the
// keyboard automatically starts a new line. Values <= 0 retain the legacy
// unlimited-row behavior; this convention is subject to change in v2.
func (in *InlineKeyboard) SetMaxRow(maxRow int) *InlineKeyboard {
	in.maxRow = maxRow
	return in
}

// GetMaxRow returns the maximum number of buttons per row.
func (in *InlineKeyboard) GetMaxRow() int { return in.maxRow }

func (in *InlineKeyboard) append(button tgapi.InlineKeyboardButton) *InlineKeyboard {
	if in.maxRow > 0 && in.CurrentLine.Len() >= in.maxRow {
		in.AddLine()
	}
	in.CurrentLine = in.CurrentLine.Push(button)
	return in
}

// AddURLButton adds a button that opens a URL when pressed.
// No callback data is attached.
func (in *InlineKeyboard) AddURLButton(text, url string) *InlineKeyboard {
	return in.append(tgapi.InlineKeyboardButton{Text: text, URL: url})
}

// AddURLButtonStyle adds a button with a visual style that opens a URL.
// Style must be one of: ButtonStyleDanger, ButtonStyleSuccess, ButtonStylePrimary.
func (in *InlineKeyboard) AddURLButtonStyle(text string, style tgapi.KeyboardButtonStyle, url string) *InlineKeyboard {
	return in.append(tgapi.InlineKeyboardButton{Text: text, Style: style, URL: url})
}

// AddCallbackButton adds a button that sends a structured callback payload to the bot.
// The command and args are serialized according to the current payloadType.
func (in *InlineKeyboard) AddCallbackButton(text, cmd string, args ...any) *InlineKeyboard {
	return in.append(tgapi.InlineKeyboardButton{
		Text:         text,
		CallbackData: NewCallbackData(cmd, args...).Encode(in.payloadType),
	})
}

// AddCallbackButtonStyle adds a styled callback button.
// Style affects visual appearance; callback data is sent to bot on press.
func (in *InlineKeyboard) AddCallbackButtonStyle(text string, style tgapi.KeyboardButtonStyle, cmd string, args ...any) *InlineKeyboard {
	return in.append(tgapi.InlineKeyboardButton{
		Text:         text,
		Style:        style,
		CallbackData: NewCallbackData(cmd, args...).Encode(in.payloadType),
	})
}

// AddButton adds a button pre-configured via InlineKeyboardButtonBuilder.
// This is the most flexible way to create buttons with custom emoji, style, URL, and callback.
func (in *InlineKeyboard) AddButton(b InlineKeyboardButtonBuilder) *InlineKeyboard {
	return in.append(b.build())
}

// AddLine manually ends the current row and starts a new one.
// If the current row is empty, nothing happens.
func (in *InlineKeyboard) AddLine() *InlineKeyboard {
	if in.CurrentLine.Len() == 0 {
		return in
	}
	in.Lines = append(in.Lines, in.CurrentLine)
	in.CurrentLine = make(extypes.Slice[tgapi.InlineKeyboardButton], 0)
	return in
}

// Get finalizes the keyboard and returns a tgapi.ReplyMarkup.
// Automatically flushes the current line if not empty.
//
// Returns a pointer to a ReplyMarkup suitable for use with tgapi.SendMessage.
func (in *InlineKeyboard) Get() *tgapi.ReplyMarkup {
	if in.CurrentLine.Len() > 0 {
		in.AddLine()
	}
	lines := make([][]tgapi.InlineKeyboardButton, len(in.Lines))
	for i := range in.Lines {
		lines[i] = append([]tgapi.InlineKeyboardButton(nil), in.Lines[i]...)
	}
	return &tgapi.ReplyMarkup{InlineKeyboard: lines}
}

// GetValidated finalizes and validates the keyboard before returning it.
//
// Existing fluent Add* methods remain error-free for v1 compatibility. Their
// signatures are subject to change in v2; new code should use GetValidated.
func (in *InlineKeyboard) GetValidated() (*tgapi.ReplyMarkup, error) {
	markup := in.Get()
	if err := in.validateMarkup(markup); err != nil {
		return nil, err
	}
	return markup, nil
}

// Validate checks completed and pending rows without finalizing the keyboard.
func (in *InlineKeyboard) Validate() error {
	lines := make([][]tgapi.InlineKeyboardButton, 0, len(in.Lines)+1)
	lines = append(lines, in.Lines...)
	if len(in.CurrentLine) > 0 {
		lines = append(lines, in.CurrentLine)
	}
	return in.validateMarkup(&tgapi.ReplyMarkup{InlineKeyboard: lines})
}

func (in *InlineKeyboard) validateMarkup(markup *tgapi.ReplyMarkup) error {
	for rowIndex, row := range markup.InlineKeyboard {
		if in.maxRow > 0 && len(row) > in.maxRow {
			return fmt.Errorf("%w: row %d has %d buttons, limit %d", ErrInlineKeyboardRowTooLong, rowIndex, len(row), in.maxRow)
		}
		for columnIndex, button := range row {
			if err := validateInlineKeyboardButton(button); err != nil {
				return fmt.Errorf("row %d button %d: %w", rowIndex, columnIndex, err)
			}
		}
	}
	return nil
}

func validateInlineKeyboardButton(button tgapi.InlineKeyboardButton) error {
	actions := 0
	if button.URL != "" {
		actions++
	}
	if button.CallbackData != "" {
		actions++
	}
	if actions != 1 {
		return ErrInlineKeyboardButtonAction
	}
	if button.CallbackData != "" {
		length := len([]byte(button.CallbackData))
		if length < 1 || length > 64 {
			return fmt.Errorf("%w: got %d", ErrCallbackDataLength, length)
		}
	}
	return nil
}

// CallbackData represents the structured payload sent when an inline button
// with callback data is pressed.
//
// This structure is serialized to JSON and sent to the bot as a string.
// The bot should parse this back to determine the command and arguments.
//
// Example:
//
//	{"cmd":"delete_user","args":["123","confirm"]}
type CallbackData struct {
	// Command is the command name used for payload routing.
	Command string `json:"cmd"`
	// Args contains the string arguments passed to the payload handler.
	Args []string `json:"args"`
}

// NewCallbackData creates a new CallbackData instance with the given command and args.
//
// All args are converted to strings using fmt.Sprint. This is safe for primitives
// (int, string, bool, float64) but may not serialize complex structs meaningfully.
//
// Use this to build callback payloads for bot command routing.
func NewCallbackData(command string, args ...any) CallbackData {
	stringArgs := make([]string, len(args))
	for i, arg := range args {
		stringArgs[i] = fmt.Sprint(arg)
	}
	return CallbackData{Command: command, Args: stringArgs}
}

// All To* encoders return an empty string when serialization fails. Telegram
// rejects empty callback_data, so an empty result surfaces a real bug rather
// than masking it with a stub payload that silently routes to no handler.
// Build CallbackData from primitives (string, []string) only — the encoders
// have no failure modes for that input.

// ToJSON serializes the CallbackData to a JSON string.
// Returns an empty string if serialization fails.
func (d CallbackData) ToJSON() string {
	data, err := encodeJSONPayload(d)
	if err != nil {
		return ""
	}
	return data
}

// ToBase64 serializes the CallbackData to a JSON string and then encodes it as Base64.
// Returns an empty string if serialization or encoding fails.
func (d CallbackData) ToBase64() string {
	data, err := encodeBase64Payload(d)
	if err != nil {
		return ""
	}
	return data
}

// ToCompact serializes the CallbackData to a compact delimited string.
// Returns an empty string if serialization fails.
//
// The compact format coalesces "no args" with "single empty arg" — both
// produce "cmd|" and decode back to nil args. Use ToJSON or ToBase64 when
// that distinction must be preserved.
func (d CallbackData) ToCompact() string {
	data, err := encodeCompactPayload(d)
	if err != nil {
		return ""
	}
	return data
}

// ToCompactBase64 serializes the CallbackData to compact text and then encodes it as Base64.
// Returns an empty string if serialization or encoding fails.
func (d CallbackData) ToCompactBase64() string {
	data, err := encodeCompactBase64Payload(d)
	if err != nil {
		return ""
	}
	return data
}

// Encode serializes the CallbackData according to the specified payload type.
// Supported types: BotPayloadJSON, BotPayloadBase64, BotPayloadCompact, and BotPayloadCompactBase64.
// For unknown types, returns an empty string.
func (d CallbackData) Encode(t BotPayloadType) string {
	switch t {
	case BotPayloadBase64:
		return d.ToBase64()
	case BotPayloadJSON:
		return d.ToJSON()
	case BotPayloadCompact:
		return d.ToCompact()
	case BotPayloadCompactBase64:
		return d.ToCompactBase64()
	}
	return ""
}

// EncodeValidated serializes callback data and enforces Telegram's 1-64 byte limit.
func (d CallbackData) EncodeValidated(t BotPayloadType) (string, error) {
	encoded := d.Encode(t)
	if encoded == "" {
		if t != BotPayloadBase64 && t != BotPayloadJSON && t != BotPayloadCompact && t != BotPayloadCompactBase64 {
			return "", ErrInvalidPayloadType
		}
		return "", ErrCallbackDataLength
	}
	if length := len([]byte(encoded)); length > 64 {
		return "", fmt.Errorf("%w: got %d", ErrCallbackDataLength, length)
	}
	return encoded, nil
}