From 7a56113d69c5a56ea333f87c5d67cd5a371aa090 Mon Sep 17 00:00:00 2001
From: littleblack111 <littleblack11111@gmail.com>
Date: Fri, 7 Mar 2025 18:02:11 +0800
Subject: fix: session not disconnected properly

perf: use server sockfd directly instead of a dupe of it
---
 src/sessionManager.hpp | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'src/sessionManager.hpp')

diff --git a/src/sessionManager.hpp b/src/sessionManager.hpp
index 2dd825a..0b447b5 100644
--- a/src/sessionManager.hpp
+++ b/src/sessionManager.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "format.hpp"
 #include "session.hpp"
 #include <memory>
 #include <thread>
@@ -9,11 +10,12 @@ class CSessionManager {
   public:
 	CSessionManager();
 	~CSessionManager();
-	void enterLoop();
-	void newSession(Hyprutils::OS::CFileDescriptor sockfd);
-	void broadcast(const std::string &message) const;
-	void kick(const CSession *session) const;
-	void addSession(const CSession &session);
+	void												enterLoop();
+	std::pair<std::jthread, std::shared_ptr<CSession>> *newSession(Hyprutils::OS::CFileDescriptor sockfd);
+	void												broadcast(const std::string &message, eFormatType type = {}) const;
+	void												kick(const CSession *session) const;
+	void												addSession(const CSession &session);
+	void												removeSession(std::pair<std::jthread, std::shared_ptr<CSession>> *session);
 
   private:
 	std::vector<std::pair<std::jthread, std::shared_ptr<CSession>>> m_vSessions;
-- 
cgit v1.2.3