aboutsummaryrefslogtreecommitdiff
path: root/ui/view-main.go
diff options
context:
space:
mode:
Diffstat (limited to 'ui/view-main.go')
-rw-r--r--ui/view-main.go387
1 files changed, 387 insertions, 0 deletions
diff --git a/ui/view-main.go b/ui/view-main.go
new file mode 100644
index 0000000..8ec482e
--- /dev/null
+++ b/ui/view-main.go
@@ -0,0 +1,387 @@
+// gomuks - A terminal Matrix client written in Go.
+// Copyright (C) 2018 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// 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
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package ui
+
+import (
+ "fmt"
+ "sort"
+ "strings"
+ "time"
+ "unicode"
+
+ "github.com/gdamore/tcell"
+ "github.com/mattn/go-runewidth"
+ "maunium.net/go/gomatrix"
+ "maunium.net/go/gomuks/config"
+ "maunium.net/go/gomuks/interface"
+ "maunium.net/go/gomuks/ui/debug"
+ "maunium.net/go/gomuks/ui/types"
+ "maunium.net/go/gomuks/ui/widget"
+ "maunium.net/go/tview"
+)
+
+type MainView struct {
+ *tview.Grid
+
+ roomList *tview.List
+ roomView *tview.Pages
+ rooms map[string]*widget.RoomView
+ input *widget.AdvancedInputField
+ currentRoomIndex int
+ roomIDs []string
+
+ matrix ifc.MatrixContainer
+ gmx ifc.Gomuks
+ config *config.Config
+ parent *GomuksUI
+}
+
+func (view *MainView) addItem(p tview.Primitive, x, y, w, h int) {
+ view.Grid.AddItem(p, x, y, w, h, 0, 0, false)
+}
+
+func (ui *GomuksUI) NewMainView() tview.Primitive {
+ mainView := &MainView{
+ Grid: tview.NewGrid(),
+ roomList: tview.NewList(),
+ roomView: tview.NewPages(),
+ rooms: make(map[string]*widget.RoomView),
+ input: widget.NewAdvancedInputField(),
+
+ matrix: ui.gmx.MatrixContainer(),
+ gmx: ui.gmx,
+ config: ui.gmx.Config(),
+ parent: ui,
+ }
+
+ mainView.SetColumns(30, 1, 0).SetRows(0, 1)
+
+ mainView.roomList.
+ ShowSecondaryText(false).
+ SetSelectedBackgroundColor(tcell.ColorDarkGreen).
+ SetSelectedTextColor(tcell.ColorWhite).
+ SetBorderPadding(0, 0, 1, 0)
+
+ mainView.input.
+ SetDoneFunc(mainView.InputDone).
+ SetChangedFunc(mainView.InputChanged).
+ SetTabCompleteFunc(mainView.InputTabComplete).
+ SetFieldBackgroundColor(tcell.ColorDefault).
+ SetPlaceholder("Send a message...").
+ SetPlaceholderExtColor(tcell.ColorGray).
+ SetInputCapture(mainView.InputCapture)
+
+ mainView.addItem(mainView.roomList, 0, 0, 2, 1)
+ mainView.addItem(widget.NewBorder(), 0, 1, 2, 1)
+ mainView.addItem(mainView.roomView, 0, 2, 1, 1)
+ mainView.AddItem(mainView.input, 1, 2, 1, 1, 0, 0, true)
+
+ ui.mainView = mainView
+
+ return mainView
+}
+
+func (view *MainView) InputChanged(text string) {
+ if len(text) == 0 {
+ go view.matrix.SendTyping(view.CurrentRoomID(), false)
+ } else if text[0] != '/' {
+ go view.matrix.SendTyping(view.CurrentRoomID(), true)
+ }
+}
+
+func findWordToTabComplete(text string) string {
+ output := ""
+ runes := []rune(text)
+ for i := len(runes) - 1; i >= 0; i-- {
+ if unicode.IsSpace(runes[i]) {
+ break
+ }
+ output = string(runes[i]) + output
+ }
+ return output
+}
+
+func (view *MainView) InputTabComplete(text string, cursorOffset int) string {
+ roomView, _ := view.rooms[view.CurrentRoomID()]
+ if roomView != nil {
+ str := runewidth.Truncate(text, cursorOffset, "")
+ word := findWordToTabComplete(str)
+ userCompletions := roomView.AutocompleteUser(word)
+ if len(userCompletions) == 1 {
+ text = str[0:len(str)-len(word)] + userCompletions[0] + text[len(str):]
+ } else if len(userCompletions) > 1 && len(userCompletions) < 6 {
+ roomView.SetStatus(fmt.Sprintf("Completions: %s", strings.Join(userCompletions, ", ")))
+ }
+ }
+ return text
+}
+
+func (view *MainView) InputDone(key tcell.Key) {
+ if key == tcell.KeyEnter {
+ room, text := view.CurrentRoomID(), view.input.GetText()
+ if len(text) == 0 {
+ return
+ } else if text[0] == '/' {
+ args := strings.SplitN(text, " ", 2)
+ command := strings.ToLower(args[0])
+ args = args[1:]
+ go view.HandleCommand(room, command, args)
+ } else {
+ go view.matrix.SendMessage(room, text)
+ }
+ view.input.SetText("")
+ }
+}
+
+func (view *MainView) HandleCommand(room, command string, args []string) {
+ view.gmx.Recover()
+ debug.Print("Handling command", command, args)
+ switch command {
+ case "/quit":
+ view.gmx.Stop()
+ case "/clearcache":
+ view.config.Session.Clear()
+ view.gmx.Stop()
+ case "/part":
+ fallthrough
+ case "/leave":
+ debug.Print(view.matrix.LeaveRoom(room))
+ case "/join":
+ if len(args) == 0 {
+ view.AddServiceMessage(room, "Usage: /join <room>")
+ break
+ }
+ debug.Print(view.matrix.JoinRoom(args[0]))
+ default:
+ view.AddServiceMessage(room, "Unknown command.")
+ }
+}
+
+func (view *MainView) InputCapture(key *tcell.EventKey) *tcell.EventKey {
+ k := key.Key()
+ if key.Modifiers() == tcell.ModCtrl {
+ if k == tcell.KeyDown {
+ view.SwitchRoom(view.currentRoomIndex + 1)
+ view.roomList.SetCurrentItem(view.currentRoomIndex)
+ } else if k == tcell.KeyUp {
+ view.SwitchRoom(view.currentRoomIndex - 1)
+ view.roomList.SetCurrentItem(view.currentRoomIndex)
+ } else {
+ return key
+ }
+ } else if k == tcell.KeyPgUp || k == tcell.KeyPgDn {
+ msgView := view.rooms[view.CurrentRoomID()].MessageView()
+ if k == tcell.KeyPgUp {
+ msgView.PageUp()
+ } else {
+ msgView.PageDown()
+ }
+ } else {
+ return key
+ }
+ return nil
+}
+
+func (view *MainView) CurrentRoomID() string {
+ if len(view.roomIDs) == 0 {
+ return ""
+ }
+ return view.roomIDs[view.currentRoomIndex]
+}
+
+func (view *MainView) SwitchRoom(roomIndex int) {
+ if roomIndex < 0 {
+ roomIndex = len(view.roomIDs) - 1
+ }
+ view.currentRoomIndex = roomIndex % len(view.roomIDs)
+ view.roomView.SwitchToPage(view.CurrentRoomID())
+ view.roomList.SetCurrentItem(roomIndex)
+ view.parent.Render()
+}
+
+func (view *MainView) addRoom(index int, room string) {
+ roomStore := view.matrix.GetRoom(room)
+
+ view.roomList.AddItem(roomStore.GetTitle(), "", 0, func() {
+ view.SwitchRoom(index)
+ })
+ if !view.roomView.HasPage(room) {
+ roomView := widget.NewRoomView(view, roomStore)
+ view.rooms[room] = roomView
+ view.roomView.AddPage(room, roomView, true, false)
+ roomView.UpdateUserList()
+ view.GetHistory(room)
+ }
+}
+
+func (view *MainView) GetRoom(id string) *widget.RoomView {
+ return view.rooms[id]
+}
+
+func (view *MainView) HasRoom(room string) bool {
+ for _, existingRoom := range view.roomIDs {
+ if existingRoom == room {
+ return true
+ }
+ }
+ return false
+}
+
+func (view *MainView) AddRoom(room string) {
+ if view.HasRoom(room) {
+ return
+ }
+ view.roomIDs = append(view.roomIDs, room)
+ view.addRoom(len(view.roomIDs)-1, room)
+}
+
+func (view *MainView) RemoveRoom(room string) {
+ if !view.HasRoom(room) {
+ return
+ }
+ removeIndex := 0
+ if view.CurrentRoomID() == room {
+ removeIndex = view.currentRoomIndex
+ view.SwitchRoom(view.currentRoomIndex - 1)
+ } else {
+ removeIndex = sort.StringSlice(view.roomIDs).Search(room)
+ }
+ view.roomList.RemoveItem(removeIndex)
+ view.roomIDs = append(view.roomIDs[:removeIndex], view.roomIDs[removeIndex+1:]...)
+ view.roomView.RemovePage(room)
+ delete(view.rooms, room)
+ view.Render()
+}
+
+func (view *MainView) SetRooms(rooms []string) {
+ view.roomIDs = rooms
+ view.roomList.Clear()
+ view.roomView.Clear()
+ view.rooms = make(map[string]*widget.RoomView)
+ for index, room := range rooms {
+ view.addRoom(index, room)
+ }
+ view.SwitchRoom(0)
+}
+
+func (view *MainView) SetTyping(room string, users []string) {
+ roomView, ok := view.rooms[room]
+ if ok {
+ roomView.SetTyping(users)
+ view.parent.Render()
+ }
+}
+
+func (view *MainView) AddServiceMessage(room, message string) {
+ roomView, ok := view.rooms[room]
+ if ok {
+ messageView := roomView.MessageView()
+ message := messageView.NewMessage("", "*", message, time.Now())
+ messageView.AddMessage(message, widget.AppendMessage)
+ view.parent.Render()
+ }
+}
+
+func (view *MainView) Render() {
+ view.parent.Render()
+}
+
+func (view *MainView) GetHistory(room string) {
+ roomView := view.rooms[room]
+ history, _, err := view.matrix.GetHistory(roomView.Room.ID, view.config.Session.NextBatch, 50)
+ if err != nil {
+ debug.Print("Failed to fetch history for", roomView.Room.ID, err)
+ return
+ }
+ for _, evt := range history {
+ var room *widget.RoomView
+ var message *types.Message
+ if evt.Type == "m.room.message" {
+ room, message = view.ProcessMessageEvent(&evt)
+ } else if evt.Type == "m.room.member" {
+ room, message = view.ProcessMembershipEvent(&evt, false)
+ }
+ if room != nil && message != nil {
+ room.AddMessage(message, widget.PrependMessage)
+ }
+ }
+}
+
+func (view *MainView) ProcessMessageEvent(evt *gomatrix.Event) (room *widget.RoomView, message *types.Message) {
+ room = view.GetRoom(evt.RoomID)
+ if room != nil {
+ text, _ := evt.Content["body"].(string)
+ message = room.NewMessage(evt.ID, evt.Sender, text, unixToTime(evt.Timestamp))
+ }
+ return
+}
+
+func (view *MainView) processOwnMembershipChange(evt *gomatrix.Event) {
+ membership, _ := evt.Content["membership"].(string)
+ prevMembership := "leave"
+ if evt.Unsigned.PrevContent != nil {
+ prevMembership, _ = evt.Unsigned.PrevContent["membership"].(string)
+ }
+ if membership == prevMembership {
+ return
+ }
+ if membership == "join" {
+ view.AddRoom(evt.RoomID)
+ } else if membership == "leave" {
+ view.RemoveRoom(evt.RoomID)
+ }
+}
+
+func (view *MainView) ProcessMembershipEvent(evt *gomatrix.Event, new bool) (room *widget.RoomView, message *types.Message) {
+ if new && evt.StateKey != nil && *evt.StateKey == view.config.Session.MXID {
+ view.processOwnMembershipChange(evt)
+ }
+
+ room = view.GetRoom(evt.RoomID)
+ if room != nil {
+ membership, _ := evt.Content["membership"].(string)
+ var sender, text string
+ if membership == "invite" {
+ sender = "---"
+ text = fmt.Sprintf("%s invited %s.", evt.Sender, *evt.StateKey)
+ } else if membership == "join" {
+ sender = "-->"
+ text = fmt.Sprintf("%s joined the room.", *evt.StateKey)
+ } else if membership == "leave" {
+ sender = "<--"
+ if evt.Sender != *evt.StateKey {
+ reason, _ := evt.Content["reason"].(string)
+ text = fmt.Sprintf("%s kicked %s: %s", evt.Sender, *evt.StateKey, reason)
+ } else {
+ text = fmt.Sprintf("%s left the room.", *evt.StateKey)
+ }
+ } else {
+ room = nil
+ return
+ }
+ message = room.NewMessage(evt.ID, sender, text, unixToTime(evt.Timestamp))
+ }
+ return
+}
+
+func unixToTime(unix int64) time.Time {
+ timestamp := time.Now()
+ if unix != 0 {
+ timestamp = time.Unix(unix/1000, unix%1000*1000)
+ }
+ return timestamp
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-26 15:31:42 +0000