# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1654084740 -7200
# Node ID cfdc8cca64d369fefedd10e94312e5ce97019976
# Parent  0f56587bb37f332a0afb59d10d451c146bdc0158
util.prosodyctl.shell: Print errors in red to highlight them

diff -r 0f56587bb37f -r cfdc8cca64d3 util/prosodyctl/shell.lua
--- a/util/prosodyctl/shell.lua	Tue May 31 13:26:44 2022 +0200
+++ b/util/prosodyctl/shell.lua	Wed Jun 01 13:59:00 2022 +0200
@@ -4,6 +4,8 @@
 local path = require "util.paths";
 local parse_args = require "util.argparse".parse;
 local unpack = table.unpack or _G.unpack;
+local tc = require "util.termcolours";
+local isatty = require "util.pposix".isatty;
 
 local have_readline, readline = pcall(require, "readline");
 
@@ -64,6 +66,7 @@
 local function start(arg) --luacheck: ignore 212/arg
 	local client = adminstream.client();
 	local opts, err, where = parse_args(arg);
+	local ttyout = isatty(io.stdout);
 
 	if not opts then
 		if err == "param-not-found" then
@@ -122,7 +125,11 @@
 	client.events.add_handler("received", function (stanza)
 		if stanza.name == "repl-output" or stanza.name == "repl-result" then
 			local result_prefix = stanza.attr.type == "error" and "!" or "|";
-			print(result_prefix.." "..stanza:get_text());
+			local out = result_prefix.." "..stanza:get_text();
+			if ttyout and stanza.attr.type == "error" then
+				out = tc.getstring(tc.getstyle("red"), out);
+			end
+			print(out);
 		end
 		if stanza.name == "repl-result" then
 			repl(client);