Add command to log out

2 files changed, 6 insertions, 0 deletions

diff --git a/ui/ui.go b/ui/ui.go
index 2f45f27..5e06d16 100644
--- a/ui/ui.go
+++ b/ui/ui.go
@@ -78,10 +78,12 @@ func (ui *GomuksUI) Render() {
 
 func (ui *GomuksUI) OnLogin() {
 	ui.SetView(ViewMain)
+	ui.app.SetFocus(ui.mainView)
 }
 
 func (ui *GomuksUI) OnLogout() {
 	ui.SetView(ViewLogin)
+	ui.app.SetFocus(ui.loginView)
 }
 
 func (ui *GomuksUI) SetView(name View) {
diff --git a/ui/view-main.go b/ui/view-main.go
index 7a94194..ac7f3d2 100644
--- a/ui/view-main.go
+++ b/ui/view-main.go
@@ -119,6 +119,7 @@ func (view *MainView) SendMessage(roomView *RoomView, text string) {
 
 func (view *MainView) sendTempMessage(roomView *RoomView, tempMessage ifc.Message, text string) {
 	defer debug.Recover()
+	debug.Print("Sending message", tempMessage.Type(), text)
 	eventID, err := view.matrix.SendMarkdownMessage(roomView.Room.ID, tempMessage.Type(), text)
 	if err != nil {
 		tempMessage.SetState(ifc.MessageStateFailed)
@@ -131,6 +132,7 @@ func (view *MainView) sendTempMessage(roomView *RoomView, tempMessage ifc.Messag
 		roomView.AddServiceMessage(fmt.Sprintf("Failed to send message: %v", err))
 		view.parent.Render()
 	} else {
+		debug.Print("Event ID received:", eventID)
 		roomView.MessageView().UpdateMessageID(tempMessage, eventID)
 	}
 }
@@ -167,6 +169,8 @@ func (view *MainView) HandleCommand(roomView *RoomView, command string, args []s
 		if err == nil {
 			view.AddRoom(room)
 		}
+	case "/logout":
+		view.matrix.Logout()
 	default:
 		roomView.AddServiceMessage("Unknown command.")
 	}