348 lines
12 KiB
C++
348 lines
12 KiB
C++
/*
|
|
* psql - the openGauss interactive terminal
|
|
*
|
|
* Copyright (c) 2000-2012, PostgreSQL Global Development Group
|
|
*
|
|
* src/bin/psql/prompt.c
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "input.h"
|
|
#include "prompt.h"
|
|
#include "postgres_fe.h"
|
|
|
|
#ifdef WIN32
|
|
#include <io.h>
|
|
#include <win32.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_UNIX_SOCKETS
|
|
#include <unistd.h>
|
|
#include <netdb.h>
|
|
#endif
|
|
|
|
#ifdef ENABLE_UT
|
|
#define static
|
|
#endif
|
|
|
|
/* --------------------------
|
|
* 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"
|
|
* %> - database server port number
|
|
* %n - database user name
|
|
* %/ - current database
|
|
* %o - gaussdb-ified current database
|
|
* %~ - like %/ but "~" when database name equals user name
|
|
* %# - "#" 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)
|
|
* %? - 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)
|
|
{
|
|
#define MAX_PROMPT_SIZE 256
|
|
static char destination[MAX_PROMPT_SIZE + 1];
|
|
char buf[MAX_PROMPT_SIZE + 1];
|
|
bool esc = false;
|
|
const char* p = NULL;
|
|
const char* prompt_string = NULL;
|
|
const char* username = NULL;
|
|
errno_t rc = EOK;
|
|
|
|
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;
|
|
|
|
default:
|
|
prompt_string = "? ";
|
|
break;
|
|
}
|
|
|
|
destination[0] = '\0';
|
|
|
|
const char* gsname = "openGauss"; /* gaussdb-ify name */
|
|
|
|
for (p = prompt_string; *p && strlen(destination) < sizeof(destination) - 1; p++) {
|
|
check_memset_s(memset_s(buf, sizeof(buf), 0, sizeof(buf)));
|
|
if (esc) {
|
|
switch (*p) {
|
|
/* Current database */
|
|
case '/':
|
|
if (pset.db != NULL) {
|
|
rc = strncpy_s(buf, sizeof(buf), PQdb(pset.db), strlen(PQdb(pset.db)) + 1);
|
|
securec_check_c(rc, "\0", "\0");
|
|
}
|
|
break;
|
|
case 'o': /* gaussdb-ify overwrite default prompt */
|
|
if (pset.db != NULL) {
|
|
if (strcmp(PQdb(pset.db), "postgres") == 0) {
|
|
rc = strncpy_s(buf, sizeof(buf), gsname, strlen(gsname) + 1);
|
|
} else {
|
|
rc = strncpy_s(buf, sizeof(buf), PQdb(pset.db), strlen(PQdb(pset.db)) + 1);
|
|
}
|
|
securec_check_c(rc, "\0", "\0");
|
|
}
|
|
break;
|
|
case '~':
|
|
if (pset.db != NULL) {
|
|
char* var = NULL;
|
|
|
|
if (PQdb(pset.db) == NULL || PQuser(pset.db) == NULL) {
|
|
psql_error("Unexpected NULL value in connection attribute.\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
var = GetEnvStr("PGDATABASE");
|
|
if (strcmp(PQdb(pset.db), PQuser(pset.db)) == 0 ||
|
|
(var != NULL && strcmp(var, PQdb(pset.db)) == 0)) {
|
|
rc = strncpy_s(buf, sizeof(buf), "~", 2);
|
|
securec_check_c(rc, "\0", "\0");
|
|
} else {
|
|
rc = strncpy_s(buf, sizeof(buf), PQdb(pset.db), strlen(PQdb(pset.db)) + 1);
|
|
securec_check_c(rc, "\0", "\0");
|
|
}
|
|
|
|
if (var != NULL)
|
|
free(var);
|
|
var = NULL;
|
|
}
|
|
break;
|
|
|
|
/* DB server hostname (long/short) */
|
|
case 'M':
|
|
case 'm':
|
|
if (pset.db != NULL) {
|
|
const char* host = PQhost(pset.db);
|
|
|
|
/* INET socket */
|
|
if (NULL != host && host[0] && !is_absolute_path(host)) {
|
|
rc = strcpy_s(buf, sizeof(buf), host);
|
|
check_strcpy_s(rc);
|
|
if (*p == 'm')
|
|
buf[strcspn(buf, ".")] = '\0';
|
|
}
|
|
#ifdef HAVE_UNIX_SOCKETS
|
|
/* UNIX socket */
|
|
else {
|
|
if ((host == NULL) || strcmp(host, DEFAULT_PGSOCKET_DIR) == 0 || *p == 'm') {
|
|
rc = strcpy_s(buf, sizeof(buf), "local");
|
|
check_strcpy_s(rc);
|
|
}
|
|
else
|
|
check_sprintf_s(sprintf_s(buf, sizeof(buf), "local:%s", host));
|
|
}
|
|
#endif
|
|
}
|
|
break;
|
|
/* DB server port number */
|
|
case '>':
|
|
if ((pset.db != NULL) && (PQport(pset.db) != NULL)) {
|
|
rc = strcpy_s(buf, sizeof(buf), PQport(pset.db));
|
|
check_strcpy_s(rc);
|
|
}
|
|
break;
|
|
/* DB server user name */
|
|
case 'n':
|
|
username = session_username();
|
|
if ((pset.db != NULL) && (username != NULL)) {
|
|
rc = strcpy_s(buf, sizeof(buf), username);
|
|
check_strcpy_s(rc);
|
|
}
|
|
break;
|
|
|
|
case '0':
|
|
case '1':
|
|
case '2':
|
|
case '3':
|
|
case '4':
|
|
case '5':
|
|
case '6':
|
|
case '7':
|
|
*buf = (char)strtol(p, (char**)&p, 8);
|
|
--p;
|
|
break;
|
|
case 'R':
|
|
switch (status) {
|
|
case PROMPT_READY:
|
|
if (pset.db == NULL)
|
|
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 == NULL)
|
|
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 '?':
|
|
/* not here yet */
|
|
break;
|
|
|
|
case '#':
|
|
if (is_superuser())
|
|
buf[0] = '#';
|
|
else
|
|
buf[0] = '>';
|
|
break;
|
|
|
|
/* execute command */
|
|
case '`': {
|
|
FILE* fd = NULL;
|
|
char* file = pg_strdup(p + 1);
|
|
int cmdend;
|
|
|
|
cmdend = (int)strcspn(file, "`");
|
|
file[cmdend] = '\0';
|
|
check_env_value(file);
|
|
fd = popen(file, "r");
|
|
if (fd != NULL) {
|
|
if (fgets(buf, sizeof(buf), fd) == NULL)
|
|
buf[0] = '\0';
|
|
pclose(fd);
|
|
}
|
|
if (strlen(buf) > 0 && buf[strlen(buf) - 1] == '\n')
|
|
buf[strlen(buf) - 1] = '\0';
|
|
free(file);
|
|
file = NULL;
|
|
p += cmdend + 1;
|
|
break;
|
|
}
|
|
|
|
/* interpolate variable */
|
|
case ':': {
|
|
char* name = NULL;
|
|
const char* val = NULL;
|
|
int nameend;
|
|
|
|
name = pg_strdup(p + 1);
|
|
nameend = (int)strcspn(name, ":");
|
|
name[nameend] = '\0';
|
|
val = GetVariable(pset.vars, name);
|
|
if (NULL != val) {
|
|
rc = strcpy_s(buf, sizeof(buf), val);
|
|
check_strcpy_s(rc);
|
|
}
|
|
free(name);
|
|
name = NULL;
|
|
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));
|
|
}
|
|
|
|
return destination;
|
|
}
|