From 6b40d9bdbdc9f873868b0ddecacd9a307fc8ee26 Mon Sep 17 00:00:00 2001 From: Fujii Masao Date: Thu, 25 Feb 2021 18:20:57 +0900 Subject: [PATCH] Improve tab-completion for TRUNCATE. Author: Kota Miyake Reviewed-by: Muhammad Usama Discussion: https://postgr.es/m/f5d30053d00dcafda3280c9e267ecb0f@oss.nttdata.com --- src/bin/psql/tab-complete.c | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c index d777992559..9f0208ac49 100644 --- a/src/bin/psql/tab-complete.c +++ b/src/bin/psql/tab-complete.c @@ -3832,7 +3832,20 @@ psql_completion(const char *text, int start, int end) /* TRUNCATE */ else if (Matches("TRUNCATE")) + COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, + " UNION SELECT 'TABLE'" + " UNION SELECT 'ONLY'"); + else if (Matches("TRUNCATE", "TABLE")) + COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, + " UNION SELECT 'ONLY'"); + else if (HeadMatches("TRUNCATE") && TailMatches("ONLY")) COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL); + else if (Matches("TRUNCATE", MatchAny) || + Matches("TRUNCATE", "TABLE|ONLY", MatchAny) || + Matches("TRUNCATE", "TABLE", "ONLY", MatchAny)) + COMPLETE_WITH("RESTART IDENTITY", "CONTINUE IDENTITY", "CASCADE", "RESTRICT"); + else if (HeadMatches("TRUNCATE") && TailMatches("IDENTITY")) + COMPLETE_WITH("CASCADE", "RESTRICT"); /* UNLISTEN */ else if (Matches("UNLISTEN"))