From e7b557f754f20353dfe12094558de401fa3df04c Mon Sep 17 00:00:00 2001
From: Luna <luna@luna4.xyz>
Date: Sat, 3 May 2025 04:12:38 +0200
Subject: [PATCH] improved the post receive file

---
 main.cpp | 35 ++++++++++++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 3 deletions(-)

diff --git a/main.cpp b/main.cpp
index 7415d3b..6c8161f 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,5 +1,7 @@
 #include "netlib/src/netlib.h"
+#ifdef __linux__
 #include <sys/sendfile.h>
+#endif
 #include <print>
 #include <fstream>
 
@@ -18,6 +20,20 @@ int atoi_newline(const char *data)
     return ret;
 }
 
+std::string get_filename(std::string_view data)
+{
+    int index = 1;
+    std::string ret;
+    while (true)
+    {
+        if (data[index] == '"')
+            break;
+        ret.push_back(data[index]);
+        index++;
+    }
+    return ret;
+}
+
 int main()
 {
     netlib::server_raw server;
@@ -53,7 +69,12 @@ int main()
                         size = file.tellg() - size;
                         file.close();
                         int filefd = open("../test.html", O_RDONLY);
+                        #ifdef __linux__
                         sendfile(user, filefd, 0, size);
+                        #endif
+                        #ifdef __APPLE__
+                        sendfile(filefd, user, 0, (long long*)&size, nullptr, 0);
+                        #endif
                     }
                     server.disconnect_user(user);
                 }
@@ -64,10 +85,18 @@ int main()
                     dat.remove_prefix(strlen("Content-Length: "));
                     int lenght = atoi_newline(dat.data());
                     std::println("{} {}", dat, lenght);
-                    pos = dat.find("image/png");
+                    pos = dat.find("filename=");
                     dat.remove_prefix(pos);
-                    dat.remove_prefix(strlen("image/png\r\n") + 2);
-                    std::ofstream a("Test.png", std::ios::binary);
+                    dat.remove_prefix(strlen("filename="));
+                    std::string_view file_data(data);
+                    std::string filename = get_filename(dat);
+                    std::println("Filename {}", filename);
+                    dat.remove_prefix(dat.find("\r\n\r\n"));
+                    dat.remove_prefix(dat.find("\r\n\r\n"));
+                    dat.remove_prefix(4);
+                    std::println("{}", dat);
+                    
+                    std::ofstream a(filename, std::ios::binary);
                     a.write(dat.data(), lenght);
                     netlib::send_packet(std::make_tuple(std::string("HTTP/1.1 201 CREATED")), user);
                     server.disconnect_user(user);