// Miniscript
// Written in 2019 by
//    Thomas Eizinger <thomas@coblox.tech>
//
// To the extent possible under law, the author(s) have dedicated all
// copyright and related and neighboring rights to this software to
// the public domain worldwide. This software is distributed without
// any warranty.
//
// You should have received a copy of the CC0 Public Domain Dedication
// along with this software.
// If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
//

//! Example: Create an HTLC with miniscript using the policy compiler

extern crate elements_miniscript as miniscript;

use std::str::FromStr;

use crate::miniscript::descriptor::Wsh;
use crate::miniscript::policy::{Concrete, Liftable};

fn main() {
    // HTLC policy with 10:1 odds for happy (co-operative) case compared to uncooperative case.
    let htlc_policy = Concrete::<bitcoin::PublicKey>::from_str(&format!("or(10@and(sha256({secret_hash}),pk({redeem_identity})),1@and(older({expiry}),pk({refund_identity})))",
                                                  secret_hash = "1111111111111111111111111111111111111111111111111111111111111111",
                                                  redeem_identity = "022222222222222222222222222222222222222222222222222222222222222222",
                                                  refund_identity = "020202020202020202020202020202020202020202020202020202020202020202",
                                                  expiry = "4444"
    )).unwrap();

    let htlc_descriptor = Wsh::new(
        htlc_policy
            .compile()
            .expect("Policy compilation only fails on resource limits or mixed timelocks"),
    )
    .expect("Resource limits");

    // Check whether the descriptor is safe. This checks whether all spend paths are accessible in
    // the Bitcoin network. It may be possible that some of the spend paths require more than 100
    // elements in Wsh scripts or they contain a combination of timelock and heightlock.
    assert!(htlc_descriptor.sanity_check().is_ok());
    assert_eq!(
        format!("{}", htlc_descriptor),
        "elwsh(andor(pk(022222222222222222222222222222222222222222222222222222222222222222),sha256(1111111111111111111111111111111111111111111111111111111111111111),and_v(v:pkh(020202020202020202020202020202020202020202020202020202020202020202),older(4444))))#mm63458x"
    );

    // Lift the descriptor into an abstract policy.
    assert_eq!(
        format!("{}", htlc_descriptor.lift().unwrap()),
        "or(and(pk(022222222222222222222222222222222222222222222222222222222222222222),sha256(1111111111111111111111111111111111111111111111111111111111111111)),and(pk(020202020202020202020202020202020202020202020202020202020202020202),older(4444)))"
    );

    // Get the scriptPpubkey for this Wsh descriptor.
    assert_eq!(
        format!("{:x}", htlc_descriptor.script_pubkey()),
        "0020d853877af928a8d2a569c9c0ed14bd16f6a80ce9cccaf8a6150fd8f7f8867ae2"
    );

    // Encode the Wsh descriptor into a Bitcoin script.
    assert_eq!(
        format!("{:x}", htlc_descriptor.inner_script()),
        "21022222222222222222222222222222222222222222222222222222222222222222ac6476a91451814f108670aced2d77c1805ddd6634bc9d473188ad025c11b26782012088a82011111111111111111111111111111111111111111111111111111111111111118768"
    );

    // Get the address for this Wsh descriptor.v
    assert_eq!(
        format!(
            "{}",
            htlc_descriptor.address(None, &elements::AddressParams::ELEMENTS)
        ),
        "ert1qmpfcw7he9z5d9ftfe8qw699azmm2sr8fen903fs4plv007yx0t3qdt0h29"
    );
}