Tab complete table names after ALTER TABLE x [NO] INHERIT.

Jeff Janes

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 6b5dd4978fe83502987749a9957069803d04282c..a1bb230d6be585bb2643a1b707700581d1e98c1f 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -1261,6 +1261,21 @@ psql_completion(char *text, int start, int end)
 
        COMPLETE_WITH_LIST(list_ALTERENABLE2);
    }
+   /* ALTER TABLE xxx INHERIT */
+   else if (pg_strcasecmp(prev4_wd, "ALTER") == 0 &&
+            pg_strcasecmp(prev3_wd, "TABLE") == 0 &&
+            pg_strcasecmp(prev_wd, "INHERIT") == 0)
+   {
+       COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, "");
+   }
+   /* ALTER TABLE xxx NO INHERIT */
+   else if (pg_strcasecmp(prev5_wd, "ALTER") == 0 &&
+            pg_strcasecmp(prev4_wd, "TABLE") == 0 &&
+            pg_strcasecmp(prev2_wd, "NO") == 0 &&
+            pg_strcasecmp(prev_wd, "INHERIT") == 0)
+   {
+       COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, "");
+   }
    else if (pg_strcasecmp(prev4_wd, "ALTER") == 0 &&
             pg_strcasecmp(prev3_wd, "TABLE") == 0 &&
             pg_strcasecmp(prev_wd, "DISABLE") == 0)