use smash::{Actor, Handler, error::Error};
use async_trait::*;

/// A Message
struct Ping(&'static str);

/// The Actor
struct Echo {
    id: i32,
}

#[async_trait]
impl Actor for Echo {
    type Arg = ();
    type Err = Error;

    async fn started(&mut self, arg: Self::Arg) -> Result<(), Self::Err> {
        println!("started {arg:?}");
        Ok(())
    }

    async fn stopping(&mut self, err: &Option<Self::Err>) -> Result<bool, Self::Err> {
        println!("stopping {err:?}");
        Ok(true)
    }

    async fn stopped(&mut self, err: &Option<Self::Err>) {
        println!("stopped {err:?}");
    }
}

#[async_trait]
impl Handler<Ping> for Echo {
    type Output = String;

    async fn handle(&mut self, msg: Ping) -> Self::Output {
        println!("{} {}", self.id, msg.0);
        msg.0.to_string()
    }
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let echo = smash::spawn!(Echo { id: 1 });
    let pong = echo.call(Ping("hi")).await?;

    assert_eq!(pong, "hi");

    echo.quit(Some(Error::Closed));

    Ok(smash::join().await)
}