392 lines
9.1 KiB
Go
392 lines
9.1 KiB
Go
package ui
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/atotto/clipboard"
|
|
"github.com/ayntgl/discordgo"
|
|
"github.com/ayntgl/discordo/discord"
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
)
|
|
|
|
func onAppInputCapture(app *App, e *tcell.EventKey) *tcell.EventKey {
|
|
if app.MessageInputField.HasFocus() {
|
|
return e
|
|
}
|
|
|
|
switch e.Name() {
|
|
case app.Config.Keybindings.ToggleGuildsList:
|
|
app.SetFocus(app.GuildsList)
|
|
return nil
|
|
case app.Config.Keybindings.ToggleChannelsTreeView:
|
|
app.SetFocus(app.ChannelsTreeView)
|
|
return nil
|
|
case app.Config.Keybindings.ToggleMessagesTextView:
|
|
app.SetFocus(app.MessagesTextView)
|
|
return nil
|
|
case app.Config.Keybindings.ToggleMessageInputField:
|
|
app.SetFocus(app.MessageInputField)
|
|
return nil
|
|
}
|
|
|
|
return e
|
|
}
|
|
|
|
func onGuildsListSelected(app *App, guildIdx int) {
|
|
rootTreeNode := app.ChannelsTreeView.GetRoot()
|
|
rootTreeNode.ClearChildren()
|
|
app.SelectedMessage = -1
|
|
app.MessagesTextView.
|
|
Highlight().
|
|
Clear()
|
|
app.MessageInputField.SetText("")
|
|
|
|
// If the user is a bot account, the direct messages item does not exist in the guilds list.
|
|
if app.Session.State.User.Bot && guildIdx == 0 {
|
|
guildIdx = 1
|
|
}
|
|
|
|
if guildIdx == 0 { // Direct Messages
|
|
cs := 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
|
|
cs := app.Session.State.Guilds[guildIdx-1].Channels
|
|
sort.Slice(cs, func(i, j int) bool {
|
|
return cs[i].Position < cs[j].Position
|
|
})
|
|
|
|
for _, c := range cs {
|
|
if (c.Type == discordgo.ChannelTypeGuildText || c.Type == discordgo.ChannelTypeGuildNews) && (c.ParentID == "") {
|
|
channelTreeNode := tview.NewTreeNode(discord.ChannelToString(c)).
|
|
SetReference(c.ID)
|
|
rootTreeNode.AddChild(channelTreeNode)
|
|
}
|
|
}
|
|
|
|
CATEGORY:
|
|
for _, c := range cs {
|
|
if c.Type == discordgo.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 == discordgo.ChannelTypeGuildText || c.Type == discordgo.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)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
app.ChannelsTreeView.SetCurrentNode(rootTreeNode)
|
|
app.SetFocus(app.ChannelsTreeView)
|
|
}
|
|
|
|
func onChannelsTreeViewSelected(app *App, n *tview.TreeNode) {
|
|
app.SelectedMessage = -1
|
|
app.MessagesTextView.
|
|
Highlight().
|
|
Clear()
|
|
app.MessageInputField.SetText("")
|
|
|
|
c, err := app.Session.State.Channel(n.GetReference().(string))
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if c.Type == discordgo.ChannelTypeGuildCategory {
|
|
n.SetExpanded(!n.IsExpanded())
|
|
return
|
|
}
|
|
|
|
app.SelectedChannel = c
|
|
app.SetFocus(app.MessageInputField)
|
|
|
|
go func() {
|
|
ms, err := app.Session.ChannelMessages(c.ID, app.Config.General.FetchMessagesLimit, "", "", "")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
for i := len(ms) - 1; i >= 0; i-- {
|
|
app.SelectedChannel.Messages = append(app.SelectedChannel.Messages, ms[i])
|
|
_, err = app.MessagesTextView.Write(buildMessage(app, ms[i]))
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
app.MessagesTextView.ScrollToEnd()
|
|
}()
|
|
}
|
|
|
|
func onMessagesTextViewInputCapture(app *App, e *tcell.EventKey) *tcell.EventKey {
|
|
if app.SelectedChannel == nil {
|
|
return nil
|
|
}
|
|
|
|
ms := app.SelectedChannel.Messages
|
|
if len(ms) == 0 {
|
|
return nil
|
|
}
|
|
|
|
switch e.Name() {
|
|
case app.Config.Keybindings.SelectPreviousMessage:
|
|
if len(app.MessagesTextView.GetHighlights()) == 0 {
|
|
app.SelectedMessage = len(ms) - 1
|
|
} else {
|
|
app.SelectedMessage--
|
|
if app.SelectedMessage < 0 {
|
|
app.SelectedMessage = 0
|
|
}
|
|
}
|
|
|
|
app.MessagesTextView.
|
|
Highlight(ms[app.SelectedMessage].ID).
|
|
ScrollToHighlight()
|
|
|
|
return nil
|
|
case app.Config.Keybindings.SelectNextMessage:
|
|
if len(app.MessagesTextView.GetHighlights()) == 0 {
|
|
app.SelectedMessage = len(ms) - 1
|
|
} else {
|
|
app.SelectedMessage++
|
|
if app.SelectedMessage >= len(ms) {
|
|
app.SelectedMessage = len(ms) - 1
|
|
}
|
|
}
|
|
|
|
app.MessagesTextView.
|
|
Highlight(ms[app.SelectedMessage].ID).
|
|
ScrollToHighlight()
|
|
|
|
return nil
|
|
case app.Config.Keybindings.SelectFirstMessage:
|
|
app.SelectedMessage = 0
|
|
app.MessagesTextView.
|
|
Highlight(ms[app.SelectedMessage].ID).
|
|
ScrollToHighlight()
|
|
|
|
return nil
|
|
case app.Config.Keybindings.SelectLastMessage:
|
|
app.SelectedMessage = len(ms) - 1
|
|
app.MessagesTextView.
|
|
Highlight(ms[app.SelectedMessage].ID).
|
|
ScrollToHighlight()
|
|
|
|
return nil
|
|
case app.Config.Keybindings.ToggleMessageActionsList:
|
|
messageActionsList := tview.NewList()
|
|
|
|
hs := app.MessagesTextView.GetHighlights()
|
|
if len(hs) == 0 {
|
|
return nil
|
|
}
|
|
|
|
_, m := discord.FindMessageByID(app.SelectedChannel.Messages, hs[0])
|
|
if m == nil {
|
|
return nil
|
|
}
|
|
|
|
if discord.HasPermission(app.Session.State, app.SelectedChannel.ID, discordgo.PermissionSendMessages) {
|
|
messageActionsList.
|
|
AddItem("Reply", "", 'r', nil).
|
|
AddItem("Mention Reply", "", 'R', nil)
|
|
}
|
|
|
|
if m.ReferencedMessage != nil {
|
|
messageActionsList.AddItem("Select Reply", "", 'm', nil)
|
|
}
|
|
|
|
messageActionsList.
|
|
ShowSecondaryText(false).
|
|
AddItem("Copy Content", "", 'c', nil).
|
|
AddItem("Copy ID", "", 'i', nil).
|
|
SetDoneFunc(func() {
|
|
app.
|
|
SetRoot(app.MainFlex, true).
|
|
SetFocus(app.MessagesTextView)
|
|
}).
|
|
SetSelectedFunc(func(_ int, mainText string, _ string, _ rune) {
|
|
onMessageActionsListSelected(app, mainText, m)
|
|
}).
|
|
SetTitle("Press the Escape key to close").
|
|
SetBorder(true)
|
|
|
|
app.SetRoot(messageActionsList, true)
|
|
|
|
return nil
|
|
case "Esc":
|
|
app.SelectedMessage = -1
|
|
app.SetFocus(app.MainFlex)
|
|
app.MessagesTextView.
|
|
Clear().
|
|
Highlight()
|
|
|
|
return nil
|
|
}
|
|
|
|
return e
|
|
}
|
|
|
|
func onMessageActionsListSelected(app *App, mainText string, m *discordgo.Message) {
|
|
switch mainText {
|
|
case "Copy Content":
|
|
if err := clipboard.WriteAll(m.Content); err != nil {
|
|
return
|
|
}
|
|
|
|
app.SetRoot(app.MainFlex, false)
|
|
case "Copy ID":
|
|
if err := clipboard.WriteAll(m.ID); err != nil {
|
|
return
|
|
}
|
|
|
|
app.SetRoot(app.MainFlex, false)
|
|
case "Reply":
|
|
app.MessageInputField.SetTitle("Replying to " + m.Author.String())
|
|
app.
|
|
SetRoot(app.MainFlex, false).
|
|
SetFocus(app.MessageInputField)
|
|
case "Mention Reply":
|
|
app.MessageInputField.SetTitle("[@] Replying to " + m.Author.String())
|
|
app.
|
|
SetRoot(app.MainFlex, false).
|
|
SetFocus(app.MessageInputField)
|
|
case "Select Reply":
|
|
app.SelectedMessage, _ = discord.FindMessageByID(app.SelectedChannel.Messages, m.ReferencedMessage.ID)
|
|
app.MessagesTextView.
|
|
Highlight(m.ReferencedMessage.ID).
|
|
ScrollToHighlight()
|
|
app.
|
|
SetRoot(app.MainFlex, false).
|
|
SetFocus(app.MessagesTextView)
|
|
}
|
|
}
|
|
|
|
func onMessageInputFieldInputCapture(app *App, e *tcell.EventKey) *tcell.EventKey {
|
|
switch e.Name() {
|
|
case "Enter":
|
|
if app.SelectedChannel == nil {
|
|
return nil
|
|
}
|
|
|
|
t := strings.TrimSpace(app.MessageInputField.GetText())
|
|
if t == "" {
|
|
return nil
|
|
}
|
|
|
|
if len(app.MessagesTextView.GetHighlights()) != 0 {
|
|
_, m := discord.FindMessageByID(app.SelectedChannel.Messages, app.MessagesTextView.GetHighlights()[0])
|
|
d := &discordgo.MessageSend{
|
|
Content: t,
|
|
Reference: m.Reference(),
|
|
AllowedMentions: &discordgo.MessageAllowedMentions{RepliedUser: false},
|
|
}
|
|
if strings.HasPrefix(app.MessageInputField.GetTitle(), "[@]") {
|
|
d.AllowedMentions.RepliedUser = true
|
|
} else {
|
|
d.AllowedMentions.RepliedUser = false
|
|
}
|
|
|
|
go app.Session.ChannelMessageSendComplex(m.ChannelID, d)
|
|
|
|
app.SelectedMessage = -1
|
|
app.MessagesTextView.Highlight()
|
|
|
|
app.MessageInputField.SetTitle("")
|
|
} else {
|
|
go app.Session.ChannelMessageSend(app.SelectedChannel.ID, t)
|
|
}
|
|
|
|
app.MessageInputField.SetText("")
|
|
|
|
return nil
|
|
case "Ctrl+V":
|
|
text, _ := clipboard.ReadAll()
|
|
text = app.MessageInputField.GetText() + text
|
|
app.MessageInputField.SetText(text)
|
|
|
|
return nil
|
|
case "Esc":
|
|
app.MessageInputField.
|
|
SetText("").
|
|
SetTitle("")
|
|
app.SetFocus(app.MainFlex)
|
|
|
|
app.SelectedMessage = -1
|
|
app.MessagesTextView.Highlight()
|
|
|
|
return nil
|
|
case app.Config.Keybindings.ToggleExternalEditor:
|
|
e := os.Getenv("EDITOR")
|
|
if e == "" {
|
|
return nil
|
|
}
|
|
|
|
f, err := os.CreateTemp(os.TempDir(), "discordo-*.md")
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
defer os.Remove(f.Name())
|
|
|
|
cmd := exec.Command(e, f.Name())
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
|
|
app.Suspend(func() {
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
return
|
|
}
|
|
})
|
|
|
|
b, err := io.ReadAll(f)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
app.MessageInputField.SetText(string(b))
|
|
|
|
return nil
|
|
}
|
|
|
|
return e
|
|
}
|