Improve tab completion for BEGIN & START|SET TRANSACTION.

Andreas Karlsson with minor change by me for SET TRANSACTION
SNAPSHOT.

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 1345e4ed80344ed52aadd9190bc8138916a73b02..019f75a37605b9c22e64e0ef05718948233d48bb 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -1894,8 +1894,11 @@ psql_completion(const char *text, int start, int end)
    else if (Matches5("ALTER", "GROUP", MatchAny, "ADD|DROP", "USER"))
        COMPLETE_WITH_QUERY(Query_for_list_of_roles);
 
-/* BEGIN, END, ABORT */
-   else if (Matches1("BEGIN|END|ABORT"))
+/* BEGIN */
+   else if (Matches1("BEGIN"))
+       COMPLETE_WITH_LIST6("WORK", "TRANSACTION", "ISOLATION LEVEL", "READ", "DEFERRABLE", "NOT DEFERRABLE");
+/* END, ABORT */
+   else if (Matches1("END|ABORT"))
        COMPLETE_WITH_LIST2("WORK", "TRANSACTION");
 /* COMMIT */
    else if (Matches1("COMMIT"))
@@ -2762,20 +2765,36 @@ psql_completion(const char *text, int start, int end)
    else if (Matches1("SHOW"))
        COMPLETE_WITH_QUERY(Query_for_list_of_show_vars);
    /* Complete "SET TRANSACTION" */
-   else if (Matches2("SET|BEGIN|START", "TRANSACTION") ||
+   else if (Matches2("SET", "TRANSACTION"))
+       COMPLETE_WITH_LIST5("SNAPSHOT", "ISOLATION LEVEL", "READ", "DEFERRABLE", "NOT DEFERRABLE");
+   else if (Matches2("BEGIN|START", "TRANSACTION") ||
             Matches2("BEGIN", "WORK") ||
+            Matches1("BEGIN") ||
          Matches5("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION"))
-       COMPLETE_WITH_LIST2("ISOLATION LEVEL", "READ");
+       COMPLETE_WITH_LIST4("ISOLATION LEVEL", "READ", "DEFERRABLE", "NOT DEFERRABLE");
+   else if (Matches3("SET|BEGIN|START", "TRANSACTION|WORK", "NOT") ||
+            Matches2("BEGIN", "NOT") ||
+            Matches6("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "NOT"))
+       COMPLETE_WITH_CONST("DEFERRABLE");
    else if (Matches3("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION") ||
+            Matches2("BEGIN", "ISOLATION") ||
             Matches6("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "ISOLATION"))
        COMPLETE_WITH_CONST("LEVEL");
-   else if (Matches4("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL"))
+   else if (Matches4("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL") ||
+            Matches3("BEGIN", "ISOLATION", "LEVEL") ||
+            Matches7("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "ISOLATION", "LEVEL"))
        COMPLETE_WITH_LIST3("READ", "REPEATABLE READ", "SERIALIZABLE");
-   else if (Matches5("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL", "READ"))
+   else if (Matches5("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL", "READ") ||
+            Matches4("BEGIN", "ISOLATION", "LEVEL", "READ") ||
+            Matches8("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "ISOLATION", "LEVEL", "READ"))
        COMPLETE_WITH_LIST2("UNCOMMITTED", "COMMITTED");
-   else if (Matches5("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL", "REPEATABLE"))
+   else if (Matches5("SET|BEGIN|START", "TRANSACTION|WORK", "ISOLATION", "LEVEL", "REPEATABLE") ||
+            Matches4("BEGIN", "ISOLATION", "LEVEL", "REPEATABLE") ||
+            Matches8("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "ISOLATION", "LEVEL", "REPEATABLE"))
        COMPLETE_WITH_CONST("READ");
-   else if (Matches3("SET|BEGIN|START", "TRANSACTION|WORK", "READ"))
+   else if (Matches3("SET|BEGIN|START", "TRANSACTION|WORK", "READ") ||
+            Matches2("BEGIN", "READ") ||
+            Matches6("SET", "SESSION", "CHARACTERISTICS", "AS", "TRANSACTION", "READ"))
        COMPLETE_WITH_LIST2("ONLY", "WRITE");
    /* SET CONSTRAINTS */
    else if (Matches2("SET", "CONSTRAINTS"))