diff --git a/crates/cdk-redb/src/mint/mod.rs b/crates/cdk-redb/src/mint/mod.rs deleted file mode 100644 index c4b32211..00000000 --- a/crates/cdk-redb/src/mint/mod.rs +++ /dev/null @@ -1,1103 +0,0 @@ -//! SQLite Storage for CDK - -use std::cmp::Ordering; -use std::collections::{HashMap, HashSet}; -use std::path::Path; -use std::str::FromStr; -use std::sync::Arc; - -use async_trait::async_trait; -use cdk_common::common::QuoteTTL; -use cdk_common::database::{ - self, MintDatabase, MintKeysDatabase, MintProofsDatabase, MintQuotesDatabase, - MintSignaturesDatabase, -}; -use cdk_common::dhke::hash_to_curve; -use cdk_common::mint::{self, MintKeySetInfo, MintQuote}; -use cdk_common::nut00::ProofsMethods; -use cdk_common::state::check_state_transition; -use cdk_common::util::unix_time; -use cdk_common::{ - BlindSignature, CurrencyUnit, Id, MeltQuoteState, MintInfo, MintQuoteState, Proof, Proofs, - PublicKey, State, -}; -use migrations::{migrate_01_to_02, migrate_04_to_05}; -use redb::{Database, MultimapTableDefinition, ReadableTable, TableDefinition}; -use uuid::Uuid; - -use super::error::Error; -use crate::migrations::migrate_00_to_01; -use crate::mint::migrations::{migrate_02_to_03, migrate_03_to_04}; - -#[cfg(feature = "auth")] -mod auth; -mod migrations; - -#[cfg(feature = "auth")] -pub use auth::MintRedbAuthDatabase; - -const ACTIVE_KEYSETS_TABLE: TableDefinition<&str, &str> = TableDefinition::new("active_keysets"); -const KEYSETS_TABLE: TableDefinition<&str, &str> = TableDefinition::new("keysets"); -const MINT_QUOTES_TABLE: TableDefinition<[u8; 16], &str> = TableDefinition::new("mint_quotes"); -const MELT_QUOTES_TABLE: TableDefinition<[u8; 16], &str> = TableDefinition::new("melt_quotes"); -const PROOFS_TABLE: TableDefinition<[u8; 33], &str> = TableDefinition::new("proofs"); -const PROOFS_STATE_TABLE: TableDefinition<[u8; 33], &str> = TableDefinition::new("proofs_state"); -const PROOF_CREATED_TIME: TableDefinition<[u8; 33], u64> = - TableDefinition::new("proof_created_time"); -const CONFIG_TABLE: TableDefinition<&str, &str> = TableDefinition::new("config"); -// Key is hex blinded_message B_ value is blinded_signature -const BLINDED_SIGNATURES: TableDefinition<[u8; 33], &str> = - TableDefinition::new("blinded_signatures"); -const BLIND_SIGNATURE_CREATED_TIME: TableDefinition<[u8; 33], u64> = - TableDefinition::new("blind_signature_created_time"); -const QUOTE_PROOFS_TABLE: MultimapTableDefinition<[u8; 16], [u8; 33]> = - MultimapTableDefinition::new("quote_proofs"); -const QUOTE_SIGNATURES_TABLE: MultimapTableDefinition<[u8; 16], [u8; 33]> = - MultimapTableDefinition::new("quote_signatures"); - -const DATABASE_VERSION: u32 = 5; - -/// Mint Redbdatabase -#[derive(Debug, Clone)] -pub struct MintRedbDatabase { - db: Arc, -} - -impl MintRedbDatabase { - /// Create new [`MintRedbDatabase`] - pub fn new(path: &Path) -> Result { - { - // Check database version - - let db = Arc::new(Database::create(path)?); - - // Check database version - let read_txn = db.begin_read()?; - let table = read_txn.open_table(CONFIG_TABLE); - - let db_version = match table { - Ok(table) => table.get("db_version")?.map(|v| v.value().to_owned()), - Err(_) => None, - }; - match db_version { - Some(db_version) => { - let mut current_file_version = u32::from_str(&db_version)?; - match current_file_version.cmp(&DATABASE_VERSION) { - Ordering::Less => { - tracing::info!( - "Database needs to be upgraded at {} current is {}", - current_file_version, - DATABASE_VERSION - ); - if current_file_version == 0 { - current_file_version = migrate_00_to_01(Arc::clone(&db))?; - } - - if current_file_version == 1 { - current_file_version = migrate_01_to_02(Arc::clone(&db))?; - } - - if current_file_version == 2 { - current_file_version = migrate_02_to_03(Arc::clone(&db))?; - } - - if current_file_version == 3 { - current_file_version = migrate_03_to_04(Arc::clone(&db))?; - } - - if current_file_version == 4 { - current_file_version = migrate_04_to_05(Arc::clone(&db))?; - } - - if current_file_version != DATABASE_VERSION { - tracing::warn!( - "Database upgrade did not complete at {} current is {}", - current_file_version, - DATABASE_VERSION - ); - return Err(Error::UnknownDatabaseVersion); - } - - let write_txn = db.begin_write()?; - { - let mut table = write_txn.open_table(CONFIG_TABLE)?; - - table - .insert("db_version", DATABASE_VERSION.to_string().as_str())?; - } - - write_txn.commit()?; - } - Ordering::Equal => { - tracing::info!("Database is at current version {}", DATABASE_VERSION); - } - Ordering::Greater => { - tracing::warn!( - "Database upgrade did not complete at {} current is {}", - current_file_version, - DATABASE_VERSION - ); - return Err(Error::UnknownDatabaseVersion); - } - } - } - None => { - let write_txn = db.begin_write()?; - { - // Open all tables to init a new db - let mut table = write_txn.open_table(CONFIG_TABLE)?; - let _ = write_txn.open_table(ACTIVE_KEYSETS_TABLE)?; - let _ = write_txn.open_table(KEYSETS_TABLE)?; - let _ = write_txn.open_table(MINT_QUOTES_TABLE)?; - let _ = write_txn.open_table(MELT_QUOTES_TABLE)?; - let _ = write_txn.open_table(PROOFS_TABLE)?; - let _ = write_txn.open_table(PROOFS_STATE_TABLE)?; - let _ = write_txn.open_table(PROOF_CREATED_TIME)?; - let _ = write_txn.open_table(BLINDED_SIGNATURES)?; - let _ = write_txn.open_table(BLIND_SIGNATURE_CREATED_TIME)?; - let _ = write_txn.open_multimap_table(QUOTE_PROOFS_TABLE)?; - let _ = write_txn.open_multimap_table(QUOTE_SIGNATURES_TABLE)?; - - table.insert("db_version", DATABASE_VERSION.to_string().as_str())?; - } - - write_txn.commit()?; - } - } - drop(db); - } - - let db = Database::create(path)?; - Ok(Self { db: Arc::new(db) }) - } -} - -#[async_trait] -impl MintKeysDatabase for MintRedbDatabase { - type Err = database::Error; - - async fn set_active_keyset(&self, unit: CurrencyUnit, id: Id) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(ACTIVE_KEYSETS_TABLE) - .map_err(Error::from)?; - table - .insert(unit.to_string().as_str(), id.to_string().as_str()) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_active_keyset_id(&self, unit: &CurrencyUnit) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(ACTIVE_KEYSETS_TABLE) - .map_err(Error::from)?; - - if let Some(id) = table.get(unit.to_string().as_str()).map_err(Error::from)? { - return Ok(Some(Id::from_str(id.value()).map_err(Error::from)?)); - } - - Ok(None) - } - - async fn get_active_keysets(&self) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(ACTIVE_KEYSETS_TABLE) - .map_err(Error::from)?; - - let mut active_keysets = HashMap::new(); - - for (unit, id) in (table.iter().map_err(Error::from)?).flatten() { - let unit = CurrencyUnit::from_str(unit.value())?; - let id = Id::from_str(id.value()).map_err(Error::from)?; - - active_keysets.insert(unit, id); - } - - Ok(active_keysets) - } - - async fn add_keyset_info(&self, keyset: MintKeySetInfo) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn.open_table(KEYSETS_TABLE).map_err(Error::from)?; - table - .insert( - keyset.id.to_string().as_str(), - serde_json::to_string(&keyset) - .map_err(Error::from)? - .as_str(), - ) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_keyset_info(&self, keyset_id: &Id) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(KEYSETS_TABLE).map_err(Error::from)?; - - match table - .get(keyset_id.to_string().as_str()) - .map_err(Error::from)? - { - Some(keyset) => Ok(serde_json::from_str(keyset.value()).map_err(Error::from)?), - None => Ok(None), - } - } - - async fn get_keyset_infos(&self) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(KEYSETS_TABLE).map_err(Error::from)?; - - let mut keysets = Vec::new(); - - for (_id, keyset) in (table.iter().map_err(Error::from)?).flatten() { - let keyset = serde_json::from_str(keyset.value()).map_err(Error::from)?; - - keysets.push(keyset) - } - - Ok(keysets) - } -} - -#[async_trait] -impl MintQuotesDatabase for MintRedbDatabase { - type Err = database::Error; - - async fn add_mint_quote(&self, quote: MintQuote) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - table - .insert( - quote.id.as_bytes(), - serde_json::to_string("e).map_err(Error::from)?.as_str(), - ) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_mint_quote(&self, quote_id: &Uuid) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - - match table.get(quote_id.as_bytes()).map_err(Error::from)? { - Some(quote) => Ok(serde_json::from_str(quote.value()).map_err(Error::from)?), - None => Ok(None), - } - } - - async fn update_mint_quote_state( - &self, - quote_id: &Uuid, - state: MintQuoteState, - ) -> Result { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - let current_state; - { - let mut mint_quote: MintQuote; - let mut table = write_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - { - let quote_guard = table - .get(quote_id.as_bytes()) - .map_err(Error::from)? - .ok_or(Error::UnknownQuote)?; - - let quote = quote_guard.value(); - - mint_quote = serde_json::from_str(quote).map_err(Error::from)?; - } - - current_state = mint_quote.state; - mint_quote.state = state; - - { - table - .insert( - quote_id.as_bytes(), - serde_json::to_string(&mint_quote) - .map_err(Error::from)? - .as_str(), - ) - .map_err(Error::from)?; - } - } - write_txn.commit().map_err(Error::from)?; - - Ok(current_state) - } - async fn get_mint_quote_by_request( - &self, - request: &str, - ) -> Result, Self::Err> { - let quotes = self.get_mint_quotes().await?; - - let quote = quotes - .into_iter() - .filter(|q| q.request.eq(request)) - .collect::>() - .first() - .cloned(); - - Ok(quote) - } - - async fn get_mint_quote_by_request_lookup_id( - &self, - request_lookup_id: &str, - ) -> Result, Self::Err> { - let quotes = self.get_mint_quotes().await?; - - let quote = quotes - .into_iter() - .filter(|q| q.request_lookup_id.eq(request_lookup_id)) - .collect::>() - .first() - .cloned(); - - Ok(quote) - } - - async fn get_mint_quotes(&self) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - - let mut quotes = Vec::new(); - - for (_id, quote) in (table.iter().map_err(Error::from)?).flatten() { - let quote = serde_json::from_str(quote.value()).map_err(Error::from)?; - - quotes.push(quote) - } - - Ok(quotes) - } - - async fn get_mint_quotes_with_state( - &self, - state: MintQuoteState, - ) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - - let mut quotes = Vec::new(); - - for (_id, quote) in (table.iter().map_err(Error::from)?).flatten() { - let quote: MintQuote = serde_json::from_str(quote.value()).map_err(Error::from)?; - - if quote.state == state { - quotes.push(quote) - } - } - - Ok(quotes) - } - - async fn remove_mint_quote(&self, quote_id: &Uuid) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(MINT_QUOTES_TABLE) - .map_err(Error::from)?; - table.remove(quote_id.as_bytes()).map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn add_melt_quote(&self, quote: mint::MeltQuote) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(MELT_QUOTES_TABLE) - .map_err(Error::from)?; - table - .insert( - quote.id.as_bytes(), - serde_json::to_string("e).map_err(Error::from)?.as_str(), - ) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_melt_quote(&self, quote_id: &Uuid) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(MELT_QUOTES_TABLE) - .map_err(Error::from)?; - - let quote = table.get(quote_id.as_bytes()).map_err(Error::from)?; - - Ok(quote.map(|q| serde_json::from_str(q.value()).unwrap())) - } - - async fn update_melt_quote_state( - &self, - quote_id: &Uuid, - state: MeltQuoteState, - ) -> Result<(MeltQuoteState, mint::MeltQuote), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - let current_state; - let mut melt_quote: mint::MeltQuote; - - { - let mut table = write_txn - .open_table(MELT_QUOTES_TABLE) - .map_err(Error::from)?; - { - let quote_guard = table - .get(quote_id.as_bytes()) - .map_err(Error::from)? - .ok_or(Error::UnknownQuote)?; - - let quote = quote_guard.value(); - - melt_quote = serde_json::from_str(quote).map_err(Error::from)?; - } - - current_state = melt_quote.state; - melt_quote.state = state; - - { - table - .insert( - quote_id.as_bytes(), - serde_json::to_string(&melt_quote) - .map_err(Error::from)? - .as_str(), - ) - .map_err(Error::from)?; - } - } - write_txn.commit().map_err(Error::from)?; - - Ok((current_state, melt_quote)) - } - - async fn get_melt_quotes(&self) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(MELT_QUOTES_TABLE) - .map_err(Error::from)?; - - let mut quotes = Vec::new(); - - for (_id, quote) in (table.iter().map_err(Error::from)?).flatten() { - let quote = serde_json::from_str(quote.value()).map_err(Error::from)?; - - quotes.push(quote) - } - - Ok(quotes) - } - - async fn remove_melt_quote(&self, quote_id: &Uuid) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(MELT_QUOTES_TABLE) - .map_err(Error::from)?; - table.remove(quote_id.as_bytes()).map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } -} - -#[async_trait] -impl MintProofsDatabase for MintRedbDatabase { - type Err = database::Error; - - async fn add_proofs(&self, proofs: Proofs, quote_id: Option) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn.open_table(PROOFS_TABLE).map_err(Error::from)?; - let mut time_table = write_txn - .open_table(PROOF_CREATED_TIME) - .map_err(Error::from)?; - let mut quote_proofs_table = write_txn - .open_multimap_table(QUOTE_PROOFS_TABLE) - .map_err(Error::from)?; - - // Get current timestamp in seconds - let current_time = unix_time(); - - for proof in proofs { - let y: PublicKey = hash_to_curve(&proof.secret.to_bytes()).map_err(Error::from)?; - let y = y.to_bytes(); - if table.get(y).map_err(Error::from)?.is_none() { - table - .insert( - y, - serde_json::to_string(&proof).map_err(Error::from)?.as_str(), - ) - .map_err(Error::from)?; - - // Store creation time - time_table.insert(y, current_time).map_err(Error::from)?; - } - - if let Some(quote_id) = "e_id { - quote_proofs_table - .insert(quote_id.as_bytes(), y) - .map_err(Error::from)?; - } - } - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn remove_proofs( - &self, - ys: &[PublicKey], - quote_id: Option, - ) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - let mut states: HashSet = HashSet::new(); - - { - let mut proof_state_table = write_txn - .open_table(PROOFS_STATE_TABLE) - .map_err(Error::from)?; - for y in ys { - let state = proof_state_table - .remove(&y.to_bytes()) - .map_err(Error::from)?; - - if let Some(state) = state { - let state: State = serde_json::from_str(state.value()).map_err(Error::from)?; - - states.insert(state); - } - } - } - - if states.contains(&State::Spent) { - tracing::warn!("Db attempted to remove spent proof"); - write_txn.abort().map_err(Error::from)?; - return Err(Self::Err::AttemptRemoveSpentProof); - } - - { - let mut proofs_table = write_txn.open_table(PROOFS_TABLE).map_err(Error::from)?; - let mut time_table = write_txn - .open_table(PROOF_CREATED_TIME) - .map_err(Error::from)?; - - for y in ys { - proofs_table.remove(&y.to_bytes()).map_err(Error::from)?; - time_table.remove(&y.to_bytes()).map_err(Error::from)?; - } - } - - if let Some(quote_id) = quote_id { - let mut quote_proofs_table = write_txn - .open_multimap_table(QUOTE_PROOFS_TABLE) - .map_err(Error::from)?; - - quote_proofs_table - .remove_all(quote_id.as_bytes()) - .map_err(Error::from)?; - } - - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_proofs_by_ys(&self, ys: &[PublicKey]) -> Result>, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(PROOFS_TABLE).map_err(Error::from)?; - - let mut proofs = Vec::with_capacity(ys.len()); - - for y in ys { - match table.get(y.to_bytes()).map_err(Error::from)? { - Some(proof) => proofs.push(Some( - serde_json::from_str(proof.value()).map_err(Error::from)?, - )), - None => proofs.push(None), - } - } - - Ok(proofs) - } - - async fn get_proof_ys_by_quote_id(&self, quote_id: &Uuid) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_multimap_table(QUOTE_PROOFS_TABLE) - .map_err(Error::from)?; - - let ys = table.get(quote_id.as_bytes()).map_err(Error::from)?; - - let proof_ys = ys.fold(Vec::new(), |mut acc, y| { - if let Ok(y) = y { - if let Ok(pubkey) = PublicKey::from_slice(&y.value()) { - acc.push(pubkey); - } - } - acc - }); - - Ok(proof_ys) - } - - async fn get_proofs_states(&self, ys: &[PublicKey]) -> Result>, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(PROOFS_STATE_TABLE) - .map_err(Error::from)?; - - let mut states = Vec::with_capacity(ys.len()); - - for y in ys { - match table.get(y.to_bytes()).map_err(Error::from)? { - Some(state) => states.push(Some( - serde_json::from_str(state.value()).map_err(Error::from)?, - )), - None => states.push(None), - } - } - - Ok(states) - } - - async fn get_proofs_by_keyset_id( - &self, - keyset_id: &Id, - ) -> Result<(Proofs, Vec>), Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(PROOFS_TABLE).map_err(Error::from)?; - - let proofs_for_id = table - .iter() - .map_err(Error::from)? - .flatten() - .map(|(_, p)| serde_json::from_str::(p.value())) - .collect::>()? - .into_iter() - .filter(|p| &p.keyset_id == keyset_id) - .collect::(); - - let proof_ys = proofs_for_id.ys()?; - - assert_eq!(proofs_for_id.len(), proof_ys.len()); - - let states = self.get_proofs_states(&proof_ys).await?; - - Ok((proofs_for_id, states)) - } - - async fn update_proofs_states( - &self, - ys: &[PublicKey], - proofs_state: State, - ) -> Result>, Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - let mut states = Vec::with_capacity(ys.len()); - { - let table = write_txn - .open_table(PROOFS_STATE_TABLE) - .map_err(Error::from)?; - { - // First collect current states - for y in ys { - let current_state = match table.get(y.to_bytes()).map_err(Error::from)? { - Some(state) => { - let current_state = - serde_json::from_str(state.value()).map_err(Error::from)?; - check_state_transition(current_state, proofs_state)?; - Some(current_state) - } - None => None, - }; - - states.push(current_state); - } - } - } - - { - let mut table = write_txn - .open_table(PROOFS_STATE_TABLE) - .map_err(Error::from)?; - { - // If no proofs are spent, proceed with update - let state_str = serde_json::to_string(&proofs_state).map_err(Error::from)?; - for y in ys { - table - .insert(y.to_bytes(), state_str.as_str()) - .map_err(Error::from)?; - } - } - } - write_txn.commit().map_err(Error::from)?; - - Ok(states) - } -} - -#[async_trait] -impl MintSignaturesDatabase for MintRedbDatabase { - type Err = database::Error; - - async fn add_blind_signatures( - &self, - blinded_messages: &[PublicKey], - blind_signatures: &[BlindSignature], - quote_id: Option, - ) -> Result<(), Self::Err> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn - .open_table(BLINDED_SIGNATURES) - .map_err(Error::from)?; - let mut time_table = write_txn - .open_table(BLIND_SIGNATURE_CREATED_TIME) - .map_err(Error::from)?; - let mut quote_sigs_table = write_txn - .open_multimap_table(QUOTE_SIGNATURES_TABLE) - .map_err(Error::from)?; - - // Get current timestamp in seconds - let current_time = unix_time(); - - for (blinded_message, blind_signature) in blinded_messages.iter().zip(blind_signatures) - { - let blind_sig = serde_json::to_string(&blind_signature).map_err(Error::from)?; - table - .insert(blinded_message.to_bytes(), blind_sig.as_str()) - .map_err(Error::from)?; - - // Store creation time - time_table - .insert(blinded_message.to_bytes(), current_time) - .map_err(Error::from)?; - - if let Some(quote_id) = "e_id { - quote_sigs_table - .insert(quote_id.as_bytes(), blinded_message.to_bytes()) - .map_err(Error::from)?; - } - } - } - - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - - async fn get_blind_signatures( - &self, - blinded_messages: &[PublicKey], - ) -> Result>, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(BLINDED_SIGNATURES) - .map_err(Error::from)?; - - let mut signatures = Vec::with_capacity(blinded_messages.len()); - - for blinded_message in blinded_messages { - match table.get(blinded_message.to_bytes()).map_err(Error::from)? { - Some(blind_signature) => signatures.push(Some( - serde_json::from_str(blind_signature.value()).map_err(Error::from)?, - )), - None => signatures.push(None), - } - } - - Ok(signatures) - } - - async fn get_blind_signatures_for_keyset( - &self, - keyset_id: &Id, - ) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn - .open_table(BLINDED_SIGNATURES) - .map_err(Error::from)?; - - Ok(table - .iter() - .map_err(Error::from)? - .flatten() - .filter_map(|(_m, s)| { - match serde_json::from_str::(s.value()).ok() { - Some(signature) if &signature.keyset_id == keyset_id => Some(signature), // Filter by keyset_id - _ => None, // Exclude non-matching entries - } - }) - .collect()) - } - - /// Get [`BlindSignature`]s for quote - async fn get_blind_signatures_for_quote( - &self, - quote_id: &Uuid, - ) -> Result, Self::Err> { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let quote_proofs_table = read_txn - .open_multimap_table(QUOTE_SIGNATURES_TABLE) - .map_err(Error::from)?; - - let ys = quote_proofs_table.get(quote_id.as_bytes()).unwrap(); - - let ys: Vec<[u8; 33]> = ys.into_iter().flatten().map(|v| v.value()).collect(); - - let mut signatures = Vec::new(); - - let signatures_table = read_txn - .open_table(BLINDED_SIGNATURES) - .map_err(Error::from)?; - - for y in ys { - if let Some(sig) = signatures_table.get(y).map_err(Error::from)? { - let sig = serde_json::from_str(sig.value())?; - signatures.push(sig); - } - } - - Ok(signatures) - } -} - -#[async_trait] -impl MintDatabase for MintRedbDatabase { - async fn set_mint_info(&self, mint_info: MintInfo) -> Result<(), database::Error> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn.open_table(CONFIG_TABLE).map_err(Error::from)?; - table - .insert("mint_info", serde_json::to_string(&mint_info)?.as_str()) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - async fn get_mint_info(&self) -> Result { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(CONFIG_TABLE).map_err(Error::from)?; - - if let Some(mint_info) = table.get("mint_info").map_err(Error::from)? { - let mint_info = serde_json::from_str(mint_info.value())?; - - return Ok(mint_info); - } - - Err(Error::UnknownMintInfo.into()) - } - - async fn set_quote_ttl(&self, quote_ttl: QuoteTTL) -> Result<(), database::Error> { - let write_txn = self.db.begin_write().map_err(Error::from)?; - - { - let mut table = write_txn.open_table(CONFIG_TABLE).map_err(Error::from)?; - table - .insert("quote_ttl", serde_json::to_string("e_ttl)?.as_str()) - .map_err(Error::from)?; - } - write_txn.commit().map_err(Error::from)?; - - Ok(()) - } - async fn get_quote_ttl(&self) -> Result { - let read_txn = self.db.begin_read().map_err(Error::from)?; - let table = read_txn.open_table(CONFIG_TABLE).map_err(Error::from)?; - - if let Some(quote_ttl) = table.get("quote_ttl").map_err(Error::from)? { - let quote_ttl = serde_json::from_str(quote_ttl.value())?; - - return Ok(quote_ttl); - } - - Err(Error::UnknownQuoteTTL.into()) - } -} - -#[cfg(test)] -mod tests { - use cdk_common::secret::Secret; - use cdk_common::{mint_db_test, Amount, SecretKey}; - use tempfile::tempdir; - - use super::*; - - #[tokio::test] - async fn test_remove_spent_proofs() { - let tmp_dir = tempdir().unwrap(); - - let db = MintRedbDatabase::new(&tmp_dir.path().join("mint.redb")).unwrap(); - // Create some test proofs - let keyset_id = Id::from_str("00916bbf7ef91a36").unwrap(); - - let proofs = vec![ - Proof { - amount: Amount::from(100), - keyset_id, - secret: Secret::generate(), - c: SecretKey::generate().public_key(), - witness: None, - dleq: None, - }, - Proof { - amount: Amount::from(200), - keyset_id, - secret: Secret::generate(), - c: SecretKey::generate().public_key(), - witness: None, - dleq: None, - }, - ]; - - // Add proofs to database - db.add_proofs(proofs.clone(), None).await.unwrap(); - - // Mark one proof as spent - db.update_proofs_states(&[proofs[0].y().unwrap()], State::Spent) - .await - .unwrap(); - - db.update_proofs_states(&[proofs[1].y().unwrap()], State::Unspent) - .await - .unwrap(); - - // Try to remove both proofs - should fail because one is spent - let result = db - .remove_proofs(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()], None) - .await; - - assert!(result.is_err()); - assert!(matches!( - result.unwrap_err(), - database::Error::AttemptRemoveSpentProof - )); - - // Verify both proofs still exist - let states = db - .get_proofs_states(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()]) - .await - .unwrap(); - - assert_eq!(states.len(), 2); - assert_eq!(states[0], Some(State::Spent)); - assert_eq!(states[1], Some(State::Unspent)); - } - - #[tokio::test] - async fn test_update_spent_proofs() { - let tmp_dir = tempdir().unwrap(); - - let db = MintRedbDatabase::new(&tmp_dir.path().join("mint.redb")).unwrap(); - // Create some test proofs - let keyset_id = Id::from_str("00916bbf7ef91a36").unwrap(); - - let proofs = vec![ - Proof { - amount: Amount::from(100), - keyset_id, - secret: Secret::generate(), - c: SecretKey::generate().public_key(), - witness: None, - dleq: None, - }, - Proof { - amount: Amount::from(200), - keyset_id, - secret: Secret::generate(), - c: SecretKey::generate().public_key(), - witness: None, - dleq: None, - }, - ]; - - // Add proofs to database - db.add_proofs(proofs.clone(), None).await.unwrap(); - - // Mark one proof as spent - db.update_proofs_states(&[proofs[0].y().unwrap()], State::Spent) - .await - .unwrap(); - - db.update_proofs_states(&[proofs[1].y().unwrap()], State::Unspent) - .await - .unwrap(); - - // Mark one proof as spent - let result = db - .update_proofs_states( - &[proofs[0].y().unwrap(), proofs[1].y().unwrap()], - State::Unspent, - ) - .await; - - assert!(result.is_err()); - assert!(matches!( - result.unwrap_err(), - database::Error::AttemptUpdateSpentProof - )); - - // Verify both proofs still exist - let states = db - .get_proofs_states(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()]) - .await - .unwrap(); - - assert_eq!(states.len(), 2); - assert_eq!(states[0], Some(State::Spent)); - assert_eq!(states[1], Some(State::Unspent)); - } - - async fn provide_db() -> MintRedbDatabase { - let tmp_dir = tempdir().unwrap(); - - MintRedbDatabase::new(&tmp_dir.path().join("mint.redb")).unwrap() - } - - mint_db_test!(provide_db); -}