use clap::Parser;
use erl_dist::term::List;

#[derive(Debug, Parser)]
struct Args {
    node_name: erl_dist::node::NodeName,
    module: String,
    function: String,

    #[clap(long)]
    cookie: Option<String>,

    #[clap(long, short)]
    port: Option<u16>,
}

fn main() -> anyhow::Result<()> {
    let args = Args::parse();
    let cookie = if let Some(cookie) = &args.cookie {
        cookie.clone()
    } else if let Some(dir) = dirs::home_dir().filter(|dir| dir.join(".erlang.cookie").exists()) {
        std::fs::read_to_string(dir.join(".erlang.cookie"))?
    } else {
        anyhow::bail!("Could not find the cookie file $HOME/.erlang.cookie. Please specify `-cookie` arg instead.");
    };

    smol::block_on(async {
        let client = if let Some(port) = args.port {
            erl_rpc::RpcClient::connect_with_port(&args.node_name.to_string(), port, &cookie)
                .await?
        } else {
            erl_rpc::RpcClient::connect(&args.node_name.to_string(), &cookie).await?
        };
        let mut handle = client.handle();
        smol::spawn(async {
            if let Err(e) = client.run().await {
                eprintln!("RpcClient Error: {}", e);
            }
        })
        .detach();

        let result = handle
            .call(args.module.into(), args.function.into(), List::nil())
            .await?;
        println!("{}", result);

        Ok(())
    })
}