From 72945c9a284b6858594f1e8a43743c397e90c380 Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Sun, 18 Mar 2018 21:24:03 +0200 Subject: Organize files --- view-main.go | 385 ----------------------------------------------------------- 1 file changed, 385 deletions(-) delete mode 100644 view-main.go (limited to 'view-main.go') diff --git a/view-main.go b/view-main.go deleted file mode 100644 index 2fd503a..0000000 --- a/view-main.go +++ /dev/null @@ -1,385 +0,0 @@ -// 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 . - -package main - -import ( - "fmt" - "sort" - "strings" - "time" - "unicode" - - "github.com/gdamore/tcell" - "github.com/mattn/go-runewidth" - "maunium.net/go/gomatrix" - "maunium.net/go/tview" -) - -type MainView struct { - *tview.Grid - - roomList *tview.List - roomView *tview.Pages - rooms map[string]*RoomView - input *AdvancedInputField - currentRoomIndex int - roomIDs []string - - matrix *MatrixContainer - debug DebugPrinter - gmx Gomuks - 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]*RoomView), - input: NewAdvancedInputField(), - - matrix: ui.matrix, - debug: ui.debug, - gmx: ui.gmx, - config: ui.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(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.status.Clear() - fmt.Fprintf(roomView.status, "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() - view.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": - view.matrix.client.LeaveRoom(room) - case "/join": - if len(args) == 0 { - view.AddServiceMessage(room, "Usage: /join ") - break - } - view.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 := 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) *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) SetRoomList(rooms []string) { - view.roomIDs = rooms - view.roomList.Clear() - view.roomView.Clear() - view.rooms = make(map[string]*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, 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 { - view.debug.Print("Failed to fetch history for", roomView.room.ID, err) - return - } - for _, evt := range history { - var room *RoomView - var message *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, PrependMessage) - } - } -} - -func (view *MainView) ProcessMessageEvent(evt *gomatrix.Event) (room *RoomView, message *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 *RoomView, message *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 -} -- cgit v1.2.3