149 lines
3.2 KiB
Go
149 lines
3.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"log/slog"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/atotto/clipboard"
|
|
"github.com/ayn2op/discordo/internal/constants"
|
|
"github.com/diamondburned/arikawa/v3/api"
|
|
"github.com/diamondburned/arikawa/v3/discord"
|
|
"github.com/diamondburned/arikawa/v3/utils/json/option"
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
)
|
|
|
|
type MessageInput struct {
|
|
*tview.TextArea
|
|
replyMessageID discord.MessageID
|
|
}
|
|
|
|
func newMessageInput() *MessageInput {
|
|
mi := &MessageInput{
|
|
TextArea: tview.NewTextArea(),
|
|
}
|
|
|
|
mi.SetTextStyle(tcell.StyleDefault.Background(tcell.GetColor(cfg.Theme.BackgroundColor)))
|
|
mi.SetClipboard(func(s string) {
|
|
_ = clipboard.WriteAll(s)
|
|
}, func() string {
|
|
text, _ := clipboard.ReadAll()
|
|
return text
|
|
})
|
|
|
|
mi.SetInputCapture(mi.onInputCapture)
|
|
mi.SetBackgroundColor(tcell.GetColor(cfg.Theme.BackgroundColor))
|
|
|
|
mi.SetTitleColor(tcell.GetColor(cfg.Theme.TitleColor))
|
|
mi.SetTitleAlign(tview.AlignLeft)
|
|
|
|
p := cfg.Theme.BorderPadding
|
|
mi.SetBorder(cfg.Theme.Border)
|
|
mi.SetBorderColor(tcell.GetColor(cfg.Theme.BorderColor))
|
|
mi.SetBorderPadding(p[0], p[1], p[2], p[3])
|
|
|
|
return mi
|
|
}
|
|
|
|
func (mi *MessageInput) reset() {
|
|
mi.replyMessageID = 0
|
|
mi.SetTitle("")
|
|
mi.SetText("", true)
|
|
}
|
|
|
|
func (mi *MessageInput) onInputCapture(event *tcell.EventKey) *tcell.EventKey {
|
|
switch event.Name() {
|
|
case cfg.Keys.MessageInput.Send:
|
|
mi.send()
|
|
return nil
|
|
case cfg.Keys.MessageInput.Editor:
|
|
mi.editor()
|
|
return nil
|
|
case cfg.Keys.MessageInput.Cancel:
|
|
mi.reset()
|
|
return nil
|
|
}
|
|
|
|
return event
|
|
}
|
|
|
|
func (mi *MessageInput) send() {
|
|
if !mainFlex.guildsTree.selectedChannelID.IsValid() {
|
|
return
|
|
}
|
|
|
|
text := strings.TrimSpace(mi.GetText())
|
|
if text == "" {
|
|
return
|
|
}
|
|
|
|
if mi.replyMessageID != 0 {
|
|
data := api.SendMessageData{
|
|
Content: text,
|
|
Reference: &discord.MessageReference{MessageID: mi.replyMessageID},
|
|
AllowedMentions: &api.AllowedMentions{RepliedUser: option.False},
|
|
}
|
|
|
|
if strings.HasPrefix(mi.GetTitle(), "[@]") {
|
|
data.AllowedMentions.RepliedUser = option.True
|
|
}
|
|
|
|
go func() {
|
|
if _, err := discordState.SendMessageComplex(mainFlex.guildsTree.selectedChannelID, data); err != nil {
|
|
slog.Error("failed to send message", "err", err)
|
|
}
|
|
}()
|
|
} else {
|
|
go func() {
|
|
if _, err := discordState.SendMessage(mainFlex.guildsTree.selectedChannelID, text); err != nil {
|
|
slog.Error("failed to send message", "err", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
mi.replyMessageID = 0
|
|
mi.reset()
|
|
|
|
mainFlex.messagesText.Highlight()
|
|
mainFlex.messagesText.ScrollToEnd()
|
|
}
|
|
|
|
func (mi *MessageInput) editor() {
|
|
e := cfg.Editor
|
|
if e == "default" {
|
|
e = os.Getenv("EDITOR")
|
|
}
|
|
|
|
f, err := os.CreateTemp("", constants.TmpFilePattern)
|
|
if err != nil {
|
|
slog.Error("failed to create temporary file", "err", err)
|
|
return
|
|
}
|
|
_, _ = f.WriteString(mi.GetText())
|
|
f.Close()
|
|
|
|
defer os.Remove(f.Name())
|
|
|
|
cmd := exec.Command(e, f.Name())
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
app.Suspend(func() {
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
slog.Error("failed to run command", "err", err, "command", cmd)
|
|
return
|
|
}
|
|
})
|
|
|
|
msg, err := os.ReadFile(f.Name())
|
|
if err != nil {
|
|
slog.Error("failed to read temporary file", "err", err, "name", f.Name())
|
|
return
|
|
}
|
|
|
|
mi.SetText(strings.TrimSpace(string(msg)), true)
|
|
}
|