...

Author: despiegk

Cargo.toml

@@ -20,3 +20,15 @@ thiserror = "1.0"
 clap = "2.33"
 git-version = "0.3.5"
 command-group = "1.0.8"
+
+[lib]
+name = "zinit"
+path = "src/lib.rs"
+
+[[bin]]
+name = "zinit"
+path = "src/main.rs"
+
+[[bin]]
+name = "testapp"
+path = "src/bin/testapp.rs"

src/app/mod.rs

@@ -1,4 +1,4 @@
-mod api;
+pub mod api;
 
 use crate::zinit;
 use anyhow::{Context, Result};

src/bin/testapp.rs

@@ -0,0 +1,107 @@
+extern crate zinit;
+
+use anyhow::Result;
+use std::path::Path;
+use tokio::time::{sleep, Duration};
+use std::env;
+
+use zinit::app::api::Client;
+use zinit::testapp;
+
+#[tokio::main]
+async fn main() -> Result<()> {
+    // Define paths for socket and config
+    let temp_dir = env::temp_dir();
+    let socket_path = temp_dir.join("zinit-test.sock").to_str().unwrap().to_string();
+    let config_dir = temp_dir.join("zinit-test-config").to_str().unwrap().to_string();
+
+    println!("Starting zinit with socket at: {}", socket_path);
+    println!("Using config directory: {}", config_dir);
+
+    // Start zinit in the background
+    testapp::start_zinit(&socket_path, &config_dir).await?;
+
+    // Wait for zinit to initialize
+    sleep(Duration::from_secs(2)).await;
+
+    // Create a client to communicate with zinit
+    let client = Client::new(&socket_path);
+
+    // Create service configurations
+    println!("Creating service configurations...");
+    
+    // Create a find service
+    testapp::create_service_config(&config_dir, "find-service", "find / -name \"*.txt\" -type f").await?;
+    
+    // Create a sleep service with echo
+    testapp::create_service_config(
+        &config_dir, 
+        "sleep-service", 
+        "sh -c 'echo Starting sleep; sleep 30; echo Finished sleep'"
+    ).await?;
+
+    // Wait for zinit to load the configurations
+    sleep(Duration::from_secs(1)).await;
+
+    // Tell zinit to monitor our services
+    println!("Monitoring services...");
+    client.monitor("find-service").await?;
+    client.monitor("sleep-service").await?;
+
+    // List all services
+    println!("\nListing all services:");
+    let services = client.list().await?;
+    for (name, status) in services {
+        println!("Service: {} - Status: {}", name, status);
+    }
+
+    // Start the find service
+    println!("\nStarting find-service...");
+    client.start("find-service").await?;
+    
+    // Wait a bit and check status
+    sleep(Duration::from_secs(2)).await;
+    let status = client.status("find-service").await?;
+    println!("find-service status: {:?}", status);
+
+    // Start the sleep service
+    println!("\nStarting sleep-service...");
+    client.start("sleep-service").await?;
+    
+    // Wait a bit and check status
+    sleep(Duration::from_secs(2)).await;
+    let status = client.status("sleep-service").await?;
+    println!("sleep-service status: {:?}", status);
+
+    // Stop the find service
+    println!("\nStopping find-service...");
+    client.stop("find-service").await?;
+    
+    // Wait a bit and check status
+    sleep(Duration::from_secs(2)).await;
+    let status = client.status("find-service").await?;
+    println!("find-service status after stopping: {:?}", status);
+
+    // Kill the sleep service with SIGTERM
+    println!("\nKilling sleep-service with SIGTERM...");
+    client.kill("sleep-service", "SIGTERM").await?;
+    
+    // Wait a bit and check status
+    sleep(Duration::from_secs(2)).await;
+    let status = client.status("sleep-service").await?;
+    println!("sleep-service status after killing: {:?}", status);
+
+    // Cleanup - forget services
+    println!("\nForgetting services...");
+    if status.pid == 0 {  // Only forget if it's not running
+        client.forget("sleep-service").await?;
+    }
+    client.forget("find-service").await?;
+
+    // Shutdown zinit
+    println!("\nShutting down zinit...");
+    client.shutdown().await?;
+
+    println!("\nTest completed successfully!");
+    Ok(())
+}

src/lib.rs

@@ -0,0 +1,11 @@
+extern crate serde;
+#[macro_use]
+extern crate anyhow;
+#[macro_use]
+extern crate log;
+extern crate tokio;
+
+pub mod app;
+pub mod manager;
+pub mod zinit;
+pub mod testapp;

src/main.rs

@@ -1,16 +1,10 @@
-extern crate serde;
-#[macro_use]
-extern crate anyhow;
-#[macro_use]
-extern crate log;
-extern crate tokio;
+extern crate zinit;
+
 use anyhow::Result;
 use clap::{App, Arg, SubCommand};
 use git_version::git_version;
 
-mod app;
-mod manager;
-mod zinit;
+use zinit::app;
 
 const GIT_VERSION: &str = git_version!(args = ["--tags", "--always", "--dirty=-modified"]);