Folks,

Please find attached patch for $subject :)

Cheers,
David.
-- 
David Fetter <da...@fetter.org> http://fetter.org/
Phone: +1 415 235 3778  AIM: dfetter666  Yahoo!: dfetter
Skype: davidfetter      XMPP: david.fet...@gmail.com
iCal: webcal://www.tripit.com/feed/ical/people/david74/tripit.ics

Remember to vote!
Consider donating to Postgres: http://www.postgresql.org/about/donate
*** a/src/bin/psql/tab-complete.c
--- b/src/bin/psql/tab-complete.c
***************
*** 303,308 **** static const SchemaQuery Query_for_list_of_tables = {
--- 303,375 ----
        NULL
  };
  
+ /* The bit masks for the following three functions come from
+  * src/include/catalog/pg_trigger.h.
+  */
+ static const SchemaQuery Query_for_list_of_insertables = {
+       /* catname */
+       "pg_catalog.pg_class c",
+       /* selcondition */
+       "c.relkind = 'r' OR (c.relkind = 'v' AND c.relhastriggers AND EXISTS "
+       "(SELECT 1 FROM pg_catalog.pg_trigger t WHERE t.tgrelid = c.oid AND 
t.tgtype | (1 << 2) = t.tgtype))",
+       /* viscondition */
+       "pg_catalog.pg_table_is_visible(c.oid)",
+       /* namespace */
+       "c.relnamespace",
+       /* result */
+       "pg_catalog.quote_ident(c.relname)",
+       /* qualresult */
+       NULL
+ };
+ 
+ static const SchemaQuery Query_for_list_of_deleteables = {
+       /* catname */
+       "pg_catalog.pg_class c",
+       /* selcondition */
+       "c.relkind = 'r' OR (c.relkind = 'v' AND c.relhastriggers AND EXISTS "
+       "(SELECT 1 FROM pg_catalog.pg_trigger t WHERE t.tgrelid = c.oid AND 
t.tgtype | (1 << 3) = t.tgtype))",
+       /* viscondition */
+       "pg_catalog.pg_table_is_visible(c.oid)",
+       /* namespace */
+       "c.relnamespace",
+       /* result */
+       "pg_catalog.quote_ident(c.relname)",
+       /* qualresult */
+       NULL
+ };
+ 
+ static const SchemaQuery Query_for_list_of_updateables = {
+       /* catname */
+       "pg_catalog.pg_class c",
+       /* selcondition */
+       "c.relkind = 'r' OR (c.relkind = 'v' AND c.relhastriggers AND EXISTS "
+       "(SELECT 1 FROM pg_catalog.pg_trigger t WHERE t.tgrelid = c.oid AND 
t.tgtype | (1 << 4) = t.tgtype))",
+       /* viscondition */
+       "pg_catalog.pg_table_is_visible(c.oid)",
+       /* namespace */
+       "c.relnamespace",
+       /* result */
+       "pg_catalog.quote_ident(c.relname)",
+       /* qualresult */
+       NULL
+ };
+ 
+ static const SchemaQuery Query_for_list_of_writeables = {
+       /* catname */
+       "pg_catalog.pg_class c",
+       /* selcondition */
+       "c.relkind = 'r' OR (c.relkind = 'v' AND c.relhastriggers AND EXISTS "
+       "(SELECT 1 FROM pg_catalog.pg_trigger t WHERE t.tgrelid = c.oid AND 
(t.tgtype & ( (1<<2) | (1<<3) | (1<<4)))::bool)",
+       /* viscondition */
+       "pg_catalog.pg_table_is_visible(c.oid)",
+       /* namespace */
+       "c.relnamespace",
+       /* result */
+       "pg_catalog.quote_ident(c.relname)",
+       /* qualresult */
+       NULL
+ };
+ 
  static const SchemaQuery Query_for_list_of_tisv = {
        /* catname */
        "pg_catalog.pg_class c",
***************
*** 333,338 **** static const SchemaQuery Query_for_list_of_tsv = {
--- 400,420 ----
        NULL
  };
  
+ static const SchemaQuery Query_for_list_of_tv = {
+       /* catname */
+       "pg_catalog.pg_class c",
+       /* selcondition */
+       "c.relkind IN ('r', 'v')",
+       /* viscondition */
+       "pg_catalog.pg_table_is_visible(c.oid)",
+       /* namespace */
+       "c.relnamespace",
+       /* result */
+       "pg_catalog.quote_ident(c.relname)",
+       /* qualresult */
+       NULL
+ };
+ 
  static const SchemaQuery Query_for_list_of_views = {
        /* catname */
        "pg_catalog.pg_class c",
***************
*** 630,636 **** psql_completion(char *text, int start, int end)
                           *prev2_wd,
                           *prev3_wd,
                           *prev4_wd,
!                          *prev5_wd;
  
        static const char *const sql_commands[] = {
                "ABORT", "ALTER", "ANALYZE", "BEGIN", "CHECKPOINT", "CLOSE", 
"CLUSTER",
--- 712,719 ----
                           *prev2_wd,
                           *prev3_wd,
                           *prev4_wd,
!                          *prev5_wd,
!                          *prev6_wd;
  
        static const char *const sql_commands[] = {
                "ABORT", "ALTER", "ANALYZE", "BEGIN", "CHECKPOINT", "CLOSE", 
"CLUSTER",
***************
*** 669,675 **** psql_completion(char *text, int start, int end)
        completion_info_charp2 = NULL;
  
        /*
!        * Scan the input line before our current position for the last five
         * words. According to those we'll make some smart decisions on what the
         * user is probably intending to type. TODO: Use strtokx() to do this.
         */
--- 752,758 ----
        completion_info_charp2 = NULL;
  
        /*
!        * Scan the input line before our current position for the last six
         * words. According to those we'll make some smart decisions on what the
         * user is probably intending to type. TODO: Use strtokx() to do this.
         */
***************
*** 678,683 **** psql_completion(char *text, int start, int end)
--- 761,767 ----
        prev3_wd = previous_word(start, 2);
        prev4_wd = previous_word(start, 3);
        prev5_wd = previous_word(start, 4);
+       prev6_wd = previous_word(start, 5);
  
        /* If a backslash command was started, continue */
        if (text[0] == '\\')
***************
*** 971,977 **** psql_completion(char *text, int start, int end)
        else if (pg_strcasecmp(prev4_wd, "ALTER") == 0 &&
                         pg_strcasecmp(prev3_wd, "TRIGGER") == 0 &&
                         pg_strcasecmp(prev_wd, "ON") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
  
        /* ALTER TRIGGER <name> ON <name> */
        else if (pg_strcasecmp(prev4_wd, "TRIGGER") == 0 &&
--- 1055,1061 ----
        else if (pg_strcasecmp(prev4_wd, "ALTER") == 0 &&
                         pg_strcasecmp(prev3_wd, "TRIGGER") == 0 &&
                         pg_strcasecmp(prev_wd, "ON") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_writeables, NULL);
  
        /* ALTER TRIGGER <name> ON <name> */
        else if (pg_strcasecmp(prev4_wd, "TRIGGER") == 0 &&
***************
*** 1579,1585 **** psql_completion(char *text, int start, int end)
        else if (pg_strcasecmp(prev4_wd, "AS") == 0 &&
                         pg_strcasecmp(prev3_wd, "ON") == 0 &&
                         pg_strcasecmp(prev_wd, "TO") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
  
  /* CREATE SERVER <name> */
        else if (pg_strcasecmp(prev3_wd, "CREATE") == 0 &&
--- 1663,1669 ----
        else if (pg_strcasecmp(prev4_wd, "AS") == 0 &&
                         pg_strcasecmp(prev3_wd, "ON") == 0 &&
                         pg_strcasecmp(prev_wd, "TO") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tv, NULL);
  
  /* CREATE SERVER <name> */
        else if (pg_strcasecmp(prev3_wd, "CREATE") == 0 &&
***************
*** 1641,1655 **** psql_completion(char *text, int start, int end)
                         pg_strcasecmp(prev2_wd, "TRIGGER") == 0)
        {
                static const char *const list_CREATETRIGGER[] =
!               {"BEFORE", "AFTER", NULL};
  
                COMPLETE_WITH_LIST(list_CREATETRIGGER);
        }
        /* complete CREATE TRIGGER <name> BEFORE,AFTER with an event */
!       else if (pg_strcasecmp(prev4_wd, "CREATE") == 0 &&
                         pg_strcasecmp(prev3_wd, "TRIGGER") == 0 &&
                         (pg_strcasecmp(prev_wd, "BEFORE") == 0 ||
!                         pg_strcasecmp(prev_wd, "AFTER") == 0))
        {
                static const char *const list_CREATETRIGGER_EVENTS[] =
                {"INSERT", "DELETE", "UPDATE", "TRUNCATE", NULL};
--- 1725,1744 ----
                         pg_strcasecmp(prev2_wd, "TRIGGER") == 0)
        {
                static const char *const list_CREATETRIGGER[] =
!               {"BEFORE", "AFTER", "INSTEAD OF", NULL};
  
                COMPLETE_WITH_LIST(list_CREATETRIGGER);
        }
        /* complete CREATE TRIGGER <name> BEFORE,AFTER with an event */
!       else if ((pg_strcasecmp(prev4_wd, "CREATE") == 0 &&
                         pg_strcasecmp(prev3_wd, "TRIGGER") == 0 &&
                         (pg_strcasecmp(prev_wd, "BEFORE") == 0 ||
!                         pg_strcasecmp(prev_wd, "AFTER") == 0)) ||
!                       (pg_strcasecmp(prev5_wd, "CREATE") == 0 &&
!                        pg_strcasecmp(prev4_wd, "TRIGGER") == 0 &&
!                        pg_strcasecmp(prev3_wd, "INSTEAD") == 0 &&
!                        pg_strcasecmp(prev2_wd, "OF") == 0))
! 
        {
                static const char *const list_CREATETRIGGER_EVENTS[] =
                {"INSERT", "DELETE", "UPDATE", "TRUNCATE", NULL};
***************
*** 1672,1682 **** psql_completion(char *text, int start, int end)
         * complete CREATE TRIGGER <name> BEFORE,AFTER event ON with a list of
         * tables
         */
!       else if (pg_strcasecmp(prev5_wd, "TRIGGER") == 0 &&
                         (pg_strcasecmp(prev3_wd, "BEFORE") == 0 ||
!                         pg_strcasecmp(prev3_wd, "AFTER") == 0) &&
                         pg_strcasecmp(prev_wd, "ON") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
        /* complete CREATE TRIGGER ... EXECUTE with PROCEDURE */
        else if (pg_strcasecmp(prev_wd, "EXECUTE") == 0)
                COMPLETE_WITH_CONST("PROCEDURE");
--- 1761,1774 ----
         * complete CREATE TRIGGER <name> BEFORE,AFTER event ON with a list of
         * tables
         */
!       else if ((pg_strcasecmp(prev5_wd, "TRIGGER") == 0 &&
                         (pg_strcasecmp(prev3_wd, "BEFORE") == 0 ||
!                         pg_strcasecmp(prev3_wd, "AFTER") == 0)) ||
!                       (pg_strcasecmp(prev6_wd, "TRIGGER") == 0 &&
!                        pg_strcasecmp(prev4_wd, "INSTEAD") == 0 &&
!                        pg_strcasecmp(prev3_wd, "OF") == 0) &&
                         pg_strcasecmp(prev_wd, "ON") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tv, NULL);
        /* complete CREATE TRIGGER ... EXECUTE with PROCEDURE */
        else if (pg_strcasecmp(prev_wd, "EXECUTE") == 0)
                COMPLETE_WITH_CONST("PROCEDURE");
***************
*** 1764,1770 **** psql_completion(char *text, int start, int end)
        /* Complete DELETE FROM with a list of tables */
        else if (pg_strcasecmp(prev2_wd, "DELETE") == 0 &&
                         pg_strcasecmp(prev_wd, "FROM") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
        /* Complete DELETE FROM <table> */
        else if (pg_strcasecmp(prev3_wd, "DELETE") == 0 &&
                         pg_strcasecmp(prev2_wd, "FROM") == 0)
--- 1856,1862 ----
        /* Complete DELETE FROM with a list of tables */
        else if (pg_strcasecmp(prev2_wd, "DELETE") == 0 &&
                         pg_strcasecmp(prev_wd, "FROM") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_deleteables, NULL);
        /* Complete DELETE FROM <table> */
        else if (pg_strcasecmp(prev3_wd, "DELETE") == 0 &&
                         pg_strcasecmp(prev2_wd, "FROM") == 0)
***************
*** 2052,2058 **** psql_completion(char *text, int start, int end)
        /* Complete INSERT INTO with table names */
        else if (pg_strcasecmp(prev2_wd, "INSERT") == 0 &&
                         pg_strcasecmp(prev_wd, "INTO") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
        /* Complete "INSERT INTO <table> (" with attribute names */
        else if (pg_strcasecmp(prev4_wd, "INSERT") == 0 &&
                         pg_strcasecmp(prev3_wd, "INTO") == 0 &&
--- 2144,2150 ----
        /* Complete INSERT INTO with table names */
        else if (pg_strcasecmp(prev2_wd, "INSERT") == 0 &&
                         pg_strcasecmp(prev_wd, "INTO") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_insertables, NULL);
        /* Complete "INSERT INTO <table> (" with attribute names */
        else if (pg_strcasecmp(prev4_wd, "INSERT") == 0 &&
                         pg_strcasecmp(prev3_wd, "INTO") == 0 &&
***************
*** 2409,2415 **** psql_completion(char *text, int start, int end)
  /* UPDATE */
        /* If prev. word is UPDATE suggest a list of tables */
        else if (pg_strcasecmp(prev_wd, "UPDATE") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_tables, NULL);
        /* Complete UPDATE <table> with "SET" */
        else if (pg_strcasecmp(prev2_wd, "UPDATE") == 0)
                COMPLETE_WITH_CONST("SET");
--- 2501,2507 ----
  /* UPDATE */
        /* If prev. word is UPDATE suggest a list of tables */
        else if (pg_strcasecmp(prev_wd, "UPDATE") == 0)
!               COMPLETE_WITH_SCHEMA_QUERY(Query_for_list_of_updateables, NULL);
        /* Complete UPDATE <table> with "SET" */
        else if (pg_strcasecmp(prev2_wd, "UPDATE") == 0)
                COMPLETE_WITH_CONST("SET");
-- 
Sent via pgsql-hackers mailing list (pgsql-hackers@postgresql.org)
To make changes to your subscription:
http://www.postgresql.org/mailpref/pgsql-hackers

Reply via email to