From 3e66411779c2794636e1921a1bb54a88f6c802a3 Mon Sep 17 00:00:00 2001
From: Antonio <antonio@vieiro.net>
Date: Mon, 30 Sep 2024 20:33:18 +0200
Subject: [PATCH] File handlers could receive window number

---
 include/webui.h   | 15 +++++++++++++++
 include/webui.hpp |  7 +++++++
 src/webui.c       | 33 +++++++++++++++++++++++++++++----
 3 files changed, 51 insertions(+), 4 deletions(-)

diff --git a/include/webui.h b/include/webui.h
index 474f51e3..9dbdba8f 100644
--- a/include/webui.h
+++ b/include/webui.h
@@ -424,6 +424,7 @@ WEBUI_EXPORT bool webui_set_default_root_folder(const char* path);
 /**
  * @brief Set a custom handler to serve files. This custom handler should
  * return full HTTP header and body.
+ * This deactivates any previous handler set with `webui_set_file_handler_window`
  *
  * @param window The window number
  * @param handler The handler function: `void myHandler(const char* filename,
@@ -433,6 +434,20 @@ WEBUI_EXPORT bool webui_set_default_root_folder(const char* path);
  */
 WEBUI_EXPORT void webui_set_file_handler(size_t window, const void* (*handler)(const char* filename, int* length));
 
+/**
+ * @brief Set a custom handler to serve files. This custom handler should
+ * return full HTTP header and body.
+ * This deactivates any previous handler set with `webui_set_file_handler`
+ *
+ * @param window The window number
+ * @param handler The handler function: `void myHandler(size_t window, const char* filename,
+ * int* length)`
+ *
+ * @example webui_set_file_handler(myWindow, myHandlerFunction);
+ */
+WEBUI_EXPORT void webui_set_file_handler_window(size_t window, const void* (*handler)(size_t window, const char* filename, int* length));
+
+
 /**
  * @brief Check if the specified window is still running.
  *
diff --git a/include/webui.hpp b/include/webui.hpp
index 77eb471d..749e1c5b 100644
--- a/include/webui.hpp
+++ b/include/webui.hpp
@@ -304,10 +304,17 @@ namespace webui {
         }
 
         // Set a custom handler to serve files. This custom handler should return full HTTP header and body.
+        // Resets previous handler set with `set_file_handler_window`.
         void set_file_handler(const void* (*handler)(const char* filename, int* length)) const {
             webui_set_file_handler(webui_window, handler);
         }
 
+        // Set a custom handler to serve files. This custom handler should return full HTTP header and body.
+        // Resets previous handler set with `set_file_handler`
+        void set_file_handler_window(const void* (*handler)(size_t window, const char* filename, int* length)) const {
+            webui_set_file_handler_window(webui_window, handler);
+        }
+
         // Set the web browser profile to use. An empty `name` and `path` means the default user profile. Need
         // to be called before `webui_show()`.
         void set_profile(const std::string_view name = {""}, const std::string_view path = {""}) const {
diff --git a/src/webui.c b/src/webui.c
index 26a401d0..b979d270 100644
--- a/src/webui.c
+++ b/src/webui.c
@@ -329,6 +329,7 @@ typedef struct _webui_window_t {
     bool position_set;
     size_t process_id;
     const void*(*files_handler)(const char* filename, int* length);
+    const void*(*files_handler_window)(size_t window, const char* filename, int* length);
     webui_event_inf_t* events[WEBUI_MAX_IDS];
     size_t events_count;
     bool is_public;
@@ -735,7 +736,29 @@ void webui_set_file_handler(size_t window, const void*(*handler)(const char* fil
         return;
     _webui_window_t* win = _webui.wins[window];
 
+    // Set the new `files_handler`
     win->files_handler = handler;
+    // And reset any previous `files_handler_window`
+    win->files_handler_window = NULL;
+}
+
+void webui_set_file_handler_window(size_t window, const void*(*handler)(size_t window, const char* filename, int* length)) {
+
+    if (handler == NULL)
+        return;
+
+    // Initialization
+    _webui_init();
+
+    // Dereference
+    if (_webui_mutex_is_exit_now(WEBUI_MUTEX_NONE) || _webui.wins[window] == NULL)
+        return;
+    _webui_window_t* win = _webui.wins[window];
+
+    // Reset any previous `files_handler`
+    win->files_handler = NULL;
+    // And set `files_handler_window`
+    win->files_handler_window = handler;
 }
 
 bool webui_script_client(webui_event_t* e, const char* script, size_t timeout,
@@ -4170,8 +4193,7 @@ static int _webui_external_file_handler(_webui_window_t* win, struct mg_connecti
     const struct mg_request_info * ri = mg_get_request_info(client);
     const char* url = ri->local_uri;
 
-    if (win->files_handler != NULL) {
-
+    if (win->files_handler != NULL || win->files_handler_window != NULL) {
         // Get file content from the external files handler
         size_t length = 0;
 
@@ -4195,7 +4217,10 @@ static int _webui_external_file_handler(_webui_window_t* win, struct mg_connecti
         printf("[Core]\t\t_webui_external_file_handler() -> Calling custom files handler callback\n");
         printf("[Call]\n");
         #endif
-        const void* callback_resp = win->files_handler(url, (int*)&length);
+
+        // True if we pass the window num to the handler, false otherwise.
+        int is_file_handler_window = win->files_handler_window != NULL;
+        const void* callback_resp = is_file_handler_window ? win->files_handler_window(win->num, url, (int*)&length) : win->files_handler(url, (int*)&length);
 
         if (callback_resp != NULL) {
 
@@ -7924,7 +7949,7 @@ static int _webui_http_handler(struct mg_connection* client, void * _win) {
                 _webui_http_send(win, client, "application/javascript", "", 0, false);
             }
         }
-        else if ((win->files_handler != NULL) && (_webui_external_file_handler(win, client, client_id) != 0)) {
+        else if ((win->files_handler != NULL || (win->files_handler_window != NULL)) && (_webui_external_file_handler(win, client, client_id) != 0)) {
 
             // File already handled by the custom external file handler
             // nothing to do now.