psql: Add tab completion for COPY with query

From: Andreas Karlsson <andreas@proxel.se>

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index c3c77bd58765371aeebf8d0540c5be3dc415d3b8..f09e65c58ddf3eacf34690dcd1d3c1c7ecee29ef 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -1934,11 +1934,18 @@ psql_completion(const char *text, int start, int end)
 /* COPY */
 
        /*
-        * If we have COPY [BINARY] (which you'd have to type yourself), offer
-        * list of tables (Also cover the analogous backslash command)
+        * If we have COPY, offer list of tables or "(" (Also cover the analogous
+        * backslash command).
         */
-       else if (Matches1("COPY|\\copy") || Matches2("COPY", "BINARY"))
+       else if (Matches1("COPY|\\copy"))
+               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables,
+                                                                  " UNION ALL SELECT '('");
+       /* If we have COPY BINARY, complete with list of tables */
+       else if (Matches2("COPY", "BINARY"))
                COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
+       /* If we have COPY (, complete it with legal commands */
+       else if (Matches2("COPY|\\copy", "("))
+               COMPLETE_WITH_LIST7("SELECT", "TABLE", "VALUES", "INSERT", "UPDATE", "DELETE", "WITH");
        /* If we have COPY [BINARY] <sth>, complete it with "TO" or "FROM" */
        else if (Matches2("COPY|\\copy", MatchAny) ||
                         Matches3("COPY", "BINARY", MatchAny))