feat: `shared_runner` for `run_after_build` in nebuild.

Signed-off-by: Amlal El Mahrouss <amlal@nekernel.org>

1 files changed, 19 insertions, 15 deletions

diff --git a/dev/src/JSONManifestBuilder.cc b/dev/src/JSONManifestBuilder.cc
index 33d7f62..f64dba4 100644
--- a/dev/src/JSONManifestBuilder.cc
+++ b/dev/src/JSONManifestBuilder.cc
@@ -4,6 +4,7 @@
 // ============================================================= //
 
 #include <BuildKit/JSONManifestBuilder.h>
+#include <dlfcn.h>
 
 using JSON = nlohmann::json;
 
@@ -99,12 +100,24 @@ bool JSONManifestBuilder::BuildTarget(const std::string& argv_val, const bool dr
 
     try {
       if (json_obj["run_after_build"].get<bool>()) {
-        if (target.ends_with(".so")) {
-          NeBuild::Logger::info()
-              << "error: can't open dynamic library, it mayn't have an entrypoint" << std::endl;
+        if (target.ends_with(".so") || target.ends_with(".dylib")) {
+#if defined(NEBUILD_POSIX)
+          auto dll = dlopen(target.c_str(), RTLD_LAZY);
 
-          return true;
-        } else if (target.ends_with(".dylib") || target.ends_with(".dll")) {
+          if (dll) {
+            int (*entrypoint)(void) = nullptr;
+            entrypoint = (decltype(entrypoint))dlsym(dll, "shared_runner");
+
+            if (entrypoint) entrypoint();
+
+            dlclose(dll);
+
+            return true;
+          }
+#endif
+
+          return false;
+        } else {
           std::ifstream     file = std::ifstream(target);
           std::stringstream ss;
 
@@ -119,17 +132,8 @@ bool JSONManifestBuilder::BuildTarget(const std::string& argv_val, const bool dr
             NeBuild::Logger::info()
                 << "error: can't open FEP dynamic library, it mayn't contain an entrypoint"
                 << std::endl;
-          else if (ss.str()[0] == 'M' && ss.str()[1] == 'Z')
-            NeBuild::Logger::info()
-                << "error: can't open MZ dynamic library, it mayn't contain an entrypoint"
-                << std::endl;
-          else if (ss.str()[0] == 0x7F && ss.str()[1] == 'E') {
-            NeBuild::Logger::info()
-                << "error: can't open ELF dynamic library, it mayn't contain an entrypoint"
-                << std::endl;
-          }
 
-          return true;
+          return false;
         }
 
 #if defined(NEBUILD_WINDOWS)