From 241ac2f59329611b4f1f90de05353652237c3c47 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Thu, 18 Sep 2025 15:13:55 -0400
Subject: [PATCH] Graceful shutdown menu

---
 daemon/HaikuDaemon.cpp | 48 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 44 insertions(+), 4 deletions(-)

diff --git a/daemon/HaikuDaemon.cpp b/daemon/HaikuDaemon.cpp
index 0a8b8a31..a1aa1e8b 100644
--- a/daemon/HaikuDaemon.cpp
+++ b/daemon/HaikuDaemon.cpp
@@ -8,36 +8,76 @@
 
 #if defined(__HAIKU__)
 
-#include "Daemon.h"
-
+#include <memory>
 #include <MenuItem.h>
 #include <MenuBar.h>
+#include <MessageRunner.h>
 #include <Window.h>
 #include <Application.h>
 
+#include "RouterContext.h"
+#include "Tunnel.h"
+#include "Daemon.h"
+
+constexpr int M_GRACEFUL_SHUTDOWN = 1;
+constexpr int C_GRACEFUL_SHUTDOWN_UPDATE = 2;
+constexpr bigtime_t GRACEFUL_SHUTDOWN_UPDATE_INTERVAL = 1000*1000; // in microseconds
+constexpr int GRACEFUL_SHUTDOWN_UPDATE_COUNT = 600; // 10 minutes
+
 class MainWindow: public BWindow
 {
 	public:
 		MainWindow ();
+	
+	private:
+		void MessageReceived (BMessage * msg) override;	
+	
+	private:
+		BMessenger m_Messenger;
+		std::unique_ptr<BMessageRunner> m_GracefulShutdownTimer;	
 };	
 
 class I2PApp: public BApplication
 {
 	public:
-		I2PApp ();
+		I2PApp ();	
 };
 
 MainWindow::MainWindow ():
-	BWindow (BRect(100, 100, 500, 400), "i2pd", B_TITLED_WINDOW, B_QUIT_ON_WINDOW_CLOSE)
+	BWindow (BRect(100, 100, 500, 400), "i2pd", B_TITLED_WINDOW, B_QUIT_ON_WINDOW_CLOSE),
+	m_Messenger (nullptr, this)
 {	
 	auto r = Bounds (); r.bottom = 20;
 	auto menuBar = new BMenuBar (r, "menubar");
 	AddChild (menuBar);
 	auto runMenu = new BMenu ("Run");
+	runMenu->AddItem (new BMenuItem ("Graceful shutdown", new BMessage (M_GRACEFUL_SHUTDOWN), 'G'));
 	runMenu->AddItem (new BMenuItem ("Quit", new BMessage (B_QUIT_REQUESTED), 'Q'));
 	menuBar->AddItem (runMenu);
 }	
 
+void MainWindow::MessageReceived (BMessage * msg)
+{
+	if (!msg) return;
+	switch (msg->what)
+	{
+		case M_GRACEFUL_SHUTDOWN:
+			i2p::context.SetAcceptsTunnels (false);
+			m_GracefulShutdownTimer = std::make_unique<BMessageRunner>(m_Messenger, 
+				BMessage (C_GRACEFUL_SHUTDOWN_UPDATE), GRACEFUL_SHUTDOWN_UPDATE_INTERVAL);
+		break;
+		case C_GRACEFUL_SHUTDOWN_UPDATE:
+			if (i2p::tunnel::tunnels.CountTransitTunnels () <= 0)
+			{
+				m_GracefulShutdownTimer = nullptr;
+				m_Messenger.SendMessage (B_QUIT_REQUESTED);
+			}	
+		break;
+		default:
+			BWindow::MessageReceived (msg);
+	}	
+}	
+
 I2PApp::I2PApp (): BApplication("application/x-vnd.purplei2p-i2pd")
 {
 	auto mainWindow = new MainWindow ();