discordo-plus/state.go

package main

import (
	"log"

	"github.com/diamondburned/arikawa/v3/gateway"
	"github.com/diamondburned/arikawa/v3/state"
	"github.com/rivo/tview"
)

type State struct {
	*state.State
}

func newState(token string) *State {
	s := &State{
		State: state.New(token),
	}

	s.AddHandler(s.onReady)

	return s
}

func (s *State) onReady(r *gateway.ReadyEvent) {
	root := guildsTree.GetRoot()

	dmNode := tview.NewTreeNode("Direct Messages")
	root.AddChild(dmNode)

	for _, gf := range r.UserSettings.GuildFolders {
		/// If the ID of the guild folder is zero, the guild folder only contains single guild.
		if gf.ID == 0 {
			if err := guildsTree.newGuild(root, gf.GuildIDs[0]); err != nil {
				log.Println(err)
				continue
			}
		} else {
			gfNode := tview.NewTreeNode("Folder")
			root.AddChild(gfNode)

			for _, gid := range gf.GuildIDs {
				if err := guildsTree.newGuild(gfNode, gid); err != nil {
					log.Println(err)
					continue
				}
			}
		}
	}

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