discordo/ui/guilds.go

package ui

import (
	"sort"

	"github.com/ayntgl/astatine"
	"github.com/ayntgl/discordo/discord"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
)

type GuildsList struct {
	*tview.List
	app *App
}

func NewGuildsList(app *App) *GuildsList {
	gl := &GuildsList{
		List: tview.NewList(),
		app:  app,
	}

	gl.AddItem("Direct Messages", "", 0, nil)
	gl.ShowSecondaryText(false)
	gl.SetTitle("Guilds")
	gl.SetTitleAlign(tview.AlignLeft)
	gl.SetBorder(true)
	gl.SetBorderPadding(0, 0, 1, 1)
	gl.SetSelectedFunc(gl.onSelected)
	gl.SetInputCapture(gl.onInputCapture)
	return gl
}

func (gl *GuildsList) onInputCapture(e *tcell.EventKey) *tcell.EventKey {
	i := gl.List.GetCurrentItem()
	switch e.Rune() {
	case 'g': // Home.
		i = 0
	case 'G': // End.
		i = gl.List.GetItemCount() - 1
	case 'j': // Down.
		i++
	case 'k': // Up.
		i--
		if i < 0 {
			i = 0
		}
	default:
		return e
	}
	gl.List.SetCurrentItem(i)
	return nil
}

func (gl *GuildsList) onSelected(idx int, mainText string, secondaryText string, shortcut rune) {
	rootTreeNode := gl.app.ChannelsTreeView.GetRoot()
	rootTreeNode.ClearChildren()
	gl.app.SelectedMessage = -1
	gl.app.MessagesTextView.
		Highlight().
		Clear()
	gl.app.MessageInputField.SetText("")

	if mainText == "Direct Messages" {
		cs := gl.app.Session.State.PrivateChannels
		sort.Slice(cs, func(i, j int) bool {
			return cs[i].LastMessageID > cs[j].LastMessageID
		})

		for _, c := range cs {
			channelTreeNode := tview.NewTreeNode(discord.ChannelToString(c)).
				SetReference(c.ID)
			rootTreeNode.AddChild(channelTreeNode)
		}
	} else { // Guild
		// Decrement the index of the selected item by one since the first item in the list is always "Direct Messages".
		cs := gl.app.Session.State.Guilds[idx-1].Channels
		sort.Slice(cs, func(i, j int) bool {
			return cs[i].Position < cs[j].Position
		})

		for _, c := range cs {
			if (c.Type == astatine.ChannelTypeGuildText || c.Type == astatine.ChannelTypeGuildNews) && (c.ParentID == "") {
				channelTreeNode := tview.NewTreeNode(discord.ChannelToString(c)).
					SetReference(c.ID)
				rootTreeNode.AddChild(channelTreeNode)
			}
		}

	CATEGORY:
		for _, c := range cs {
			if c.Type == astatine.ChannelTypeGuildCategory {
				for _, nestedChannel := range cs {
					if nestedChannel.ParentID == c.ID {
						channelTreeNode := tview.NewTreeNode(c.Name).
							SetReference(c.ID)
						rootTreeNode.AddChild(channelTreeNode)
						continue CATEGORY
					}
				}

				channelTreeNode := tview.NewTreeNode(c.Name).
					SetReference(c.ID)
				rootTreeNode.AddChild(channelTreeNode)
			}
		}

		for _, c := range cs {
			if (c.Type == astatine.ChannelTypeGuildText || c.Type == astatine.ChannelTypeGuildNews) && (c.ParentID != "") {
				var parentTreeNode *tview.TreeNode
				rootTreeNode.Walk(func(node, _ *tview.TreeNode) bool {
					if node.GetReference() == c.ParentID {
						parentTreeNode = node
						return false
					}

					return true
				})

				if parentTreeNode != nil {
					channelTreeNode := tview.NewTreeNode(discord.ChannelToString(c)).
						SetReference(c.ID)
					parentTreeNode.AddChild(channelTreeNode)
				}
			}
		}
	}

	gl.app.ChannelsTreeView.SetCurrentNode(rootTreeNode)
	gl.app.SetFocus(gl.app.ChannelsTreeView)
}