|
@@ -1,66 +1,22 @@
|
|
|
package chat
|
|
package chat
|
|
|
|
|
|
|
|
import (
|
|
import (
|
|
|
- "context"
|
|
|
|
|
"log/slog"
|
|
"log/slog"
|
|
|
"slices"
|
|
"slices"
|
|
|
|
|
|
|
|
- "github.com/ayn2op/discordo/internal/http"
|
|
|
|
|
"github.com/ayn2op/discordo/internal/notifications"
|
|
"github.com/ayn2op/discordo/internal/notifications"
|
|
|
"github.com/ayn2op/tview"
|
|
"github.com/ayn2op/tview"
|
|
|
"github.com/diamondburned/arikawa/v3/discord"
|
|
"github.com/diamondburned/arikawa/v3/discord"
|
|
|
"github.com/diamondburned/arikawa/v3/gateway"
|
|
"github.com/diamondburned/arikawa/v3/gateway"
|
|
|
- "github.com/diamondburned/arikawa/v3/session"
|
|
|
|
|
- "github.com/diamondburned/arikawa/v3/state"
|
|
|
|
|
- "github.com/diamondburned/arikawa/v3/state/store/defaultstore"
|
|
|
|
|
- "github.com/diamondburned/arikawa/v3/utils/handler"
|
|
|
|
|
- "github.com/diamondburned/arikawa/v3/utils/httputil"
|
|
|
|
|
"github.com/diamondburned/arikawa/v3/utils/httputil/httpdriver"
|
|
"github.com/diamondburned/arikawa/v3/utils/httputil/httpdriver"
|
|
|
"github.com/diamondburned/arikawa/v3/utils/ws"
|
|
"github.com/diamondburned/arikawa/v3/utils/ws"
|
|
|
- "github.com/diamondburned/ningen/v3"
|
|
|
|
|
|
|
+ "github.com/diamondburned/ningen/v3/states/read"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
-func (m *Model) OpenState(token string) error {
|
|
|
|
|
- identifyProps := http.IdentifyProperties()
|
|
|
|
|
- gateway.DefaultIdentity = identifyProps
|
|
|
|
|
- gateway.DefaultPresence = &gateway.UpdatePresenceCommand{
|
|
|
|
|
- Status: m.cfg.Status,
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- id := gateway.DefaultIdentifier(token)
|
|
|
|
|
- id.Compress = false
|
|
|
|
|
-
|
|
|
|
|
- session := session.NewCustom(id, http.NewClient(token), handler.New())
|
|
|
|
|
- state := state.NewFromSession(session, defaultstore.New())
|
|
|
|
|
- m.state = ningen.FromState(state)
|
|
|
|
|
-
|
|
|
|
|
- // Handlers
|
|
|
|
|
- m.state.AddHandler(m.onRaw)
|
|
|
|
|
- m.state.AddHandler(m.onReady)
|
|
|
|
|
- m.state.AddHandler(m.onMessageCreate)
|
|
|
|
|
- m.state.AddHandler(m.onMessageUpdate)
|
|
|
|
|
- m.state.AddHandler(m.onMessageDelete)
|
|
|
|
|
- m.state.AddHandler(m.onReadUpdate)
|
|
|
|
|
- m.state.AddHandler(m.onGuildMembersChunk)
|
|
|
|
|
- m.state.AddHandler(m.onGuildMemberRemove)
|
|
|
|
|
-
|
|
|
|
|
- if m.cfg.TypingIndicator.Receive {
|
|
|
|
|
- m.state.AddHandler(m.onTypingStart)
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- m.state.StateLog = func(err error) {
|
|
|
|
|
- slog.Error("state log", "err", err)
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- m.state.OnRequest = append(m.state.OnRequest, httputil.WithHeaders(http.Headers()), m.onRequest)
|
|
|
|
|
- return m.state.Open(context.Background())
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
func (m *Model) onRequest(r httpdriver.Request) error {
|
|
func (m *Model) onRequest(r httpdriver.Request) error {
|
|
|
if req, ok := r.(*httpdriver.DefaultRequest); ok {
|
|
if req, ok := r.(*httpdriver.DefaultRequest); ok {
|
|
|
slog.Debug("new HTTP request", "method", req.Method, "url", req.URL)
|
|
slog.Debug("new HTTP request", "method", req.Method, "url", req.URL)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
return nil
|
|
return nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -74,79 +30,76 @@ func (m *Model) onRaw(event *ws.RawEvent) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func (m *Model) onReady(event *gateway.ReadyEvent) {
|
|
func (m *Model) onReady(event *gateway.ReadyEvent) {
|
|
|
- m.app.QueueUpdateDraw(func() {
|
|
|
|
|
- // Rebuild indexes from scratch so reconnects and account switches do not
|
|
|
|
|
- // retain pointers to detached tree nodes.
|
|
|
|
|
- m.guildsTree.resetNodeIndex()
|
|
|
|
|
-
|
|
|
|
|
- dmNode := tview.NewTreeNode("Direct Messages").SetReference(dmNode{}).SetExpandable(true).SetExpanded(false)
|
|
|
|
|
- m.guildsTree.dmRootNode = dmNode
|
|
|
|
|
-
|
|
|
|
|
- root := m.guildsTree.
|
|
|
|
|
- GetRoot().
|
|
|
|
|
- ClearChildren().
|
|
|
|
|
- AddChild(dmNode)
|
|
|
|
|
-
|
|
|
|
|
- // Track guilds already in folders to find orphans (newly joined guilds may not be synced to GuildFolders yet but always appear in GuildPositions)
|
|
|
|
|
- guildsInFolders := make(map[discord.GuildID]bool)
|
|
|
|
|
- for _, folder := range event.UserSettings.GuildFolders {
|
|
|
|
|
- for _, guildID := range folder.GuildIDs {
|
|
|
|
|
- guildsInFolders[guildID] = true
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ // Rebuild indexes from scratch so reconnects and account switches do not
|
|
|
|
|
+ // retain pointers to detached tree nodes.
|
|
|
|
|
+ m.guildsTree.resetNodeIndex()
|
|
|
|
|
+
|
|
|
|
|
+ dmNode := tview.NewTreeNode("Direct Messages").SetReference(dmNode{}).SetExpandable(true).SetExpanded(false)
|
|
|
|
|
+ m.guildsTree.dmRootNode = dmNode
|
|
|
|
|
+
|
|
|
|
|
+ root := m.guildsTree.
|
|
|
|
|
+ GetRoot().
|
|
|
|
|
+ ClearChildren().
|
|
|
|
|
+ AddChild(dmNode)
|
|
|
|
|
+
|
|
|
|
|
+ // Track guilds already in folders to find orphans.
|
|
|
|
|
+ // Newly joined guilds may not be synced to GuildFolders yet but always appear in guild positions.
|
|
|
|
|
+ guildsInFolders := make(map[discord.GuildID]bool)
|
|
|
|
|
+ for _, folder := range event.UserSettings.GuildFolders {
|
|
|
|
|
+ for _, guildID := range folder.GuildIDs {
|
|
|
|
|
+ guildsInFolders[guildID] = true
|
|
|
}
|
|
}
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- // Build index of all available guilds.
|
|
|
|
|
- guildsByID := make(map[discord.GuildID]*gateway.GuildCreateEvent, len(event.Guilds))
|
|
|
|
|
- for index := range event.Guilds {
|
|
|
|
|
- guildsByID[event.Guilds[index].ID] = &event.Guilds[index]
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ // Build index of all available guilds.
|
|
|
|
|
+ guildsByID := make(map[discord.GuildID]*gateway.GuildCreateEvent, len(event.Guilds))
|
|
|
|
|
+ for index := range event.Guilds {
|
|
|
|
|
+ guildsByID[event.Guilds[index].ID] = &event.Guilds[index]
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- // Use GuildPositions for ordering (it's the canonical order).
|
|
|
|
|
- // Guilds not in any folder are "orphans" - add them directly to root.
|
|
|
|
|
- positions := event.UserSettings.GuildPositions
|
|
|
|
|
- // Fallback: GuildPositions shouldn't be nil but handle gracefully
|
|
|
|
|
- if len(positions) == 0 {
|
|
|
|
|
- positions = make([]discord.GuildID, 0, len(event.Guilds))
|
|
|
|
|
- for _, guildEvent := range event.Guilds {
|
|
|
|
|
- positions = append(positions, guildEvent.ID)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ // Use GuildPositions for ordering (it's the canonical order).
|
|
|
|
|
+ // Guilds not in any folder are "orphans" - add them directly to root.
|
|
|
|
|
+ positions := event.UserSettings.GuildPositions
|
|
|
|
|
+ // Fallback: GuildPositions shouldn't be nil but handle gracefully
|
|
|
|
|
+ if len(positions) == 0 {
|
|
|
|
|
+ positions = make([]discord.GuildID, 0, len(event.Guilds))
|
|
|
|
|
+ for _, guildEvent := range event.Guilds {
|
|
|
|
|
+ positions = append(positions, guildEvent.ID)
|
|
|
}
|
|
}
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- for _, guildID := range positions {
|
|
|
|
|
- // Already handled in folder processing below
|
|
|
|
|
- if guildsInFolders[guildID] {
|
|
|
|
|
- continue
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ for _, guildID := range positions {
|
|
|
|
|
+ // Already handled in folder processing below
|
|
|
|
|
+ if guildsInFolders[guildID] {
|
|
|
|
|
+ continue
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- // Orphan guild - add directly to root in order
|
|
|
|
|
- if guildEvent, ok := guildsByID[guildID]; ok {
|
|
|
|
|
- m.guildsTree.createGuildNode(root, guildEvent.Guild)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ // Orphan guild - add directly to root in order
|
|
|
|
|
+ if guildEvent, ok := guildsByID[guildID]; ok {
|
|
|
|
|
+ m.guildsTree.createGuildNode(root, guildEvent.Guild)
|
|
|
}
|
|
}
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- // Process folders (real folders and single-guild "folders")
|
|
|
|
|
- for _, folder := range event.UserSettings.GuildFolders {
|
|
|
|
|
- if folder.ID == 0 && len(folder.GuildIDs) == 1 {
|
|
|
|
|
- if guild, ok := guildsByID[folder.GuildIDs[0]]; ok {
|
|
|
|
|
- m.guildsTree.createGuildNode(root, guild.Guild)
|
|
|
|
|
- }
|
|
|
|
|
- } else {
|
|
|
|
|
- m.guildsTree.createFolderNode(folder, guildsByID)
|
|
|
|
|
|
|
+ // Process folders (real folders and single-guild "folders")
|
|
|
|
|
+ for _, folder := range event.UserSettings.GuildFolders {
|
|
|
|
|
+ if folder.ID == 0 && len(folder.GuildIDs) == 1 {
|
|
|
|
|
+ if guild, ok := guildsByID[folder.GuildIDs[0]]; ok {
|
|
|
|
|
+ m.guildsTree.createGuildNode(root, guild.Guild)
|
|
|
}
|
|
}
|
|
|
|
|
+ } else {
|
|
|
|
|
+ m.guildsTree.createFolderNode(folder, guildsByID)
|
|
|
}
|
|
}
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- m.guildsTree.SetCurrentNode(root)
|
|
|
|
|
- m.app.SetFocus(m.guildsTree)
|
|
|
|
|
- })
|
|
|
|
|
|
|
+ m.guildsTree.SetCurrentNode(root)
|
|
|
|
|
+ m.app.SetFocus(m.guildsTree)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func (m *Model) onMessageCreate(message *gateway.MessageCreateEvent) {
|
|
func (m *Model) onMessageCreate(message *gateway.MessageCreateEvent) {
|
|
|
selectedChannel := m.SelectedChannel()
|
|
selectedChannel := m.SelectedChannel()
|
|
|
if selectedChannel != nil && selectedChannel.ID == message.ChannelID {
|
|
if selectedChannel != nil && selectedChannel.ID == message.ChannelID {
|
|
|
m.removeTyper(message.Author.ID)
|
|
m.removeTyper(message.Author.ID)
|
|
|
- m.app.QueueUpdateDraw(func() {
|
|
|
|
|
- m.messagesList.addMessage(message.Message)
|
|
|
|
|
- })
|
|
|
|
|
|
|
+ m.messagesList.addMessage(message.Message)
|
|
|
} else {
|
|
} else {
|
|
|
if err := notifications.Notify(m.state, message, m.cfg); err != nil {
|
|
if err := notifications.Notify(m.state, message, m.cfg); err != nil {
|
|
|
slog.Error("failed to notify", "err", err, "channel_id", message.ChannelID, "message_id", message.ID)
|
|
slog.Error("failed to notify", "err", err, "channel_id", message.ChannelID, "message_id", message.ID)
|
|
@@ -163,14 +116,17 @@ func (m *Model) onMessageUpdate(message *gateway.MessageUpdateEvent) {
|
|
|
return
|
|
return
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- m.app.QueueUpdateDraw(func() {
|
|
|
|
|
- m.messagesList.setMessage(index, message.Message)
|
|
|
|
|
- })
|
|
|
|
|
|
|
+ m.messagesList.setMessage(index, message.Message)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func (m *Model) onMessageDelete(message *gateway.MessageDeleteEvent) {
|
|
func (m *Model) onMessageDelete(message *gateway.MessageDeleteEvent) {
|
|
|
- if selected := m.SelectedChannel(); selected != nil && selected.ID == message.ChannelID {
|
|
|
|
|
|
|
+ selectedChannel := m.SelectedChannel()
|
|
|
|
|
+ if selectedChannel == nil {
|
|
|
|
|
+ return
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ if selectedChannel.ID == message.ChannelID {
|
|
|
prevCursor := m.messagesList.Cursor()
|
|
prevCursor := m.messagesList.Cursor()
|
|
|
deletedIndex := slices.IndexFunc(m.messagesList.messages, func(m discord.Message) bool {
|
|
deletedIndex := slices.IndexFunc(m.messagesList.messages, func(m discord.Message) bool {
|
|
|
return m.ID == message.ID
|
|
return m.ID == message.ID
|
|
@@ -179,9 +135,7 @@ func (m *Model) onMessageDelete(message *gateway.MessageDeleteEvent) {
|
|
|
return
|
|
return
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- m.app.QueueUpdateDraw(func() {
|
|
|
|
|
- m.messagesList.deleteMessage(deletedIndex)
|
|
|
|
|
- })
|
|
|
|
|
|
|
+ m.messagesList.deleteMessage(deletedIndex)
|
|
|
|
|
|
|
|
// Keep cursor stable when possible after removal.
|
|
// Keep cursor stable when possible after removal.
|
|
|
newCursor := prevCursor
|
|
newCursor := prevCursor
|
|
@@ -201,9 +155,7 @@ func (m *Model) onMessageDelete(message *gateway.MessageDeleteEvent) {
|
|
|
}
|
|
}
|
|
|
if newCursor != prevCursor {
|
|
if newCursor != prevCursor {
|
|
|
// Avoid redundant cursor updates if nothing changed.
|
|
// Avoid redundant cursor updates if nothing changed.
|
|
|
- m.app.QueueUpdateDraw(func() {
|
|
|
|
|
- m.messagesList.SetCursor(newCursor)
|
|
|
|
|
- })
|
|
|
|
|
|
|
+ m.messagesList.SetCursor(newCursor)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -233,3 +185,19 @@ func (m *Model) onTypingStart(event *gateway.TypingStartEvent) {
|
|
|
|
|
|
|
|
m.addTyper(event.UserID)
|
|
m.addTyper(event.UserID)
|
|
|
}
|
|
}
|
|
|
|
|
+
|
|
|
|
|
+func (m *Model) onReadUpdate(event *read.UpdateEvent) {
|
|
|
|
|
+ // Use indexed node lookup to avoid walking the whole tree on every read
|
|
|
|
|
+ // event. This runs frequently while reading/typing across channels.
|
|
|
|
|
+ if event.GuildID.IsValid() {
|
|
|
|
|
+ if guildNode := m.guildsTree.findNodeByReference(event.GuildID); guildNode != nil {
|
|
|
|
|
+ m.guildsTree.setNodeLineStyle(guildNode, m.guildsTree.getGuildNodeStyle(event.GuildID))
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ // Channel style is always updated for the target channel regardless of
|
|
|
|
|
+ // whether it's in a guild or DM.
|
|
|
|
|
+ if channelNode := m.guildsTree.findNodeByReference(event.ChannelID); channelNode != nil {
|
|
|
|
|
+ m.guildsTree.setNodeLineStyle(channelNode, m.guildsTree.getChannelNodeStyle(event.ChannelID))
|
|
|
|
|
+ }
|
|
|
|
|
+}
|