Tab complete CREATE EXTENSION .. VERSION.

Jeff Janes

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 1619de52aeadec5a71cc914329053409f7c53267..4eb5058416c1dd3d24463e5511ddea5e6acccd98 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -729,6 +729,13 @@ static const SchemaQuery Query_for_list_of_matviews = {
 "   FROM pg_catalog.pg_available_extensions "\
 "  WHERE substring(pg_catalog.quote_ident(name),1,%d)='%s' AND installed_version IS NULL"
 
+/* the silly-looking length condition is just to eat up the current word */
+#define Query_for_list_of_available_extension_versions \
+" SELECT pg_catalog.quote_ident(version) "\
+"   FROM pg_catalog.pg_available_extension_versions "\
+"  WHERE (%d = pg_catalog.length('%s'))"\
+"    AND pg_catalog.quote_ident(name)='%s'"
+
 #define Query_for_list_of_prepared_statements \
 " SELECT pg_catalog.quote_ident(name) "\
 "   FROM pg_catalog.pg_prepared_statements "\
@@ -2266,10 +2273,18 @@ psql_completion(const char *text, int start, int end)
                         pg_strcasecmp(prev2_wd, "EXTENSION") == 0)
        {
                static const char *const list_CREATE_EXTENSION[] =
-               {"WITH SCHEMA", "CASCADE", NULL};
+               {"WITH SCHEMA", "CASCADE", "VERSION", NULL};
 
                COMPLETE_WITH_LIST(list_CREATE_EXTENSION);
        }
+       /* CREATE EXTENSION <name> VERSION */
+       else if (pg_strcasecmp(prev4_wd, "CREATE") == 0 &&
+                        pg_strcasecmp(prev3_wd, "EXTENSION") == 0 &&
+                        pg_strcasecmp(prev_wd, "VERSION") == 0)
+       {
+               completion_info_charp = prev2_wd;
+               COMPLETE_WITH_QUERY(Query_for_list_of_available_extension_versions);
+       }
 
        /* CREATE FOREIGN */
        else if (pg_strcasecmp(prev2_wd, "CREATE") == 0 &&