154 lines
3.4 KiB
Go
154 lines
3.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/atotto/clipboard"
|
|
"github.com/ayn2op/discordo/internal/config"
|
|
"github.com/charmbracelet/log"
|
|
"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"
|
|
)
|
|
|
|
const tmpFilePattern = config.Name + "_*.md"
|
|
|
|
type MessageInput struct {
|
|
*tview.TextArea
|
|
cfg *config.Config
|
|
app *tview.Application
|
|
replyMessageID discord.MessageID
|
|
}
|
|
|
|
func newMessageInput(app *tview.Application, cfg *config.Config) *MessageInput {
|
|
mi := &MessageInput{
|
|
TextArea: tview.NewTextArea(),
|
|
cfg: cfg,
|
|
app: app,
|
|
}
|
|
|
|
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 mi.cfg.Keys.MessageInput.Send:
|
|
mi.send()
|
|
return nil
|
|
case mi.cfg.Keys.MessageInput.Editor:
|
|
mi.editor()
|
|
return nil
|
|
case mi.cfg.Keys.MessageInput.Cancel:
|
|
mi.reset()
|
|
return nil
|
|
}
|
|
|
|
return event
|
|
}
|
|
|
|
func (mi *MessageInput) send() {
|
|
if !layout.guildsTree.selectedChannelID.IsValid() {
|
|
return
|
|
}
|
|
|
|
text := strings.TrimSpace(mi.GetText())
|
|
if text == "" {
|
|
return
|
|
}
|
|
|
|
data := api.SendMessageData{
|
|
Content: text,
|
|
}
|
|
if mi.replyMessageID != 0 {
|
|
data.Reference = &discord.MessageReference{MessageID: mi.replyMessageID}
|
|
data.AllowedMentions = &api.AllowedMentions{RepliedUser: option.False}
|
|
|
|
if strings.HasPrefix(mi.GetTitle(), "[@]") {
|
|
data.AllowedMentions.RepliedUser = option.True
|
|
}
|
|
|
|
go func() {
|
|
if _, err := discordState.SendMessageComplex(layout.guildsTree.selectedChannelID, data); err != nil {
|
|
}
|
|
}()
|
|
}
|
|
|
|
go func() {
|
|
if _, err := discordState.SendMessageComplex(layout.guildsTree.selectedChannelID, data); err != nil {
|
|
log.Error("failed to send message in channel", "channel_id", layout.guildsTree.selectedChannelID, "err", err)
|
|
}
|
|
}()
|
|
|
|
mi.replyMessageID = 0
|
|
mi.reset()
|
|
|
|
layout.messagesText.Highlight()
|
|
layout.messagesText.ScrollToEnd()
|
|
}
|
|
|
|
func (mi *MessageInput) editor() {
|
|
e := mi.cfg.Editor
|
|
if e == "default" {
|
|
e = os.Getenv("EDITOR")
|
|
}
|
|
|
|
f, err := os.CreateTemp("", tmpFilePattern)
|
|
if err != nil {
|
|
log.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
|
|
|
|
mi.app.Suspend(func() {
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Error("failed to run command", "args", cmd.Args, "err", err)
|
|
return
|
|
}
|
|
})
|
|
|
|
msg, err := os.ReadFile(f.Name())
|
|
if err != nil {
|
|
log.Error("failed to read temporary file", "name", f.Name(), "err", err)
|
|
return
|
|
}
|
|
|
|
mi.SetText(strings.TrimSpace(string(msg)), true)
|
|
}
|