From 5a2c1b2c3aeb99e7a703b0c5f6fe1a21d29b3be4 Mon Sep 17 00:00:00 2001
From: Lorenz Kästle <12514511+RincewindsHat@users.noreply.github.com>
Date: Mon, 15 Sep 2025 02:30:42 +0200
Subject: Add output formatting option

---
 plugins/check_curl.c | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'plugins/check_curl.c')

diff --git a/plugins/check_curl.c b/plugins/check_curl.c
index 6d50568f..fc704171 100644
--- a/plugins/check_curl.c
+++ b/plugins/check_curl.c
@@ -139,6 +139,10 @@ int main(int argc, char **argv) {
 
 	const check_curl_config config = tmp_config.config;
 
+	if (config.output_format_is_set) {
+		mp_set_format(config.output_format);
+	}
+
 	check_curl_working_state working_state = config.initial_config;
 
 	mp_check overall = mp_check_init();
@@ -828,7 +832,8 @@ check_curl_config_wrapper process_arguments(int argc, char **argv) {
 		AUTOMATIC_DECOMPRESSION,
 		COOKIE_JAR,
 		HAPROXY_PROTOCOL,
-		STATE_REGEX
+		STATE_REGEX,
+		OUTPUT_FORMAT
 	};
 
 	static struct option longopts[] = {
@@ -875,6 +880,7 @@ check_curl_config_wrapper process_arguments(int argc, char **argv) {
 		{"enable-automatic-decompression", no_argument, 0, AUTOMATIC_DECOMPRESSION},
 		{"cookie-jar", required_argument, 0, COOKIE_JAR},
 		{"haproxy-protocol", no_argument, 0, HAPROXY_PROTOCOL},
+		{"output-format", required_argument, 0, OUTPUT_FORMAT},
 		{0, 0, 0, 0}};
 
 	check_curl_config_wrapper result = {
@@ -1301,6 +1307,18 @@ check_curl_config_wrapper process_arguments(int argc, char **argv) {
 			/* print short usage statement if args not parsable */
 			usage5();
 			break;
+		case OUTPUT_FORMAT: {
+			parsed_output_format parser = mp_parse_output_format(optarg);
+			if (!parser.parsing_success) {
+				// TODO List all available formats here, maybe add anothoer usage function
+				printf("Invalid output format: %s\n", optarg);
+				exit(STATE_UNKNOWN);
+			}
+
+			result.config.output_format_is_set = true;
+			result.config.output_format = parser.output_format;
+			break;
+		}
 		}
 	}
 
@@ -1602,6 +1620,8 @@ void print_help(void) {
 
 	printf(UT_VERBOSE);
 
+	printf(UT_OUTPUT_FORMAT);
+
 	printf("\n");
 	printf("%s\n", _("Notes:"));
 	printf(" %s\n", _("This plugin will attempt to open an HTTP connection with the host."));
-- 
cgit v1.2.3-74-g34f1