util.prosodyctl.shell: Allow setting custom prompt (admin_shell_prompt)

1 files changed, 6 insertions, 4 deletions

diff --git a/util/prosodyctl/shell.lua b/util/prosodyctl/shell.lua
index 79d96f15..214779ec 100644
--- a/util/prosodyctl/shell.lua
+++ b/util/prosodyctl/shell.lua
@@ -17,11 +17,11 @@ if have_readline then
 		});
 end
 
-local function read_line()
+local function read_line(prompt_string)
 	if have_readline then
-		return readline.readline("prosody> ");
+		return readline.readline(prompt_string);
 	else
-		io.write("prosody> ");
+		io.write(prompt_string);
 		return io.read("*line");
 	end
 end
@@ -31,7 +31,7 @@ local function send_line(client, line)
 end
 
 local function repl(client)
-	local line = read_line();
+	local line = read_line(client.prompt_string or "prosody> ");
 	if not line or line == "quit" or line == "exit" or line == "bye" then
 		if not line then
 			print("");
@@ -123,6 +123,8 @@ local function start(arg) --luacheck: ignore 212/arg
 		end
 	end);
 
+	client.prompt_string = config.get("*", "admin_shell_prompt");
+
 	local socket_path = path.resolve_relative_path(prosody.paths.data, opts.socket or config.get("*", "admin_socket") or "prosody.sock");
 	local conn = adminstream.connection(socket_path, client.listeners);
 	local ok, err = conn:connect();