discordo-plus/cmd/state.go

package cmd

import (
	"context"
	"log"
	"runtime"

	"github.com/ayn2op/discordo/internal/constants"
	"github.com/diamondburned/arikawa/v3/api"
	"github.com/diamondburned/arikawa/v3/gateway"
	"github.com/diamondburned/arikawa/v3/state"
	"github.com/diamondburned/arikawa/v3/utils/httputil/httpdriver"
	"github.com/diamondburned/ningen/v3"
	"github.com/rivo/tview"
)

func init() {
	api.UserAgent = constants.UserAgent
	gateway.DefaultIdentity = gateway.IdentifyProperties{
		OS:      runtime.GOOS,
		Browser: constants.Name,
		Device:  "",
	}
}

type State struct {
	*ningen.State
}

func openState(token string) error {
	s := state.New(token)
	discordState = &State{
		State: ningen.FromState(s),
	}

	// Handlers
	discordState.AddHandler(discordState.onReady)
	discordState.AddHandler(discordState.onMessageCreate)
	discordState.AddHandler(discordState.onMessageDelete)

	discordState.OnRequest = append(discordState.Client.OnRequest, discordState.onRequest)
	return discordState.Open(context.TODO())
}

func (s *State) onRequest(r httpdriver.Request) error {
	req, ok := r.(*httpdriver.DefaultRequest)
	if ok {
		log.Printf("method = %s; url = %s\n", req.Method, req.URL)
	}

	return nil
}

func (s *State) onReady(r *gateway.ReadyEvent) {
	root := mainFlex.guildsTree.GetRoot()
	dmNode := tview.NewTreeNode("Direct Messages")
	root.AddChild(dmNode)

	folders := r.UserSettings.GuildFolders
	if len(folders) == 0 {
		for _, g := range r.Guilds {
			mainFlex.guildsTree.createGuildNode(root, g.Guild)
		}
	} else {
		for _, folder := range folders {
			// If the ID of the guild folder is zero, the guild folder only contains single guild.
			if folder.ID == 0 {
				gID := folder.GuildIDs[0]
				g, err := discordState.Cabinet.Guild(gID)
				if err != nil {
					log.Printf("guild %v not found in state: %v\n", gID, err)
					continue
				}

				mainFlex.guildsTree.createGuildNode(root, *g)
			} else {
				mainFlex.guildsTree.createFolderNode(folder)
			}
		}
	}

	mainFlex.guildsTree.SetCurrentNode(root)
	app.SetFocus(mainFlex.guildsTree)
}

func (s *State) onMessageCreate(m *gateway.MessageCreateEvent) {
	if mainFlex.guildsTree.selectedChannelID.IsValid() && mainFlex.guildsTree.selectedChannelID == m.ChannelID {
		mainFlex.messagesText.createMessage(m.Message)
	}
}

func (s *State) onMessageDelete(m *gateway.MessageDeleteEvent) {
	if mainFlex.guildsTree.selectedChannelID == m.ChannelID {
		mainFlex.messagesText.selectedMessage = -1
		mainFlex.messagesText.Highlight()
		mainFlex.messagesText.Clear()

		mainFlex.messagesText.drawMsgs(m.ChannelID)
	}
}