Add psql \prompt capability.

Chad Wagner

diff --git a/doc/src/sgml/ref/psql-ref.sgml b/doc/src/sgml/ref/psql-ref.sgml
index ce936a0f98ed917e32841d59052febbebd47bcea..3075d6957594be543afe11242d15bd35d4106538 100644 (file)
--- a/doc/src/sgml/ref/psql-ref.sgml
+++ b/doc/src/sgml/ref/psql-ref.sgml
@@ -1,5 +1,5 @@
 <!--
-$PostgreSQL: pgsql/doc/src/sgml/ref/psql-ref.sgml,v 1.186 2007/02/21 23:22:42 momjian Exp $
+$PostgreSQL: pgsql/doc/src/sgml/ref/psql-ref.sgml,v 1.187 2007/02/23 18:20:58 momjian Exp $
 PostgreSQL documentation
 -->
 
@@ -1429,6 +1429,24 @@ lo_import 152801
         </listitem>
       </varlistentry>
 
+      <varlistentry>
+        <term><literal>\prompt [ <replaceable class="parameter">text</replaceable> ] <replaceable class="parameter">name</replaceable></literal></term>
+        <listitem>
+        <para>
+         Prompts the user to set variable <replaceable
+         class="parameter">name</>.  An optional prompt, <replaceable
+         class="parameter">text</>, can be specified.  (For multi-word
+         prompts, use single-quotes.)
+        </para>
+
+        <para>
+         By default, <literal>\prompt</> uses the terminal for input and
+         output.  However, if the <option>-f</> command line switch is
+         used, <literal>\prompt</> uses standard input and standard output.
+        </para>
+        </listitem>
+      </varlistentry>
+
       <varlistentry>
         <term><literal>\pset <replaceable class="parameter">parameter</replaceable> [ <replaceable class="parameter">value</replaceable> ]</literal></term>
 

diff --git a/src/bin/psql/command.c b/src/bin/psql/command.c
index c25a5e6f0f7a4b3ab27d407d3b5ed076449d4554..bcaf684a8ecac9b61c0cfbedace34a953a480e1a 100644 (file)
--- a/src/bin/psql/command.c
+++ b/src/bin/psql/command.c
@@ -3,7 +3,7 @@
  *
  * Copyright (c) 2000-2007, PostgreSQL Global Development Group
  *
- * $PostgreSQL: pgsql/src/bin/psql/command.c,v 1.177 2007/01/05 22:19:49 momjian Exp $
+ * $PostgreSQL: pgsql/src/bin/psql/command.c,v 1.178 2007/02/23 18:20:58 momjian Exp $
  */
 #include "postgres_fe.h"
 #include "command.h"
@@ -712,6 +712,57 @@ exec_command(const char *cmd,
                free(pw2);
        }
 
+       /* \prompt -- prompt and set variable */
+       else if (strcmp(cmd, "prompt") == 0)
+       {
+               char       *opt, *prompt_text = NULL;
+               char       *arg1, *arg2;
+
+               arg1 = psql_scan_slash_option(scan_state, OT_NORMAL, NULL, false);
+               arg2 = psql_scan_slash_option(scan_state, OT_NORMAL, NULL, false);
+
+               if (!arg1)
+               {
+                       psql_error("\\%s: missing required argument\n", cmd);
+                       success = false;
+               }
+               else
+               {
+                       char       *result;
+
+                       if (arg2)
+                       {
+                               prompt_text = arg1;
+                               opt = arg2;
+                       }
+                       else
+                               opt = arg1;
+
+                       if (!pset.inputfile)
+                               result = simple_prompt(prompt_text, 4096, true);
+                       else
+                       {
+                               if (prompt_text)
+                               {
+                                       fputs(prompt_text, stdout);
+                                       fflush(stdout);
+                               }
+                               result = gets_fromFile(stdin);
+                       }
+
+                       if (!SetVariable(pset.vars, opt, result))
+                       {
+                               psql_error("\\%s: error\n", cmd);
+                               success = false;
+                       }
+
+                       free(result);
+                       if (prompt_text)
+                               free(prompt_text);
+                       free(opt);
+               }
+       }
+
        /* \pset -- set printing parameters */
        else if (strcmp(cmd, "pset") == 0)
        {

diff --git a/src/bin/psql/help.c b/src/bin/psql/help.c
index e16c0289975d8818ea6ec1834954e524357db4ec..027b5cb0207cde13e967e1fcd6230675cdc36b27 100644 (file)
--- a/src/bin/psql/help.c
+++ b/src/bin/psql/help.c
@@ -3,7 +3,7 @@
  *
  * Copyright (c) 2000-2007, PostgreSQL Global Development Group
  *
- * $PostgreSQL: pgsql/src/bin/psql/help.c,v 1.116 2007/01/05 22:19:49 momjian Exp $
+ * $PostgreSQL: pgsql/src/bin/psql/help.c,v 1.117 2007/02/23 18:20:59 momjian Exp $
  */
 #include "postgres_fe.h"
 
@@ -161,7 +161,7 @@ slashUsage(unsigned short int pager)
 {
        FILE       *output;
 
-       output = PageOutput(67, pager);
+       output = PageOutput(69, pager);
 
        /* if you add/remove a line here, change the row count above */
 
@@ -184,6 +184,8 @@ slashUsage(unsigned short int pager)
        fprintf(output, _("  \\timing        toggle timing of commands (currently %s)\n"),
                        ON(pset.timing));
        fprintf(output, _("  \\unset NAME    unset (delete) internal variable\n"));
+       fprintf(output, _("  \\prompt [TEXT] NAME\n"
+                                         "                 prompt user to set internal variable\n"));
        fprintf(output, _("  \\! [COMMAND]   execute command in shell or start interactive shell\n"));
        fprintf(output, "\n");
 

diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
index 8012b70a4dc70d3a6c4e923d223274392d312fec..63db48546bf3c7e59a82b152d23e490af812467d 100644 (file)
--- a/src/bin/psql/tab-complete.c
+++ b/src/bin/psql/tab-complete.c
@@ -3,7 +3,7 @@
  *
  * Copyright (c) 2000-2007, PostgreSQL Global Development Group
  *
- * $PostgreSQL: pgsql/src/bin/psql/tab-complete.c,v 1.158 2007/02/07 00:52:35 petere Exp $
+ * $PostgreSQL: pgsql/src/bin/psql/tab-complete.c,v 1.159 2007/02/23 18:20:59 momjian Exp $
  */
 
 /*----------------------------------------------------------------------
@@ -542,7 +542,7 @@ psql_completion(char *text, int start, int end)
                "\\e", "\\echo", "\\encoding",
                "\\f", "\\g", "\\h", "\\help", "\\H", "\\i", "\\l",
                "\\lo_import", "\\lo_export", "\\lo_list", "\\lo_unlink",
-               "\\o", "\\p", "\\password", "\\pset", "\\q", "\\qecho", "\\r",
+               "\\o", "\\p", "\\password", "\\prompt", "\\pset", "\\q", "\\qecho", "\\r",
                "\\set", "\\t", "\\T",
                "\\timing", "\\unset", "\\x", "\\w", "\\z", "\\!", NULL
        };