goMuffin/utils/paginationEmbed.go

package utils

import (
	"fmt"
	"math/rand"

	"github.com/bwmarrin/discordgo"
)

// PaginationEmbed is embed with page
type PaginationEmbed struct {
	Container  *discordgo.Container
	Components []discordgo.MessageComponent
	Current    int
	Total      int
	Id         string
	m          any
}

var PaginationEmbeds = make(map[string]*PaginationEmbed)

func NewPaginationEmbedBuilder(m any) *PaginationEmbed {
	var userId string

	switch m := m.(type) {
	case *MessageCreate:
		userId = m.Author.ID
	case *InteractionCreate:
		userId = m.Member.User.ID
	}

	id := fmt.Sprintf("%s/%d", userId, rand.Intn(100))
	return &PaginationEmbed{
		Current: 1,
		Id:      id,
		m:       m,
	}
}

func (p *PaginationEmbed) SetContainer(container discordgo.Container) *PaginationEmbed {
	p.Container = &container
	return p
}

func (p *PaginationEmbed) AddComponents(components ...discordgo.MessageComponent) *PaginationEmbed {
	p.Total += len(components)
	p.Components = append(p.Components, components...)
	return p
}

func (p *PaginationEmbed) Start() {
	startPaginationEmbed(p)
}

func makeComponents(id string, current, total int) *discordgo.ActionsRow {
	disabled := false

	if total == 1 {
		disabled = true
	}

	return &discordgo.ActionsRow{
		Components: []discordgo.MessageComponent{
			discordgo.Button{
				Style:    discordgo.PrimaryButton,
				Label:    "이전",
				CustomID: MakePaginationEmbedPrev(id),
				Disabled: disabled,
			},
			discordgo.Button{
				Style:    discordgo.SecondaryButton,
				Label:    fmt.Sprintf("(%d/%d)", current, total),
				CustomID: MakePaginationEmbedPages(id),
				Disabled: disabled,
			},
			discordgo.Button{
				Style:    discordgo.PrimaryButton,
				Label:    "다음",
				CustomID: MakePaginationEmbedNext(id),
				Disabled: disabled,
			},
		},
	}
}

func MakeDesc(desc, item string) string {
	var newDesc string

	if desc == "" {
		newDesc = item
	} else {
		newDesc = fmt.Sprintf(desc, item)
	}
	return newDesc
}

func startPaginationEmbed(p *PaginationEmbed) error {
	p.Container.Components = append(p.Container.Components, p.Components[0], makeComponents(p.Id, p.Current, p.Total))

	PaginationEmbeds[p.Id] = p

	err := NewMessageSender(p.m).
		AddComponents(p.Container).
		SetReply(true).
		SetEphemeral(true).
		SetComponentsV2(true).
		Send()
	return err
}

func GetPaginationEmbed(id string) *PaginationEmbed {
	if p, ok := PaginationEmbeds[id]; ok {
		return p
	}
	return nil
}

func (p *PaginationEmbed) Prev(i *InteractionCreate) {
	if p.Current == 1 {
		i.Reply(&discordgo.InteractionResponseData{
			Embeds: []*discordgo.MessageEmbed{
				{
					Title:       "❌ 오류",
					Description: "해당 페이지가 처음ㅇ이에요.",
					Color:       EmbedFail,
				},
			},
			Flags: discordgo.MessageFlagsEphemeral,
		})
		return
	}

	p.Current -= 1

	p.Set(i, p.Current)
}

func (p *PaginationEmbed) Next(i *InteractionCreate) {
	if p.Current >= p.Total {
		i.Reply(&discordgo.InteractionResponseData{
			Embeds: []*discordgo.MessageEmbed{
				{
					Title:       "❌ 오류",
					Description: "해당 페이지가 마지막ㅇ이에요.",
					Color:       EmbedFail,
				},
			},
			Flags: discordgo.MessageFlagsEphemeral,
		})
		return
	}

	p.Current += 1

	p.Set(i, p.Current)
}

func (p *PaginationEmbed) Set(i *InteractionCreate, page int) error {
	if page <= 0 {
		i.Reply(&discordgo.InteractionResponseData{
			Embeds: []*discordgo.MessageEmbed{
				{
					Title:       "❌ 오류",
					Description: "해당 값은 0보다 커야해요.",
					Color:       EmbedFail,
				},
			},
			Flags: discordgo.MessageFlagsEphemeral,
		})
		return nil
	}

	if page > p.Total {
		i.Reply(&discordgo.InteractionResponseData{
			Embeds: []*discordgo.MessageEmbed{
				{
					Title:       "❌ 오류",
					Description: "해당 값은 총 페이지의 수보다 작아야해요.",
					Color:       EmbedFail,
				},
			},
			Flags: discordgo.MessageFlagsEphemeral,
		})
		return nil
	}

	p.Current = page

	p.Container.Components = []discordgo.MessageComponent{p.Components[p.Current-1], makeComponents(p.Id, p.Current, p.Total)}

	err := i.Update(&discordgo.InteractionResponseData{
		Flags:      discordgo.MessageFlagsIsComponentsV2,
		Components: []discordgo.MessageComponent{p.Container},
	})
	return err
}

func (p *PaginationEmbed) ShowModal(i *InteractionCreate) {
	i.ShowModal(&ModalData{
		CustomId: MakePaginationEmbedModal(p.Id),
		Title:    fmt.Sprintf("%s의 리스트", i.Session.State.User.Username),
		Components: []discordgo.MessageComponent{
			discordgo.ActionsRow{
				Components: []discordgo.MessageComponent{
					discordgo.TextInput{
						CustomID:    MakePaginationEmbedSetPage(p.Id),
						Label:       "페이지",
						Style:       discordgo.TextInputShort,
						Placeholder: "이동할 페이지를 여기에 적어주세요.",
						Required:    true,
					},
				},
			},
		},
	})
}