mitoxide-agent 0.1.0

Remote agent binary for Mitoxide
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

//! Mitoxide Agent Binary
//!
//! The remote agent that executes operations on behalf of the client.

use anyhow::Result;
use std::sync::Arc;
use tracing::{info, error};

mod agent;
mod handlers;
mod router;
mod bootstrap;

use agent::AgentLoop;
use handlers::{ProcessHandler, FileHandler, PtyHandler, PingHandler, WasmHandler};

#[tokio::main]
async fn main() -> Result<()> {
    // Initialize tracing
    tracing_subscriber::fmt()
        .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
        .init();

    info!("Starting Mitoxide agent");

    // Create and run the agent loop
    let mut agent = AgentLoop::new();
    
    // Register handlers
    agent.register_handler("process_exec".to_string(), Arc::new(ProcessHandler)).await;
    agent.register_handler("file_get".to_string(), Arc::new(FileHandler)).await;
    agent.register_handler("file_put".to_string(), Arc::new(FileHandler)).await;
    agent.register_handler("dir_list".to_string(), Arc::new(FileHandler)).await;
    agent.register_handler("pty_exec".to_string(), Arc::new(PtyHandler)).await;
    agent.register_handler("ping".to_string(), Arc::new(PingHandler)).await;
    
    // Register WASM handler
    match WasmHandler::new() {
        Ok(wasm_handler) => {
            agent.register_handler("wasm_exec".to_string(), Arc::new(wasm_handler)).await;
            info!("WASM handler registered successfully");
        }
        Err(e) => {
            error!("Failed to create WASM handler: {}", e);
            // Continue without WASM support
        }
    }
    
    info!("All handlers registered, starting agent loop");
    
    if let Err(e) = agent.run().await {
        error!("Agent error: {}", e);
        std::process::exit(1);
    }

    info!("Agent shutting down");
    Ok(())
}