mod.rs 63 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990
  1. //! SQLite Mint
  2. use std::collections::{HashMap, HashSet};
  3. use std::path::Path;
  4. use std::str::FromStr;
  5. use async_trait::async_trait;
  6. use bitcoin::bip32::DerivationPath;
  7. use cdk_common::common::{PaymentProcessorKey, QuoteTTL};
  8. use cdk_common::database::{
  9. self, MintDatabase, MintKeysDatabase, MintProofsDatabase, MintQuotesDatabase,
  10. MintSignaturesDatabase,
  11. };
  12. use cdk_common::mint::{self, MintKeySetInfo, MintQuote};
  13. use cdk_common::nut00::ProofsMethods;
  14. use cdk_common::nut05::QuoteState;
  15. use cdk_common::secret::Secret;
  16. use cdk_common::state::check_state_transition;
  17. use cdk_common::util::unix_time;
  18. use cdk_common::{
  19. Amount, BlindSignature, BlindSignatureDleq, CurrencyUnit, Id, MeltBolt11Request,
  20. MeltQuoteState, MintInfo, MintQuoteState, PaymentMethod, Proof, Proofs, PublicKey, SecretKey,
  21. State,
  22. };
  23. use error::Error;
  24. use lightning_invoice::Bolt11Invoice;
  25. use sqlx::sqlite::SqliteRow;
  26. use sqlx::{Executor, Pool, Row, Sqlite};
  27. use uuid::fmt::Hyphenated;
  28. use uuid::Uuid;
  29. use crate::common::create_sqlite_pool;
  30. #[cfg(feature = "auth")]
  31. mod auth;
  32. pub mod error;
  33. pub mod memory;
  34. #[cfg(feature = "auth")]
  35. pub use auth::MintSqliteAuthDatabase;
  36. /// Mint SQLite Database
  37. #[derive(Debug, Clone)]
  38. pub struct MintSqliteDatabase {
  39. pool: Pool<Sqlite>,
  40. }
  41. impl MintSqliteDatabase {
  42. /// Check if any proofs are spent
  43. async fn check_for_spent_proofs<'e, 'c: 'e, E>(
  44. &self,
  45. transaction: E,
  46. ys: &[PublicKey],
  47. ) -> Result<bool, database::Error>
  48. where
  49. E: Executor<'c, Database = Sqlite>,
  50. {
  51. if ys.is_empty() {
  52. return Ok(false);
  53. }
  54. let check_sql = format!(
  55. "SELECT state FROM proof WHERE y IN ({}) AND state = 'SPENT'",
  56. std::iter::repeat("?")
  57. .take(ys.len())
  58. .collect::<Vec<_>>()
  59. .join(",")
  60. );
  61. let spent_count = ys
  62. .iter()
  63. .fold(sqlx::query(&check_sql), |query, y| {
  64. query.bind(y.to_bytes().to_vec())
  65. })
  66. .fetch_all(transaction)
  67. .await
  68. .map_err(Error::from)?
  69. .len();
  70. Ok(spent_count > 0)
  71. }
  72. /// Create new [`MintSqliteDatabase`]
  73. #[cfg(not(feature = "sqlcipher"))]
  74. pub async fn new<P: AsRef<Path>>(path: P) -> Result<Self, Error> {
  75. let db = Self {
  76. pool: create_sqlite_pool(path.as_ref().to_str().ok_or(Error::InvalidDbPath)?).await?,
  77. };
  78. db.migrate().await?;
  79. Ok(db)
  80. }
  81. /// Create new [`MintSqliteDatabase`]
  82. #[cfg(feature = "sqlcipher")]
  83. pub async fn new<P: AsRef<Path>>(path: P, password: String) -> Result<Self, Error> {
  84. let db = Self {
  85. pool: create_sqlite_pool(
  86. path.as_ref().to_str().ok_or(Error::InvalidDbPath)?,
  87. password,
  88. )
  89. .await?,
  90. };
  91. db.migrate().await?;
  92. Ok(db)
  93. }
  94. /// Migrate [`MintSqliteDatabase`]
  95. async fn migrate(&self) -> Result<(), Error> {
  96. sqlx::migrate!("./src/mint/migrations")
  97. .run(&self.pool)
  98. .await
  99. .map_err(|_| Error::CouldNotInitialize)?;
  100. Ok(())
  101. }
  102. }
  103. #[async_trait]
  104. impl MintKeysDatabase for MintSqliteDatabase {
  105. type Err = database::Error;
  106. async fn set_active_keyset(&self, unit: CurrencyUnit, id: Id) -> Result<(), Self::Err> {
  107. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  108. let update_res = sqlx::query(
  109. r#"
  110. UPDATE keyset
  111. SET active=FALSE
  112. WHERE unit IS ?;
  113. "#,
  114. )
  115. .bind(unit.to_string())
  116. .execute(&mut *transaction)
  117. .await;
  118. match update_res {
  119. Ok(_) => (),
  120. Err(err) => {
  121. tracing::error!("SQLite Could not update keyset");
  122. if let Err(err) = transaction.rollback().await {
  123. tracing::error!("Could not rollback sql transaction: {}", err);
  124. }
  125. return Err(Error::from(err).into());
  126. }
  127. };
  128. let update_res = sqlx::query(
  129. r#"
  130. UPDATE keyset
  131. SET active=TRUE
  132. WHERE unit IS ?
  133. AND id IS ?;
  134. "#,
  135. )
  136. .bind(unit.to_string())
  137. .bind(id.to_string())
  138. .execute(&mut *transaction)
  139. .await;
  140. match update_res {
  141. Ok(_) => (),
  142. Err(err) => {
  143. tracing::error!("SQLite Could not update keyset");
  144. if let Err(err) = transaction.rollback().await {
  145. tracing::error!("Could not rollback sql transaction: {}", err);
  146. }
  147. return Err(Error::from(err).into());
  148. }
  149. };
  150. transaction.commit().await.map_err(Error::from)?;
  151. Ok(())
  152. }
  153. async fn get_active_keyset_id(&self, unit: &CurrencyUnit) -> Result<Option<Id>, Self::Err> {
  154. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  155. let rec = sqlx::query(
  156. r#"
  157. SELECT id
  158. FROM keyset
  159. WHERE active = 1
  160. AND unit IS ?
  161. "#,
  162. )
  163. .bind(unit.to_string())
  164. .fetch_one(&mut *transaction)
  165. .await;
  166. let rec = match rec {
  167. Ok(rec) => {
  168. transaction.commit().await.map_err(Error::from)?;
  169. rec
  170. }
  171. Err(err) => match err {
  172. sqlx::Error::RowNotFound => {
  173. transaction.commit().await.map_err(Error::from)?;
  174. return Ok(None);
  175. }
  176. _ => {
  177. return {
  178. if let Err(err) = transaction.rollback().await {
  179. tracing::error!("Could not rollback sql transaction: {}", err);
  180. }
  181. Err(Error::SQLX(err).into())
  182. }
  183. }
  184. },
  185. };
  186. Ok(Some(
  187. Id::from_str(rec.try_get("id").map_err(Error::from)?).map_err(Error::from)?,
  188. ))
  189. }
  190. async fn get_active_keysets(&self) -> Result<HashMap<CurrencyUnit, Id>, Self::Err> {
  191. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  192. let recs = sqlx::query(
  193. r#"
  194. SELECT id, unit
  195. FROM keyset
  196. WHERE active = 1
  197. "#,
  198. )
  199. .fetch_all(&mut *transaction)
  200. .await;
  201. match recs {
  202. Ok(recs) => {
  203. transaction.commit().await.map_err(Error::from)?;
  204. let keysets = recs
  205. .iter()
  206. .filter_map(|r| match Id::from_str(r.get("id")) {
  207. Ok(id) => Some((
  208. CurrencyUnit::from_str(r.get::<'_, &str, &str>("unit")).unwrap(),
  209. id,
  210. )),
  211. Err(_) => None,
  212. })
  213. .collect();
  214. Ok(keysets)
  215. }
  216. Err(err) => {
  217. tracing::error!("SQLite could not get active keyset");
  218. if let Err(err) = transaction.rollback().await {
  219. tracing::error!("Could not rollback sql transaction: {}", err);
  220. }
  221. Err(Error::from(err).into())
  222. }
  223. }
  224. }
  225. async fn add_keyset_info(&self, keyset: MintKeySetInfo) -> Result<(), Self::Err> {
  226. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  227. let res = sqlx::query(
  228. r#"
  229. INSERT INTO keyset
  230. (id, unit, active, valid_from, valid_to, derivation_path, max_order, input_fee_ppk, derivation_path_index)
  231. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
  232. ON CONFLICT(id) DO UPDATE SET
  233. unit = excluded.unit,
  234. active = excluded.active,
  235. valid_from = excluded.valid_from,
  236. valid_to = excluded.valid_to,
  237. derivation_path = excluded.derivation_path,
  238. max_order = excluded.max_order,
  239. input_fee_ppk = excluded.input_fee_ppk,
  240. derivation_path_index = excluded.derivation_path_index
  241. "#,
  242. )
  243. .bind(keyset.id.to_string())
  244. .bind(keyset.unit.to_string())
  245. .bind(keyset.active)
  246. .bind(keyset.valid_from as i64)
  247. .bind(keyset.valid_to.map(|v| v as i64))
  248. .bind(keyset.derivation_path.to_string())
  249. .bind(keyset.max_order)
  250. .bind(keyset.input_fee_ppk as i64)
  251. .bind(keyset.derivation_path_index)
  252. .execute(&mut *transaction)
  253. .await;
  254. match res {
  255. Ok(_) => {
  256. transaction.commit().await.map_err(Error::from)?;
  257. Ok(())
  258. }
  259. Err(err) => {
  260. tracing::error!("SQLite could not add keyset info");
  261. if let Err(err) = transaction.rollback().await {
  262. tracing::error!("Could not rollback sql transaction: {}", err);
  263. }
  264. Err(Error::from(err).into())
  265. }
  266. }
  267. }
  268. async fn get_keyset_info(&self, id: &Id) -> Result<Option<MintKeySetInfo>, Self::Err> {
  269. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  270. let rec = sqlx::query(
  271. r#"
  272. SELECT *
  273. FROM keyset
  274. WHERE id=?;
  275. "#,
  276. )
  277. .bind(id.to_string())
  278. .fetch_one(&mut *transaction)
  279. .await;
  280. match rec {
  281. Ok(rec) => {
  282. transaction.commit().await.map_err(Error::from)?;
  283. Ok(Some(sqlite_row_to_keyset_info(rec)?))
  284. }
  285. Err(err) => match err {
  286. sqlx::Error::RowNotFound => {
  287. transaction.commit().await.map_err(Error::from)?;
  288. return Ok(None);
  289. }
  290. _ => {
  291. tracing::error!("SQLite could not get keyset info");
  292. if let Err(err) = transaction.rollback().await {
  293. tracing::error!("Could not rollback sql transaction: {}", err);
  294. }
  295. return Err(Error::SQLX(err).into());
  296. }
  297. },
  298. }
  299. }
  300. async fn get_keyset_infos(&self) -> Result<Vec<MintKeySetInfo>, Self::Err> {
  301. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  302. let recs = sqlx::query(
  303. r#"
  304. SELECT *
  305. FROM keyset;
  306. "#,
  307. )
  308. .fetch_all(&mut *transaction)
  309. .await
  310. .map_err(Error::from);
  311. match recs {
  312. Ok(recs) => {
  313. transaction.commit().await.map_err(Error::from)?;
  314. Ok(recs
  315. .into_iter()
  316. .map(sqlite_row_to_keyset_info)
  317. .collect::<Result<_, _>>()?)
  318. }
  319. Err(err) => {
  320. tracing::error!("SQLite could not get keyset info");
  321. if let Err(err) = transaction.rollback().await {
  322. tracing::error!("Could not rollback sql transaction: {}", err);
  323. }
  324. Err(err.into())
  325. }
  326. }
  327. }
  328. }
  329. #[async_trait]
  330. impl MintQuotesDatabase for MintSqliteDatabase {
  331. type Err = database::Error;
  332. async fn add_mint_quote(&self, quote: MintQuote) -> Result<(), Self::Err> {
  333. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  334. let res = sqlx::query(
  335. r#"
  336. INSERT INTO mint_quote
  337. (id, amount, unit, request, state, expiry, request_lookup_id, pubkey, created_time, paid_time, issued_time)
  338. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
  339. ON CONFLICT(id) DO UPDATE SET
  340. amount = excluded.amount,
  341. unit = excluded.unit,
  342. request = excluded.request,
  343. state = excluded.state,
  344. expiry = excluded.expiry,
  345. request_lookup_id = excluded.request_lookup_id,
  346. created_time = excluded.created_time,
  347. paid_time = excluded.paid_time,
  348. issued_time = excluded.issued_time
  349. ON CONFLICT(request_lookup_id) DO UPDATE SET
  350. amount = excluded.amount,
  351. unit = excluded.unit,
  352. request = excluded.request,
  353. state = excluded.state,
  354. expiry = excluded.expiry,
  355. id = excluded.id,
  356. created_time = excluded.created_time,
  357. paid_time = excluded.paid_time,
  358. issued_time = excluded.issued_time
  359. "#,
  360. )
  361. .bind(quote.id.to_string())
  362. .bind(u64::from(quote.amount) as i64)
  363. .bind(quote.unit.to_string())
  364. .bind(quote.request)
  365. .bind(quote.state.to_string())
  366. .bind(quote.expiry as i64)
  367. .bind(quote.request_lookup_id)
  368. .bind(quote.pubkey.map(|p| p.to_string()))
  369. .bind(quote.created_time as i64)
  370. .bind(quote.paid_time.map(|t| t as i64))
  371. .bind(quote.issued_time.map(|t| t as i64))
  372. .execute(&mut *transaction)
  373. .await;
  374. match res {
  375. Ok(_) => {
  376. transaction.commit().await.map_err(Error::from)?;
  377. Ok(())
  378. }
  379. Err(err) => {
  380. tracing::error!("SQLite Could not update keyset");
  381. if let Err(err) = transaction.rollback().await {
  382. tracing::error!("Could not rollback sql transaction: {}", err);
  383. }
  384. Err(Error::from(err).into())
  385. }
  386. }
  387. }
  388. async fn get_mint_quote(&self, quote_id: &Uuid) -> Result<Option<MintQuote>, Self::Err> {
  389. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  390. let rec = sqlx::query(
  391. r#"
  392. SELECT *
  393. FROM mint_quote
  394. WHERE id=?;
  395. "#,
  396. )
  397. .bind(quote_id.as_hyphenated())
  398. .fetch_one(&mut *transaction)
  399. .await;
  400. match rec {
  401. Ok(rec) => {
  402. transaction.commit().await.map_err(Error::from)?;
  403. Ok(Some(sqlite_row_to_mint_quote(rec)?))
  404. }
  405. Err(err) => match err {
  406. sqlx::Error::RowNotFound => {
  407. transaction.commit().await.map_err(Error::from)?;
  408. Ok(None)
  409. }
  410. _ => {
  411. if let Err(err) = transaction.rollback().await {
  412. tracing::error!("Could not rollback sql transaction: {}", err);
  413. }
  414. Err(Error::SQLX(err).into())
  415. }
  416. },
  417. }
  418. }
  419. async fn get_mint_quote_by_request(
  420. &self,
  421. request: &str,
  422. ) -> Result<Option<MintQuote>, Self::Err> {
  423. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  424. let rec = sqlx::query(
  425. r#"
  426. SELECT *
  427. FROM mint_quote
  428. WHERE request=?;
  429. "#,
  430. )
  431. .bind(request)
  432. .fetch_one(&mut *transaction)
  433. .await;
  434. match rec {
  435. Ok(rec) => {
  436. transaction.commit().await.map_err(Error::from)?;
  437. Ok(Some(sqlite_row_to_mint_quote(rec)?))
  438. }
  439. Err(err) => match err {
  440. sqlx::Error::RowNotFound => {
  441. transaction.commit().await.map_err(Error::from)?;
  442. Ok(None)
  443. }
  444. _ => {
  445. if let Err(err) = transaction.rollback().await {
  446. tracing::error!("Could not rollback sql transaction: {}", err);
  447. }
  448. Err(Error::SQLX(err).into())
  449. }
  450. },
  451. }
  452. }
  453. async fn get_mint_quote_by_request_lookup_id(
  454. &self,
  455. request_lookup_id: &str,
  456. ) -> Result<Option<MintQuote>, Self::Err> {
  457. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  458. let rec = sqlx::query(
  459. r#"
  460. SELECT *
  461. FROM mint_quote
  462. WHERE request_lookup_id=?;
  463. "#,
  464. )
  465. .bind(request_lookup_id)
  466. .fetch_one(&mut *transaction)
  467. .await;
  468. match rec {
  469. Ok(rec) => {
  470. transaction.commit().await.map_err(Error::from)?;
  471. Ok(Some(sqlite_row_to_mint_quote(rec)?))
  472. }
  473. Err(err) => match err {
  474. sqlx::Error::RowNotFound => {
  475. transaction.commit().await.map_err(Error::from)?;
  476. Ok(None)
  477. }
  478. _ => {
  479. if let Err(err) = transaction.rollback().await {
  480. tracing::error!("Could not rollback sql transaction: {}", err);
  481. }
  482. Err(Error::SQLX(err).into())
  483. }
  484. },
  485. }
  486. }
  487. async fn update_mint_quote_state(
  488. &self,
  489. quote_id: &Uuid,
  490. state: MintQuoteState,
  491. ) -> Result<MintQuoteState, Self::Err> {
  492. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  493. let rec = sqlx::query(
  494. r#"
  495. SELECT *
  496. FROM mint_quote
  497. WHERE id=?;
  498. "#,
  499. )
  500. .bind(quote_id.as_hyphenated())
  501. .fetch_one(&mut *transaction)
  502. .await;
  503. let quote = match rec {
  504. Ok(row) => sqlite_row_to_mint_quote(row)?,
  505. Err(err) => {
  506. tracing::error!("SQLite Could not update keyset");
  507. if let Err(err) = transaction.rollback().await {
  508. tracing::error!("Could not rollback sql transaction: {}", err);
  509. }
  510. return Err(Error::from(err).into());
  511. }
  512. };
  513. let update_query = match state {
  514. MintQuoteState::Paid => {
  515. r#"UPDATE mint_quote SET state = ?, paid_time = ? WHERE id = ?"#
  516. }
  517. MintQuoteState::Issued => {
  518. r#"UPDATE mint_quote SET state = ?, issued_time = ? WHERE id = ?"#
  519. }
  520. _ => r#"UPDATE mint_quote SET state = ? WHERE id = ?"#,
  521. };
  522. let current_time = unix_time();
  523. let update = match state {
  524. MintQuoteState::Paid => {
  525. sqlx::query(update_query)
  526. .bind(state.to_string())
  527. .bind(current_time as i64)
  528. .bind(quote_id.as_hyphenated())
  529. .execute(&mut *transaction)
  530. .await
  531. }
  532. MintQuoteState::Issued => {
  533. sqlx::query(update_query)
  534. .bind(state.to_string())
  535. .bind(current_time as i64)
  536. .bind(quote_id.as_hyphenated())
  537. .execute(&mut *transaction)
  538. .await
  539. }
  540. _ => {
  541. sqlx::query(update_query)
  542. .bind(state.to_string())
  543. .bind(quote_id.as_hyphenated())
  544. .execute(&mut *transaction)
  545. .await
  546. }
  547. };
  548. match update {
  549. Ok(_) => {
  550. transaction.commit().await.map_err(Error::from)?;
  551. Ok(quote.state)
  552. }
  553. Err(err) => {
  554. tracing::error!("SQLite Could not update keyset");
  555. if let Err(err) = transaction.rollback().await {
  556. tracing::error!("Could not rollback sql transaction: {}", err);
  557. }
  558. return Err(Error::from(err).into());
  559. }
  560. }
  561. }
  562. async fn get_mint_quotes(&self) -> Result<Vec<MintQuote>, Self::Err> {
  563. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  564. let rec = sqlx::query(
  565. r#"
  566. SELECT *
  567. FROM mint_quote
  568. "#,
  569. )
  570. .fetch_all(&mut *transaction)
  571. .await;
  572. match rec {
  573. Ok(rows) => {
  574. transaction.commit().await.map_err(Error::from)?;
  575. let mint_quotes = rows
  576. .into_iter()
  577. .map(sqlite_row_to_mint_quote)
  578. .collect::<Result<Vec<MintQuote>, _>>()?;
  579. Ok(mint_quotes)
  580. }
  581. Err(err) => {
  582. tracing::error!("SQLite get mint quotes");
  583. if let Err(err) = transaction.rollback().await {
  584. tracing::error!("Could not rollback sql transaction: {}", err);
  585. }
  586. return Err(Error::from(err).into());
  587. }
  588. }
  589. }
  590. async fn get_mint_quotes_with_state(
  591. &self,
  592. state: MintQuoteState,
  593. ) -> Result<Vec<MintQuote>, Self::Err> {
  594. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  595. let rec = sqlx::query(
  596. r#"
  597. SELECT *
  598. FROM mint_quote
  599. WHERE state = ?
  600. "#,
  601. )
  602. .bind(state.to_string())
  603. .fetch_all(&mut *transaction)
  604. .await;
  605. match rec {
  606. Ok(rows) => {
  607. transaction.commit().await.map_err(Error::from)?;
  608. let mint_quotes = rows
  609. .into_iter()
  610. .map(sqlite_row_to_mint_quote)
  611. .collect::<Result<Vec<MintQuote>, _>>()?;
  612. Ok(mint_quotes)
  613. }
  614. Err(err) => {
  615. tracing::error!("SQLite get mint quotes with state");
  616. if let Err(err) = transaction.rollback().await {
  617. tracing::error!("Could not rollback sql transaction: {}", err);
  618. }
  619. return Err(Error::from(err).into());
  620. }
  621. }
  622. }
  623. async fn remove_mint_quote(&self, quote_id: &Uuid) -> Result<(), Self::Err> {
  624. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  625. let res = sqlx::query(
  626. r#"
  627. DELETE FROM mint_quote
  628. WHERE id=?
  629. "#,
  630. )
  631. .bind(quote_id.as_hyphenated())
  632. .execute(&mut *transaction)
  633. .await;
  634. match res {
  635. Ok(_) => {
  636. transaction.commit().await.map_err(Error::from)?;
  637. Ok(())
  638. }
  639. Err(err) => {
  640. tracing::error!("SQLite Could not remove mint quote");
  641. if let Err(err) = transaction.rollback().await {
  642. tracing::error!("Could not rollback sql transaction: {}", err);
  643. }
  644. Err(Error::from(err).into())
  645. }
  646. }
  647. }
  648. async fn add_melt_quote(&self, quote: mint::MeltQuote) -> Result<(), Self::Err> {
  649. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  650. let res = sqlx::query(
  651. r#"
  652. INSERT INTO melt_quote
  653. (id, unit, amount, request, fee_reserve, state, expiry, payment_preimage, request_lookup_id, msat_to_pay, created_time, paid_time)
  654. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
  655. ON CONFLICT(id) DO UPDATE SET
  656. unit = excluded.unit,
  657. amount = excluded.amount,
  658. request = excluded.request,
  659. fee_reserve = excluded.fee_reserve,
  660. state = excluded.state,
  661. expiry = excluded.expiry,
  662. payment_preimage = excluded.payment_preimage,
  663. request_lookup_id = excluded.request_lookup_id,
  664. msat_to_pay = excluded.msat_to_pay,
  665. created_time = excluded.created_time,
  666. paid_time = excluded.paid_time
  667. ON CONFLICT(request_lookup_id) DO UPDATE SET
  668. unit = excluded.unit,
  669. amount = excluded.amount,
  670. request = excluded.request,
  671. fee_reserve = excluded.fee_reserve,
  672. state = excluded.state,
  673. expiry = excluded.expiry,
  674. payment_preimage = excluded.payment_preimage,
  675. id = excluded.id,
  676. created_time = excluded.created_time,
  677. paid_time = excluded.paid_time;
  678. "#,
  679. )
  680. .bind(quote.id.to_string())
  681. .bind(quote.unit.to_string())
  682. .bind(u64::from(quote.amount) as i64)
  683. .bind(quote.request)
  684. .bind(u64::from(quote.fee_reserve) as i64)
  685. .bind(quote.state.to_string())
  686. .bind(quote.expiry as i64)
  687. .bind(quote.payment_preimage)
  688. .bind(quote.request_lookup_id)
  689. .bind(quote.msat_to_pay.map(|a| u64::from(a) as i64))
  690. .bind(quote.created_time as i64)
  691. .bind(quote.paid_time.map(|t| t as i64))
  692. .execute(&mut *transaction)
  693. .await;
  694. match res {
  695. Ok(_) => {
  696. transaction.commit().await.map_err(Error::from)?;
  697. Ok(())
  698. }
  699. Err(err) => {
  700. tracing::error!("SQLite Could not remove mint quote");
  701. if let Err(err) = transaction.rollback().await {
  702. tracing::error!("Could not rollback sql transaction: {}", err);
  703. }
  704. Err(Error::from(err).into())
  705. }
  706. }
  707. }
  708. async fn get_melt_quote(&self, quote_id: &Uuid) -> Result<Option<mint::MeltQuote>, Self::Err> {
  709. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  710. let rec = sqlx::query(
  711. r#"
  712. SELECT *
  713. FROM melt_quote
  714. WHERE id=?;
  715. "#,
  716. )
  717. .bind(quote_id.as_hyphenated())
  718. .fetch_one(&mut *transaction)
  719. .await;
  720. match rec {
  721. Ok(rec) => {
  722. transaction.commit().await.map_err(Error::from)?;
  723. Ok(Some(sqlite_row_to_melt_quote(rec)?))
  724. }
  725. Err(err) => match err {
  726. sqlx::Error::RowNotFound => {
  727. transaction.commit().await.map_err(Error::from)?;
  728. Ok(None)
  729. }
  730. _ => {
  731. if let Err(err) = transaction.rollback().await {
  732. tracing::error!("Could not rollback sql transaction: {}", err);
  733. }
  734. Err(Error::SQLX(err).into())
  735. }
  736. },
  737. }
  738. }
  739. async fn get_melt_quotes(&self) -> Result<Vec<mint::MeltQuote>, Self::Err> {
  740. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  741. let rec = sqlx::query(
  742. r#"
  743. SELECT *
  744. FROM melt_quote
  745. "#,
  746. )
  747. .fetch_all(&mut *transaction)
  748. .await
  749. .map_err(Error::from);
  750. match rec {
  751. Ok(rec) => {
  752. let melt_quotes = rec
  753. .into_iter()
  754. .map(sqlite_row_to_melt_quote)
  755. .collect::<Result<Vec<mint::MeltQuote>, _>>()?;
  756. Ok(melt_quotes)
  757. }
  758. Err(err) => {
  759. if let Err(err) = transaction.rollback().await {
  760. tracing::error!("Could not rollback sql transaction: {}", err);
  761. }
  762. Err(err.into())
  763. }
  764. }
  765. }
  766. async fn update_melt_quote_state(
  767. &self,
  768. quote_id: &Uuid,
  769. state: MeltQuoteState,
  770. ) -> Result<MeltQuoteState, Self::Err> {
  771. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  772. let rec = sqlx::query(
  773. r#"
  774. SELECT *
  775. FROM melt_quote
  776. WHERE id=?;
  777. "#,
  778. )
  779. .bind(quote_id.as_hyphenated())
  780. .fetch_one(&mut *transaction)
  781. .await;
  782. let quote = match rec {
  783. Ok(rec) => sqlite_row_to_melt_quote(rec)?,
  784. Err(err) => {
  785. tracing::error!("SQLite Could not update keyset");
  786. if let Err(err) = transaction.rollback().await {
  787. tracing::error!("Could not rollback sql transaction: {}", err);
  788. }
  789. return Err(Error::from(err).into());
  790. }
  791. };
  792. let update_query = if state == MeltQuoteState::Paid {
  793. r#"UPDATE melt_quote SET state = ?, paid_time = ? WHERE id = ?"#
  794. } else {
  795. r#"UPDATE melt_quote SET state = ? WHERE id = ?"#
  796. };
  797. let current_time = unix_time();
  798. let rec = if state == MeltQuoteState::Paid {
  799. sqlx::query(update_query)
  800. .bind(state.to_string())
  801. .bind(current_time as i64)
  802. .bind(quote_id.as_hyphenated())
  803. .execute(&mut *transaction)
  804. .await
  805. } else {
  806. sqlx::query(update_query)
  807. .bind(state.to_string())
  808. .bind(quote_id.as_hyphenated())
  809. .execute(&mut *transaction)
  810. .await
  811. };
  812. match rec {
  813. Ok(_) => {
  814. transaction.commit().await.map_err(Error::from)?;
  815. }
  816. Err(err) => {
  817. tracing::error!("SQLite Could not update melt quote");
  818. if let Err(err) = transaction.rollback().await {
  819. tracing::error!("Could not rollback sql transaction: {}", err);
  820. }
  821. return Err(Error::from(err).into());
  822. }
  823. };
  824. Ok(quote.state)
  825. }
  826. async fn remove_melt_quote(&self, quote_id: &Uuid) -> Result<(), Self::Err> {
  827. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  828. let res = sqlx::query(
  829. r#"
  830. DELETE FROM melt_quote
  831. WHERE id=?
  832. "#,
  833. )
  834. .bind(quote_id.as_hyphenated())
  835. .execute(&mut *transaction)
  836. .await;
  837. match res {
  838. Ok(_) => {
  839. transaction.commit().await.map_err(Error::from)?;
  840. Ok(())
  841. }
  842. Err(err) => {
  843. tracing::error!("SQLite Could not update melt quote");
  844. if let Err(err) = transaction.rollback().await {
  845. tracing::error!("Could not rollback sql transaction: {}", err);
  846. }
  847. Err(Error::from(err).into())
  848. }
  849. }
  850. }
  851. async fn add_melt_request(
  852. &self,
  853. melt_request: MeltBolt11Request<Uuid>,
  854. ln_key: PaymentProcessorKey,
  855. ) -> Result<(), Self::Err> {
  856. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  857. let res = sqlx::query(
  858. r#"
  859. INSERT INTO melt_request
  860. (id, inputs, outputs, method, unit)
  861. VALUES (?, ?, ?, ?, ?)
  862. ON CONFLICT(id) DO UPDATE SET
  863. inputs = excluded.inputs,
  864. outputs = excluded.outputs,
  865. method = excluded.method,
  866. unit = excluded.unit
  867. "#,
  868. )
  869. .bind(melt_request.quote())
  870. .bind(serde_json::to_string(&melt_request.inputs())?)
  871. .bind(serde_json::to_string(&melt_request.outputs())?)
  872. .bind(ln_key.method.to_string())
  873. .bind(ln_key.unit.to_string())
  874. .execute(&mut *transaction)
  875. .await;
  876. match res {
  877. Ok(_) => {
  878. transaction.commit().await.map_err(Error::from)?;
  879. Ok(())
  880. }
  881. Err(err) => {
  882. tracing::error!("SQLite Could not update keyset");
  883. if let Err(err) = transaction.rollback().await {
  884. tracing::error!("Could not rollback sql transaction: {}", err);
  885. }
  886. Err(Error::from(err).into())
  887. }
  888. }
  889. }
  890. async fn get_melt_request(
  891. &self,
  892. quote_id: &Uuid,
  893. ) -> Result<Option<(MeltBolt11Request<Uuid>, PaymentProcessorKey)>, Self::Err> {
  894. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  895. let rec = sqlx::query(
  896. r#"
  897. SELECT *
  898. FROM melt_request
  899. WHERE id=?;
  900. "#,
  901. )
  902. .bind(quote_id.as_hyphenated())
  903. .fetch_one(&mut *transaction)
  904. .await;
  905. match rec {
  906. Ok(rec) => {
  907. transaction.commit().await.map_err(Error::from)?;
  908. let (request, key) = sqlite_row_to_melt_request(rec)?;
  909. Ok(Some((request, key)))
  910. }
  911. Err(err) => match err {
  912. sqlx::Error::RowNotFound => {
  913. transaction.commit().await.map_err(Error::from)?;
  914. return Ok(None);
  915. }
  916. _ => {
  917. return {
  918. if let Err(err) = transaction.rollback().await {
  919. tracing::error!("Could not rollback sql transaction: {}", err);
  920. }
  921. Err(Error::SQLX(err).into())
  922. }
  923. }
  924. },
  925. }
  926. }
  927. }
  928. #[async_trait]
  929. impl MintProofsDatabase for MintSqliteDatabase {
  930. type Err = database::Error;
  931. async fn add_proofs(&self, proofs: Proofs, quote_id: Option<Uuid>) -> Result<(), Self::Err> {
  932. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  933. let current_time = unix_time();
  934. for proof in proofs {
  935. let result = sqlx::query(
  936. r#"
  937. INSERT OR IGNORE INTO proof
  938. (y, amount, keyset_id, secret, c, witness, state, quote_id, created_time)
  939. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?);
  940. "#,
  941. )
  942. .bind(proof.y()?.to_bytes().to_vec())
  943. .bind(u64::from(proof.amount) as i64)
  944. .bind(proof.keyset_id.to_string())
  945. .bind(proof.secret.to_string())
  946. .bind(proof.c.to_bytes().to_vec())
  947. .bind(proof.witness.map(|w| serde_json::to_string(&w).unwrap()))
  948. .bind("UNSPENT")
  949. .bind(quote_id.map(|q| q.hyphenated()))
  950. .bind(current_time as i64)
  951. .execute(&mut *transaction)
  952. .await;
  953. // We still need to check for foreign key constraint errors
  954. if let Err(err) = result {
  955. if let sqlx::Error::Database(db_err) = &err {
  956. if db_err.message().contains("FOREIGN KEY constraint failed") {
  957. tracing::error!(
  958. "Foreign key constraint failed when adding proof: {:?}",
  959. err
  960. );
  961. transaction.rollback().await.map_err(Error::from)?;
  962. return Err(database::Error::InvalidKeysetId);
  963. }
  964. }
  965. // For any other error, roll back and return the error
  966. tracing::error!("Error adding proof: {:?}", err);
  967. transaction.rollback().await.map_err(Error::from)?;
  968. return Err(Error::from(err).into());
  969. }
  970. }
  971. transaction.commit().await.map_err(Error::from)?;
  972. Ok(())
  973. }
  974. async fn remove_proofs(
  975. &self,
  976. ys: &[PublicKey],
  977. _quote_id: Option<Uuid>,
  978. ) -> Result<(), Self::Err> {
  979. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  980. if self.check_for_spent_proofs(&mut *transaction, ys).await? {
  981. transaction.rollback().await.map_err(Error::from)?;
  982. return Err(Self::Err::AttemptRemoveSpentProof);
  983. }
  984. // If no proofs are spent, proceed with deletion
  985. let delete_sql = format!(
  986. "DELETE FROM proof WHERE y IN ({})",
  987. std::iter::repeat("?")
  988. .take(ys.len())
  989. .collect::<Vec<_>>()
  990. .join(",")
  991. );
  992. ys.iter()
  993. .fold(sqlx::query(&delete_sql), |query, y| {
  994. query.bind(y.to_bytes().to_vec())
  995. })
  996. .execute(&mut *transaction)
  997. .await
  998. .map_err(Error::from)?;
  999. transaction.commit().await.map_err(Error::from)?;
  1000. Ok(())
  1001. }
  1002. async fn get_proofs_by_ys(&self, ys: &[PublicKey]) -> Result<Vec<Option<Proof>>, Self::Err> {
  1003. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1004. let sql = format!(
  1005. "SELECT * FROM proof WHERE y IN ({})",
  1006. "?,".repeat(ys.len()).trim_end_matches(',')
  1007. );
  1008. let mut proofs = ys
  1009. .iter()
  1010. .fold(sqlx::query(&sql), |query, y| {
  1011. query.bind(y.to_bytes().to_vec())
  1012. })
  1013. .fetch_all(&mut *transaction)
  1014. .await
  1015. .map_err(|err| {
  1016. tracing::error!("SQLite could not get state of proof: {err:?}");
  1017. Error::SQLX(err)
  1018. })?
  1019. .into_iter()
  1020. .map(|row| {
  1021. PublicKey::from_slice(row.get("y"))
  1022. .map_err(Error::from)
  1023. .and_then(|y| sqlite_row_to_proof(row).map(|proof| (y, proof)))
  1024. })
  1025. .collect::<Result<HashMap<_, _>, _>>()?;
  1026. Ok(ys.iter().map(|y| proofs.remove(y)).collect())
  1027. }
  1028. async fn get_proof_ys_by_quote_id(&self, quote_id: &Uuid) -> Result<Vec<PublicKey>, Self::Err> {
  1029. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1030. let rec = sqlx::query(
  1031. r#"
  1032. SELECT *
  1033. FROM proof
  1034. WHERE quote_id=?;
  1035. "#,
  1036. )
  1037. .bind(quote_id.as_hyphenated())
  1038. .fetch_all(&mut *transaction)
  1039. .await;
  1040. let ys = match rec {
  1041. Ok(rec) => {
  1042. transaction.commit().await.map_err(Error::from)?;
  1043. let proofs = rec
  1044. .into_iter()
  1045. .map(sqlite_row_to_proof)
  1046. .collect::<Result<Vec<Proof>, _>>()?;
  1047. proofs.ys()?
  1048. }
  1049. Err(err) => match err {
  1050. sqlx::Error::RowNotFound => {
  1051. transaction.commit().await.map_err(Error::from)?;
  1052. vec![]
  1053. }
  1054. _ => {
  1055. if let Err(err) = transaction.rollback().await {
  1056. tracing::error!("Could not rollback sql transaction: {}", err);
  1057. }
  1058. return Err(Error::SQLX(err).into());
  1059. }
  1060. },
  1061. };
  1062. Ok(ys)
  1063. }
  1064. async fn get_proofs_states(&self, ys: &[PublicKey]) -> Result<Vec<Option<State>>, Self::Err> {
  1065. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1066. let sql = format!(
  1067. "SELECT y, state FROM proof WHERE y IN ({})",
  1068. "?,".repeat(ys.len()).trim_end_matches(',')
  1069. );
  1070. let mut current_states = ys
  1071. .iter()
  1072. .fold(sqlx::query(&sql), |query, y| {
  1073. query.bind(y.to_bytes().to_vec())
  1074. })
  1075. .fetch_all(&mut *transaction)
  1076. .await
  1077. .map_err(|err| {
  1078. tracing::error!("SQLite could not get state of proof: {err:?}");
  1079. Error::SQLX(err)
  1080. })?
  1081. .into_iter()
  1082. .map(|row| {
  1083. PublicKey::from_slice(row.get("y"))
  1084. .map_err(Error::from)
  1085. .and_then(|y| {
  1086. let state: String = row.get("state");
  1087. State::from_str(&state)
  1088. .map_err(Error::from)
  1089. .map(|state| (y, state))
  1090. })
  1091. })
  1092. .collect::<Result<HashMap<_, _>, _>>()?;
  1093. Ok(ys.iter().map(|y| current_states.remove(y)).collect())
  1094. }
  1095. async fn get_proofs_by_keyset_id(
  1096. &self,
  1097. keyset_id: &Id,
  1098. ) -> Result<(Proofs, Vec<Option<State>>), Self::Err> {
  1099. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1100. let rec = sqlx::query(
  1101. r#"
  1102. SELECT *
  1103. FROM proof
  1104. WHERE keyset_id=?;
  1105. "#,
  1106. )
  1107. .bind(keyset_id.to_string())
  1108. .fetch_all(&mut *transaction)
  1109. .await;
  1110. match rec {
  1111. Ok(rec) => {
  1112. transaction.commit().await.map_err(Error::from)?;
  1113. let mut proofs_for_id = vec![];
  1114. let mut states = vec![];
  1115. for row in rec {
  1116. let (proof, state) = sqlite_row_to_proof_with_state(row)?;
  1117. proofs_for_id.push(proof);
  1118. states.push(state);
  1119. }
  1120. Ok((proofs_for_id, states))
  1121. }
  1122. Err(err) => {
  1123. tracing::error!("SQLite could not get proofs by keysets id");
  1124. if let Err(err) = transaction.rollback().await {
  1125. tracing::error!("Could not rollback sql transaction: {}", err);
  1126. }
  1127. return Err(Error::from(err).into());
  1128. }
  1129. }
  1130. }
  1131. async fn update_proofs_states(
  1132. &self,
  1133. ys: &[PublicKey],
  1134. proofs_state: State,
  1135. ) -> Result<Vec<Option<State>>, Self::Err> {
  1136. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1137. let sql = format!(
  1138. "SELECT y, state FROM proof WHERE y IN ({})",
  1139. "?,".repeat(ys.len()).trim_end_matches(',')
  1140. );
  1141. let rows = ys
  1142. .iter()
  1143. .fold(sqlx::query(&sql), |query, y| {
  1144. query.bind(y.to_bytes().to_vec())
  1145. })
  1146. .fetch_all(&mut *transaction)
  1147. .await
  1148. .map_err(|err| {
  1149. tracing::error!("SQLite could not get state of proof: {err:?}");
  1150. Error::SQLX(err)
  1151. })?;
  1152. // Check if all proofs exist
  1153. if rows.len() != ys.len() {
  1154. transaction.rollback().await.map_err(Error::from)?;
  1155. tracing::warn!("Attempted to update state of non-existent proof");
  1156. return Err(database::Error::ProofNotFound);
  1157. }
  1158. let mut current_states = rows
  1159. .into_iter()
  1160. .map(|row| {
  1161. PublicKey::from_slice(row.get("y"))
  1162. .map_err(Error::from)
  1163. .and_then(|y| {
  1164. let state: String = row.get("state");
  1165. State::from_str(&state)
  1166. .map_err(Error::from)
  1167. .map(|state| (y, state))
  1168. })
  1169. })
  1170. .collect::<Result<HashMap<_, _>, _>>()?;
  1171. let states = current_states.values().collect::<HashSet<_>>();
  1172. for state in states {
  1173. check_state_transition(*state, proofs_state)?;
  1174. }
  1175. // If no proofs are spent, proceed with update
  1176. let update_sql = format!(
  1177. "UPDATE proof SET state = ? WHERE y IN ({})",
  1178. "?,".repeat(ys.len()).trim_end_matches(',')
  1179. );
  1180. ys.iter()
  1181. .fold(
  1182. sqlx::query(&update_sql).bind(proofs_state.to_string()),
  1183. |query, y| query.bind(y.to_bytes().to_vec()),
  1184. )
  1185. .execute(&mut *transaction)
  1186. .await
  1187. .map_err(|err| {
  1188. tracing::error!("SQLite could not update proof state: {err:?}");
  1189. Error::SQLX(err)
  1190. })?;
  1191. transaction.commit().await.map_err(Error::from)?;
  1192. Ok(ys.iter().map(|y| current_states.remove(y)).collect())
  1193. }
  1194. }
  1195. #[async_trait]
  1196. impl MintSignaturesDatabase for MintSqliteDatabase {
  1197. type Err = database::Error;
  1198. async fn add_blind_signatures(
  1199. &self,
  1200. blinded_messages: &[PublicKey],
  1201. blinded_signatures: &[BlindSignature],
  1202. quote_id: Option<Uuid>,
  1203. ) -> Result<(), Self::Err> {
  1204. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1205. let current_time = unix_time();
  1206. for (message, signature) in blinded_messages.iter().zip(blinded_signatures) {
  1207. let res = sqlx::query(
  1208. r#"
  1209. INSERT INTO blind_signature
  1210. (y, amount, keyset_id, c, quote_id, dleq_e, dleq_s, created_time)
  1211. VALUES (?, ?, ?, ?, ?, ?, ?, ?);
  1212. "#,
  1213. )
  1214. .bind(message.to_bytes().to_vec())
  1215. .bind(u64::from(signature.amount) as i64)
  1216. .bind(signature.keyset_id.to_string())
  1217. .bind(signature.c.to_bytes().to_vec())
  1218. .bind(quote_id.map(|q| q.hyphenated()))
  1219. .bind(signature.dleq.as_ref().map(|dleq| dleq.e.to_secret_hex()))
  1220. .bind(signature.dleq.as_ref().map(|dleq| dleq.s.to_secret_hex()))
  1221. .bind(current_time as i64)
  1222. .execute(&mut *transaction)
  1223. .await;
  1224. if let Err(err) = res {
  1225. tracing::error!("SQLite could not add blind signature");
  1226. if let Err(err) = transaction.rollback().await {
  1227. tracing::error!("Could not rollback sql transaction: {}", err);
  1228. }
  1229. return Err(Error::SQLX(err).into());
  1230. }
  1231. }
  1232. transaction.commit().await.map_err(Error::from)?;
  1233. Ok(())
  1234. }
  1235. async fn get_blind_signatures(
  1236. &self,
  1237. blinded_messages: &[PublicKey],
  1238. ) -> Result<Vec<Option<BlindSignature>>, Self::Err> {
  1239. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1240. let sql = format!(
  1241. "SELECT * FROM blind_signature WHERE y IN ({})",
  1242. "?,".repeat(blinded_messages.len()).trim_end_matches(',')
  1243. );
  1244. let mut blinded_signatures = blinded_messages
  1245. .iter()
  1246. .fold(sqlx::query(&sql), |query, y| {
  1247. query.bind(y.to_bytes().to_vec())
  1248. })
  1249. .fetch_all(&mut *transaction)
  1250. .await
  1251. .map_err(|err| {
  1252. tracing::error!("SQLite could not get state of proof: {err:?}");
  1253. Error::SQLX(err)
  1254. })?
  1255. .into_iter()
  1256. .map(|row| {
  1257. PublicKey::from_slice(row.get("y"))
  1258. .map_err(Error::from)
  1259. .and_then(|y| sqlite_row_to_blind_signature(row).map(|blinded| (y, blinded)))
  1260. })
  1261. .collect::<Result<HashMap<_, _>, _>>()?;
  1262. Ok(blinded_messages
  1263. .iter()
  1264. .map(|y| blinded_signatures.remove(y))
  1265. .collect())
  1266. }
  1267. async fn get_blind_signatures_for_keyset(
  1268. &self,
  1269. keyset_id: &Id,
  1270. ) -> Result<Vec<BlindSignature>, Self::Err> {
  1271. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1272. let rec = sqlx::query(
  1273. r#"
  1274. SELECT *
  1275. FROM blind_signature
  1276. WHERE keyset_id=?;
  1277. "#,
  1278. )
  1279. .bind(keyset_id.to_string())
  1280. .fetch_all(&mut *transaction)
  1281. .await;
  1282. match rec {
  1283. Ok(rec) => {
  1284. transaction.commit().await.map_err(Error::from)?;
  1285. let sigs = rec
  1286. .into_iter()
  1287. .map(sqlite_row_to_blind_signature)
  1288. .collect::<Result<Vec<BlindSignature>, _>>()?;
  1289. Ok(sigs)
  1290. }
  1291. Err(err) => {
  1292. tracing::error!("SQLite could not get vlinf signatures for keyset");
  1293. if let Err(err) = transaction.rollback().await {
  1294. tracing::error!("Could not rollback sql transaction: {}", err);
  1295. }
  1296. return Err(Error::from(err).into());
  1297. }
  1298. }
  1299. }
  1300. /// Get [`BlindSignature`]s for quote
  1301. async fn get_blind_signatures_for_quote(
  1302. &self,
  1303. quote_id: &Uuid,
  1304. ) -> Result<Vec<BlindSignature>, Self::Err> {
  1305. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1306. let recs = sqlx::query(
  1307. r#"
  1308. SELECT *
  1309. FROM blind_signature
  1310. WHERE quote_id=?;
  1311. "#,
  1312. )
  1313. .bind(quote_id.as_hyphenated())
  1314. .fetch_all(&mut *transaction)
  1315. .await;
  1316. match recs {
  1317. Ok(recs) => {
  1318. transaction.commit().await.map_err(Error::from)?;
  1319. let keysets = recs
  1320. .into_iter()
  1321. .map(sqlite_row_to_blind_signature)
  1322. .collect::<Result<Vec<_>, _>>()?;
  1323. Ok(keysets)
  1324. }
  1325. Err(err) => {
  1326. tracing::error!("SQLite could not get active keyset");
  1327. if let Err(err) = transaction.rollback().await {
  1328. tracing::error!("Could not rollback sql transaction: {}", err);
  1329. }
  1330. Err(Error::from(err).into())
  1331. }
  1332. }
  1333. }
  1334. }
  1335. #[async_trait]
  1336. impl MintDatabase<database::Error> for MintSqliteDatabase {
  1337. async fn set_mint_info(&self, mint_info: MintInfo) -> Result<(), database::Error> {
  1338. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1339. let res = sqlx::query(
  1340. r#"
  1341. INSERT INTO config
  1342. (id, value)
  1343. VALUES (?, ?)
  1344. ON CONFLICT(id) DO UPDATE SET
  1345. value = excluded.value
  1346. ;
  1347. "#,
  1348. )
  1349. .bind("mint_info")
  1350. .bind(serde_json::to_string(&mint_info)?)
  1351. .execute(&mut *transaction)
  1352. .await;
  1353. match res {
  1354. Ok(_) => {
  1355. transaction.commit().await.map_err(Error::from)?;
  1356. Ok(())
  1357. }
  1358. Err(err) => {
  1359. tracing::error!("SQLite Could not update mint info");
  1360. if let Err(err) = transaction.rollback().await {
  1361. tracing::error!("Could not rollback sql transaction: {}", err);
  1362. }
  1363. Err(Error::from(err).into())
  1364. }
  1365. }
  1366. }
  1367. async fn get_mint_info(&self) -> Result<MintInfo, database::Error> {
  1368. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1369. let rec = sqlx::query(
  1370. r#"
  1371. SELECT *
  1372. FROM config
  1373. WHERE id=?;
  1374. "#,
  1375. )
  1376. .bind("mint_info")
  1377. .fetch_one(&mut *transaction)
  1378. .await;
  1379. match rec {
  1380. Ok(rec) => {
  1381. transaction.commit().await.map_err(Error::from)?;
  1382. let value: String = rec.try_get("value").map_err(Error::from)?;
  1383. let mint_info = serde_json::from_str(&value)?;
  1384. Ok(mint_info)
  1385. }
  1386. Err(err) => match err {
  1387. sqlx::Error::RowNotFound => {
  1388. transaction.commit().await.map_err(Error::from)?;
  1389. return Err(Error::UnknownMintInfo.into());
  1390. }
  1391. _ => {
  1392. return {
  1393. if let Err(err) = transaction.rollback().await {
  1394. tracing::error!("Could not rollback sql transaction: {}", err);
  1395. }
  1396. Err(Error::SQLX(err).into())
  1397. }
  1398. }
  1399. },
  1400. }
  1401. }
  1402. async fn set_quote_ttl(&self, quote_ttl: QuoteTTL) -> Result<(), database::Error> {
  1403. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1404. let res = sqlx::query(
  1405. r#"
  1406. INSERT INTO config
  1407. (id, value)
  1408. VALUES (?, ?)
  1409. ON CONFLICT(id) DO UPDATE SET
  1410. value = excluded.value
  1411. ;
  1412. "#,
  1413. )
  1414. .bind("quote_ttl")
  1415. .bind(serde_json::to_string(&quote_ttl)?)
  1416. .execute(&mut *transaction)
  1417. .await;
  1418. match res {
  1419. Ok(_) => {
  1420. transaction.commit().await.map_err(Error::from)?;
  1421. Ok(())
  1422. }
  1423. Err(err) => {
  1424. tracing::error!("SQLite Could not update mint info");
  1425. if let Err(err) = transaction.rollback().await {
  1426. tracing::error!("Could not rollback sql transaction: {}", err);
  1427. }
  1428. Err(Error::from(err).into())
  1429. }
  1430. }
  1431. }
  1432. async fn get_quote_ttl(&self) -> Result<QuoteTTL, database::Error> {
  1433. let mut transaction = self.pool.begin().await.map_err(Error::from)?;
  1434. let rec = sqlx::query(
  1435. r#"
  1436. SELECT *
  1437. FROM config
  1438. WHERE id=?;
  1439. "#,
  1440. )
  1441. .bind("quote_ttl")
  1442. .fetch_one(&mut *transaction)
  1443. .await;
  1444. match rec {
  1445. Ok(rec) => {
  1446. transaction.commit().await.map_err(Error::from)?;
  1447. let value: String = rec.try_get("value").map_err(Error::from)?;
  1448. let quote_ttl = serde_json::from_str(&value)?;
  1449. Ok(quote_ttl)
  1450. }
  1451. Err(err) => match err {
  1452. sqlx::Error::RowNotFound => {
  1453. transaction.commit().await.map_err(Error::from)?;
  1454. return Err(Error::UnknownQuoteTTL.into());
  1455. }
  1456. _ => {
  1457. return {
  1458. if let Err(err) = transaction.rollback().await {
  1459. tracing::error!("Could not rollback sql transaction: {}", err);
  1460. }
  1461. Err(Error::SQLX(err).into())
  1462. }
  1463. }
  1464. },
  1465. }
  1466. }
  1467. }
  1468. fn sqlite_row_to_keyset_info(row: SqliteRow) -> Result<MintKeySetInfo, Error> {
  1469. let row_id: String = row.try_get("id").map_err(Error::from)?;
  1470. let row_unit: String = row.try_get("unit").map_err(Error::from)?;
  1471. let row_active: bool = row.try_get("active").map_err(Error::from)?;
  1472. let row_valid_from: i64 = row.try_get("valid_from").map_err(Error::from)?;
  1473. let row_valid_to: Option<i64> = row.try_get("valid_to").map_err(Error::from)?;
  1474. let row_derivation_path: String = row.try_get("derivation_path").map_err(Error::from)?;
  1475. let row_max_order: u8 = row.try_get("max_order").map_err(Error::from)?;
  1476. let row_keyset_ppk: Option<i64> = row.try_get("input_fee_ppk").ok();
  1477. let row_derivation_path_index: Option<i64> =
  1478. row.try_get("derivation_path_index").map_err(Error::from)?;
  1479. Ok(MintKeySetInfo {
  1480. id: Id::from_str(&row_id).map_err(Error::from)?,
  1481. unit: CurrencyUnit::from_str(&row_unit).map_err(Error::from)?,
  1482. active: row_active,
  1483. valid_from: row_valid_from as u64,
  1484. valid_to: row_valid_to.map(|v| v as u64),
  1485. derivation_path: DerivationPath::from_str(&row_derivation_path).map_err(Error::from)?,
  1486. derivation_path_index: row_derivation_path_index.map(|d| d as u32),
  1487. max_order: row_max_order,
  1488. input_fee_ppk: row_keyset_ppk.unwrap_or(0) as u64,
  1489. })
  1490. }
  1491. fn sqlite_row_to_mint_quote(row: SqliteRow) -> Result<MintQuote, Error> {
  1492. let row_id: Hyphenated = row.try_get("id").map_err(Error::from)?;
  1493. let row_amount: i64 = row.try_get("amount").map_err(Error::from)?;
  1494. let row_unit: String = row.try_get("unit").map_err(Error::from)?;
  1495. let row_request: String = row.try_get("request").map_err(Error::from)?;
  1496. let row_state: String = row.try_get("state").map_err(Error::from)?;
  1497. let row_expiry: i64 = row.try_get("expiry").map_err(Error::from)?;
  1498. let row_request_lookup_id: Option<String> =
  1499. row.try_get("request_lookup_id").map_err(Error::from)?;
  1500. let row_pubkey: Option<String> = row.try_get("pubkey").map_err(Error::from)?;
  1501. let row_created_time: i64 = row.try_get("created_time").map_err(Error::from)?;
  1502. let row_paid_time: Option<i64> = row.try_get("paid_time").map_err(Error::from)?;
  1503. let row_issued_time: Option<i64> = row.try_get("issued_time").map_err(Error::from)?;
  1504. let request_lookup_id = match row_request_lookup_id {
  1505. Some(id) => id,
  1506. None => match Bolt11Invoice::from_str(&row_request) {
  1507. Ok(invoice) => invoice.payment_hash().to_string(),
  1508. Err(_) => row_request.clone(),
  1509. },
  1510. };
  1511. let pubkey = row_pubkey
  1512. .map(|key| PublicKey::from_str(&key))
  1513. .transpose()?;
  1514. Ok(MintQuote {
  1515. id: row_id.into_uuid(),
  1516. amount: Amount::from(row_amount as u64),
  1517. unit: CurrencyUnit::from_str(&row_unit).map_err(Error::from)?,
  1518. request: row_request,
  1519. state: MintQuoteState::from_str(&row_state).map_err(Error::from)?,
  1520. expiry: row_expiry as u64,
  1521. request_lookup_id,
  1522. pubkey,
  1523. created_time: row_created_time as u64,
  1524. paid_time: row_paid_time.map(|p| p as u64),
  1525. issued_time: row_issued_time.map(|p| p as u64),
  1526. })
  1527. }
  1528. fn sqlite_row_to_melt_quote(row: SqliteRow) -> Result<mint::MeltQuote, Error> {
  1529. let row_id: Hyphenated = row.try_get("id").map_err(Error::from)?;
  1530. let row_unit: String = row.try_get("unit").map_err(Error::from)?;
  1531. let row_amount: i64 = row.try_get("amount").map_err(Error::from)?;
  1532. let row_request: String = row.try_get("request").map_err(Error::from)?;
  1533. let row_fee_reserve: i64 = row.try_get("fee_reserve").map_err(Error::from)?;
  1534. let row_state: String = row.try_get("state").map_err(Error::from)?;
  1535. let row_expiry: i64 = row.try_get("expiry").map_err(Error::from)?;
  1536. let row_preimage: Option<String> = row.try_get("payment_preimage").map_err(Error::from)?;
  1537. let row_request_lookup: Option<String> =
  1538. row.try_get("request_lookup_id").map_err(Error::from)?;
  1539. let request_lookup_id = row_request_lookup.unwrap_or(row_request.clone());
  1540. let row_msat_to_pay: Option<i64> = row.try_get("msat_to_pay").map_err(Error::from)?;
  1541. let row_created_time: i64 = row.try_get("created_time").map_err(Error::from)?;
  1542. let row_paid_time: Option<i64> = row.try_get("paid_time").map_err(Error::from)?;
  1543. Ok(mint::MeltQuote {
  1544. id: row_id.into_uuid(),
  1545. amount: Amount::from(row_amount as u64),
  1546. unit: CurrencyUnit::from_str(&row_unit).map_err(Error::from)?,
  1547. request: row_request,
  1548. fee_reserve: Amount::from(row_fee_reserve as u64),
  1549. state: QuoteState::from_str(&row_state)?,
  1550. expiry: row_expiry as u64,
  1551. payment_preimage: row_preimage,
  1552. request_lookup_id,
  1553. msat_to_pay: row_msat_to_pay.map(|a| Amount::from(a as u64)),
  1554. created_time: row_created_time as u64,
  1555. paid_time: row_paid_time.map(|p| p as u64),
  1556. })
  1557. }
  1558. fn sqlite_row_to_proof(row: SqliteRow) -> Result<Proof, Error> {
  1559. let row_amount: i64 = row.try_get("amount").map_err(Error::from)?;
  1560. let keyset_id: String = row.try_get("keyset_id").map_err(Error::from)?;
  1561. let row_secret: String = row.try_get("secret").map_err(Error::from)?;
  1562. let row_c: Vec<u8> = row.try_get("c").map_err(Error::from)?;
  1563. let row_witness: Option<String> = row.try_get("witness").map_err(Error::from)?;
  1564. Ok(Proof {
  1565. amount: Amount::from(row_amount as u64),
  1566. keyset_id: Id::from_str(&keyset_id)?,
  1567. secret: Secret::from_str(&row_secret)?,
  1568. c: PublicKey::from_slice(&row_c)?,
  1569. witness: row_witness.and_then(|w| serde_json::from_str(&w).ok()),
  1570. dleq: None,
  1571. })
  1572. }
  1573. fn sqlite_row_to_proof_with_state(row: SqliteRow) -> Result<(Proof, Option<State>), Error> {
  1574. let row_amount: i64 = row.try_get("amount").map_err(Error::from)?;
  1575. let keyset_id: String = row.try_get("keyset_id").map_err(Error::from)?;
  1576. let row_secret: String = row.try_get("secret").map_err(Error::from)?;
  1577. let row_c: Vec<u8> = row.try_get("c").map_err(Error::from)?;
  1578. let row_witness: Option<String> = row.try_get("witness").map_err(Error::from)?;
  1579. let row_state: Option<String> = row.try_get("state").map_err(Error::from)?;
  1580. let state = row_state.and_then(|s| State::from_str(&s).ok());
  1581. Ok((
  1582. Proof {
  1583. amount: Amount::from(row_amount as u64),
  1584. keyset_id: Id::from_str(&keyset_id)?,
  1585. secret: Secret::from_str(&row_secret)?,
  1586. c: PublicKey::from_slice(&row_c)?,
  1587. witness: row_witness.and_then(|w| serde_json::from_str(&w).ok()),
  1588. dleq: None,
  1589. },
  1590. state,
  1591. ))
  1592. }
  1593. fn sqlite_row_to_blind_signature(row: SqliteRow) -> Result<BlindSignature, Error> {
  1594. let row_amount: i64 = row.try_get("amount").map_err(Error::from)?;
  1595. let keyset_id: String = row.try_get("keyset_id").map_err(Error::from)?;
  1596. let row_c: Vec<u8> = row.try_get("c").map_err(Error::from)?;
  1597. let row_dleq_e: Option<String> = row.try_get("dleq_e").map_err(Error::from)?;
  1598. let row_dleq_s: Option<String> = row.try_get("dleq_s").map_err(Error::from)?;
  1599. let dleq = match (row_dleq_e, row_dleq_s) {
  1600. (Some(e), Some(s)) => Some(BlindSignatureDleq {
  1601. e: SecretKey::from_hex(e)?,
  1602. s: SecretKey::from_hex(s)?,
  1603. }),
  1604. _ => None,
  1605. };
  1606. Ok(BlindSignature {
  1607. amount: Amount::from(row_amount as u64),
  1608. keyset_id: Id::from_str(&keyset_id)?,
  1609. c: PublicKey::from_slice(&row_c)?,
  1610. dleq,
  1611. })
  1612. }
  1613. fn sqlite_row_to_melt_request(
  1614. row: SqliteRow,
  1615. ) -> Result<(MeltBolt11Request<Uuid>, PaymentProcessorKey), Error> {
  1616. let quote_id: Hyphenated = row.try_get("id").map_err(Error::from)?;
  1617. let row_inputs: String = row.try_get("inputs").map_err(Error::from)?;
  1618. let row_outputs: Option<String> = row.try_get("outputs").map_err(Error::from)?;
  1619. let row_method: String = row.try_get("method").map_err(Error::from)?;
  1620. let row_unit: String = row.try_get("unit").map_err(Error::from)?;
  1621. let melt_request = MeltBolt11Request::new(
  1622. quote_id.into_uuid(),
  1623. serde_json::from_str(&row_inputs)?,
  1624. row_outputs.and_then(|o| serde_json::from_str(&o).ok()),
  1625. );
  1626. let ln_key = PaymentProcessorKey {
  1627. unit: CurrencyUnit::from_str(&row_unit)?,
  1628. method: PaymentMethod::from_str(&row_method)?,
  1629. };
  1630. Ok((melt_request, ln_key))
  1631. }
  1632. #[cfg(test)]
  1633. mod tests {
  1634. use cdk_common::mint::MintKeySetInfo;
  1635. use cdk_common::{mint_db_test, Amount};
  1636. use super::*;
  1637. #[tokio::test]
  1638. async fn test_remove_spent_proofs() {
  1639. let db = memory::empty().await.unwrap();
  1640. // Create a keyset and add it to the database
  1641. let keyset_id = Id::from_str("00916bbf7ef91a36").unwrap();
  1642. let keyset_info = MintKeySetInfo {
  1643. id: keyset_id,
  1644. unit: CurrencyUnit::Sat,
  1645. active: true,
  1646. valid_from: 0,
  1647. valid_to: None,
  1648. derivation_path: bitcoin::bip32::DerivationPath::from_str("m/0'/0'/0'").unwrap(),
  1649. derivation_path_index: Some(0),
  1650. max_order: 32,
  1651. input_fee_ppk: 0,
  1652. };
  1653. db.add_keyset_info(keyset_info).await.unwrap();
  1654. let proofs = vec![
  1655. Proof {
  1656. amount: Amount::from(100),
  1657. keyset_id,
  1658. secret: Secret::generate(),
  1659. c: SecretKey::generate().public_key(),
  1660. witness: None,
  1661. dleq: None,
  1662. },
  1663. Proof {
  1664. amount: Amount::from(200),
  1665. keyset_id,
  1666. secret: Secret::generate(),
  1667. c: SecretKey::generate().public_key(),
  1668. witness: None,
  1669. dleq: None,
  1670. },
  1671. ];
  1672. // Add proofs to database
  1673. db.add_proofs(proofs.clone(), None).await.unwrap();
  1674. // Mark one proof as spent
  1675. db.update_proofs_states(&[proofs[0].y().unwrap()], State::Spent)
  1676. .await
  1677. .unwrap();
  1678. // Try to remove both proofs - should fail because one is spent
  1679. let result = db
  1680. .remove_proofs(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()], None)
  1681. .await;
  1682. assert!(result.is_err());
  1683. assert!(matches!(
  1684. result.unwrap_err(),
  1685. database::Error::AttemptRemoveSpentProof
  1686. ));
  1687. // Verify both proofs still exist
  1688. let states = db
  1689. .get_proofs_states(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()])
  1690. .await
  1691. .unwrap();
  1692. assert_eq!(states.len(), 2);
  1693. assert_eq!(states[0], Some(State::Spent));
  1694. assert_eq!(states[1], Some(State::Unspent));
  1695. }
  1696. #[tokio::test]
  1697. async fn test_update_spent_proofs() {
  1698. let db = memory::empty().await.unwrap();
  1699. // Create a keyset and add it to the database
  1700. let keyset_id = Id::from_str("00916bbf7ef91a36").unwrap();
  1701. let keyset_info = MintKeySetInfo {
  1702. id: keyset_id,
  1703. unit: CurrencyUnit::Sat,
  1704. active: true,
  1705. valid_from: 0,
  1706. valid_to: None,
  1707. derivation_path: bitcoin::bip32::DerivationPath::from_str("m/0'/0'/0'").unwrap(),
  1708. derivation_path_index: Some(0),
  1709. max_order: 32,
  1710. input_fee_ppk: 0,
  1711. };
  1712. db.add_keyset_info(keyset_info).await.unwrap();
  1713. let proofs = vec![
  1714. Proof {
  1715. amount: Amount::from(100),
  1716. keyset_id,
  1717. secret: Secret::generate(),
  1718. c: SecretKey::generate().public_key(),
  1719. witness: None,
  1720. dleq: None,
  1721. },
  1722. Proof {
  1723. amount: Amount::from(200),
  1724. keyset_id,
  1725. secret: Secret::generate(),
  1726. c: SecretKey::generate().public_key(),
  1727. witness: None,
  1728. dleq: None,
  1729. },
  1730. ];
  1731. // Add proofs to database
  1732. db.add_proofs(proofs.clone(), None).await.unwrap();
  1733. // Mark one proof as spent
  1734. db.update_proofs_states(&[proofs[0].y().unwrap()], State::Spent)
  1735. .await
  1736. .unwrap();
  1737. // Try to update both proofs - should fail because one is spent
  1738. let result = db
  1739. .update_proofs_states(&[proofs[0].y().unwrap()], State::Unspent)
  1740. .await;
  1741. assert!(result.is_err());
  1742. assert!(matches!(
  1743. result.unwrap_err(),
  1744. database::Error::AttemptUpdateSpentProof
  1745. ));
  1746. // Verify states haven't changed
  1747. let states = db
  1748. .get_proofs_states(&[proofs[0].y().unwrap(), proofs[1].y().unwrap()])
  1749. .await
  1750. .unwrap();
  1751. assert_eq!(states.len(), 2);
  1752. assert_eq!(states[0], Some(State::Spent));
  1753. assert_eq!(states[1], Some(State::Unspent));
  1754. }
  1755. async fn provide_db() -> MintSqliteDatabase {
  1756. memory::empty().await.unwrap()
  1757. }
  1758. mint_db_test!(provide_db);
  1759. }