From b282c1eb6d3e2b8caa6552fd41a31cd6aea85355 Mon Sep 17 00:00:00 2001
From: strawberry <strawberry@puppygock.gay>
Date: Sun, 28 Apr 2024 18:29:48 -0400
Subject: [PATCH] add (probably messy) support for hot lib reload via admin
 command

`!admin test test1`

Signed-off-by: strawberry <strawberry@puppygock.gay>
---
 Cargo.lock                        |  5 +++++
 Cargo.toml                        |  8 +++++++-
 lib/Cargo.toml                    |  7 +++++++
 lib/src/lib.rs                    |  4 ++++
 src/service/admin/mod.rs          | 12 ++++++++++++
 src/service/admin/test_cmd/mod.rs | 31 +++++++++++++++++++++++++++++++
 6 files changed, 66 insertions(+), 1 deletion(-)
 create mode 100644 lib/Cargo.toml
 create mode 100644 lib/src/lib.rs
 create mode 100644 src/service/admin/test_cmd/mod.rs

diff --git a/Cargo.lock b/Cargo.lock
index 0f862072..9bee6c43 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -575,6 +575,7 @@ dependencies = [
  "ipaddress",
  "itertools 0.12.1",
  "jsonwebtoken",
+ "lib",
  "log",
  "loole",
  "lru-cache",
@@ -1796,6 +1797,10 @@ version = "1.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "830d08ce1d1d941e6b30645f1a0eb5643013d835ce3779a5fc208261dbe10f55"
 
+[[package]]
+name = "lib"
+version = "0.1.0"
+
 [[package]]
 name = "libc"
 version = "0.2.153"
diff --git a/Cargo.toml b/Cargo.toml
index 2302972a..4fddac7c 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -16,11 +16,17 @@ edition = "2021"
 # See also `rust-toolchain.toml`
 rust-version = "1.76.0"
 
+# for hot lib reload
+[workspace]
+members = ["lib"]
 
 [dependencies]
 console-subscriber = { version = "0.1", optional = true }
 
+# for hot lib reload
 hot-lib-reloader = { version = "^0.7", optional = true }
+lib = { path = "lib", optional = true }
+#no-mangle-if-debug = { version = "*" }
 
 # Used for secure identifiers
 rand = "0.8.5"
@@ -400,7 +406,7 @@ perf_measurements = [
 # incompatible with release_max_log_level
 tokio_console = ["console-subscriber", "tokio/tracing"]
 
-hot_reload = ["dep:hot-lib-reloader"]
+hot_reload = ["dep:hot-lib-reloader", "lib"]
 
 hardened_malloc = ["hardened_malloc-rs"]
 
diff --git a/lib/Cargo.toml b/lib/Cargo.toml
new file mode 100644
index 00000000..b301b690
--- /dev/null
+++ b/lib/Cargo.toml
@@ -0,0 +1,7 @@
+[package]
+name = "lib"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["rlib", "dylib"]
diff --git a/lib/src/lib.rs b/lib/src/lib.rs
new file mode 100644
index 00000000..91e76586
--- /dev/null
+++ b/lib/src/lib.rs
@@ -0,0 +1,4 @@
+#[no_mangle]
+pub fn test_command() {
+	println!("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
+}
diff --git a/src/service/admin/mod.rs b/src/service/admin/mod.rs
index 64ca48c4..b6b810bd 100644
--- a/src/service/admin/mod.rs
+++ b/src/service/admin/mod.rs
@@ -27,6 +27,9 @@ use tokio::sync::{Mutex, MutexGuard};
 use tracing::{error, warn};
 
 use self::fsck::FsckCommand;
+#[cfg(feature = "hot_reload")]
+#[allow(unused_imports)]
+use self::test_cmd::TestCommands;
 use super::pdu::PduBuilder;
 use crate::{
 	service::admin::{
@@ -44,6 +47,9 @@ pub(crate) mod media;
 pub(crate) mod query;
 pub(crate) mod room;
 pub(crate) mod server;
+#[cfg(feature = "hot_reload")]
+#[allow(unused_imports)]
+pub(crate) mod test_cmd;
 pub(crate) mod user;
 
 const PAGE_SIZE: usize = 100;
@@ -87,6 +93,10 @@ enum AdminCommand {
 	#[command(subcommand)]
 	/// - Query all the database getters and iterators
 	Fsck(FsckCommand),
+
+	#[cfg(feature = "hot_reload")]
+	#[command(subcommand)]
+	Test(TestCommands),
 }
 
 #[derive(Debug)]
@@ -304,6 +314,8 @@ impl Service {
 			AdminCommand::Debug(command) => debug::process(command, body).await?,
 			AdminCommand::Query(command) => query::process(command, body).await?,
 			AdminCommand::Fsck(command) => fsck::process(command, body).await?,
+			#[cfg(feature = "hot_reload")]
+			AdminCommand::Test(command) => test_cmd::process(command, body).await?,
 		};
 
 		Ok(reply_message_content)
diff --git a/src/service/admin/test_cmd/mod.rs b/src/service/admin/test_cmd/mod.rs
new file mode 100644
index 00000000..852afbe7
--- /dev/null
+++ b/src/service/admin/test_cmd/mod.rs
@@ -0,0 +1,31 @@
+use clap::Subcommand;
+#[cfg(feature = "hot_reload")]
+#[allow(unused_imports)]
+#[allow(clippy::wildcard_imports)]
+use hot_lib::*;
+use ruma::events::room::message::RoomMessageEventContent;
+
+use crate::{debug_error, Result};
+
+#[cfg(feature = "hot_reload")]
+#[hot_lib_reloader::hot_module(dylib = "lib")]
+mod hot_lib {
+	hot_functions_from_file!("lib/src/lib.rs");
+}
+
+#[cfg_attr(test, derive(Debug))]
+#[derive(Subcommand)]
+pub(crate) enum TestCommands {
+	Test1,
+}
+
+pub(crate) async fn process(command: TestCommands, _body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	Ok(match command {
+		TestCommands::Test1 => {
+			debug_error!("before calling test_command");
+			test_command();
+			debug_error!("after calling test_command");
+			RoomMessageEventContent::notice_plain(String::from("loaded"))
+		},
+	})
+}