[PATCH] Tab completion for ALTER TYPE … RENAMEVALUE …

Hi hackers,

Here's a patch to add psql tab completion for the recently-added ALTER
TYPE … RENAME VALUE feature (thanks to Tom for fixing it up and
committing it).

It's modelled on the ALTER TYPE … RENAME ATTRIBUTE completion, but
tweaked to return string literals instead of identifiers.

- ilmari

>From f4fa474262e6e65f02095f9de09205bff7ea2a1d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dagfinn=20Ilmari=20Manns=C3=A5ker?= <ilmari@ilmari.org>
Date: Mon, 12 Sep 2016 12:17:37 +0100
Subject: [PATCH] =?UTF-8?q?Add=20psql=20tab=20completion=20for=20ALTER=20T?=
 =?UTF-8?q?YPE=20=E2=80=A6=20RENAME=20VALUE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Modelled on the completion for attributes, tweaked to return string
literals intead of identifiers.
---
 src/bin/psql/tab-complete.c | 57 +++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 55 insertions(+), 2 deletions(-)

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 3e2f084..40790c9 100644
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -202,6 +202,31 @@ do { \
     matches = completion_matches(text, complete_from_query); \
 } while (0)

+#define COMPLETE_WITH_ENUM_VALUE(type) \
+do { \
+    char   *_completion_schema; \
+    char   *_completion_type; \
+\
+    _completion_schema = strtokx(type, " \t\n\r", ".", "\"", 0, \
+                                 false, false, pset.encoding); \
+    (void) strtokx(NULL, " \t\n\r", ".", "\"", 0, \
+                   false, false, pset.encoding); \
+    _completion_type = strtokx(NULL, " \t\n\r", ".", "\"", 0, \
+                               false, false, pset.encoding);  \
+    if (_completion_type == NULL)\
+    { \
+        completion_charp = Query_for_list_of_enum_values; \
+        completion_info_charp = type; \
+    } \
+    else \
+    { \
+        completion_charp = Query_for_list_of_enum_values_with_schema; \
+        completion_info_charp = _completion_type; \
+        completion_info_charp2 = _completion_schema; \
+    } \
+    matches = completion_matches(text, complete_from_query); \
+} while (0)
+
 #define COMPLETE_WITH_FUNCTION_ARG(function) \
 do { \
     char   *_completion_schema; \
@@ -598,6 +623,26 @@ static const SchemaQuery Query_for_list_of_matviews = {
 "   AND (pg_catalog.quote_ident(nspname)='%s' "\
 "        OR '\"' || nspname || '\"' ='%s') "

+#define Query_for_list_of_enum_values \
+"SELECT pg_catalog.quote_literal(enumlabel) "\
+"  FROM pg_catalog.pg_enum e, pg_catalog.pg_type t "\
+" WHERE t.oid = e.enumtypid "\
+"   AND substring(pg_catalog.quote_literal(enumlabel),1,%d)='%s' "\
+"   AND (pg_catalog.quote_ident(typname)='%s' "\
+"        OR '\"' || typname || '\"'='%s') "\
+"   AND pg_catalog.pg_type_is_visible(t.oid)"
+
+#define Query_for_list_of_enum_values_with_schema \
+"SELECT pg_catalog.quote_literal(enumlabel) "\
+"  FROM pg_catalog.pg_enum e, pg_catalog.pg_type t, pg_catalog.pg_namespace n "\
+" WHERE t.oid = e.enumtypid "\
+"   AND n.oid = t.typnamespace "\
+"   AND substring(pg_catalog.quote_literal(enumlabel),1,%d)='%s' "\
+"   AND (pg_catalog.quote_ident(typname)='%s' "\
+"        OR '\"' || typname || '\"'='%s') "\
+"   AND (pg_catalog.quote_ident(nspname)='%s' "\
+"        OR '\"' || nspname || '\"' ='%s') "
+
 #define Query_for_list_of_template_databases \
 "SELECT pg_catalog.quote_ident(d.datname) "\
 "  FROM pg_catalog.pg_database d "\
@@ -1873,11 +1918,13 @@ psql_completion(const char *text, int start, int end)
         COMPLETE_WITH_LIST2("ATTRIBUTE", "VALUE");
     /* ALTER TYPE <foo> RENAME    */
     else if (Matches4("ALTER", "TYPE", MatchAny, "RENAME"))
-        COMPLETE_WITH_LIST2("ATTRIBUTE", "TO");
+        COMPLETE_WITH_LIST3("ATTRIBUTE", "TO", "VALUE");
     /* ALTER TYPE xxx RENAME ATTRIBUTE yyy */
     else if (Matches6("ALTER", "TYPE", MatchAny, "RENAME", "ATTRIBUTE", MatchAny))
         COMPLETE_WITH_CONST("TO");
-
+    /* ALTER TYPE xxx RENAME VALUE yyy */
+    else if (Matches6("ALTER", "TYPE", MatchAny, "RENAME", "VALUE", MatchAny))
+        COMPLETE_WITH_CONST("TO");
     /*
      * If we have ALTER TYPE <sth> ALTER/DROP/RENAME ATTRIBUTE, provide list
      * of attributes
@@ -1897,6 +1944,12 @@ psql_completion(const char *text, int start, int end)
     else if (Matches5("ALTER", "GROUP", MatchAny, "ADD|DROP", "USER"))
         COMPLETE_WITH_QUERY(Query_for_list_of_roles);

+    /*
+     * If we have ALTER TYPE <sth> RENAME VALUE, provide list of enum values
+     */
+    else if (Matches5("ALTER", "TYPE", MatchAny, "RENAME", "VALUE"))
+        COMPLETE_WITH_ENUM_VALUE(prev3_wd);
+
 /* BEGIN */
     else if (Matches1("BEGIN"))
         COMPLETE_WITH_LIST6("WORK", "TRANSACTION", "ISOLATION LEVEL", "READ", "DEFERRABLE", "NOT DEFERRABLE");
--
2.9.3


--
- Twitter seems more influential [than blogs] in the 'gets reported in
  the mainstream press' sense at least.               - Matt McLeod
- That'd be because the content of a tweet is easier to condense down
  to a mainstream media article.                      - Calle Dybedahl