From 194c99a86e2b17a9be90f301fcded5e2d1e82bea Mon Sep 17 00:00:00 2001
From: Michael Friedrich <michael.friedrich@icinga.com>
Date: Fri, 6 Apr 2018 14:49:15 +0200
Subject: [PATCH] Allow to disable brackets for the Url class and Format()

This commit also adds unit tests.

refs #5706
---
 lib/remote/url.cpp  | 12 ++++++++++--
 lib/remote/url.hpp  |  2 ++
 test/remote-url.cpp |  8 ++++++++
 3 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/lib/remote/url.cpp b/lib/remote/url.cpp
index 262de301b..f6c84ece1 100644
--- a/lib/remote/url.cpp
+++ b/lib/remote/url.cpp
@@ -205,6 +205,11 @@ void Url::SetQuery(const std::map<String, std::vector<String> >& query)
 	m_Query = query;
 }
 
+void Url::SetArrayFormatUseBrackets(bool useBrackets)
+{
+	m_ArrayFormatUseBrackets = useBrackets;
+}
+
 void Url::AddQueryElement(const String& name, const String& value)
 {
 	auto it = m_Query.find(name);
@@ -272,8 +277,11 @@ String Url::Format(bool onlyPathAndQuery, bool printCredentials) const
 					temp += "&";
 
 				temp += key;
-				if (kv.second.size() > 1)
-					temp += "[]";
+
+				if (m_ArrayFormatUseBrackets) {
+					if (kv.second.size() > 1)
+						temp += "[]";
+				}
 
 				if (!s.IsEmpty())
 					temp += "=" + Utility::EscapeString(s, ACQUERY_ENCODE, false);
diff --git a/lib/remote/url.hpp b/lib/remote/url.hpp
index ac022f427..20386980c 100644
--- a/lib/remote/url.hpp
+++ b/lib/remote/url.hpp
@@ -65,6 +65,7 @@ public:
 	void SetPort(const String& port);
 	void SetPath(const std::vector<String>& path);
 	void SetQuery(const std::map<String, std::vector<String> >& query);
+	void SetArrayFormatUseBrackets(bool useBrackets = true);
 
 	void AddQueryElement(const String& name, const String& query);
 	void SetQueryElements(const String& name, const std::vector<String>& query);
@@ -78,6 +79,7 @@ private:
 	String m_Port;
 	std::vector<String> m_Path;
 	std::map<String, std::vector<String> > m_Query;
+	bool m_ArrayFormatUseBrackets;
 	String m_Fragment;
 
 	bool ParseScheme(const String& scheme);
diff --git a/test/remote-url.cpp b/test/remote-url.cpp
index d70751872..60df7c130 100644
--- a/test/remote-url.cpp
+++ b/test/remote-url.cpp
@@ -95,6 +95,14 @@ BOOST_AUTO_TEST_CASE(format)
 
 	url = new Url("/");
 	BOOST_CHECK(url->Format(false, false) == "/");
+
+	url = new Url("https://nsclient:8443/query/check_cpu?time%5B%5D=1m&time=5m&time%5B%5D=15m");
+	url->SetArrayFormatUseBrackets(false);
+	BOOST_CHECK(url2 = new Url(url->Format(false, false)));
+
+	url = new Url("https://icinga2/query?a[]=1&a[]=2&a[]=3");
+	url->SetArrayFormatUseBrackets(true);
+	BOOST_CHECK(url2 = new Url(url->Format(false, false)));
 }
 
 BOOST_AUTO_TEST_CASE(illegal_legal_strings)
-- 
2.40.0