From 509f0c9a8a83e10a8a2a677980612fadbf4009f2 Mon Sep 17 00:00:00 2001
From: vaxerski <43317083+vaxerski@users.noreply.github.com>
Date: Mon, 21 Mar 2022 18:29:41 +0100
Subject: IPC is now a socket

---
 src/debug/HyprCtl.cpp          | 75 ++++++++++++++++++++++++++++++------------
 src/debug/HyprCtl.hpp          |  1 +
 src/managers/ThreadManager.cpp |  4 +--
 3 files changed, 57 insertions(+), 23 deletions(-)

(limited to 'src')

diff --git a/src/debug/HyprCtl.cpp b/src/debug/HyprCtl.cpp
index 1b06617a..56319e53 100644
--- a/src/debug/HyprCtl.cpp
+++ b/src/debug/HyprCtl.cpp
@@ -1,6 +1,12 @@
 #include "HyprCtl.hpp"
 
+#include <netinet/in.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
 #include <sys/stat.h>
+#include <sys/types.h>
 #include <unistd.h>
 
 #include <string>
@@ -46,31 +52,58 @@ std::string workspacesRequest() {
     return result;
 }
 
-void HyprCtl::tickHyprCtl() {
-    struct stat buf;
+void HyprCtl::startHyprCtlSocket() {
+    int port = 9187;
 
-    if (stat("/tmp/hypr/.hyprlandrq", &buf) == 0) {
-        // file exists, let's open it
+    std::thread([&]() {
+        const auto SOCKET = socket(AF_INET, SOCK_STREAM, 0);
 
-        requestStream.open("/tmp/hypr/.hyprlandrq");
-
-        std::string request = "";
-        std::getline(requestStream, request);
+        if (SOCKET < 0) {
+            Debug::log(ERR, "Couldn't start the Hyprland Socket. (1) IPC will not work.");
+            return;
+        }
 
-        requestStream.close();
-        
-        std::string reply = "";
-        if (request == "R>monitors") reply = monitorsRequest();
-        if (request == "R>workspaces") reply = workspacesRequest();
-        if (request == "R>clients") reply = clientsRequest();
+        const sockaddr_in SERVERADDRESS = {.sin_family = AF_INET, .sin_port = port, .sin_addr = (in_addr)INADDR_ANY};
 
-        if (reply != "") {
-            std::ofstream ofs;
-            ofs.open("/tmp/hypr/.hyprlandrq", std::ios::trunc);
-            ofs << "RPLY:\n" << reply;
-            ofs.close();
+        if (bind(SOCKET, (sockaddr*)&SERVERADDRESS, sizeof(SERVERADDRESS)) < 0) {
+            Debug::log(ERR, "Couldn't start the Hyprland Socket. (2) IPC will not work.");
+            return;
         }
 
-        // the hyprctl app deletes the file when done.
-    }
+        // 10 max queued.
+        listen(SOCKET, 10);
+
+        sockaddr_in clientAddress;
+        socklen_t clientSize = sizeof(clientAddress);
+
+        char readBuffer[1024] = {0};
+
+        Debug::log(LOG, "Hypr socket started on port %i", SERVERADDRESS.sin_port);
+
+        std::string cmd = "rm -f /tmp/hypr/.socket && echo \"" + std::to_string(SERVERADDRESS.sin_port) + "\" > /tmp/hypr/.socket";
+        system(cmd.c_str()); // forgive me for using system() but it works and it doesnt matter here that much
+
+        while(1) {
+            const auto ACCEPTEDCONNECTION = accept(SOCKET, (sockaddr*)&clientAddress, &clientSize);
+
+            if (ACCEPTEDCONNECTION < 0) {
+                Debug::log(ERR, "Couldn't listen on the Hyprland Socket. (3) IPC will not work.");
+                break;
+            }
+
+            auto messageSize = read(ACCEPTEDCONNECTION, readBuffer, 1024);
+            readBuffer[messageSize == 1024 ? 1024 : messageSize] = '\0';
+
+            std::string request(readBuffer);
+
+            std::string reply = "";
+            if (request == "R>monitors") reply = monitorsRequest();
+            if (request == "R>workspaces") reply = workspacesRequest();
+            if (request == "R>clients") reply = clientsRequest();
+
+            write(ACCEPTEDCONNECTION, reply.c_str(), reply.length());
+
+            close(ACCEPTEDCONNECTION);
+        }
+    }).detach();
 }
\ No newline at end of file
diff --git a/src/debug/HyprCtl.hpp b/src/debug/HyprCtl.hpp
index 41303364..1a8851c7 100644
--- a/src/debug/HyprCtl.hpp
+++ b/src/debug/HyprCtl.hpp
@@ -4,6 +4,7 @@
 #include <fstream>
 
 namespace HyprCtl {
+    void            startHyprCtlSocket();
     void            tickHyprCtl();
 
     inline std::ifstream requestStream;
diff --git a/src/managers/ThreadManager.cpp b/src/managers/ThreadManager.cpp
index 6ad6339b..fa3d0d4e 100644
--- a/src/managers/ThreadManager.cpp
+++ b/src/managers/ThreadManager.cpp
@@ -20,6 +20,8 @@ void CThreadManager::handle() {
 
     g_pConfigManager->init();
 
+    HyprCtl::startHyprCtlSocket();
+
     while (3.1415f) {
         slowUpdate++;
         if (slowUpdate >= g_pConfigManager->getInt("general:max_fps")){
@@ -27,8 +29,6 @@ void CThreadManager::handle() {
             slowUpdate = 0;
         }
 
-        HyprCtl::tickHyprCtl();
-
         std::this_thread::sleep_for(std::chrono::microseconds(1000000 / g_pConfigManager->getInt("general:max_fps")));
     }
 }
\ No newline at end of file
-- 
cgit v1.2.3