1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use std::convert::{TryFrom, TryInto};

use super::{BootstrapWitness, Vkeywitness};
use crate::byron::AddressContent;

use cml_crypto::{
    chain_crypto::{self, derive::combine_pk_and_chaincode},
    CryptoError, PrivateKey, RawBytesEncoding, TransactionHash,
};

impl BootstrapWitness {
    // pub fn to_public_key(&self) -> Result<crypto::Bip32PublicKey, crypto::CryptoError> {
    //     crypto::chain_crypto::PublicKey::<chain_crypto::ed25519_derive::Ed25519Bip32>::try_from(self.clone())
    //         .map(crypto::Bip32PublicKey)
    //         .map_err(Into::into)
    // }

    pub fn to_address(&self) -> Result<AddressContent, CryptoError> {
        AddressContent::try_from(self.clone()).map_err(Into::into)
    }
}

impl TryInto<chain_crypto::PublicKey<chain_crypto::ed25519_derive::Ed25519Bip32>>
    for BootstrapWitness
{
    type Error = CryptoError;

    fn try_into(
        self,
    ) -> Result<chain_crypto::PublicKey<chain_crypto::ed25519_derive::Ed25519Bip32>, Self::Error>
    {
        combine_pk_and_chaincode(self.public_key.0, &self.chain_code).map_err(Into::into)
    }
}

impl TryFrom<BootstrapWitness> for AddressContent {
    type Error = CryptoError;

    fn try_from(wit: BootstrapWitness) -> Result<Self, Self::Error> {
        let protocol_magic = wit.attributes.protocol_magic;
        let key: chain_crypto::PublicKey<chain_crypto::ed25519_derive::Ed25519Bip32> =
            wit.try_into()?;
        let address_content =
            AddressContent::new_simple(cml_crypto::Bip32PublicKey::from(key), protocol_magic);
        Ok(address_content)
    }
}

pub fn make_vkey_witness(tx_body_hash: &TransactionHash, sk: &PrivateKey) -> Vkeywitness {
    let sig = sk.sign(tx_body_hash.to_raw_bytes());
    Vkeywitness::new(sk.to_public(), sig)
}