diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c index eb018854a5c..8b735476ade 100644 --- a/src/bin/psql/tab-complete.c +++ b/src/bin/psql/tab-complete.c @@ -2316,19 +2316,14 @@ psql_completion(const char *text, int start, int end) else if (Matches("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 (Matches("COPY", "BINARY")) - COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL); - /* If we have COPY (, complete it with legal commands */ + /* Complete COPY ( with legal query commands */ else if (Matches("COPY|\\copy", "(")) COMPLETE_WITH("SELECT", "TABLE", "VALUES", "INSERT", "UPDATE", "DELETE", "WITH"); - /* If we have COPY [BINARY] , complete it with "TO" or "FROM" */ - else if (Matches("COPY|\\copy", MatchAny) || - Matches("COPY", "BINARY", MatchAny)) + /* Complete COPY */ + else if (Matches("COPY|\\copy", MatchAny)) COMPLETE_WITH("FROM", "TO"); - /* If we have COPY [BINARY] FROM|TO, complete with filename */ - else if (Matches("COPY", MatchAny, "FROM|TO") || - Matches("COPY", "BINARY", MatchAny, "FROM|TO")) + /* Complete COPY FROM|TO with filename */ + else if (Matches("COPY", MatchAny, "FROM|TO")) { completion_charp = ""; completion_force_quote = true; /* COPY requires quoted filename */ @@ -2340,17 +2335,28 @@ psql_completion(const char *text, int start, int end) completion_force_quote = false; matches = rl_completion_matches(text, complete_from_files); } - /* Offer options after COPY [BINARY] FROM|TO filename */ - else if (Matches("COPY|\\copy", MatchAny, "FROM|TO", MatchAny) || - Matches("COPY", "BINARY", MatchAny, "FROM|TO", MatchAny)) - COMPLETE_WITH("BINARY", "DELIMITER", "NULL", "CSV", - "ENCODING"); - /* Offer options after COPY [BINARY] FROM|TO filename CSV */ - else if (Matches("COPY|\\copy", MatchAny, "FROM|TO", MatchAny, "CSV") || - Matches("COPY", "BINARY", MatchAny, "FROM|TO", MatchAny, "CSV")) - COMPLETE_WITH("HEADER", "QUOTE", "ESCAPE", "FORCE QUOTE", - "FORCE NOT NULL"); + /* Complete COPY TO */ + else if (Matches("COPY|\\copy", MatchAny, "TO", MatchAny)) + COMPLETE_WITH("WITH ("); + + /* Complete COPY FROM */ + else if (Matches("COPY|\\copy", MatchAny, "FROM", MatchAny)) + COMPLETE_WITH("WITH (", "WHERE"); + + /* Complete COPY FROM|TO filename WITH ( */ + else if (Matches("COPY|\\copy", MatchAny, "FROM|TO", MatchAny, "WITH", "(")) + COMPLETE_WITH("FORMAT", "FREEZE", "DELIMITER", "NULL", + "HEADER", "QUOTE", "ESCAPE", "FORCE_QUOTE", + "FORCE_NOT_NULL", "FORCE_NULL", "ENCODING"); + + /* Complete COPY FROM|TO filename WITH (FORMAT */ + else if (Matches("COPY|\\copy", MatchAny, "FROM|TO", MatchAny, "WITH", "(", "FORMAT")) + COMPLETE_WITH("binary", "csv", "text"); + + /* Complete COPY FROM WITH () */ + else if (Matches("COPY|\\copy", MatchAny, "FROM", MatchAny, "WITH", MatchAny)) + COMPLETE_WITH("WHERE"); /* CREATE ACCESS METHOD */ /* Complete "CREATE ACCESS METHOD " */