messages.go

package ui

import (
	"bytes"
	"io/ioutil"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"regexp"
	"strings"

	"github.com/atotto/clipboard"
	"github.com/ayntgl/astatine"
	"github.com/craftxbox/discordo/discord"
	"github.com/gdamore/tcell/v2"
	"github.com/rainu/go-command-chain"
	"github.com/rivo/tview"
	"github.com/skratchdot/open-golang/open"
)

var linkRegex = regexp.MustCompile("https?://.+")

type MessagesTextView struct {
	*tview.TextView
	app *App
}

func NewMessagesTextView(app *App) *MessagesTextView {
	mtv := &MessagesTextView{
		TextView: tview.NewTextView(),
		app:      app,
	}

	mtv.SetDynamicColors(true)
	mtv.SetRegions(true)
	mtv.SetWordWrap(true)
	mtv.SetChangedFunc(func() {
		mtv.app.Draw()
	})
	mtv.SetTitleAlign(tview.AlignLeft)
	mtv.SetBorder(true)
	mtv.SetBorderPadding(0, 0, 1, 1)
	mtv.SetInputCapture(mtv.onInputCapture)
	return mtv
}

func (mtv *MessagesTextView) onInputCapture(e *tcell.EventKey) *tcell.EventKey {
	if mtv.app.SelectedChannel == nil {
		return nil
	}

	ms := mtv.app.SelectedChannel.Messages
	if len(ms) == 0 {
		return nil
	}

	switch e.Name() {
	case mtv.app.Config.Keys.SelectPreviousMessage:
		if len(mtv.app.MessagesTextView.GetHighlights()) == 0 {
			mtv.app.SelectedMessage = len(ms) - 1
		} else {
			mtv.app.SelectedMessage--
			if mtv.app.SelectedMessage < 0 {
				mtv.app.SelectedMessage = 0
			}
		}

		mtv.app.MessagesTextView.
			Highlight(ms[mtv.app.SelectedMessage].ID).
			ScrollToHighlight()
		return nil
	case mtv.app.Config.Keys.SelectNextMessage:
		if len(mtv.app.MessagesTextView.GetHighlights()) == 0 {
			mtv.app.SelectedMessage = len(ms) - 1
		} else {
			mtv.app.SelectedMessage++
			if mtv.app.SelectedMessage >= len(ms) {
				mtv.app.SelectedMessage = len(ms) - 1
			}
		}

		mtv.app.MessagesTextView.
			Highlight(ms[mtv.app.SelectedMessage].ID).
			ScrollToHighlight()
		return nil
	case mtv.app.Config.Keys.SelectFirstMessage:
		mtv.app.SelectedMessage = 0
		mtv.app.MessagesTextView.
			Highlight(ms[mtv.app.SelectedMessage].ID).
			ScrollToHighlight()
		return nil
	case mtv.app.Config.Keys.SelectLastMessage:
		mtv.app.SelectedMessage = len(ms) - 1
		mtv.app.MessagesTextView.
			Highlight(ms[mtv.app.SelectedMessage].ID).
			ScrollToHighlight()
		return nil
	case mtv.app.Config.Keys.OpenMessageActionsList:
		hs := mtv.app.MessagesTextView.GetHighlights()
		if len(hs) == 0 {
			return nil
		}

		_, m := discord.FindMessageByID(mtv.app.SelectedChannel.Messages, hs[0])
		if m == nil {
			return nil
		}

		actionsList := tview.NewList()
		actionsList.ShowSecondaryText(false)
		actionsList.SetDoneFunc(func() {
			mtv.app.
				SetRoot(mtv.app.MainFlex, true).
				SetFocus(mtv.app.MessagesTextView)
		})
		actionsList.SetTitle("Press the Escape key to close")
		actionsList.SetTitleAlign(tview.AlignLeft)
		actionsList.SetBorder(true)
		actionsList.SetBorderPadding(0, 0, 1, 1)

		if m.Author.ID == mtv.app.Session.State.User.ID {
			actionsList.AddItem("Delete", "", 'D', func() {
				mtv.app.Session.ChannelMessageDelete(mtv.app.SelectedChannel.ID, hs[0])
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessageInputField)
			})
			actionsList.AddItem("Edit", "", 'e', func() {
				mtv.app.MessageInputField.SetTitle("[E] Editing...")
				mtv.app.MessageInputField.SetText(m.Content)
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessageInputField)
			})
		} else if discord.HasPermission(mtv.app.Session.State, mtv.app.SelectedChannel.ID, astatine.PermissionManageMessages) {
			actionsList.AddItem("Delete", "", 'D', func() {
				mtv.app.Session.ChannelMessageDelete(mtv.app.SelectedChannel.ID, hs[0])
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessageInputField)
			})
		}

		// If the client user has `SEND_MESSAGES` permission, add the appropriate actions to the list.
		if discord.HasPermission(mtv.app.Session.State, mtv.app.SelectedChannel.ID, astatine.PermissionSendMessages) {
			actionsList.AddItem("Reply", "", 'r', func() {
				mtv.app.MessageInputField.SetTitle("Replying to " + m.Author.String())
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessageInputField)
			})

			actionsList.AddItem("Mention Reply", "", 'R', func() {
				mtv.app.MessageInputField.SetTitle("[@] Replying to " + m.Author.String())
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessageInputField)
			})
		}

		// If the referenced message exists, add the appropriate actions to the list.
		if m.ReferencedMessage != nil {
			actionsList.AddItem("Select Reply", "", 'm', func() {
				mtv.app.SelectedMessage, _ = discord.FindMessageByID(mtv.app.SelectedChannel.Messages, m.ReferencedMessage.ID)
				mtv.app.MessagesTextView.
					Highlight(m.ReferencedMessage.ID).
					ScrollToHighlight()
				mtv.app.
					SetRoot(mtv.app.MainFlex, true).
					SetFocus(mtv.app.MessagesTextView)
			})
		}

		// If the content of the message contains link(s), add the appropriate actions to the list.
		links := linkRegex.FindAllString(m.Content, -1)
		if len(links) != 0 {
			actionsList.AddItem("Open Link", "", 'l', func() {
				for _, l := range links {
					go open.Run(l)
				}
			})
		}

		// If the message contains attachments, add the appropriate actions to the actions list.
		if len(m.Attachments) != 0 {
			actionsList.AddItem("Download Attachment", "", 'd', func() {
				go mtv.downloadAttachment(m.Attachments)
				mtv.app.SetRoot(mtv.app.MainFlex, true)
			})
			actionsList.AddItem("Open Attachment", "", 'o', func() {
				go mtv.openAttachment(m.Attachments)
				mtv.app.SetRoot(mtv.app.MainFlex, true)
			})
		}

		actionsList.AddItem("Copy Content", "", 'c', func() {
			if err := clipboard.WriteAll(m.Content); err != nil {
				return
			}

			mtv.app.SetRoot(mtv.app.MainFlex, true)
			mtv.app.SetFocus(mtv.app.MessagesTextView)
		})
		actionsList.AddItem("Copy ID: "+m.ID, "", 'i', func() {
			if err := clipboard.WriteAll(m.ID); err != nil {
				return
			}

			mtv.app.SetRoot(mtv.app.MainFlex, true)
			mtv.app.SetFocus(mtv.app.MessagesTextView)
		})
		actionsList.AddItem("User ID: "+m.Author.ID, "", 'u', func() {
			if err := clipboard.WriteAll(m.Author.ID); err != nil {
				return
			}

			mtv.app.SetRoot(mtv.app.MainFlex, true)
			mtv.app.SetFocus(mtv.app.MessagesTextView)
		})

		mtv.app.SetRoot(actionsList, true)
		return nil
	case "Esc":
		mtv.app.SelectedMessage = -1
		mtv.app.SetFocus(mtv.app.MainFlex)
		mtv.app.MessagesTextView.
			Clear().
			Highlight()
		return nil
	}

	return e
}

func (mtv *MessagesTextView) downloadAttachment(as []*astatine.MessageAttachment) error {
	for _, a := range as {
		f, err := os.Create(filepath.Join(mtv.app.Config.General.AttachmentDownloadsDir, a.Filename))
		if err != nil {
			return err
		}
		defer f.Close()

		resp, err := http.Get(a.URL)
		if err != nil {
			return err
		}

		d, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			return err
		}

		f.Write(d)
	}

	return nil
}

func (mtv *MessagesTextView) openAttachment(as []*astatine.MessageAttachment) error {
	for _, a := range as {
		cacheDirPath, _ := os.UserCacheDir()
		f, err := os.Create(filepath.Join(cacheDirPath, a.Filename))
		if err != nil {
			return err
		}
		defer f.Close()

		resp, err := http.Get(a.URL)
		if err != nil {
			return err
		}

		d, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			return err
		}

		f.Write(d)
		go open.Run(f.Name())
	}

	return nil
}

type MessageInputField struct {
	*tview.InputField
	app *App
}

func NewMessageInputField(app *App) *MessageInputField {
	mi := &MessageInputField{
		InputField: tview.NewInputField(),
		app:        app,
	}

	mi.SetFieldBackgroundColor(tview.Styles.PrimitiveBackgroundColor)
	mi.SetPlaceholder("Message...")
	mi.SetPlaceholderStyle(tcell.StyleDefault.Background(tview.Styles.PrimitiveBackgroundColor))
	mi.SetTitleAlign(tview.AlignLeft)
	mi.SetBorder(true)
	mi.SetBorderPadding(0, 0, 1, 1)
	mi.SetInputCapture(mi.onInputCapture)
	return mi
}

func (mi *MessageInputField) onInputCapture(e *tcell.EventKey) *tcell.EventKey {
	switch e.Name() {
	case "Enter":
		if mi.app.SelectedChannel == nil {
			return nil
		}

		t := strings.TrimSpace(mi.app.MessageInputField.GetText())
		if t == "" {
			return nil
		}

		// I know the variable T already exists, but this is so people can use a space before their messages
		// to bypass the command handler and send a message starting with /
		if strings.HasPrefix(mi.app.MessageInputField.GetText(), "/") {
			var m *astatine.Message = nil
			if len(mi.app.MessagesTextView.GetHighlights()) > 0 {
				_, m = discord.FindMessageByID(mi.app.SelectedChannel.Messages, mi.app.MessagesTextView.GetHighlights()[0])
			}
			mi.SetTitle("Processing...")
			if HandleCommand(mi, t, m) {
				return nil
			} else {
				// recalcuate t because the text field has been modified by a non-terminating command
				t = strings.TrimSpace(mi.app.MessageInputField.GetText())
				if t == "" {
					return nil
				}
			}
		}

		if len(mi.app.MessagesTextView.GetHighlights()) != 0 {
			_, m := discord.FindMessageByID(mi.app.SelectedChannel.Messages, mi.app.MessagesTextView.GetHighlights()[0])
			if strings.HasPrefix(mi.app.MessageInputField.GetTitle(), "[E]") {
				d := &astatine.MessageEdit{
					ID:      m.ID,
					Channel: m.ChannelID,
					Content: &t,
				}
				go mi.app.Session.ChannelMessageEditComplex(d)
			} else {
				if m.Author.ID == mi.app.Session.State.User.ID && strings.HasPrefix(t, "s/") {
					go func() {
						output := &bytes.Buffer{}

						err := cmdchain.Builder().
							Join("echo", m.Content).
							Join("sed", "-Ee", t).
							Finalize().WithOutput(output).Run()
						if err == nil {
							c := output.String()
							d := &astatine.MessageEdit{
								ID:      m.ID,
								Channel: m.ChannelID,
								Content: &c,
							}
							mi.app.Session.ChannelMessageEditComplex(d)
						}
					}()
				} else {
					d := &astatine.MessageSend{
						Content:         t,
						Reference:       m.Reference(),
						AllowedMentions: &astatine.MessageAllowedMentions{RepliedUser: false},
					}
					if strings.HasPrefix(mi.app.MessageInputField.GetTitle(), "[@]") {
						d.AllowedMentions.RepliedUser = true
					} else {
						d.AllowedMentions.RepliedUser = false
					}

					go mi.app.Session.ChannelMessageSendComplex(m.ChannelID, d)
				}
			}

			mi.app.SelectedMessage = -1
			mi.app.MessagesTextView.Highlight()

			mi.app.MessageInputField.SetTitle("")
		} else {
			if strings.HasPrefix(t, "s/") {
				go func() {
					output := &bytes.Buffer{}
					_, m := discord.FindLatestMessageFrom(mi.app.SelectedChannel.Messages, mi.app.Session.State.User.ID)

					err := cmdchain.Builder().
						Join("echo", m.Content).
						Join("sed", "-Ee", t).
						Finalize().WithOutput(output).Run()
					if err == nil {
						c := output.String()
						d := &astatine.MessageEdit{
							ID:      m.ID,
							Channel: m.ChannelID,
							Content: &c,
						}
						mi.app.Session.ChannelMessageEditComplex(d)
					}
				}()
			} else {
				go mi.app.Session.ChannelMessageSend(mi.app.SelectedChannel.ID, t)
			}
		}

		mi.app.MessageInputField.SetText("")

		return nil
	case "Ctrl+V":
		text, _ := clipboard.ReadAll()
		text = mi.app.MessageInputField.GetText() + text
		mi.app.MessageInputField.SetText(text)

		return nil
	case "Esc":
		mi.app.MessageInputField.
			SetText("").
			SetTitle("")
		mi.app.SetFocus(mi.app.MainFlex)

		mi.app.SelectedMessage = -1
		mi.app.MessagesTextView.Highlight()

		return nil
	case mi.app.Config.Keys.OpenExternalEditor:
		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

		mi.app.Suspend(func() {
			err = cmd.Run()
			if err != nil {
				return
			}
		})

		b, err := ioutil.ReadAll(f)
		if err != nil {
			return nil
		}

		mi.app.MessageInputField.SetText(string(b))

		return nil
	}

	return e
}