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
53
54
55
56
57
58
59
60
61
62
63
64
65
66

// Descriptor wallet library extending bitcoin & miniscript functionality
// by LNP/BP Association (https://lnp-bp.org)
// Written in 2020-2021 by
//     Dr. Maxim Orlovsky <orlovsky@pandoracore.com>
//
// 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 Apache-2.0 License
// along with this software.
// If not, see <https://opensource.org/licenses/Apache-2.0>.

use bitcoin::PublicKey;
use std::collections::BTreeMap;

use crate::psbt::{self, ProprietaryKey};
use crate::Slice32;

pub const PSBT_WALLET_PREFIX: &'static [u8] = b"descriptor-wallet";
pub const PSBT_WALLET_IN_TWEAK: u8 = 0;

pub trait ProprietaryWalletInput {
    fn p2c_tweak_add(&mut self, pubkey: PublicKey, tweak: Slice32);
    fn p2c_tweak(&self) -> BTreeMap<PublicKey, Slice32>;
}

impl ProprietaryWalletInput for psbt::Input {
    fn p2c_tweak_add(&mut self, pubkey: PublicKey, tweak: Slice32) {
        self.proprietary.insert(
            ProprietaryKey {
                prefix: PSBT_WALLET_PREFIX.to_vec(),
                subtype: PSBT_WALLET_IN_TWEAK,
                key: pubkey.to_bytes(),
            },
            tweak.to_vec(),
        );
    }

    fn p2c_tweak(&self) -> BTreeMap<PublicKey, Slice32> {
        self.proprietary
            .iter()
            .filter_map(
                |(
                    ProprietaryKey {
                        prefix,
                        subtype,
                        key,
                    },
                    value,
                )| {
                    if prefix.as_slice() == PSBT_WALLET_PREFIX
                        && *subtype == PSBT_WALLET_IN_TWEAK
                    {
                        PublicKey::from_slice(&key).ok().and_then(|pk| {
                            Slice32::from_slice(value).map(|tweak| (pk, tweak))
                        })
                    } else {
                        None
                    }
                },
            )
            .collect()
    }
}