//! SQL database implementation of the Mint //! //! This is a generic SQL implementation for the mint storage layer. Any database can be plugged in //! as long as standard ANSI SQL is used, as Postgres and SQLite would understand it. //! //! This implementation also has a rudimentary but standard migration and versioning system. //! //! The trait expects an asynchronous interaction, but it also provides tools to spawn blocking //! clients in a pool and expose them to an asynchronous environment, making them compatible with //! Mint. use std::collections::HashMap; use std::fmt::Debug; use std::str::FromStr; use std::sync::Arc; use async_trait::async_trait; use bitcoin::bip32::DerivationPath; use cdk_common::common::QuoteTTL; use cdk_common::database::{ self, ConversionError, Error, MintDatabase, MintDbWriterFinalizer, MintKeyDatabaseTransaction, MintKeysDatabase, MintProofsDatabase, MintQuotesDatabase, MintQuotesTransaction, MintSignatureTransaction, MintSignaturesDatabase, }; use cdk_common::mint::{ self, IncomingPayment, Issuance, MeltPaymentRequest, MeltQuote, MintKeySetInfo, MintQuote, }; use cdk_common::nut00::ProofsMethods; use cdk_common::payment::PaymentIdentifier; use cdk_common::secret::Secret; use cdk_common::state::check_state_transition; use cdk_common::util::unix_time; use cdk_common::{ Amount, BlindSignature, BlindSignatureDleq, CurrencyUnit, Id, MeltQuoteState, MintInfo, PaymentMethod, Proof, Proofs, PublicKey, SecretKey, State, }; use lightning_invoice::Bolt11Invoice; use migrations::MIGRATIONS; use tracing::instrument; use uuid::Uuid; use crate::common::migrate; use crate::database::{ConnectionWithTransaction, DatabaseExecutor}; use crate::pool::{DatabasePool, Pool, PooledResource}; use crate::stmt::{query, Column}; use crate::{ column_as_nullable_number, column_as_nullable_string, column_as_number, column_as_string, unpack_into, }; #[cfg(feature = "auth")] mod auth; #[rustfmt::skip] mod migrations; #[cfg(feature = "auth")] pub use auth::SQLMintAuthDatabase; /// Mint SQL Database #[derive(Debug, Clone)] pub struct SQLMintDatabase where RM: DatabasePool + 'static, { pool: Arc>, } /// SQL Transaction Writer pub struct SQLTransaction where RM: DatabasePool + 'static, { inner: ConnectionWithTransaction>, } #[inline(always)] async fn get_current_states( conn: &C, ys: &[PublicKey], ) -> Result, Error> where C: DatabaseExecutor + Send + Sync, { if ys.is_empty() { return Ok(Default::default()); } query(r#"SELECT y, state FROM proof WHERE y IN (:ys)"#)? .bind_vec("ys", ys.iter().map(|y| y.to_bytes().to_vec()).collect()) .fetch_all(conn) .await? .into_iter() .map(|row| { Ok(( column_as_string!(&row[0], PublicKey::from_hex, PublicKey::from_slice), column_as_string!(&row[1], State::from_str), )) }) .collect::, _>>() } #[inline(always)] async fn set_to_config(conn: &C, id: &str, value: &V) -> Result<(), Error> where C: DatabaseExecutor + Send + Sync, V: ?Sized + serde::Serialize, { query( r#" INSERT INTO config (id, value) VALUES (:id, :value) ON CONFLICT(id) DO UPDATE SET value = excluded.value "#, )? .bind("id", id.to_owned()) .bind("value", serde_json::to_string(&value)?) .execute(conn) .await?; Ok(()) } impl SQLMintDatabase where RM: DatabasePool + 'static, { /// Creates a new instance pub async fn new(db: X) -> Result where X: Into, { let pool = Pool::new(db.into()); Self::migrate(pool.get().map_err(|e| Error::Database(Box::new(e)))?).await?; Ok(Self { pool }) } /// Migrate async fn migrate(conn: PooledResource) -> Result<(), Error> { let tx = ConnectionWithTransaction::new(conn).await?; migrate(&tx, RM::Connection::name(), MIGRATIONS).await?; tx.commit().await?; Ok(()) } #[inline(always)] async fn fetch_from_config(&self, id: &str) -> Result where R: serde::de::DeserializeOwned, { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let value = column_as_string!(query(r#"SELECT value FROM config WHERE id = :id LIMIT 1"#)? .bind("id", id.to_owned()) .pluck(&*conn) .await? .ok_or(Error::UnknownQuoteTTL)?); Ok(serde_json::from_str(&value)?) } } #[async_trait] impl database::MintProofsTransaction<'_> for SQLTransaction where RM: DatabasePool + 'static, { type Err = Error; async fn add_proofs( &mut self, proofs: Proofs, quote_id: Option, ) -> Result<(), Self::Err> { let current_time = unix_time(); // Check any previous proof, this query should return None in order to proceed storing // Any result here would error match query(r#"SELECT state FROM proof WHERE y IN (:ys) LIMIT 1 FOR UPDATE"#)? .bind_vec( "ys", proofs .iter() .map(|y| y.y().map(|y| y.to_bytes().to_vec())) .collect::>()?, ) .pluck(&self.inner) .await? .map(|state| Ok::<_, Error>(column_as_string!(&state, State::from_str))) .transpose()? { Some(State::Spent) => Err(database::Error::AttemptUpdateSpentProof), Some(_) => Err(database::Error::Duplicate), None => Ok(()), // no previous record }?; for proof in proofs { query( r#" INSERT INTO proof (y, amount, keyset_id, secret, c, witness, state, quote_id, created_time) VALUES (:y, :amount, :keyset_id, :secret, :c, :witness, :state, :quote_id, :created_time) "#, )? .bind("y", proof.y()?.to_bytes().to_vec()) .bind("amount", proof.amount.to_i64()) .bind("keyset_id", proof.keyset_id.to_string()) .bind("secret", proof.secret.to_string()) .bind("c", proof.c.to_bytes().to_vec()) .bind( "witness", proof.witness.map(|w| serde_json::to_string(&w).unwrap()), ) .bind("state", "UNSPENT".to_string()) .bind("quote_id", quote_id.map(|q| q.hyphenated().to_string())) .bind("created_time", current_time as i64) .execute(&self.inner) .await?; } Ok(()) } async fn update_proofs_states( &mut self, ys: &[PublicKey], new_state: State, ) -> Result>, Self::Err> { let mut current_states = get_current_states(&self.inner, ys).await?; if current_states.len() != ys.len() { tracing::warn!( "Attempted to update state of non-existent proof {} {}", current_states.len(), ys.len() ); return Err(database::Error::ProofNotFound); } for state in current_states.values() { check_state_transition(*state, new_state)?; } query(r#"UPDATE proof SET state = :new_state WHERE y IN (:ys)"#)? .bind("new_state", new_state.to_string()) .bind_vec("ys", ys.iter().map(|y| y.to_bytes().to_vec()).collect()) .execute(&self.inner) .await?; Ok(ys.iter().map(|y| current_states.remove(y)).collect()) } async fn remove_proofs( &mut self, ys: &[PublicKey], _quote_id: Option, ) -> Result<(), Self::Err> { let total_deleted = query( r#" DELETE FROM proof WHERE y IN (:ys) AND state NOT IN (:exclude_state) "#, )? .bind_vec("ys", ys.iter().map(|y| y.to_bytes().to_vec()).collect()) .bind_vec("exclude_state", vec![State::Spent.to_string()]) .execute(&self.inner) .await?; if total_deleted != ys.len() { return Err(Self::Err::AttemptRemoveSpentProof); } Ok(()) } } #[async_trait] impl database::MintTransaction<'_, Error> for SQLTransaction where RM: DatabasePool + 'static, { async fn set_mint_info(&mut self, mint_info: MintInfo) -> Result<(), Error> { Ok(set_to_config(&self.inner, "mint_info", &mint_info).await?) } async fn set_quote_ttl(&mut self, quote_ttl: QuoteTTL) -> Result<(), Error> { Ok(set_to_config(&self.inner, "quote_ttl", "e_ttl).await?) } } #[async_trait] impl MintDbWriterFinalizer for SQLTransaction where RM: DatabasePool + 'static, { type Err = Error; async fn commit(self: Box) -> Result<(), Error> { self.inner.commit().await } async fn rollback(self: Box) -> Result<(), Error> { self.inner.rollback().await } } #[inline(always)] async fn get_mint_quote_payments( conn: &C, quote_id: &Uuid, ) -> Result, Error> where C: DatabaseExecutor + Send + Sync, { // Get payment IDs and timestamps from the mint_quote_payments table query( r#" SELECT payment_id, timestamp, amount FROM mint_quote_payments WHERE quote_id=:quote_id; "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .fetch_all(conn) .await? .into_iter() .map(|row| { let amount: u64 = column_as_number!(row[2].clone()); let time: u64 = column_as_number!(row[1].clone()); Ok(IncomingPayment::new( amount.into(), column_as_string!(&row[0]), time, )) }) .collect() } #[inline(always)] async fn get_mint_quote_issuance(conn: &C, quote_id: &Uuid) -> Result, Error> where C: DatabaseExecutor + Send + Sync, { // Get payment IDs and timestamps from the mint_quote_payments table query( r#" SELECT amount, timestamp FROM mint_quote_issued WHERE quote_id=:quote_id "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .fetch_all(conn) .await? .into_iter() .map(|row| { let time: u64 = column_as_number!(row[1].clone()); Ok(Issuance::new( Amount::from_i64(column_as_number!(row[0].clone())) .expect("Is amount when put into db"), time, )) }) .collect() } #[async_trait] impl MintKeyDatabaseTransaction<'_, Error> for SQLTransaction where RM: DatabasePool + 'static, { async fn add_keyset_info(&mut self, keyset: MintKeySetInfo) -> Result<(), Error> { query( r#" INSERT INTO keyset ( id, unit, active, valid_from, valid_to, derivation_path, max_order, input_fee_ppk, derivation_path_index ) VALUES ( :id, :unit, :active, :valid_from, :valid_to, :derivation_path, :max_order, :input_fee_ppk, :derivation_path_index ) ON CONFLICT(id) DO UPDATE SET unit = excluded.unit, active = excluded.active, valid_from = excluded.valid_from, valid_to = excluded.valid_to, derivation_path = excluded.derivation_path, max_order = excluded.max_order, input_fee_ppk = excluded.input_fee_ppk, derivation_path_index = excluded.derivation_path_index "#, )? .bind("id", keyset.id.to_string()) .bind("unit", keyset.unit.to_string()) .bind("active", keyset.active) .bind("valid_from", keyset.valid_from as i64) .bind("valid_to", keyset.final_expiry.map(|v| v as i64)) .bind("derivation_path", keyset.derivation_path.to_string()) .bind("max_order", keyset.max_order) .bind("input_fee_ppk", keyset.input_fee_ppk as i64) .bind("derivation_path_index", keyset.derivation_path_index) .execute(&self.inner) .await?; Ok(()) } async fn set_active_keyset(&mut self, unit: CurrencyUnit, id: Id) -> Result<(), Error> { query(r#"UPDATE keyset SET active=FALSE WHERE unit IS :unit"#)? .bind("unit", unit.to_string()) .execute(&self.inner) .await?; query(r#"UPDATE keyset SET active=TRUE WHERE unit IS :unit AND id IS :id"#)? .bind("unit", unit.to_string()) .bind("id", id.to_string()) .execute(&self.inner) .await?; Ok(()) } } #[async_trait] impl MintKeysDatabase for SQLMintDatabase where RM: DatabasePool + 'static, { type Err = Error; async fn begin_transaction<'a>( &'a self, ) -> Result + Send + Sync + 'a>, Error> { Ok(Box::new(SQLTransaction { inner: ConnectionWithTransaction::new( self.pool.get().map_err(|e| Error::Database(Box::new(e)))?, ) .await?, })) } async fn get_active_keyset_id(&self, unit: &CurrencyUnit) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok( query(r#" SELECT id FROM keyset WHERE active = 1 AND unit IS :unit"#)? .bind("unit", unit.to_string()) .pluck(&*conn) .await? .map(|id| match id { Column::Text(text) => Ok(Id::from_str(&text)?), Column::Blob(id) => Ok(Id::from_bytes(&id)?), _ => Err(Error::InvalidKeysetId), }) .transpose()?, ) } async fn get_active_keysets(&self) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query(r#"SELECT id, unit FROM keyset WHERE active = 1"#)? .fetch_all(&*conn) .await? .into_iter() .map(|row| { Ok(( column_as_string!(&row[1], CurrencyUnit::from_str), column_as_string!(&row[0], Id::from_str, Id::from_bytes), )) }) .collect::, Error>>()?) } async fn get_keyset_info(&self, id: &Id) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#"SELECT id, unit, active, valid_from, valid_to, derivation_path, derivation_path_index, max_order, input_fee_ppk FROM keyset WHERE id=:id"#, )? .bind("id", id.to_string()) .fetch_one(&*conn) .await? .map(sql_row_to_keyset_info) .transpose()?) } async fn get_keyset_infos(&self) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#"SELECT id, unit, active, valid_from, valid_to, derivation_path, derivation_path_index, max_order, input_fee_ppk FROM keyset "#, )? .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_keyset_info) .collect::, _>>()?) } } #[async_trait] impl MintQuotesTransaction<'_> for SQLTransaction where RM: DatabasePool + 'static, { type Err = Error; #[instrument(skip(self))] async fn increment_mint_quote_amount_paid( &mut self, quote_id: &Uuid, amount_paid: Amount, payment_id: String, ) -> Result { // Check if payment_id already exists in mint_quote_payments let exists = query( r#" SELECT payment_id FROM mint_quote_payments WHERE payment_id = :payment_id FOR UPDATE "#, )? .bind("payment_id", payment_id.clone()) .fetch_one(&self.inner) .await?; if exists.is_some() { tracing::error!("Payment ID already exists: {}", payment_id); return Err(database::Error::Duplicate); } // Get current amount_paid from quote let current_amount = query( r#" SELECT amount_paid FROM mint_quote WHERE id = :quote_id FOR UPDATE "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .fetch_one(&self.inner) .await .inspect_err(|err| { tracing::error!("SQLite could not get mint quote amount_paid: {}", err); })?; let current_amount_paid = if let Some(current_amount) = current_amount { let amount: u64 = column_as_number!(current_amount[0].clone()); Amount::from(amount) } else { Amount::ZERO }; // Calculate new amount_paid with overflow check let new_amount_paid = current_amount_paid .checked_add(amount_paid) .ok_or_else(|| database::Error::AmountOverflow)?; // Update the amount_paid query( r#" UPDATE mint_quote SET amount_paid = :amount_paid WHERE id = :quote_id "#, )? .bind("amount_paid", new_amount_paid.to_i64()) .bind("quote_id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await .inspect_err(|err| { tracing::error!("SQLite could not update mint quote amount_paid: {}", err); })?; // Add payment_id to mint_quote_payments table query( r#" INSERT INTO mint_quote_payments (quote_id, payment_id, amount, timestamp) VALUES (:quote_id, :payment_id, :amount, :timestamp) "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .bind("payment_id", payment_id) .bind("amount", amount_paid.to_i64()) .bind("timestamp", unix_time() as i64) .execute(&self.inner) .await .map_err(|err| { tracing::error!("SQLite could not insert payment ID: {}", err); err })?; Ok(new_amount_paid) } #[instrument(skip_all)] async fn increment_mint_quote_amount_issued( &mut self, quote_id: &Uuid, amount_issued: Amount, ) -> Result { // Get current amount_issued from quote let current_amount = query( r#" SELECT amount_issued FROM mint_quote WHERE id = :quote_id FOR UPDATE "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .fetch_one(&self.inner) .await .inspect_err(|err| { tracing::error!("SQLite could not get mint quote amount_issued: {}", err); })?; let current_amount_issued = if let Some(current_amount) = current_amount { let amount: u64 = column_as_number!(current_amount[0].clone()); Amount::from(amount) } else { Amount::ZERO }; // Calculate new amount_issued with overflow check let new_amount_issued = current_amount_issued .checked_add(amount_issued) .ok_or_else(|| database::Error::AmountOverflow)?; // Update the amount_issued query( r#" UPDATE mint_quote SET amount_issued = :amount_issued WHERE id = :quote_id FOR UPDATE "#, )? .bind("amount_issued", new_amount_issued.to_i64()) .bind("quote_id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await .inspect_err(|err| { tracing::error!("SQLite could not update mint quote amount_issued: {}", err); })?; let current_time = unix_time(); query( r#" INSERT INTO mint_quote_issued (quote_id, amount, timestamp) VALUES (:quote_id, :amount, :timestamp); "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .bind("amount", amount_issued.to_i64()) .bind("timestamp", current_time as i64) .execute(&self.inner) .await?; Ok(new_amount_issued) } #[instrument(skip_all)] async fn add_mint_quote(&mut self, quote: MintQuote) -> Result<(), Self::Err> { tracing::debug!("Adding quote with: {}", quote.payment_method.to_string()); println!("Adding quote with: {}", quote.payment_method.to_string()); query( r#" INSERT INTO mint_quote ( id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, payment_method, request_lookup_id_kind ) VALUES ( :id, :amount, :unit, :request, :expiry, :request_lookup_id, :pubkey, :created_time, :payment_method, :request_lookup_id_kind ) "#, )? .bind("id", quote.id.to_string()) .bind("amount", quote.amount.map(|a| a.to_i64())) .bind("unit", quote.unit.to_string()) .bind("request", quote.request) .bind("expiry", quote.expiry as i64) .bind( "request_lookup_id", quote.request_lookup_id.to_string(), ) .bind("pubkey", quote.pubkey.map(|p| p.to_string())) .bind("created_time", quote.created_time as i64) .bind("payment_method", quote.payment_method.to_string()) .bind("request_lookup_id_kind", quote.request_lookup_id.kind()) .execute(&self.inner) .await?; Ok(()) } async fn remove_mint_quote(&mut self, quote_id: &Uuid) -> Result<(), Self::Err> { query(r#"DELETE FROM mint_quote WHERE id=:id"#)? .bind("id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await?; Ok(()) } async fn add_melt_quote(&mut self, quote: mint::MeltQuote) -> Result<(), Self::Err> { // First try to find and replace any expired UNPAID quotes with the same request_lookup_id let current_time = unix_time(); let row_affected = query( r#" DELETE FROM melt_quote WHERE request_lookup_id = :request_lookup_id AND state = :state AND expiry < :current_time "#, )? .bind("request_lookup_id", quote.request_lookup_id.to_string()) .bind("state", MeltQuoteState::Unpaid.to_string()) .bind("current_time", current_time as i64) .execute(&self.inner) .await?; if row_affected > 0 { tracing::info!("Received new melt quote for existing invoice with expired quote."); } // Now insert the new quote query( r#" INSERT INTO melt_quote ( id, unit, amount, request, fee_reserve, state, expiry, payment_preimage, request_lookup_id, created_time, paid_time, options, request_lookup_id_kind, payment_method ) VALUES ( :id, :unit, :amount, :request, :fee_reserve, :state, :expiry, :payment_preimage, :request_lookup_id, :created_time, :paid_time, :options, :request_lookup_id_kind, :payment_method ) "#, )? .bind("id", quote.id.to_string()) .bind("unit", quote.unit.to_string()) .bind("amount", quote.amount.to_i64()) .bind("request", serde_json::to_string("e.request)?) .bind("fee_reserve", quote.fee_reserve.to_i64()) .bind("state", quote.state.to_string()) .bind("expiry", quote.expiry as i64) .bind("payment_preimage", quote.payment_preimage) .bind("request_lookup_id", quote.request_lookup_id.to_string()) .bind("created_time", quote.created_time as i64) .bind("paid_time", quote.paid_time.map(|t| t as i64)) .bind( "options", quote.options.map(|o| serde_json::to_string(&o).ok()), ) .bind("request_lookup_id_kind", quote.request_lookup_id.kind()) .bind("payment_method", quote.payment_method.to_string()) .execute(&self.inner) .await?; Ok(()) } async fn update_melt_quote_request_lookup_id( &mut self, quote_id: &Uuid, new_request_lookup_id: &PaymentIdentifier, ) -> Result<(), Self::Err> { query(r#"UPDATE melt_quote SET request_lookup_id = :new_req_id, request_lookup_id_kind = :new_kind WHERE id = :id"#)? .bind("new_req_id", new_request_lookup_id.to_string()) .bind("new_kind",new_request_lookup_id.kind() ) .bind("id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await?; Ok(()) } async fn update_melt_quote_state( &mut self, quote_id: &Uuid, state: MeltQuoteState, payment_proof: Option, ) -> Result<(MeltQuoteState, mint::MeltQuote), Self::Err> { let mut quote = query( r#" SELECT id, unit, amount, request, fee_reserve, expiry, state, payment_preimage, request_lookup_id, created_time, paid_time, payment_method, options, request_lookup_id_kind FROM melt_quote WHERE id=:id AND state != :state "#, )? .bind("id", quote_id.as_hyphenated().to_string()) .bind("state", state.to_string()) .fetch_one(&self.inner) .await? .map(sql_row_to_melt_quote) .transpose()? .ok_or(Error::QuoteNotFound)?; let rec = if state == MeltQuoteState::Paid { let current_time = unix_time(); query(r#"UPDATE melt_quote SET state = :state, paid_time = :paid_time, payment_preimage = :payment_preimage WHERE id = :id"#)? .bind("state", state.to_string()) .bind("paid_time", current_time as i64) .bind("payment_preimage", payment_proof) .bind("id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await } else { query(r#"UPDATE melt_quote SET state = :state WHERE id = :id"#)? .bind("state", state.to_string()) .bind("id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await }; match rec { Ok(_) => {} Err(err) => { tracing::error!("SQLite Could not update melt quote"); return Err(err); } }; let old_state = quote.state; quote.state = state; Ok((old_state, quote)) } async fn remove_melt_quote(&mut self, quote_id: &Uuid) -> Result<(), Self::Err> { query( r#" DELETE FROM melt_quote WHERE id=? "#, )? .bind("id", quote_id.as_hyphenated().to_string()) .execute(&self.inner) .await?; Ok(()) } async fn get_mint_quote(&mut self, quote_id: &Uuid) -> Result, Self::Err> { let payments = get_mint_quote_payments(&self.inner, quote_id).await?; let issuance = get_mint_quote_issuance(&self.inner, quote_id).await?; Ok(query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE id = :id FOR UPDATE "#, )? .bind("id", quote_id.as_hyphenated().to_string()) .fetch_one(&self.inner) .await? .map(|row| sql_row_to_mint_quote(row, payments, issuance)) .transpose()?) } async fn get_melt_quote( &mut self, quote_id: &Uuid, ) -> Result, Self::Err> { Ok(query( r#" SELECT id, unit, amount, request, fee_reserve, expiry, state, payment_preimage, request_lookup_id, created_time, paid_time, payment_method, options, request_lookup_id FROM melt_quote WHERE id=:id "#, )? .bind("id", quote_id.as_hyphenated().to_string()) .fetch_one(&self.inner) .await? .map(sql_row_to_melt_quote) .transpose()?) } async fn get_mint_quote_by_request( &mut self, request: &str, ) -> Result, Self::Err> { let mut mint_quote = query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE request = :request FOR UPDATE "#, )? .bind("request", request.to_string()) .fetch_one(&self.inner) .await? .map(|row| sql_row_to_mint_quote(row, vec![], vec![])) .transpose()?; if let Some(quote) = mint_quote.as_mut() { let payments = get_mint_quote_payments(&self.inner, "e.id).await?; let issuance = get_mint_quote_issuance(&self.inner, "e.id).await?; quote.issuance = issuance; quote.payments = payments; } Ok(mint_quote) } async fn get_mint_quote_by_request_lookup_id( &mut self, request_lookup_id: &PaymentIdentifier, ) -> Result, Self::Err> { let mut mint_quote = query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE request_lookup_id = :request_lookup_id AND request_lookup_id_kind = :request_lookup_id_kind FOR UPDATE "#, )? .bind("request_lookup_id", request_lookup_id.to_string()) .bind("request_lookup_id_kind", request_lookup_id.kind()) .fetch_one(&self.inner) .await? .map(|row| sql_row_to_mint_quote(row, vec![], vec![])) .transpose()?; if let Some(quote) = mint_quote.as_mut() { let payments = get_mint_quote_payments(&self.inner, "e.id).await?; let issuance = get_mint_quote_issuance(&self.inner, "e.id).await?; quote.issuance = issuance; quote.payments = payments; } Ok(mint_quote) } } #[async_trait] impl MintQuotesDatabase for SQLMintDatabase where RM: DatabasePool + 'static, { type Err = Error; async fn get_mint_quote(&self, quote_id: &Uuid) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let payments = get_mint_quote_payments(&*conn, quote_id).await?; let issuance = get_mint_quote_issuance(&*conn, quote_id).await?; Ok(query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE id = :id"#, )? .bind("id", quote_id.as_hyphenated().to_string()) .fetch_one(&*conn) .await? .map(|row| sql_row_to_mint_quote(row, payments, issuance)) .transpose()?) } async fn get_mint_quote_by_request( &self, request: &str, ) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut mint_quote = query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE request = :request"#, )? .bind("request", request.to_owned()) .fetch_one(&*conn) .await? .map(|row| sql_row_to_mint_quote(row, vec![], vec![])) .transpose()?; if let Some(quote) = mint_quote.as_mut() { let payments = get_mint_quote_payments(&*conn, "e.id).await?; let issuance = get_mint_quote_issuance(&*conn, "e.id).await?; quote.issuance = issuance; quote.payments = payments; } Ok(mint_quote) } async fn get_mint_quote_by_request_lookup_id( &self, request_lookup_id: &PaymentIdentifier, ) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut mint_quote = query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote WHERE request_lookup_id = :request_lookup_id AND request_lookup_id_kind = :request_lookup_id_kind "#, )? .bind("request_lookup_id", request_lookup_id.to_string()) .bind("request_lookup_id_kind", request_lookup_id.kind()) .fetch_one(&*conn) .await? .map(|row| sql_row_to_mint_quote(row, vec![], vec![])) .transpose()?; // TODO: these should use an sql join so they can be done in one query if let Some(quote) = mint_quote.as_mut() { let payments = get_mint_quote_payments(&*conn, "e.id).await?; let issuance = get_mint_quote_issuance(&*conn, "e.id).await?; quote.issuance = issuance; quote.payments = payments; } Ok(mint_quote) } async fn get_mint_quotes(&self) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut mint_quotes = query( r#" SELECT id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind FROM mint_quote "#, )? .fetch_all(&*conn) .await? .into_iter() .map(|row| sql_row_to_mint_quote(row, vec![], vec![])) .collect::, _>>()?; for quote in mint_quotes.as_mut_slice() { let payments = get_mint_quote_payments(&*conn, "e.id).await?; let issuance = get_mint_quote_issuance(&*conn, "e.id).await?; quote.issuance = issuance; quote.payments = payments; } Ok(mint_quotes) } async fn get_melt_quote(&self, quote_id: &Uuid) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT id, unit, amount, request, fee_reserve, expiry, state, payment_preimage, request_lookup_id, created_time, paid_time, payment_method, options, request_lookup_id_kind FROM melt_quote WHERE id=:id "#, )? .bind("id", quote_id.as_hyphenated().to_string()) .fetch_one(&*conn) .await? .map(sql_row_to_melt_quote) .transpose()?) } async fn get_melt_quotes(&self) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT id, unit, amount, request, fee_reserve, expiry, state, payment_preimage, request_lookup_id, created_time, paid_time, payment_method, options, request_lookup_id_kind FROM melt_quote "#, )? .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_melt_quote) .collect::, _>>()?) } } #[async_trait] impl MintProofsDatabase for SQLMintDatabase where RM: DatabasePool + 'static, { type Err = Error; async fn get_proofs_by_ys(&self, ys: &[PublicKey]) -> Result>, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut proofs = query( r#" SELECT amount, keyset_id, secret, c, witness, y FROM proof WHERE y IN (:ys) "#, )? .bind_vec("ys", ys.iter().map(|y| y.to_bytes().to_vec()).collect()) .fetch_all(&*conn) .await? .into_iter() .map(|mut row| { Ok(( column_as_string!( row.pop().ok_or(Error::InvalidDbResponse)?, PublicKey::from_hex, PublicKey::from_slice ), sql_row_to_proof(row)?, )) }) .collect::, Error>>()?; Ok(ys.iter().map(|y| proofs.remove(y)).collect()) } async fn get_proof_ys_by_quote_id(&self, quote_id: &Uuid) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT amount, keyset_id, secret, c, witness FROM proof WHERE quote_id = :quote_id "#, )? .bind("quote_id", quote_id.as_hyphenated().to_string()) .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_proof) .collect::, _>>()? .ys()?) } async fn get_proofs_states(&self, ys: &[PublicKey]) -> Result>, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut current_states = get_current_states(&*conn, ys).await?; Ok(ys.iter().map(|y| current_states.remove(y)).collect()) } async fn get_proofs_by_keyset_id( &self, keyset_id: &Id, ) -> Result<(Proofs, Vec>), Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT keyset_id, amount, secret, c, witness, state FROM proof WHERE keyset_id=? "#, )? .bind("keyset_id", keyset_id.to_string()) .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_proof_with_state) .collect::, _>>()? .into_iter() .unzip()) } } #[async_trait] impl MintSignatureTransaction<'_> for SQLTransaction where RM: DatabasePool + 'static, { type Err = Error; async fn add_blind_signatures( &mut self, blinded_messages: &[PublicKey], blind_signatures: &[BlindSignature], quote_id: Option, ) -> Result<(), Self::Err> { let current_time = unix_time(); for (message, signature) in blinded_messages.iter().zip(blind_signatures) { query( r#" INSERT INTO blind_signature (blinded_message, amount, keyset_id, c, quote_id, dleq_e, dleq_s, created_time) VALUES (:blinded_message, :amount, :keyset_id, :c, :quote_id, :dleq_e, :dleq_s, :created_time) "#, )? .bind("blinded_message", message.to_bytes().to_vec()) .bind("amount", u64::from(signature.amount) as i64) .bind("keyset_id", signature.keyset_id.to_string()) .bind("c", signature.c.to_bytes().to_vec()) .bind("quote_id", quote_id.map(|q| q.hyphenated().to_string())) .bind( "dleq_e", signature.dleq.as_ref().map(|dleq| dleq.e.to_secret_hex()), ) .bind( "dleq_s", signature.dleq.as_ref().map(|dleq| dleq.s.to_secret_hex()), ) .bind("created_time", current_time as i64) .execute(&self.inner) .await?; } Ok(()) } async fn get_blind_signatures( &mut self, blinded_messages: &[PublicKey], ) -> Result>, Self::Err> { let mut blinded_signatures = query( r#"SELECT keyset_id, amount, c, dleq_e, dleq_s, blinded_message FROM blind_signature WHERE blinded_message IN (:y) "#, )? .bind_vec( "y", blinded_messages .iter() .map(|y| y.to_bytes().to_vec()) .collect(), ) .fetch_all(&self.inner) .await? .into_iter() .map(|mut row| { Ok(( column_as_string!( &row.pop().ok_or(Error::InvalidDbResponse)?, PublicKey::from_hex, PublicKey::from_slice ), sql_row_to_blind_signature(row)?, )) }) .collect::, Error>>()?; Ok(blinded_messages .iter() .map(|y| blinded_signatures.remove(y)) .collect()) } } #[async_trait] impl MintSignaturesDatabase for SQLMintDatabase where RM: DatabasePool + 'static, { type Err = Error; async fn get_blind_signatures( &self, blinded_messages: &[PublicKey], ) -> Result>, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; let mut blinded_signatures = query( r#"SELECT keyset_id, amount, c, dleq_e, dleq_s, blinded_message FROM blind_signature WHERE blinded_message IN (:blinded_message) "#, )? .bind_vec( "blinded_message", blinded_messages .iter() .map(|b_| b_.to_bytes().to_vec()) .collect(), ) .fetch_all(&*conn) .await? .into_iter() .map(|mut row| { Ok(( column_as_string!( &row.pop().ok_or(Error::InvalidDbResponse)?, PublicKey::from_hex, PublicKey::from_slice ), sql_row_to_blind_signature(row)?, )) }) .collect::, Error>>()?; Ok(blinded_messages .iter() .map(|y| blinded_signatures.remove(y)) .collect()) } async fn get_blind_signatures_for_keyset( &self, keyset_id: &Id, ) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT keyset_id, amount, c, dleq_e, dleq_s FROM blind_signature WHERE keyset_id=:keyset_id "#, )? .bind("keyset_id", keyset_id.to_string()) .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_blind_signature) .collect::, _>>()?) } /// Get [`BlindSignature`]s for quote async fn get_blind_signatures_for_quote( &self, quote_id: &Uuid, ) -> Result, Self::Err> { let conn = self.pool.get().map_err(|e| Error::Database(Box::new(e)))?; Ok(query( r#" SELECT keyset_id, amount, c, dleq_e, dleq_s FROM blind_signature WHERE quote_id=:quote_id "#, )? .bind("quote_id", quote_id.to_string()) .fetch_all(&*conn) .await? .into_iter() .map(sql_row_to_blind_signature) .collect::, _>>()?) } } #[async_trait] impl MintDatabase for SQLMintDatabase where RM: DatabasePool + 'static, { async fn begin_transaction<'a>( &'a self, ) -> Result + Send + Sync + 'a>, Error> { Ok(Box::new(SQLTransaction { inner: ConnectionWithTransaction::new( self.pool.get().map_err(|e| Error::Database(Box::new(e)))?, ) .await?, })) } async fn get_mint_info(&self) -> Result { Ok(self.fetch_from_config("mint_info").await?) } async fn get_quote_ttl(&self) -> Result { Ok(self.fetch_from_config("quote_ttl").await?) } } fn sql_row_to_keyset_info(row: Vec) -> Result { unpack_into!( let ( id, unit, active, valid_from, valid_to, derivation_path, derivation_path_index, max_order, row_keyset_ppk ) = row ); Ok(MintKeySetInfo { id: column_as_string!(id, Id::from_str, Id::from_bytes), unit: column_as_string!(unit, CurrencyUnit::from_str), active: matches!(active, Column::Integer(1)), valid_from: column_as_number!(valid_from), derivation_path: column_as_string!(derivation_path, DerivationPath::from_str), derivation_path_index: column_as_nullable_number!(derivation_path_index), max_order: column_as_number!(max_order), input_fee_ppk: column_as_number!(row_keyset_ppk), final_expiry: column_as_nullable_number!(valid_to), }) } #[instrument(skip_all)] fn sql_row_to_mint_quote( row: Vec, payments: Vec, issueances: Vec, ) -> Result { unpack_into!( let ( id, amount, unit, request, expiry, request_lookup_id, pubkey, created_time, amount_paid, amount_issued, payment_method, request_lookup_id_kind ) = row ); let request_str = column_as_string!(&request); let request_lookup_id = column_as_nullable_string!(&request_lookup_id).unwrap_or_else(|| { Bolt11Invoice::from_str(&request_str) .map(|invoice| invoice.payment_hash().to_string()) .unwrap_or_else(|_| request_str.clone()) }); let request_lookup_id_kind = column_as_string!(request_lookup_id_kind); let pubkey = column_as_nullable_string!(&pubkey) .map(|pk| PublicKey::from_hex(&pk)) .transpose()?; let id = column_as_string!(id); let amount: Option = column_as_nullable_number!(amount); let amount_paid: u64 = column_as_number!(amount_paid); let amount_issued: u64 = column_as_number!(amount_issued); let payment_method = column_as_string!(payment_method, PaymentMethod::from_str); Ok(MintQuote::new( Some(Uuid::parse_str(&id).map_err(|_| Error::InvalidUuid(id))?), request_str, column_as_string!(unit, CurrencyUnit::from_str), amount.map(Amount::from), column_as_number!(expiry), PaymentIdentifier::new(&request_lookup_id_kind, &request_lookup_id) .map_err(|_| ConversionError::MissingParameter("Payment id".to_string()))?, pubkey, amount_paid.into(), amount_issued.into(), payment_method, column_as_number!(created_time), payments, issueances, )) } fn sql_row_to_melt_quote(row: Vec) -> Result { unpack_into!( let ( id, unit, amount, request, fee_reserve, expiry, state, payment_preimage, request_lookup_id, created_time, paid_time, payment_method, options, request_lookup_id_kind ) = row ); let id = column_as_string!(id); let amount: u64 = column_as_number!(amount); let fee_reserve: u64 = column_as_number!(fee_reserve); let expiry = column_as_number!(expiry); let payment_preimage = column_as_nullable_string!(payment_preimage); let options = column_as_nullable_string!(options); let options = options.and_then(|o| serde_json::from_str(&o).ok()); let created_time: i64 = column_as_number!(created_time); let paid_time = column_as_nullable_number!(paid_time); let payment_method = PaymentMethod::from_str(&column_as_string!(payment_method))?; let state = MeltQuoteState::from_str(&column_as_string!(&state)).map_err(ConversionError::from)?; let unit = column_as_string!(unit); let request = column_as_string!(request); let mut request_lookup_id_kind = column_as_string!(request_lookup_id_kind); let request_lookup_id = column_as_nullable_string!(&request_lookup_id).unwrap_or_else(|| { Bolt11Invoice::from_str(&request) .map(|invoice| invoice.payment_hash().to_string()) .unwrap_or_else(|_| { request_lookup_id_kind = "custom".to_string(); request.clone() }) }); let request_lookup_id = PaymentIdentifier::new(&request_lookup_id_kind, &request_lookup_id) .map_err(|_| ConversionError::MissingParameter("Payment id".to_string()))?; let request = match serde_json::from_str(&request) { Ok(req) => req, Err(err) => { tracing::debug!( "Melt quote from pre migrations defaulting to bolt11 {}.", err ); let bolt11 = Bolt11Invoice::from_str(&request).unwrap(); MeltPaymentRequest::Bolt11 { bolt11 } } }; Ok(MeltQuote { id: Uuid::parse_str(&id).map_err(|_| Error::InvalidUuid(id))?, unit: CurrencyUnit::from_str(&unit)?, amount: Amount::from(amount), request, fee_reserve: Amount::from(fee_reserve), state, expiry, payment_preimage, request_lookup_id, options, created_time: created_time as u64, paid_time, payment_method, }) } fn sql_row_to_proof(row: Vec) -> Result { unpack_into!( let ( amount, keyset_id, secret, c, witness ) = row ); let amount: u64 = column_as_number!(amount); Ok(Proof { amount: Amount::from(amount), keyset_id: column_as_string!(keyset_id, Id::from_str), secret: column_as_string!(secret, Secret::from_str), c: column_as_string!(c, PublicKey::from_hex, PublicKey::from_slice), witness: column_as_nullable_string!(witness).and_then(|w| serde_json::from_str(&w).ok()), dleq: None, }) } fn sql_row_to_proof_with_state(row: Vec) -> Result<(Proof, Option), Error> { unpack_into!( let ( keyset_id, amount, secret, c, witness, state ) = row ); let amount: u64 = column_as_number!(amount); let state = column_as_nullable_string!(state).and_then(|s| State::from_str(&s).ok()); Ok(( Proof { amount: Amount::from(amount), keyset_id: column_as_string!(keyset_id, Id::from_str, Id::from_bytes), secret: column_as_string!(secret, Secret::from_str), c: column_as_string!(c, PublicKey::from_hex, PublicKey::from_slice), witness: column_as_nullable_string!(witness) .and_then(|w| serde_json::from_str(&w).ok()), dleq: None, }, state, )) } fn sql_row_to_blind_signature(row: Vec) -> Result { unpack_into!( let ( keyset_id, amount, c, dleq_e, dleq_s ) = row ); let dleq = match ( column_as_nullable_string!(dleq_e), column_as_nullable_string!(dleq_s), ) { (Some(e), Some(s)) => Some(BlindSignatureDleq { e: SecretKey::from_hex(e)?, s: SecretKey::from_hex(s)?, }), _ => None, }; let amount: u64 = column_as_number!(amount); Ok(BlindSignature { amount: Amount::from(amount), keyset_id: column_as_string!(keyset_id, Id::from_str, Id::from_bytes), c: column_as_string!(c, PublicKey::from_hex, PublicKey::from_slice), dleq, }) }