Add support for CMP_POINT

Fixes #72

1 files changed, 21 insertions, 9 deletions

diff --git a/complete.go b/complete.go
index 185d1e8..168657e 100644
--- a/complete.go
+++ b/complete.go
@@ -10,14 +10,16 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"strconv"
 
 	"github.com/posener/complete/cmd"
 	"github.com/posener/complete/match"
 )
 
 const (
-	envComplete = "COMP_LINE"
-	envDebug    = "COMP_DEBUG"
+	envLine  = "COMP_LINE"
+	envPoint = "COMP_POINT"
+	envDebug = "COMP_DEBUG"
 )
 
 // Complete structs define completion for a command with CLI options
@@ -55,14 +57,17 @@ func (c *Complete) Run() bool {
 // For installation: it assumes that flags were added and parsed before
 // it was called.
 func (c *Complete) Complete() bool {
-	line, ok := getLine()
+	line, point, ok := getEnv()
 	if !ok {
 		// make sure flags parsed,
 		// in case they were not added in the main program
 		return c.CLI.Run()
 	}
-	Log("Completing line: %s", line)
-	a := newArgs(line)
+
+	completePhrase := line[:point]
+
+	Log("Completing phrase: %s", completePhrase)
+	a := newArgs(completePhrase)
 	Log("Completing last field: %s", a.Last)
 	options := c.Command.Predict(a)
 	Log("Options: %s", options)
@@ -79,12 +84,19 @@ func (c *Complete) Complete() bool {
 	return true
 }
 
-func getLine() (string, bool) {
-	line := os.Getenv(envComplete)
+func getEnv() (line string, point int, ok bool) {
+	line = os.Getenv(envLine)
 	if line == "" {
-		return "", false
+		return
+	}
+	point, err := strconv.Atoi(os.Getenv(envPoint))
+	if err != nil {
+		// If failed parsing point for some reason, set it to point
+		// on the end of the line.
+		Log("Failed parsing point %s: %v", os.Getenv(envPoint), err)
+		point = len(line)
 	}
-	return line, true
+	return line, point, true
 }
 
 func (c *Complete) output(options []string) {