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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use penumbra_asset::Value;
use penumbra_keys::{keys::AddressIndex, Address, AddressView};
use penumbra_proto::{view::v1 as pb, DomainType};
use penumbra_sct::{CommitmentSource, Nullifier};
use penumbra_shielded_pool::{note, Note, Rseed};
use penumbra_tct::Position;

use r2d2_sqlite::rusqlite::Row;
use serde::{Deserialize, Serialize};

/// Corresponds to the SpendableNoteRecord proto
#[derive(Serialize, Deserialize, Debug, Clone)]
#[serde(try_from = "pb::SpendableNoteRecord", into = "pb::SpendableNoteRecord")]
pub struct SpendableNoteRecord {
    pub note_commitment: note::StateCommitment,
    pub note: Note,
    pub address_index: AddressIndex,
    pub nullifier: Nullifier,
    pub height_created: u64,
    pub height_spent: Option<u64>,
    pub position: Position,
    pub source: CommitmentSource,
    pub return_address: Option<AddressView>,
}

impl DomainType for SpendableNoteRecord {
    type Proto = pb::SpendableNoteRecord;
}

impl From<SpendableNoteRecord> for pb::SpendableNoteRecord {
    fn from(v: SpendableNoteRecord) -> Self {
        pb::SpendableNoteRecord {
            note_commitment: Some(v.note_commitment.into()),
            note: Some(v.note.into()),
            address_index: Some(v.address_index.into()),
            nullifier: Some(v.nullifier.into()),
            height_created: v.height_created,
            height_spent: v.height_spent.unwrap_or(0),
            position: v.position.into(),
            source: Some(v.source.into()),
            return_address: v.return_address.map(Into::into),
        }
    }
}

impl TryFrom<pb::SpendableNoteRecord> for SpendableNoteRecord {
    type Error = anyhow::Error;
    fn try_from(v: pb::SpendableNoteRecord) -> Result<Self, Self::Error> {
        Ok(SpendableNoteRecord {
            note_commitment: v
                .note_commitment
                .ok_or_else(|| anyhow::anyhow!("missing note commitment"))?
                .try_into()?,
            note: v
                .note
                .ok_or_else(|| anyhow::anyhow!("missing note"))?
                .try_into()?,
            address_index: v
                .address_index
                .ok_or_else(|| anyhow::anyhow!("missing address index"))?
                .try_into()?,
            nullifier: v
                .nullifier
                .ok_or_else(|| anyhow::anyhow!("missing nullifier"))?
                .try_into()?,
            height_created: v.height_created,
            height_spent: if v.height_spent > 0 {
                Some(v.height_spent)
            } else {
                None
            },
            position: v.position.into(),
            source: v
                .source
                .ok_or_else(|| anyhow::anyhow!("missing note source"))?
                .try_into()?,
            return_address: v.return_address.map(TryInto::try_into).transpose()?,
        })
    }
}

impl TryFrom<&Row<'_>> for SpendableNoteRecord {
    type Error = anyhow::Error;

    fn try_from(row: &Row<'_>) -> Result<Self, Self::Error> {
        let return_address_bytes = row
            .get::<_, Option<Vec<u8>>>("return_address")
            // If there's no return_address column, fill in None
            .ok()
            .flatten();
        let return_address = return_address_bytes
            .map(|b| {
                // Address is not proto-encoded
                Address::try_from(b)
            })
            .transpose()?
            .map(|a| AddressView::Opaque { address: a });
        Ok(SpendableNoteRecord {
            address_index: row.get::<_, Vec<u8>>("address_index")?[..].try_into()?,
            nullifier: row.get::<_, Vec<u8>>("nullifier")?[..].try_into()?,
            height_created: row.get("height_created")?,
            height_spent: row.get("height_spent")?,
            position: row.get::<_, u64>("position")?.into(),
            note_commitment: row.get::<_, Vec<u8>>("note_commitment")?[..].try_into()?,
            note: Note::from_parts(
                row.get::<_, Vec<u8>>("address")?[..].try_into()?,
                Value {
                    amount: u128::from_be_bytes(row.get::<_, [u8; 16]>("amount")?).into(),
                    asset_id: row.get::<_, Vec<u8>>("asset_id")?[..].try_into()?,
                },
                Rseed(row.get::<_, [u8; 32]>("rseed")?),
            )?,
            source: CommitmentSource::decode(&row.get::<_, Vec<u8>>("source")?[..])?,
            return_address,
        })
    }
}