From 2eeee3e0d0d3aeca5e6487cc2431c1f65e35a6d8 Mon Sep 17 00:00:00 2001
From: Tavian Barnes <tavianator@tavianator.com>
Date: Fri, 16 Jun 2023 13:44:21 -0400
Subject: color: New %pq formatter for shell-escaped strings

---
 src/color.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'src/color.c')

diff --git a/src/color.c b/src/color.c
index 87575c9..1edd8b5 100644
--- a/src/color.c
+++ b/src/color.c
@@ -858,6 +858,18 @@ static int print_link_target(CFILE *cfile, const struct BFTW *ftwbuf) {
 	return ret;
 }
 
+/** Print an shell-escaped string. */
+static int print_wordesc(CFILE *cfile, const char *str) {
+	char *esc = wordesc(str);
+	if (!esc) {
+		return -1;
+	}
+
+	int ret = dstrcat(&cfile->buffer, esc);
+	free(esc);
+	return ret;
+}
+
 /** Format some colored output to the buffer. */
 BFS_FORMATTER(2, 3)
 static int cbuff(CFILE *cfile, const char *format, ...);
@@ -994,6 +1006,12 @@ static int cvbuff(CFILE *cfile, const char *format, va_list args) {
 
 			case 'p':
 				switch (*++i) {
+				case 'q':
+					if (print_wordesc(cfile, va_arg(args, const char *)) != 0) {
+						return -1;
+					}
+					break;
+
 				case 'F':
 					if (print_name(cfile, va_arg(args, const struct BFTW *)) != 0) {
 						return -1;
-- 
cgit v1.2.3