use bindle::client;
use tempfile::tempdir;
use tokio::io::AsyncWriteExt;
use tokio_stream::StreamExt;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let url = std::env::var("BINDLE_URL")?;
    let root = std::env::var("CARGO_MANIFEST_DIR")?;
    let root_path = std::path::PathBuf::from(root);

    let bindle_client = client::Client::new(
        &url,
        client::tokens::NoToken,
        std::sync::Arc::new(bindle::signature::KeyRing::default()),
    )?;

    // Load an invoice manually and send it to the server
    println!("Creating invoice 1");
    let inv = toml::from_slice(
        &tokio::fs::read(root_path.join("tests/scaffolds/valid_v1/invoice.toml")).await?,
    )?;
    let inv = bindle_client.create_invoice(inv).await?;
    println!("{:?}", inv);

    // Upload a parcel by loading the file into memory
    println!("Creating parcel 1");
    let first_sha = inv.invoice.parcel.expect("Parcel list shouldn't be empty")[0]
        .label
        .sha256
        .clone();
    let data =
        tokio::fs::read(root_path.join("tests/scaffolds/valid_v1/parcels/parcel.dat")).await?;
    bindle_client
        .create_parcel(&inv.invoice.bindle.id, &first_sha, data)
        .await?;

    // Load an invoice from file and stream it to the API
    println!("Creating invoice 2");
    let inv = bindle_client
        .create_invoice_from_file(root_path.join("tests/scaffolds/valid_v2/invoice.toml"))
        .await?;
    println!("{:?}", inv);

    // Get the missing sha from the response
    let second_sha = inv.missing.expect("Should have missing parcels")[0]
        .sha256
        .clone();

    // Get one of the created invoices
    println!("Getting invoice 2");
    let inv = bindle_client
        .get_invoice("enterprise.com/warpcore/2.0.0")
        .await?;
    println!("{:?}", inv);

    // Query the API for a specific version
    println!("Querying for invoice 1");
    let matches = bindle_client
        .query_invoices(bindle::QueryOptions {
            query: Some("enterprise.com/warpcore".to_string()),
            version: Some("1.0.0".to_string()),
            ..Default::default()
        })
        .await?;
    println!("{:?}", matches);

    // Upload a parcel using a stream instead of loading into memory
    println!("Creating parcel 2");
    bindle_client
        .create_parcel_from_file(
            "enterprise.com/warpcore/2.0.0",
            &second_sha,
            root_path.join("tests/scaffolds/valid_v2/parcels/parcel.dat"),
        )
        .await?;

    // Get a parcel and load its bytes into memory
    println!("Loading parcel 1");
    let data = bindle_client
        .get_parcel("enterprise.com/warpcore/2.0.0", &first_sha)
        .await?;
    println!("{}", data.len());

    // Get a parcel as a stream, and write it into a file somewhere
    println!("Loading parcel 2 as stream");
    let temp = tempdir()?;
    let mut stream = bindle_client
        .get_parcel_stream("enterprise.com/warpcore/2.0.0", &second_sha)
        .await?;

    let file_path = temp.path().join("foo");
    let mut file = tokio::fs::OpenOptions::new()
        .write(true)
        .create(true)
        .open(&file_path)
        .await?;

    while let Some(data) = stream.next().await {
        let data = data?;
        file.write_all(&data).await?;
    }
    file.flush().await?;

    // Read the whole file and make sure we got it
    assert_eq!(tokio::fs::read(file_path).await?, b"a green one");

    Ok(())
}