From ba379a1b4a97ee7eb56bed726f1c4edd76f76bb8 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Sat, 17 Mar 2018 14:32:01 +0200
Subject: Add basic username/id autocompletion

---
 view-main.go | 39 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

(limited to 'view-main.go')

diff --git a/view-main.go b/view-main.go
index 5a3a70c..219e603 100644
--- a/view-main.go
+++ b/view-main.go
@@ -17,10 +17,13 @@
 package main
 
 import (
+	"fmt"
 	"strings"
 	"time"
+	"unicode"
 
 	"github.com/gdamore/tcell"
+	"github.com/mattn/go-runewidth"
 	"maunium.net/go/gomatrix"
 	"maunium.net/go/tview"
 )
@@ -96,11 +99,43 @@ func (view *MainView) InputChanged(text string) {
 	}
 }
 
-func (view *MainView) InputTabComplete(text string, cursorOffset int) {
+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 *RoomView) AutocompleteUser(existingText string) (completions []string) {
+	for _, user := range view.room.GetMembers() {
+		if strings.HasPrefix(user.DisplayName, existingText) {
+			completions = append(completions, user.DisplayName)
+		} else if strings.HasPrefix(user.UserID, existingText) {
+			completions = append(completions, user.UserID)
+		}
+	}
+	return
+}
+
+func (view *MainView) InputTabComplete(text string, cursorOffset int) string {
 	roomView, _ := view.rooms[view.CurrentRoomID()]
 	if roomView != nil {
-		// text[0:cursorOffset]
+		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) {
-- 
cgit v1.2.3