Adding upstream version 15.5.upstream/15.5

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 382 insertions, 0 deletions

diff --git a/src/bin/psql/prompt.c b/src/bin/psql/prompt.c
new file mode 100644
index 0000000..509e642
--- /dev/null
+++ b/src/bin/psql/prompt.c
@@ -0,0 +1,382 @@
+/*
+ * psql - the PostgreSQL interactive terminal
+ *
+ * Copyright (c) 2000-2022, PostgreSQL Global Development Group
+ *
+ * src/bin/psql/prompt.c
+ */
+#include "postgres_fe.h"
+
+#ifdef WIN32
+#include <io.h>
+#include <win32.h>
+#endif
+
+#include "common.h"
+#include "common/string.h"
+#include "input.h"
+#include "libpq/pqcomm.h"
+#include "prompt.h"
+#include "settings.h"
+
+/*--------------------------
+ * get_prompt
+ *
+ * Returns a statically allocated prompt made by interpolating certain
+ * tcsh style escape sequences into pset.vars "PROMPT1|2|3".
+ * (might not be completely multibyte safe)
+ *
+ * Defined interpolations are:
+ * %M - database server "hostname.domainname", "[local]" for AF_UNIX
+ *		sockets, "[local:/dir/name]" if not default
+ * %m - like %M, but hostname only (before first dot), or always "[local]"
+ * %p - backend pid
+ * %> - database server port number
+ * %n - database user name
+ * %/ - current database
+ * %~ - like %/ but "~" when database name equals user name
+ * %w - whitespace of the same width as the most recent output of PROMPT1
+ * %# - "#" if superuser, ">" otherwise
+ * %R - in prompt1 normally =, or ^ if single line mode,
+ *			or a ! if session is not connected to a database;
+ *		in prompt2 -, *, ', or ";
+ *		in prompt3 nothing
+ * %x - transaction status: empty, *, !, ? (unknown or no connection)
+ * %l - The line number inside the current statement, starting from 1.
+ * %? - the error code of the last query (not yet implemented)
+ * %% - a percent sign
+ *
+ * %[0-9]		   - the character with the given decimal code
+ * %0[0-7]		   - the character with the given octal code
+ * %0x[0-9A-Fa-f]  - the character with the given hexadecimal code
+ *
+ * %`command`	   - The result of executing command in /bin/sh with trailing
+ *					 newline stripped.
+ * %:name:		   - The value of the psql variable 'name'
+ * (those will not be rescanned for more escape sequences!)
+ *
+ * %[ ... %]	   - tell readline that the contained text is invisible
+ *
+ * If the application-wide prompts become NULL somehow, the returned string
+ * will be empty (not NULL!).
+ *--------------------------
+ */
+
+char *
+get_prompt(promptStatus_t status, ConditionalStack cstack)
+{
+#define MAX_PROMPT_SIZE 256
+	static char destination[MAX_PROMPT_SIZE + 1];
+	char		buf[MAX_PROMPT_SIZE + 1];
+	bool		esc = false;
+	const char *p;
+	const char *prompt_string = "? ";
+	static size_t last_prompt1_width = 0;
+
+	switch (status)
+	{
+		case PROMPT_READY:
+			prompt_string = pset.prompt1;
+			break;
+
+		case PROMPT_CONTINUE:
+		case PROMPT_SINGLEQUOTE:
+		case PROMPT_DOUBLEQUOTE:
+		case PROMPT_DOLLARQUOTE:
+		case PROMPT_COMMENT:
+		case PROMPT_PAREN:
+			prompt_string = pset.prompt2;
+			break;
+
+		case PROMPT_COPY:
+			prompt_string = pset.prompt3;
+			break;
+	}
+
+	destination[0] = '\0';
+
+	for (p = prompt_string;
+		 *p && strlen(destination) < sizeof(destination) - 1;
+		 p++)
+	{
+		memset(buf, 0, sizeof(buf));
+		if (esc)
+		{
+			switch (*p)
+			{
+					/* Current database */
+				case '/':
+					if (pset.db)
+						strlcpy(buf, PQdb(pset.db), sizeof(buf));
+					break;
+				case '~':
+					if (pset.db)
+					{
+						const char *var;
+
+						if (strcmp(PQdb(pset.db), PQuser(pset.db)) == 0 ||
+							((var = getenv("PGDATABASE")) && strcmp(var, PQdb(pset.db)) == 0))
+							strlcpy(buf, "~", sizeof(buf));
+						else
+							strlcpy(buf, PQdb(pset.db), sizeof(buf));
+					}
+					break;
+
+					/* Whitespace of the same width as the last PROMPT1 */
+				case 'w':
+					if (pset.db)
+						memset(buf, ' ',
+							   Min(last_prompt1_width, sizeof(buf) - 1));
+					break;
+
+					/* DB server hostname (long/short) */
+				case 'M':
+				case 'm':
+					if (pset.db)
+					{
+						const char *host = PQhost(pset.db);
+
+						/* INET socket */
+						if (host && host[0] && !is_unixsock_path(host))
+						{
+							strlcpy(buf, host, sizeof(buf));
+							if (*p == 'm')
+								buf[strcspn(buf, ".")] = '\0';
+						}
+						/* UNIX socket */
+						else
+						{
+							if (!host
+								|| strcmp(host, DEFAULT_PGSOCKET_DIR) == 0
+								|| *p == 'm')
+								strlcpy(buf, "[local]", sizeof(buf));
+							else
+								snprintf(buf, sizeof(buf), "[local:%s]", host);
+						}
+					}
+					break;
+					/* DB server port number */
+				case '>':
+					if (pset.db && PQport(pset.db))
+						strlcpy(buf, PQport(pset.db), sizeof(buf));
+					break;
+					/* DB server user name */
+				case 'n':
+					if (pset.db)
+						strlcpy(buf, session_username(), sizeof(buf));
+					break;
+					/* backend pid */
+				case 'p':
+					if (pset.db)
+					{
+						int			pid = PQbackendPID(pset.db);
+
+						if (pid)
+							snprintf(buf, sizeof(buf), "%d", pid);
+					}
+					break;
+
+				case '0':
+				case '1':
+				case '2':
+				case '3':
+				case '4':
+				case '5':
+				case '6':
+				case '7':
+					*buf = (char) strtol(p, unconstify(char **, &p), 8);
+					--p;
+					break;
+				case 'R':
+					switch (status)
+					{
+						case PROMPT_READY:
+							if (cstack != NULL && !conditional_active(cstack))
+								buf[0] = '@';
+							else if (!pset.db)
+								buf[0] = '!';
+							else if (!pset.singleline)
+								buf[0] = '=';
+							else
+								buf[0] = '^';
+							break;
+						case PROMPT_CONTINUE:
+							buf[0] = '-';
+							break;
+						case PROMPT_SINGLEQUOTE:
+							buf[0] = '\'';
+							break;
+						case PROMPT_DOUBLEQUOTE:
+							buf[0] = '"';
+							break;
+						case PROMPT_DOLLARQUOTE:
+							buf[0] = '$';
+							break;
+						case PROMPT_COMMENT:
+							buf[0] = '*';
+							break;
+						case PROMPT_PAREN:
+							buf[0] = '(';
+							break;
+						default:
+							buf[0] = '\0';
+							break;
+					}
+					break;
+
+				case 'x':
+					if (!pset.db)
+						buf[0] = '?';
+					else
+						switch (PQtransactionStatus(pset.db))
+						{
+							case PQTRANS_IDLE:
+								buf[0] = '\0';
+								break;
+							case PQTRANS_ACTIVE:
+							case PQTRANS_INTRANS:
+								buf[0] = '*';
+								break;
+							case PQTRANS_INERROR:
+								buf[0] = '!';
+								break;
+							default:
+								buf[0] = '?';
+								break;
+						}
+					break;
+
+				case 'l':
+					snprintf(buf, sizeof(buf), UINT64_FORMAT, pset.stmt_lineno);
+					break;
+
+				case '?':
+					/* not here yet */
+					break;
+
+				case '#':
+					if (is_superuser())
+						buf[0] = '#';
+					else
+						buf[0] = '>';
+					break;
+
+					/* execute command */
+				case '`':
+					{
+						int			cmdend = strcspn(p + 1, "`");
+						char	   *file = pnstrdup(p + 1, cmdend);
+						FILE	   *fd = popen(file, "r");
+
+						if (fd)
+						{
+							if (fgets(buf, sizeof(buf), fd) == NULL)
+								buf[0] = '\0';
+							pclose(fd);
+						}
+
+						/* strip trailing newline and carriage return */
+						(void) pg_strip_crlf(buf);
+
+						free(file);
+						p += cmdend + 1;
+						break;
+					}
+
+					/* interpolate variable */
+				case ':':
+					{
+						int			nameend = strcspn(p + 1, ":");
+						char	   *name = pnstrdup(p + 1, nameend);
+						const char *val;
+
+						val = GetVariable(pset.vars, name);
+						if (val)
+							strlcpy(buf, val, sizeof(buf));
+						free(name);
+						p += nameend + 1;
+						break;
+					}
+
+				case '[':
+				case ']':
+#if defined(USE_READLINE) && defined(RL_PROMPT_START_IGNORE)
+
+					/*
+					 * readline >=4.0 undocumented feature: non-printing
+					 * characters in prompt strings must be marked as such, in
+					 * order to properly display the line during editing.
+					 */
+					buf[0] = (*p == '[') ? RL_PROMPT_START_IGNORE : RL_PROMPT_END_IGNORE;
+					buf[1] = '\0';
+#endif							/* USE_READLINE */
+					break;
+
+				default:
+					buf[0] = *p;
+					buf[1] = '\0';
+					break;
+			}
+			esc = false;
+		}
+		else if (*p == '%')
+			esc = true;
+		else
+		{
+			buf[0] = *p;
+			buf[1] = '\0';
+			esc = false;
+		}
+
+		if (!esc)
+			strlcat(destination, buf, sizeof(destination));
+	}
+
+	/* Compute the visible width of PROMPT1, for PROMPT2's %w */
+	if (prompt_string == pset.prompt1)
+	{
+		char	   *p = destination;
+		char	   *end = p + strlen(p);
+		bool		visible = true;
+
+		last_prompt1_width = 0;
+		while (*p)
+		{
+#if defined(USE_READLINE) && defined(RL_PROMPT_START_IGNORE)
+			if (*p == RL_PROMPT_START_IGNORE)
+			{
+				visible = false;
+				++p;
+			}
+			else if (*p == RL_PROMPT_END_IGNORE)
+			{
+				visible = true;
+				++p;
+			}
+			else
+#endif
+			{
+				int			chlen,
+							chwidth;
+
+				chlen = PQmblen(p, pset.encoding);
+				if (p + chlen > end)
+					break;		/* Invalid string */
+
+				if (visible)
+				{
+					chwidth = PQdsplen(p, pset.encoding);
+
+					if (*p == '\n')
+						last_prompt1_width = 0;
+					else if (chwidth > 0)
+						last_prompt1_width += chwidth;
+				}
+
+				p += chlen;
+			}
+		}
+	}
+
+	return destination;
+}