gomuks/ui/room-view.go

441 lines
12 KiB
Go
Raw Normal View History

2018-03-15 18:45:52 +01:00
// gomuks - A terminal Matrix client written in Go.
2019-01-17 13:13:25 +01:00
// Copyright (C) 2019 Tulir Asokan
2018-03-15 18:45:52 +01:00
//
// This program is free software: you can redistribute it and/or modify
2019-01-17 13:13:25 +01:00
// it under the terms of the GNU Affero General Public License as published by
2018-03-15 18:45:52 +01:00
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2019-01-17 13:13:25 +01:00
// GNU Affero General Public License for more details.
2018-03-15 18:45:52 +01:00
//
2019-01-17 13:13:25 +01:00
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
2018-03-15 18:45:52 +01:00
package ui
2018-03-15 18:45:52 +01:00
import (
"fmt"
2018-03-22 18:51:20 +01:00
"path/filepath"
2018-04-21 18:41:19 +02:00
"sort"
2018-03-22 20:44:46 +01:00
"strconv"
2018-03-15 18:45:52 +01:00
"strings"
"time"
2018-03-15 18:45:52 +01:00
2019-03-30 17:51:26 +01:00
"github.com/kyokomi/emoji"
2018-04-21 18:41:19 +02:00
"github.com/mattn/go-runewidth"
2019-01-17 13:13:25 +01:00
2019-03-26 18:57:44 +01:00
"maunium.net/go/gomuks/debug"
2019-03-25 23:37:35 +01:00
"maunium.net/go/mauview"
2019-01-17 13:13:25 +01:00
"maunium.net/go/mautrix"
"maunium.net/go/tcell"
2018-06-01 23:44:21 +02:00
"maunium.net/go/gomuks/config"
"maunium.net/go/gomuks/interface"
2018-04-21 18:41:19 +02:00
"maunium.net/go/gomuks/lib/util"
2018-03-22 22:46:43 +01:00
"maunium.net/go/gomuks/matrix/rooms"
"maunium.net/go/gomuks/ui/messages"
"maunium.net/go/gomuks/ui/widget"
2018-03-15 18:45:52 +01:00
)
type RoomView struct {
2019-03-25 23:37:35 +01:00
topic *mauview.TextView
2018-03-17 00:27:30 +01:00
content *MessageView
2019-03-25 23:37:35 +01:00
status *mauview.TextField
userList *mauview.TextView
ulBorder *widget.Border
2019-03-25 23:37:35 +01:00
input *mauview.InputArea
2018-03-18 20:24:03 +01:00
Room *rooms.Room
2018-04-21 18:41:19 +02:00
2019-03-25 23:37:35 +01:00
topicScreen *mauview.ProxyScreen
contentScreen *mauview.ProxyScreen
statusScreen *mauview.ProxyScreen
inputScreen *mauview.ProxyScreen
ulBorderScreen *mauview.ProxyScreen
ulScreen *mauview.ProxyScreen
2019-03-30 17:51:26 +01:00
inputSubmitFunc func(room *RoomView, text string)
2019-03-25 23:37:35 +01:00
prevScreen mauview.Screen
2018-04-22 20:05:42 +02:00
parent *MainView
2018-06-01 23:43:56 +02:00
config *config.Config
2018-04-22 20:05:42 +02:00
2018-04-21 18:41:19 +02:00
typing []string
2018-04-22 20:05:42 +02:00
2018-04-21 18:41:19 +02:00
completions struct {
list []string
textCache string
time time.Time
}
2018-03-15 19:33:01 +01:00
}
2018-04-22 20:05:42 +02:00
func NewRoomView(parent *MainView, room *rooms.Room) *RoomView {
2018-03-15 18:45:52 +01:00
view := &RoomView{
2019-03-25 23:37:35 +01:00
topic: mauview.NewTextView(),
status: mauview.NewTextField(),
userList: mauview.NewTextView(),
ulBorder: widget.NewBorder(),
2019-03-25 23:37:35 +01:00
input: mauview.NewInputArea(),
2018-03-22 16:36:06 +01:00
Room: room,
2019-03-25 23:37:35 +01:00
2019-03-26 18:57:44 +01:00
topicScreen: &mauview.ProxyScreen{OffsetX: 0, OffsetY: 0, Height: TopicBarHeight},
contentScreen: &mauview.ProxyScreen{OffsetX: 0, OffsetY: StatusBarHeight},
statusScreen: &mauview.ProxyScreen{OffsetX: 0, Height: StatusBarHeight},
inputScreen: &mauview.ProxyScreen{OffsetX: 0},
2019-03-25 23:37:35 +01:00
ulBorderScreen: &mauview.ProxyScreen{OffsetY: StatusBarHeight, Width: UserListBorderWidth},
2019-03-26 18:57:44 +01:00
ulScreen: &mauview.ProxyScreen{OffsetY: StatusBarHeight, Width: UserListWidth},
2019-03-25 23:37:35 +01:00
2019-03-26 18:57:44 +01:00
parent: parent,
config: parent.config,
2018-03-15 18:45:52 +01:00
}
2018-04-14 14:33:20 +02:00
view.content = NewMessageView(view)
2018-03-22 15:44:24 +01:00
view.input.
2019-03-25 23:37:35 +01:00
SetBackgroundColor(tcell.ColorDefault).
2018-03-22 15:44:24 +01:00
SetPlaceholder("Send a message...").
2019-03-25 23:37:35 +01:00
SetPlaceholderTextColor(tcell.ColorGray).
2018-04-21 18:41:19 +02:00
SetTabCompleteFunc(view.InputTabComplete)
2018-03-22 15:44:24 +01:00
2018-03-15 18:45:52 +01:00
view.topic.
2018-03-16 15:24:11 +01:00
SetText(strings.Replace(room.GetTopic(), "\n", " ", -1)).
2019-03-26 18:57:44 +01:00
SetTextColor(tcell.ColorWhite).
SetBackgroundColor(tcell.ColorDarkGreen)
2018-03-22 15:44:24 +01:00
2018-03-15 18:45:52 +01:00
view.status.SetBackgroundColor(tcell.ColorDimGray)
2018-03-22 15:44:24 +01:00
2018-03-26 21:03:30 +02:00
view.userList.
SetDynamicColors(true).
SetWrap(false)
2018-03-22 15:44:24 +01:00
return view
}
2018-03-22 18:51:20 +01:00
func (view *RoomView) logPath(dir string) string {
return filepath.Join(dir, fmt.Sprintf("%s.gmxlog", view.Room.ID))
}
2018-03-22 15:44:24 +01:00
func (view *RoomView) SetInputSubmitFunc(fn func(room *RoomView, text string)) *RoomView {
2019-03-30 17:51:26 +01:00
view.inputSubmitFunc = fn
2018-03-22 15:44:24 +01:00
return view
}
func (view *RoomView) SetInputChangedFunc(fn func(room *RoomView, text string)) *RoomView {
view.input.SetChangedFunc(func(text string) {
fn(view, text)
})
return view
}
func (view *RoomView) SetInputText(newText string) *RoomView {
2019-03-25 23:37:35 +01:00
view.input.SetTextAndMoveCursor(newText)
2018-03-15 18:45:52 +01:00
return view
}
2018-03-22 15:44:24 +01:00
func (view *RoomView) GetInputText() string {
return view.input.GetText()
}
2019-03-25 23:37:35 +01:00
func (view *RoomView) Focus() {
view.input.Focus()
2018-03-22 15:44:24 +01:00
}
2019-03-25 23:37:35 +01:00
func (view *RoomView) Blur() {
view.input.Blur()
2018-03-22 15:44:24 +01:00
}
2018-04-21 18:41:19 +02:00
func (view *RoomView) GetStatus() string {
var buf strings.Builder
if len(view.completions.list) > 0 {
2019-01-17 13:13:25 +01:00
if view.completions.textCache != view.input.GetText() || view.completions.time.Add(10 * time.Second).Before(time.Now()) {
2018-04-21 18:41:19 +02:00
view.completions.list = []string{}
} else {
buf.WriteString(strings.Join(view.completions.list, ", "))
buf.WriteString(" - ")
}
}
if len(view.typing) == 1 {
buf.WriteString("Typing: " + view.typing[0])
buf.WriteString(" - ")
} else if len(view.typing) > 1 {
2019-03-25 23:37:35 +01:00
_, _ = fmt.Fprintf(&buf,
2018-04-21 18:41:19 +02:00
"Typing: %s and %s - ",
strings.Join(view.typing[:len(view.typing)-1], ", "), view.typing[len(view.typing)-1])
}
return strings.TrimSuffix(buf.String(), " - ")
}
2019-03-25 23:37:35 +01:00
// Constants defining the size of the room view grid.
const (
UserListBorderWidth = 1
UserListWidth = 20
StaticHorizontalSpace = UserListBorderWidth + UserListWidth
2019-03-26 18:57:44 +01:00
TopicBarHeight = 1
StatusBarHeight = 1
2019-03-25 23:37:35 +01:00
2019-03-26 18:57:44 +01:00
MaxInputHeight = 5
2019-03-25 23:37:35 +01:00
)
func (view *RoomView) Draw(screen mauview.Screen) {
width, height := screen.Size()
if width <= 0 || height <= 0 {
return
}
2018-03-17 00:27:30 +01:00
2019-03-25 23:37:35 +01:00
if view.prevScreen != screen {
view.topicScreen.Parent = screen
view.contentScreen.Parent = screen
view.statusScreen.Parent = screen
view.inputScreen.Parent = screen
view.ulBorderScreen.Parent = screen
view.ulScreen.Parent = screen
view.prevScreen = screen
}
2019-03-25 23:37:35 +01:00
view.input.PrepareDraw(width)
inputHeight := view.input.GetTextHeight()
if inputHeight > MaxInputHeight {
inputHeight = MaxInputHeight
} else if inputHeight < 1 {
inputHeight = 1
}
2019-03-26 18:57:44 +01:00
contentHeight := height - inputHeight - TopicBarHeight - StatusBarHeight
contentWidth := width - StaticHorizontalSpace
if view.config.Preferences.HideUserList {
contentWidth = width
}
2019-03-25 23:37:35 +01:00
view.topicScreen.Width = width
view.contentScreen.Width = contentWidth
view.contentScreen.Height = contentHeight
view.statusScreen.OffsetY = view.contentScreen.YEnd()
view.statusScreen.Width = width
view.inputScreen.Width = width
view.inputScreen.OffsetY = view.statusScreen.YEnd()
view.inputScreen.Height = inputHeight
view.ulBorderScreen.OffsetX = view.contentScreen.XEnd()
view.ulBorderScreen.Height = contentHeight
view.ulScreen.OffsetX = view.ulBorderScreen.XEnd()
view.ulScreen.Height = contentHeight
2018-03-15 18:45:52 +01:00
// Draw everything
2019-03-25 23:37:35 +01:00
view.topic.Draw(view.topicScreen)
view.content.Draw(view.contentScreen)
2018-04-21 18:41:19 +02:00
view.status.SetText(view.GetStatus())
2019-03-25 23:37:35 +01:00
view.status.Draw(view.statusScreen)
view.input.Draw(view.inputScreen)
2018-06-01 23:43:56 +02:00
if !view.config.Preferences.HideUserList {
2019-03-25 23:37:35 +01:00
view.ulBorder.Draw(view.ulBorderScreen)
view.userList.Draw(view.ulScreen)
}
2018-03-15 18:45:52 +01:00
}
2019-03-25 23:37:35 +01:00
func (view *RoomView) OnKeyEvent(event mauview.KeyEvent) bool {
2019-03-30 17:51:26 +01:00
msgView := view.MessageView()
switch event.Key() {
case tcell.KeyPgUp:
if msgView.IsAtTop() {
go view.parent.LoadHistory(view.Room.ID)
}
msgView.AddScrollOffset(+msgView.Height() / 2)
return true
case tcell.KeyPgDn:
msgView.AddScrollOffset(-msgView.Height() / 2)
return true
case tcell.KeyEnter:
if event.Modifiers() & tcell.ModShift == 0 && event.Modifiers() & tcell.ModCtrl == 0 && view.inputSubmitFunc != nil {
view.inputSubmitFunc(view, view.input.GetText())
return true
}
}
2019-03-25 23:37:35 +01:00
return view.input.OnKeyEvent(event)
}
func (view *RoomView) OnPasteEvent(event mauview.PasteEvent) bool {
return view.input.OnPasteEvent(event)
}
func (view *RoomView) OnMouseEvent(event mauview.MouseEvent) bool {
2019-03-26 18:57:44 +01:00
switch {
case view.contentScreen.IsInArea(event.Position()):
return view.content.OnMouseEvent(view.contentScreen.OffsetMouseEvent(event))
case view.topicScreen.IsInArea(event.Position()):
return view.topic.OnMouseEvent(view.topicScreen.OffsetMouseEvent(event))
case view.inputScreen.IsInArea(event.Position()):
return view.input.OnMouseEvent(view.inputScreen.OffsetMouseEvent(event))
}
return false
2019-03-25 23:37:35 +01:00
}
2018-04-21 18:41:19 +02:00
func (view *RoomView) SetCompletions(completions []string) {
view.completions.list = completions
view.completions.textCache = view.input.GetText()
view.completions.time = time.Now()
2018-03-18 20:24:03 +01:00
}
2018-03-15 18:45:52 +01:00
func (view *RoomView) SetTyping(users []string) {
2018-03-16 15:24:11 +01:00
for index, user := range users {
2018-03-18 20:24:03 +01:00
member := view.Room.GetMember(user)
2018-03-16 15:24:11 +01:00
if member != nil {
users[index] = member.Displayname
2018-03-16 15:24:11 +01:00
}
}
2018-04-21 18:41:19 +02:00
view.typing = users
}
type completion struct {
displayName string
id string
2018-03-15 18:45:52 +01:00
}
2018-04-22 19:13:57 +02:00
func (view *RoomView) autocompleteUser(existingText string) (completions []completion) {
textWithoutPrefix := strings.TrimPrefix(existingText, "@")
for userID, user := range view.Room.GetMembers() {
if user.Displayname == textWithoutPrefix || userID == existingText {
2018-04-21 18:41:19 +02:00
// Exact match, return that.
return []completion{{user.Displayname, userID}}
2018-04-21 18:41:19 +02:00
}
if strings.HasPrefix(user.Displayname, textWithoutPrefix) || strings.HasPrefix(userID, existingText) {
completions = append(completions, completion{user.Displayname, userID})
}
}
return
}
2018-04-22 19:13:57 +02:00
func (view *RoomView) autocompleteRoom(existingText string) (completions []completion) {
2018-04-22 20:05:42 +02:00
for _, room := range view.parent.rooms {
alias := room.Room.GetCanonicalAlias()
if alias == existingText {
// Exact match, return that.
return []completion{{alias, room.Room.ID}}
}
if strings.HasPrefix(alias, existingText) {
completions = append(completions, completion{alias, room.Room.ID})
continue
}
}
return
2018-04-21 18:41:19 +02:00
}
2019-03-30 17:51:26 +01:00
func (view *RoomView) autocompleteEmoji(word string) (completions []string) {
if len(word) == 0 || word[0] != ':' {
return
}
for name, value := range emoji.CodeMap() {
if name == word {
return []string{value}
} else if strings.HasPrefix(name, word) {
completions = append(completions, name)
}
}
return
}
2018-04-21 18:41:19 +02:00
func (view *RoomView) InputTabComplete(text string, cursorOffset int) {
2019-03-26 21:09:10 +01:00
debug.Print("Tab completing", cursorOffset, text)
2018-04-21 18:41:19 +02:00
str := runewidth.Truncate(text, cursorOffset, "")
word := findWordToTabComplete(str)
startIndex := len(str) - len(word)
var strCompletions []string
var strCompletion string
2018-04-22 19:13:57 +02:00
completions := view.autocompleteUser(word)
completions = append(completions, view.autocompleteRoom(word)...)
2018-04-21 18:41:19 +02:00
if len(completions) == 1 {
completion := completions[0]
strCompletion = fmt.Sprintf("[%s](https://matrix.to/#/%s)", completion.displayName, completion.id)
if startIndex == 0 {
strCompletion = strCompletion + ": "
}
} else if len(completions) > 1 {
for _, completion := range completions {
strCompletions = append(strCompletions, completion.displayName)
}
}
2019-03-30 17:51:26 +01:00
strCompletions = append(strCompletions, view.autocompleteEmoji(word)...)
2018-04-21 18:41:19 +02:00
if len(strCompletions) > 0 {
strCompletion = util.LongestCommonPrefix(strCompletions)
sort.Sort(sort.StringSlice(strCompletions))
}
if len(strCompletion) > 0 {
text = str[0:startIndex] + strCompletion + text[len(str):]
}
view.input.SetTextAndMoveCursor(text)
view.SetCompletions(strCompletions)
}
2018-03-17 00:27:30 +01:00
func (view *RoomView) MessageView() *MessageView {
return view.content
}
2018-03-15 19:33:01 +01:00
func (view *RoomView) MxRoom() *rooms.Room {
return view.Room
}
2018-03-16 15:24:11 +01:00
func (view *RoomView) UpdateUserList() {
var joined strings.Builder
var invited strings.Builder
for userID, user := range view.Room.GetMembers() {
2018-03-16 15:24:11 +01:00
if user.Membership == "join" {
joined.WriteString(widget.AddColor(user.Displayname, widget.GetHashColorName(userID)))
joined.WriteRune('\n')
} else if user.Membership == "invite" {
invited.WriteString(widget.AddColor(user.Displayname, widget.GetHashColorName(userID)))
invited.WriteRune('\n')
2018-03-16 15:24:11 +01:00
}
}
view.userList.Clear()
fmt.Fprintf(view.userList, "%s\n", joined.String())
if invited.Len() > 0 {
fmt.Fprintf(view.userList, "\nInvited:\n%s", invited.String())
}
}
2018-11-13 23:00:35 +01:00
func (view *RoomView) newUIMessage(id, sender string, msgtype mautrix.MessageType, text string, timestamp time.Time) messages.UIMessage {
2018-03-18 20:24:03 +01:00
member := view.Room.GetMember(sender)
displayname := sender
if member != nil {
displayname = member.Displayname
}
msg := messages.NewTextMessage(id, sender, displayname, msgtype, text, timestamp)
return msg
2018-03-22 20:44:46 +01:00
}
2018-11-13 23:00:35 +01:00
func (view *RoomView) NewTempMessage(msgtype mautrix.MessageType, text string) ifc.Message {
2018-03-22 20:44:46 +01:00
now := time.Now()
id := strconv.FormatInt(now.UnixNano(), 10)
sender := ""
if ownerMember := view.Room.GetMember(view.Room.GetSessionOwner()); ownerMember != nil {
sender = ownerMember.Displayname
}
message := view.newUIMessage(id, sender, msgtype, text, now)
message.SetState(ifc.MessageStateSending)
view.AddMessage(message, ifc.AppendMessage)
2018-03-22 20:44:46 +01:00
return message
}
func (view *RoomView) AddServiceMessage(text string) {
message := view.newUIMessage(view.parent.matrix.Client().TxnID(), "*", "gomuks.service", text, time.Now())
message.SetIsService(true)
view.AddMessage(message, ifc.AppendMessage)
}
func (view *RoomView) AddMessage(message ifc.Message, direction ifc.MessageDirection) {
view.content.AddMessage(message, direction)
2018-03-15 18:45:52 +01:00
}