add 'headers' argument to network.get, network.get_binary and network.post

src/logic/scripting/lua/libs/libnetwork.cpp

@@ -12,7 +12,7 @@ static int l_get(lua::State* L, network::Network& network) {
     auto onResponse = lua::create_lambda_nothrow(L);
 
     network::OnReject onReject = nullptr;
-    if (lua::gettop(L) >= 3) {
+    if (!lua::isnoneornil(L, 3)) {
         lua::pushvalue(L, 3);
         auto callback = lua::create_lambda_nothrow(L);
         onReject = [callback](int code) {
@@ -20,11 +20,22 @@ static int l_get(lua::State* L, network::Network& network) {
         };
     }
 
+    std::vector<std::string> headers;
+
+    if (lua::istable(L, 4)) {
+        int len = lua::objlen(L, 4);    
+        for (int i = 1; i <= len; i++) {
+            lua::rawgeti(L, i, 4);
+            headers.push_back(lua::tostring(L, -1));
+            lua::pop(L);
+        }
+    }
+
     network.get(url, [onResponse](std::vector<char> bytes) {
         engine->postRunnable([=]() {
             onResponse({std::string(bytes.data(), bytes.size())});
         });
-    }, std::move(onReject));
+    }, std::move(onReject), std::move(headers));
     return 0;
 }
 
@@ -35,7 +46,7 @@ static int l_get_binary(lua::State* L, network::Network& network) {
     auto onResponse = lua::create_lambda_nothrow(L);
 
     network::OnReject onReject = nullptr;
-    if (lua::gettop(L) >= 3) {
+    if (!lua::isnoneornil(L, 3)) {
         lua::pushvalue(L, 3);
         auto callback = lua::create_lambda_nothrow(L);
         onReject = [callback](int code) {
@@ -43,6 +54,17 @@ static int l_get_binary(lua::State* L, network::Network& network) {
         };
     }
 
+    std::vector<std::string> headers;
+
+    if (lua::istable(L, 4)) {
+        int len = lua::objlen(L, 4);    
+        for (int i = 1; i <= len; i++) {
+            lua::rawgeti(L, i, 4);
+            headers.push_back(lua::tostring(L, -1));
+            lua::pop(L);
+        }
+    }
+
     network.get(url, [onResponse](std::vector<char> bytes) {
         auto buffer = std::make_shared<util::Buffer<ubyte>>(
             reinterpret_cast<const ubyte*>(bytes.data()), bytes.size()
@@ -50,7 +72,8 @@ static int l_get_binary(lua::State* L, network::Network& network) {
         engine->postRunnable([=]() {
             onResponse({buffer});
         });
-    }, std::move(onReject));
+    }, std::move(onReject), std::move(headers));
+
     return 0;
 }
 
@@ -62,7 +85,7 @@ static int l_post(lua::State* L, network::Network& network) {
     auto onResponse = lua::create_lambda_nothrow(L);
 
     network::OnReject onReject = nullptr;
-    if (lua::gettop(L) >= 4) {
+    if (!lua::isnoneornil(L, 4)) {
         lua::pushvalue(L, 4);
         auto callback = lua::create_lambda_nothrow(L);
         onReject = [callback](int code) {
@@ -77,6 +100,16 @@ static int l_post(lua::State* L, network::Network& network) {
         string = json::stringify(data, false);
     }
 
+    std::vector<std::string> headers;
+    if (lua::istable(L, 5)) {
+        int len = lua::objlen(L, 5);    
+        for (int i = 1; i <= len; i++) {
+            lua::rawgeti(L, i, 5);
+            headers.push_back(lua::tostring(L, -1));
+            lua::pop(L);
+        }
+    }
+
     engine->getNetwork().post(url, string, [onResponse](std::vector<char> bytes) {
         auto buffer = std::make_shared<util::Buffer<ubyte>>(
             reinterpret_cast<const ubyte*>(bytes.data()), bytes.size()
@@ -84,7 +117,7 @@ static int l_post(lua::State* L, network::Network& network) {
         engine->postRunnable([=]() {
             onResponse({std::string(bytes.data(), bytes.size())});
         });
-    }, std::move(onReject));
+    }, std::move(onReject), std::move(headers));
     return 0;
 }
 

src/network/Network.cpp

@@ -57,6 +57,7 @@ struct Request {
     long maxSize;
     bool followLocation = false;
     std::string data;
+    std::vector<std::string> headers;
 };
 
 class CurlRequests : public Requests {
@@ -86,10 +87,18 @@ public:
         const std::string& url,
         OnResponse onResponse,
         OnReject onReject,
+        std::vector<std::string> headers,
         long maxSize
     ) override {
         Request request {
-            RequestType::GET, url, onResponse, onReject, maxSize, false, ""};
+            RequestType::GET,
+            url,
+            onResponse,
+            onReject,
+            maxSize,
+            false,
+            "",
+            std::move(headers)};
         processRequest(std::move(request));
     }
 
@@ -98,10 +107,18 @@ public:
         const std::string& data,
         OnResponse onResponse,
         OnReject onReject=nullptr,
+        std::vector<std::string> headers = {},
         long maxSize=0
     ) override {
         Request request {
-            RequestType::POST, url, onResponse, onReject, maxSize, false, ""};
+            RequestType::POST,
+            url,
+            onResponse,
+            onReject,
+            maxSize,
+            false,
+            "",
+            std::move(headers)};
         request.data = data;
         processRequest(std::move(request));
     }
@@ -121,6 +138,10 @@ public:
         curl_easy_setopt(curl, CURLOPT_POST, request.type == RequestType::POST);
         
         curl_slist* hs = NULL;
+        
+        for (const auto& header : request.headers) {
+            hs = curl_slist_append(hs, header.c_str());
+        }
 
         switch (request.type) {
             case RequestType::GET:
@@ -817,9 +838,10 @@ void Network::get(
     const std::string& url,
     OnResponse onResponse,
     OnReject onReject,
+    std::vector<std::string> headers,
     long maxSize
 ) {
-    requests->get(url, onResponse, onReject, maxSize);
+    requests->get(url, onResponse, onReject, std::move(headers), maxSize);
 }
 
 void Network::post(
@@ -827,9 +849,12 @@ void Network::post(
     const std::string& fieldsData,
     OnResponse onResponse,
     OnReject onReject,
+    std::vector<std::string> headers,
     long maxSize
 ) {
-    requests->post(url, fieldsData, onResponse, onReject, maxSize);
+    requests->post(
+        url, fieldsData, onResponse, onReject, std::move(headers), maxSize
+    );
 }
 
 Connection* Network::getConnection(u64id_t id) {

src/network/Network.hpp

@@ -25,6 +25,7 @@ namespace network {
             const std::string& url,
             OnResponse onResponse,
             OnReject onReject=nullptr,
+            std::vector<std::string> headers = {},
             long maxSize=0
         ) = 0;
 
@@ -33,6 +34,7 @@ namespace network {
             const std::string& data,
             OnResponse onResponse,
             OnReject onReject=nullptr,
+            std::vector<std::string> headers = {},
             long maxSize=0
         ) = 0;
 
@@ -144,6 +146,7 @@ namespace network {
             const std::string& url,
             OnResponse onResponse,
             OnReject onReject = nullptr,
+            std::vector<std::string> headers = {},
             long maxSize=0
         );
 
@@ -152,6 +155,7 @@ namespace network {
             const std::string& fieldsData,
             OnResponse onResponse,
             OnReject onReject = nullptr,
+            std::vector<std::string> headers = {},
             long maxSize=0
         );