Added support for autocompleting CREATE queries

xo · Aug 2, 2021 · ae8d47e · ae8d47e
1 parent 8ffb02d
commit ae8d47e
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/drivers/completer/completer.go b/drivers/completer/completer.go
@@ -322,6 +322,13 @@ func (c completer) complete(previousWords []string, text []rune) [][]rune {
 	}
 	/* XXX: implement tab completion for DELETE ... USING */
 
+	/* Complete CREATE */
+	if tailMatches(IGNORE_CASE, previousWords, "CREATE") || tailMatches(IGNORE_CASE, previousWords, "CREATE", "TEMP|TEMPORARY") {
+		return completeFromList(text, "DATABASE", "SEQUENCE", "TABLE", "VIEW", "TEMPORARY")
+	}
+	if tailMatches(IGNORE_CASE, previousWords, "CREATE", "TABLE", "*") || tailMatches(IGNORE_CASE, previousWords, "CREATE", "TEMP|TEMPORARY", "TABLE", "*") {
+		return completeFromList(text, "(")
+	}
 	/* INSERT --- can be inside EXPLAIN, RULE, etc */
 	/* Complete INSERT with "INTO" */
 	if tailMatches(IGNORE_CASE, previousWords, "INSERT") {

diff --git a/drivers/completer/completer_test.go b/drivers/completer/completer_test.go
@@ -221,6 +221,28 @@ func TestCompleter(t *testing.T) {
 			[]string{},
 			2,
 		},
+		{
+			"type on create",
+			"CREATE ",
+			7,
+			[]string{
+				"DATABASE",
+				"TABLE",
+				"SEQUENCE",
+				"VIEW",
+				"TEMPORARY",
+			},
+			0,
+		},
+		{
+			"brackets on create table",
+			"CREATE TABLE p ",
+			15,
+			[]string{
+				"(",
+			},
+			0,
+		},
 	}
 
 	completer := NewDefaultCompleter(WithReader(mockReader{}), WithConnStrings([]string{"pg://"}))