From 8e5f34d8599f32b1221c81942f51347db1b9310c Mon Sep 17 00:00:00 2001 From: Haiying Tang Date: Wed, 14 Apr 2021 21:50:12 +0900 Subject: [PATCH] Support tab completion with a query result for upper character inputs in psql diff --git a/src/bin/psql/t/010_tab_completion.pl b/src/bin/psql/t/010_tab_completion.pl index c27f216d39..a3853ea995 100644 --- a/src/bin/psql/t/010_tab_completion.pl +++ b/src/bin/psql/t/010_tab_completion.pl @@ -39,7 +39,8 @@ $node->start; $node->safe_psql('postgres', "CREATE TABLE tab1 (f1 int, f2 text);\n" . "CREATE TABLE mytab123 (f1 int, f2 text);\n" - . "CREATE TABLE mytab246 (f1 int, f2 text);\n"); + . "CREATE TABLE mytab246 (f1 int, f2 text);\n" + . "CREATE TABLE \"myTAB123\" (\"aF1\" int, f2 text);\n"); # Developers would not appreciate this test adding a bunch of junk to # their ~/.psql_history, so be sure to redirect history into a temp file. @@ -142,6 +143,41 @@ check_completion("SEL\t", qr/SELECT /, "complete SEL to SELECT"); clear_query(); +# check set query command(lower case) completion for upper character inputs +check_completion("set bYT\t", qq/set bYT\b\bytea_output /, "complete set bYT to set bytea_output"); + +clear_query(); + +# check set query command(upper case) completion for upper character inputs +check_completion("set BYT\t", qr/set BYTEA_OUTPUT /, "complete set BYT to set BYTEA_OUTPUT"); + +clear_query(); + +# check query command completion for upper character ralation name +check_completion("update TAB1 SET \t", qr/update TAB1 SET \af/, "complete column name for TAB1"); + +clear_query(); + +# check quoted identifiers in table +check_completion("update \"my\t", qr/update \"myTAB123\" /, "complete quoted string1"); + +clear_query(); + +# check quoted identifiers in column +check_completion("update \"myTAB123\" SET \"aF\t", qr/update \"myTAB123\" SET \"aF1\" /, "complete quoted string2"); + +clear_query(); + +# check schema query(lower case) which is case-insensitive +check_completion("select oid from pg_Cla\t", qq/select oid from pg_Cla\b\b\bclass /, "complete schema query with lower case string"); + +clear_query(); + +# check schema query(upper case) which is case-insensitiveq +check_completion("select oid from Pg_cla\t", qq/select oid from Pg_cla\b\b\b\b\bG_CLASS /, "complete schema query with uppper case string"); + +clear_query(); + # check case variation is honored check_completion("sel\t", qr/select /, "complete sel to select"); diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c index d34271e3b8..08512dfc4c 100644 --- a/src/bin/psql/tab-complete.c +++ b/src/bin/psql/tab-complete.c @@ -197,18 +197,21 @@ static bool completion_force_quote; /* true to force-quote filenames */ */ #define COMPLETE_WITH_QUERY(query) \ do { \ + completion_case_sensitive = false; \ completion_charp = query; \ matches = rl_completion_matches(text, complete_from_query); \ } while (0) #define COMPLETE_WITH_VERSIONED_QUERY(query) \ do { \ + completion_case_sensitive = false; \ completion_vquery = query; \ matches = rl_completion_matches(text, complete_from_versioned_query); \ } while (0) #define COMPLETE_WITH_SCHEMA_QUERY(query, addon) \ do { \ + completion_case_sensitive = false; \ completion_squery = &(query); \ completion_charp = addon; \ matches = rl_completion_matches(text, complete_from_schema_query); \ @@ -216,6 +219,7 @@ do { \ #define COMPLETE_WITH_VERSIONED_SCHEMA_QUERY(query, addon) \ do { \ + completion_case_sensitive = false; \ completion_squery = query; \ completion_vquery = addon; \ matches = rl_completion_matches(text, complete_from_versioned_schema_query); \ @@ -1028,7 +1032,7 @@ static const VersionedQuery Query_for_list_of_subscriptions[] = { }; /* - * This is a list of all "things" in Pgsql, which can show up after CREATE or + * This is a list of all "things" in pgsql, which can show up after CREATE or * DROP; and there is also a query to get a list of them. */ @@ -1178,6 +1182,7 @@ static char *complete_from_files(const char *text, int state); static char *pg_strdup_keyword_case(const char *s, const char *ref); static char *escape_string(const char *text); +static char *pg_string_tolower(const char *text); static PGresult *exec_query(const char *query); static char **get_previous_words(int point, char **buffer, int *nwords); @@ -4420,16 +4425,37 @@ _complete_from_query(const char *simple_query, PQclear(result); result = NULL; - /* Set up suitably-escaped copies of textual inputs */ + /* Set up suitably-escaped copies of textual inputs, + * then change the textual inputs to lower case. + */ e_text = escape_string(text); + if(e_text != NULL) + { + if(e_text[0] == '"') + completion_case_sensitive = true; + else + e_text = pg_string_tolower(e_text); + } if (completion_info_charp) + { e_info_charp = escape_string(completion_info_charp); + if(e_info_charp[0] == '"') + completion_case_sensitive = true; + else + e_info_charp = pg_string_tolower(e_info_charp); + } else e_info_charp = NULL; if (completion_info_charp2) + { e_info_charp2 = escape_string(completion_info_charp2); + if(e_info_charp2[0] == '"') + completion_case_sensitive = true; + else + e_info_charp2 = pg_string_tolower(e_info_charp2); + } else e_info_charp2 = NULL; @@ -4464,7 +4490,7 @@ _complete_from_query(const char *simple_query, */ if (strcmp(schema_query->catname, "pg_catalog.pg_class c") == 0 && - strncmp(text, "pg_", 3) != 0) + strncmp(pg_string_tolower(text), "pg_", 3) != 0) { appendPQExpBufferStr(&query_buffer, " AND c.relnamespace <> (SELECT oid FROM" @@ -4556,7 +4582,16 @@ _complete_from_query(const char *simple_query, while (list_index < PQntuples(result) && (item = PQgetvalue(result, list_index++, 0))) if (pg_strncasecmp(text, item, byte_length) == 0) - return pg_strdup(item); + { + if (byte_length == 0 || completion_case_sensitive) + return pg_strdup(item); + else + /* + * If case insensitive matching was requested initially, + * adjust the case according to setting. + */ + return pg_strdup_keyword_case(item, text); + } } /* If nothing matches, free the db structure and return null */ @@ -4607,7 +4642,6 @@ complete_from_list(const char *text, int state) if (completion_case_sensitive) return pg_strdup(item); else - /* * If case insensitive matching was requested initially, * adjust the case according to setting. @@ -4660,7 +4694,6 @@ complete_from_const(const char *text, int state) if (completion_case_sensitive) return pg_strdup(completion_charp); else - /* * If case insensitive matching was requested initially, adjust * the case according to setting. @@ -4893,6 +4926,24 @@ escape_string(const char *text) } +/* + * pg_string_tolower - Fold a string to lower case. + */ +static char * +pg_string_tolower(const char *text) +{ + char *ret, + *p; + + ret = pg_strdup(text); + + for (p = ret; *p; p++) + *p = pg_tolower((unsigned char) *p); + + return ret; +} + + /* * Execute a query and report any errors. This should be the preferred way of * talking to the database in this file. -- 2.30.0.windows.2