psql: Tab completion for ALTER TYPE ... CASCADE/RESTRICT

Updates table completion for ALTER TYPE to offer CASCADE/RESTRICT for a
number of actions on attributes:

    ALTER TYPE ... ADD/DROP/RENAME ATTRIBUTE ... [CASCADE|RESTRICT]
    ALTER TYPE ... TYPE ... [CASCADE|RESTRICT]

Author: Kirill Reshke
Reviewed-By: Karina Litskevich
Discussion: https://postgr.es/m/CALdSSPhVELkvutquqrDB=Ujfq_Pjz=6jn-kzh+291KPNViLTfw@mail.gmail.com

diff --git a/src/bin/psql/tab-complete.in.c b/src/bin/psql/tab-complete.in.c
index d2c7f940008820efb7742871955bc38292f115d1..79b0204717450f6ecf1411d0536590532773afd7 100644 (file)
--- a/src/bin/psql/tab-complete.in.c
+++ b/src/bin/psql/tab-complete.in.c
@@ -2992,6 +2992,9 @@ match_previous_words(int pattern_id,
    /* ALTER TYPE xxx RENAME (ATTRIBUTE|VALUE) yyy */
    else if (Matches("ALTER", "TYPE", MatchAny, "RENAME", "ATTRIBUTE|VALUE", MatchAny))
        COMPLETE_WITH("TO");
+   /* ALTER TYPE xxx RENAME ATTRIBUTE yyy TO zzz */
+   else if (Matches("ALTER", "TYPE", MatchAny, "RENAME", "ATTRIBUTE", MatchAny, "TO", MatchAny))
+       COMPLETE_WITH("CASCADE", "RESTRICT");
 
    /*
     * If we have ALTER TYPE  ALTER/DROP/RENAME ATTRIBUTE, provide list
@@ -3002,9 +3005,18 @@ match_previous_words(int pattern_id,
    /* complete ALTER TYPE ADD ATTRIBUTE  with list of types */
    else if (Matches("ALTER", "TYPE", MatchAny, "ADD", "ATTRIBUTE", MatchAny))
        COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_datatypes);
+   /* complete ALTER TYPE ADD ATTRIBUTE   with CASCADE/RESTRICT */
+   else if (Matches("ALTER", "TYPE", MatchAny, "ADD", "ATTRIBUTE", MatchAny, MatchAny))
+       COMPLETE_WITH("CASCADE", "RESTRICT");
+   /* complete ALTER TYPE DROP ATTRIBUTE  with CASCADE/RESTRICT */
+   else if (Matches("ALTER", "TYPE", MatchAny, "DROP", "ATTRIBUTE", MatchAny))
+       COMPLETE_WITH("CASCADE", "RESTRICT");
    /* ALTER TYPE ALTER ATTRIBUTE  */
    else if (Matches("ALTER", "TYPE", MatchAny, "ALTER", "ATTRIBUTE", MatchAny))
        COMPLETE_WITH("TYPE");
+   /* ALTER TYPE ALTER ATTRIBUTE  TYPE  */
+   else if (Matches("ALTER", "TYPE", MatchAny, "ALTER", "ATTRIBUTE", MatchAny, "TYPE", MatchAny))
+       COMPLETE_WITH("CASCADE", "RESTRICT");
    /* complete ALTER TYPE  RENAME VALUE with list of enum values */
    else if (Matches("ALTER", "TYPE", MatchAny, "RENAME", "VALUE"))
        COMPLETE_WITH_ENUM_VALUE(prev3_wd);