Tab completion for CREATE TYPE.

Author: Thomas Munro
Reviewed-by: Kyotaro Horiguchi
Discussion: https://postgr.es/m/CA%2BhUKGLk%3D0yLDjfviONJLzcHEzygj%3Dx6VbGH43LnXbBUvQb52g%40mail.gmail.com

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 9009b05e122d773ab8da370cae025a5b2d3d0b77..9b9202e84c0fa94e0861fea40f9daacd663b3f60 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -2686,6 +2686,45 @@ psql_completion(const char *text, int start, int end)
        else if (Matches("CREATE", "ROLE|USER|GROUP", MatchAny, "IN"))
                COMPLETE_WITH("GROUP", "ROLE");
 
+/* CREATE TYPE */
+       else if (Matches("CREATE", "TYPE", MatchAny))
+               COMPLETE_WITH("(", "AS");
+       else if (Matches("CREATE", "TYPE", MatchAny, "AS"))
+               COMPLETE_WITH("ENUM", "RANGE", "(");
+       else if (HeadMatches("CREATE", "TYPE", MatchAny, "AS", "("))
+       {
+               if (TailMatches("(|*,", MatchAny))
+                       COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_datatypes, NULL);
+               else if (TailMatches("(|*,", MatchAny, MatchAnyExcept("*)")))
+                       COMPLETE_WITH("COLLATE", ",", ")");
+       }
+       else if (Matches("CREATE", "TYPE", MatchAny, "AS", "ENUM|RANGE"))
+               COMPLETE_WITH("(");
+       else if (HeadMatches("CREATE", "TYPE", MatchAny, "("))
+       {
+               if (TailMatches("(|*,"))
+                       COMPLETE_WITH("INPUT", "OUTPUT", "RECEIVE", "SEND",
+                                                 "TYPMOD_IN", "TYPMOD_OUT", "ANALYZE",
+                                                 "INTERNALLENGTH", "PASSBYVALUE", "ALIGNMENT",
+                                                 "STORAGE", "LIKE", "CATEGORY", "PREFERRED",
+                                                 "DEFAULT", "ELEMENT", "DELIMITER",
+                                                 "COLLATABLE");
+               else if (TailMatches("(*|*,", MatchAnyExcept("*=")))
+                       COMPLETE_WITH("=");
+               else if (TailMatches("=", MatchAnyExcept("*)")))
+                       COMPLETE_WITH(",", ")");
+       }
+       else if (HeadMatches("CREATE", "TYPE", MatchAny, "AS", "RANGE", "("))
+       {
+               if (TailMatches("(|*,"))
+                       COMPLETE_WITH("SUBTYPE", "SUBTYPE_OPCLASS", "COLLATION",
+                                                 "CANONICAL", "SUBTYPE_DIFF");
+               else if (TailMatches("(*|*,", MatchAnyExcept("*=")))
+                       COMPLETE_WITH("=");
+               else if (TailMatches("=", MatchAnyExcept("*)")))
+                       COMPLETE_WITH(",", ")");
+       }
+
 /* CREATE VIEW --- is allowed inside CREATE SCHEMA, so use TailMatches */
        /* Complete CREATE VIEW <name> with AS */
        else if (TailMatches("CREATE", "VIEW", MatchAny))