160 lines
3.3 KiB
Go
160 lines
3.3 KiB
Go
package ui
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/atotto/clipboard"
|
|
"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.InputField
|
|
|
|
app *Application
|
|
}
|
|
|
|
func newMessageInput(app *Application) *MessageInput {
|
|
mi := &MessageInput{
|
|
InputField: tview.NewInputField(),
|
|
|
|
app: app,
|
|
}
|
|
|
|
mi.SetFieldBackgroundColor(tview.Styles.PrimitiveBackgroundColor)
|
|
mi.SetPlaceholder("Message...")
|
|
mi.SetPlaceholderStyle(tcell.StyleDefault.Background(tview.Styles.PrimitiveBackgroundColor))
|
|
mi.SetInputCapture(mi.inputCapture)
|
|
|
|
mi.SetTitleAlign(tview.AlignLeft)
|
|
mi.SetBorder(true)
|
|
mi.SetBorderPadding(0, 0, 1, 1)
|
|
|
|
return mi
|
|
}
|
|
|
|
func (v *MessageInput) inputCapture(event *tcell.EventKey) *tcell.EventKey {
|
|
switch event.Name() {
|
|
case v.app.config.Keys.MessageInput.Send:
|
|
return v.sendMessage()
|
|
case v.app.config.Keys.MessageInput.LaunchEditor:
|
|
return v.openExternalEditor()
|
|
case v.app.config.Keys.MessageInput.Paste:
|
|
return v.pasteClipboard()
|
|
case "Esc":
|
|
v.
|
|
SetText("").
|
|
SetTitle("")
|
|
v.app.view.MessagesText.selected = -1
|
|
v.app.view.MessagesText.Highlight()
|
|
return nil
|
|
}
|
|
|
|
return event
|
|
}
|
|
|
|
func (v *MessageInput) sendMessage() *tcell.EventKey {
|
|
if v.app.view.ChannelsTree.selected == nil {
|
|
return nil
|
|
}
|
|
|
|
t := strings.TrimSpace(v.GetText())
|
|
if t == "" {
|
|
return nil
|
|
}
|
|
|
|
ms, err := v.app.state.Messages(v.app.view.ChannelsTree.selected.ID, v.app.config.MessagesLimit)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil
|
|
}
|
|
|
|
if len(v.app.view.MessagesText.GetHighlights()) != 0 {
|
|
mID, err := discord.ParseSnowflake(v.app.view.MessagesText.GetHighlights()[0])
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil
|
|
}
|
|
|
|
_, m := findMessageByID(ms, discord.MessageID(mID))
|
|
d := api.SendMessageData{
|
|
Content: t,
|
|
Reference: &discord.MessageReference{
|
|
MessageID: m.ID,
|
|
},
|
|
AllowedMentions: &api.AllowedMentions{RepliedUser: option.False},
|
|
}
|
|
|
|
// If the title of the input view has "[@]" as a prefix, send the message as a reply and mention the replied user.
|
|
if strings.HasPrefix(v.GetTitle(), "[@]") {
|
|
d.AllowedMentions.RepliedUser = option.True
|
|
}
|
|
|
|
go v.app.state.SendMessageComplex(m.ChannelID, d)
|
|
|
|
v.app.view.MessagesText.selected = -1
|
|
v.app.view.MessagesText.Highlight()
|
|
|
|
v.SetTitle("")
|
|
} else {
|
|
go v.app.state.SendMessage(v.app.view.ChannelsTree.selected.ID, t)
|
|
}
|
|
|
|
v.SetText("")
|
|
return nil
|
|
}
|
|
|
|
func (v *MessageInput) pasteClipboard() *tcell.EventKey {
|
|
text, err := clipboard.ReadAll()
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil
|
|
}
|
|
|
|
text = v.GetText() + text
|
|
v.SetText(text)
|
|
return nil
|
|
}
|
|
|
|
func (v *MessageInput) openExternalEditor() *tcell.EventKey {
|
|
e := os.Getenv("EDITOR")
|
|
if e == "" {
|
|
log.Println("environment variable EDITOR is empty")
|
|
return nil
|
|
}
|
|
|
|
f, err := os.CreateTemp(os.TempDir(), "discordo-*.md")
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil
|
|
}
|
|
defer os.Remove(f.Name())
|
|
|
|
cmd := exec.Command(e, f.Name())
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
|
|
v.app.Suspend(func() {
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
})
|
|
|
|
b, err := io.ReadAll(f)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil
|
|
}
|
|
|
|
v.SetText(string(b))
|
|
return nil
|
|
}
|