Add get_runtime() helper for runtime initialization (#134)

Author: penberg

cli/Cargo.toml

@@ -4,6 +4,10 @@ version = "0.4.0-pre.4"
 edition = "2021"
 repository = "https://github.com/tursodatabase/agentfs"
 
+[lib]
+name = "agentfs"
+path = "src/lib.rs"
+
 [[bin]]
 name = "agentfs"
 path = "src/main.rs"

cli/src/cmd/mount.rs

@@ -62,7 +62,7 @@ pub fn mount(args: MountArgs) -> Result<()> {
     };
 
     let mount = move || {
-        let rt = tokio::runtime::Runtime::new()?;
+        let rt = crate::get_runtime();
         let agentfs = rt.block_on(AgentFS::open(opts))?;
 
         // Check for overlay configuration

cli/src/lib.rs

@@ -0,0 +1,16 @@
+pub mod cmd;
+pub mod parser;
+pub mod sandbox;
+
+#[cfg(any(target_os = "linux", target_os = "macos"))]
+pub mod daemon;
+
+#[cfg(any(target_os = "linux", target_os = "macos"))]
+pub mod fuse;
+
+#[cfg(target_os = "macos")]
+pub mod nfs;
+
+pub fn get_runtime() -> tokio::runtime::Runtime {
+    tokio::runtime::Runtime::new().expect("Internal error: failed to initialize runtime")
+}

cli/src/main.rs

@@ -1,21 +1,9 @@
-mod cmd;
-mod parser;
-mod sandbox;
-
-#[cfg(any(target_os = "linux", target_os = "macos"))]
-mod daemon;
-
-#[cfg(any(target_os = "linux", target_os = "macos"))]
-mod fuse;
-
-#[cfg(target_os = "macos")]
-mod nfs;
-
 use clap::{CommandFactory, Parser};
 use clap_complete::CompleteEnv;
 
-use crate::{
-    cmd::completions::handle_completions,
+use agentfs::{
+    cmd::{self, completions::handle_completions},
+    get_runtime,
     parser::{Args, Command, FsCommand},
 };
 
@@ -27,7 +15,7 @@ fn main() {
 
     match args.command {
         Command::Init { id, force, base } => {
-            let rt = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
+            let rt = get_runtime();
             if let Err(e) = rt.block_on(cmd::init::init_database(id, force, base)) {
                 eprintln!("Error: {}", e);
                 std::process::exit(1);
@@ -41,7 +29,7 @@ fn main() {
             command,
             args,
         } => {
-            let rt = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
+            let rt = get_runtime();
             if let Err(e) = rt.block_on(cmd::handle_run_command(
                 allow,
                 no_default_allows,
@@ -77,14 +65,14 @@ fn main() {
             }
         }
         Command::Diff { id_or_path } => {
-            let rt = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
+            let rt = get_runtime();
             if let Err(e) = rt.block_on(cmd::fs::diff_filesystem(id_or_path)) {
                 eprintln!("Error: {}", e);
                 std::process::exit(1);
             }
         }
         Command::Fs { command } => {
-            let rt = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
+            let rt = get_runtime();
             match command {
                 FsCommand::Ls {
                     id_or_path,

cli/src/sandbox/overlay.rs

@@ -116,7 +116,7 @@ pub async fn run_cmd(
 
     // Start FUSE in a separate thread
     let fuse_handle = std::thread::spawn(move || {
-        let rt = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
+        let rt = crate::get_runtime();
         crate::fuse::mount(overlay, fuse_opts, rt)
     });