// Copyright 2023 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

//! In this example we will build outputs with all possible unlock conditions.
//!
//! Rename `.env.example` to `.env` first, then run the command:
//! ```sh
//! cargo run --release --all-features --example output_unlock_conditions
//! ```

use iota_sdk::{
    client::{Client, Result},
    types::block::{
        address::Address,
        output::{
            dto::OutputDto,
            unlock_condition::{
                AddressUnlockCondition, ExpirationUnlockCondition, GovernorAddressUnlockCondition,
                ImmutableAliasAddressUnlockCondition, StateControllerAddressUnlockCondition,
                StorageDepositReturnUnlockCondition, TimelockUnlockCondition,
            },
            AliasId, AliasOutputBuilder, BasicOutputBuilder, FoundryOutputBuilder, SimpleTokenScheme, TokenScheme,
        },
    },
};

#[tokio::main]
async fn main() -> Result<()> {
    // This example uses secrets in environment variables for simplicity which should not be done in production.
    dotenvy::dotenv().ok();

    for var in ["NODE_URL"] {
        std::env::var(var).expect(&format!(".env variable '{var}' is undefined, see .env.example"));
    }

    let node_url = std::env::var("NODE_URL").unwrap();

    // Create a client instance.
    let client = Client::builder().with_node(&node_url)?.finish().await?;

    let token_supply = client.get_token_supply().await?;
    let rent_structure = client.get_rent_structure().await?;

    let address = Address::try_from_bech32("rms1qpllaj0pyveqfkwxmnngz2c488hfdtmfrj3wfkgxtk4gtyrax0jaxzt70zy")?;
    let alias_address = Address::try_from_bech32("rms1pr59qm43mjtvhcajfmupqf23x29llam88yecn6pyul80rx099krmv2fnnux")?;

    let token_scheme = TokenScheme::Simple(SimpleTokenScheme::new(50, 0, 100)?);

    let basic_output_builder = BasicOutputBuilder::new_with_minimum_storage_deposit(rent_structure)
        .add_unlock_condition(AddressUnlockCondition::new(address));
    let alias_output_builder = AliasOutputBuilder::new_with_minimum_storage_deposit(rent_structure, AliasId::null());
    let foundry_output_builder =
        FoundryOutputBuilder::new_with_minimum_storage_deposit(rent_structure, 1, token_scheme);

    let outputs = [
        //// most simple output
        basic_output_builder.clone().finish_output(token_supply)?,
        // with storage deposit return unlock condition
        basic_output_builder
            .clone()
            .add_unlock_condition(StorageDepositReturnUnlockCondition::new(
                address,
                1000000,
                token_supply,
            )?)
            .finish_output(token_supply)?,
        // with timeout unlock condition
        basic_output_builder
            .clone()
            .add_unlock_condition(TimelockUnlockCondition::new(1)?)
            .finish_output(token_supply)?,
        // with expiration unlock condition
        basic_output_builder
            .add_unlock_condition(ExpirationUnlockCondition::new(address, 1)?)
            .finish_output(token_supply)?,
        // with governor and state controller unlock condition
        alias_output_builder
            .add_unlock_condition(GovernorAddressUnlockCondition::new(address))
            .add_unlock_condition(StateControllerAddressUnlockCondition::new(address))
            .finish_output(token_supply)?,
        // with immutable alias unlock condition
        foundry_output_builder
            .add_unlock_condition(ImmutableAliasAddressUnlockCondition::new(*alias_address.as_alias()))
            .finish_output(token_supply)?,
    ];

    // Convert output array to json array
    let json_outputs = serde_json::to_string_pretty(&outputs.iter().map(OutputDto::from).collect::<Vec<OutputDto>>())?;
    println!("{json_outputs}");

    Ok(())
}