[PATCH v1 13/22] chnet: node: Add set request header function in NodeJS

[Ammar Faizi <ammarfaizi2@gnuweeb.org>]

This adds set request header functionality. Also, update the ring.js
example to perform an HTTP request that executes this new function.

Signed-off-by: Ammar Faizi <ammarfaizi2@gnuweeb.org>
---
 chnet/chnet_node.cc | 41 +++++++++++++++++++++++++++++++++++++++++
 tests/js/ring.js    | 12 ++++++++++++
 2 files changed, 53 insertions(+)

diff --git a/chnet/chnet_node.cc b/chnet/chnet_node.cc
index cfa919b..1269577 100644
--- a/chnet/chnet_node.cc
+++ b/chnet/chnet_node.cc
@@ -352,6 +352,46 @@ static void CHN_NetSetMethod(const Napi::CallbackInfo &info)
 	ch->ch_.SetMethod(method.c_str());
 }
 
+static void CHN_NetSetRequestHeader(const Napi::CallbackInfo &info)
+{
+	constexpr static const char err_msg[] =
+		"chnet.set_request_header must be at least given 2 string arguments";
+	constexpr static const char err_msg2[] =
+		"chnet.set_request_header can only receive 3 arguments with the third argument being a boolean";
+
+	Napi::Env env = info.Env();
+	bool overwrite = true;
+	NodeCHNet *ch;
+	int nr_arg;
+
+
+	nr_arg = info.Length();
+	if (unlikely(nr_arg < 2 || !info[0].IsString() ||
+		     !info[1].IsString())) {
+		throw_js_exception(env, err_msg);
+		return;
+	}
+
+	if (unlikely(nr_arg > 3)) {
+		throw_js_exception(env, err_msg2);
+		return;
+	}
+
+	if (nr_arg == 3) {
+		if (unlikely(!info[2].IsBoolean())) {
+			throw_js_exception(env, err_msg2);
+			return;
+		}
+		overwrite = info[2].As<Napi::Boolean>();
+	}
+
+	ch = (NodeCHNet *)info.Data();
+	const std::string &key = info[0].ToString().Utf8Value();
+	const std::string &val = info[1].ToString().Utf8Value();
+	ch->ch_.SetRequestHeader(key.c_str(), val.c_str(), overwrite);
+}
+
+
 static Napi::Value CHN_NetReadBuf(const Napi::CallbackInfo &info)
 {
 	Napi::Env env = info.Env();
@@ -422,6 +462,7 @@ static Napi::Object CHN_CreateNet(const Napi::CallbackInfo &info)
 
 	obj_add_func(env, obj, ch, CHN_NetSetURL, "set_url");
 	obj_add_func(env, obj, ch, CHN_NetSetMethod, "set_method");
+	obj_add_func(env, obj, ch, CHN_NetSetRequestHeader, "set_request_header");
 	obj_add_func(env, obj, ch, CHN_NetReadRet, "read_ret");
 	obj_add_func(env, obj, ch, CHN_NetReadBuf, "read_buf");
 	obj_add_func(env, obj, ch, CHN_NetGetError, "get_error");
diff --git a/tests/js/ring.js b/tests/js/ring.js
index 3a4d25d..4b081ae 100644
--- a/tests/js/ring.js
+++ b/tests/js/ring.js
@@ -140,11 +140,23 @@ function test_simple_http()
 	assert(h.ch.read_ret() === payload.length);
 }
 
+function test_simple_http_set_header()
+{
+	const ua = "This is just a test user agent!";
+	let h = new SimpleHttp("http://127.0.0.1:8000/index.php?action=user_agent", "GET");
+	h.ch.set_request_header("User-Agent", ua);
+	h.prep_read(1024);
+	h.run();
+	assert(h.get_buffer() === ua);
+	assert(h.ch.read_ret() === ua.length);
+}
+
 function main()
 {
 	test_nop();
 	test_chnet_ring_multiple();
 	test_simple_http();
+	test_simple_http_set_header();
 }
 
 main();
-- 
Ammar Faizi