[dependencies]
rotastellar-agent = { git = "https://github.com/rotastellar/rotastellar-agent" }

git clone https://github.com/rotastellar/rotastellar-agent.git
cd rotastellar-agent
cargo build --release

#[async_trait]
pub trait Agent: Send + Sync {
    /// Poll the Console API for pending workloads.
    async fn poll(&self) -> Result<Option<WorkloadSpec>, AgentError>;

    /// Report an execution event.
    async fn report_event(&self, event: &AgentEvent) -> Result<(), AgentError>;

    /// Report telemetry data (heartbeat, resource usage).
    async fn report_telemetry(&self, telemetry: &AgentTelemetry) -> Result<(), AgentError>;

    /// Execute a workload.
    async fn execute(&self, workload: &WorkloadSpec) -> Result<(), AgentError>;

    /// Start the agent run loop.
    async fn start(&self) -> Result<(), AgentError>;

    /// Stop the agent gracefully.
    async fn stop(&self) -> Result<(), AgentError>;
}

use rotastellar_agent::{AgentConfig, SimulatedSatellite, Agent};

#[tokio::main]
async fn main() {
    let config = AgentConfig {
        agent_id: "sat-25544".into(),
        api_url: "https://console.rotastellar.com".into(),
        api_key: "rs_live_...".into(),
        poll_interval_s: 30,
    };

    // 100x speed — a 90-minute orbit plays in ~1 minute
    let agent = SimulatedSatellite::new(config, 100.0).unwrap();
    agent.start().await.unwrap();
}

use async_trait::async_trait;
use rotastellar_agent::{Agent, AgentError, AgentEvent, AgentTelemetry, WorkloadSpec};

struct MyAgent {
    config: AgentConfig,
    client: ConsoleClient,
}

#[async_trait]
impl Agent for MyAgent {
    async fn poll(&self) -> Result<Option<WorkloadSpec>, AgentError> {
        self.client.poll_workloads().await
    }

    async fn report_event(&self, event: &AgentEvent) -> Result<(), AgentError> {
        self.client.report_event(&event.job_id, event).await
    }

    async fn report_telemetry(&self, telemetry: &AgentTelemetry) -> Result<(), AgentError> {
        self.client.report_telemetry(telemetry).await
    }

    async fn execute(&self, workload: &WorkloadSpec) -> Result<(), AgentError> {
        // Run your computation here.
        // Report events for each step transition.
        for event in &workload.events {
            self.report_event(event).await?;
        }
        Ok(())
    }

    async fn start(&self) -> Result<(), AgentError> {
        // Implement your run loop
        todo!()
    }

    async fn stop(&self) -> Result<(), AgentError> {
        // Signal graceful shutdown
        todo!()
    }
}

pub struct AgentConfig {
    pub agent_id: String,      // Unique agent identifier (e.g., "sat-25544")
    pub api_url: String,       // Console API URL
    pub api_key: String,       // API key (rs_live_...)
    pub poll_interval_s: u64,  // Seconds between polls (default: 30)
}

pub struct WorkloadSpec {
    pub plan_id: String,          // CAE plan ID
    pub deployment_id: String,    // Console deployment ID
    pub satellite_id: String,     // NORAD catalog ID
    pub plan_data: serde_json::Value,  // Full CAE plan data
    pub events: Vec<AgentEvent>,  // Pre-computed event timeline
}

pub struct AgentEvent {
    pub event_type: String,            // Serializes as "type" in JSON
    pub timestamp: String,             // ISO 8601
    pub job_id: String,
    pub step_id: Option<String>,       // Omitted from JSON when None
    pub payload: serde_json::Value,
}

pub struct AgentTelemetry {
    pub agent_id: String,
    pub status: AgentStatus,          // idle, executing, transferring, offline
    pub timestamp: String,
    pub cpu_percent: Option<f64>,     // Omitted when None
    pub memory_mb: Option<f64>,
    pub battery_percent: Option<f64>,
    pub temperature_c: Option<f64>,
}

pub enum AgentError {
    ApiError(String),              // API request failed
    NetworkError(reqwest::Error),  // Network error
    SerializationError(serde_json::Error),
    ExecutionError(String),        // Execution error
    Stopped,                       // Agent stopped
}

rotastellar-agent simulate \
  --plan plan.json \
  --speed 100 \
  --api-url https://console.rotastellar.com \
  --api-key rs_live_...

rotastellar-agent run \
  --agent-id sat-25544 \
  --api-url https://console.rotastellar.com \
  --api-key rs_live_...