123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800 |
- //! Rexie Browser Database
- use std::collections::{HashMap, HashSet};
- use std::rc::Rc;
- use std::result::Result;
- use async_trait::async_trait;
- use cdk::cdk_database::{self, WalletDatabase};
- use cdk::mint_url::MintUrl;
- use cdk::nuts::{
- CurrencyUnit, Id, KeySetInfo, Keys, MintInfo, PublicKey, SpendingConditions, State,
- };
- use cdk::types::ProofInfo;
- use cdk::util::unix_time;
- use cdk::wallet::{MeltQuote, MintQuote};
- use rexie::*;
- use thiserror::Error;
- use tokio::sync::Mutex;
- // Tables
- const MINTS: &str = "mints";
- const MINT_KEYSETS: &str = "keysets_by_mint";
- const KEYSETS: &str = "keysets";
- const MINT_KEYS: &str = "mint_keys";
- const MINT_QUOTES: &str = "mint_quotes";
- const MELT_QUOTES: &str = "melt_quotes";
- const PROOFS: &str = "proofs";
- const CONFIG: &str = "config";
- const KEYSET_COUNTER: &str = "keyset_counter";
- const NOSTR_LAST_CHECKED: &str = "nostr_last_check";
- const DATABASE_VERSION: u32 = 3;
- /// Rexie Database Error
- #[derive(Debug, Error)]
- pub enum Error {
- /// CDK Database Error
- #[error(transparent)]
- CDKDatabase(#[from] cdk::cdk_database::Error),
- /// Rexie Error
- #[error(transparent)]
- Rexie(#[from] rexie::Error),
- /// Serde Wasm Error
- #[error(transparent)]
- SerdeBindgen(#[from] serde_wasm_bindgen::Error),
- /// NUT00 Error
- #[error(transparent)]
- NUT00(cdk::nuts::nut00::Error),
- #[error("Not found")]
- /// Not Found
- NotFound,
- }
- impl From<Error> for cdk::cdk_database::Error {
- fn from(e: Error) -> Self {
- Self::Database(Box::new(e))
- }
- }
- // These are okay because we never actually send across threads in the browser
- unsafe impl Send for Error {}
- unsafe impl Sync for Error {}
- /// Wallet Rexie Database
- #[derive(Debug, Clone)]
- pub struct WalletRexieDatabase {
- db: Rc<Mutex<Rexie>>,
- }
- // These are okay because we never actually send across threads in the browser
- unsafe impl Send for WalletRexieDatabase {}
- unsafe impl Sync for WalletRexieDatabase {}
- impl WalletRexieDatabase {
- /// Create new [`WalletRexieDatabase`]
- pub async fn new() -> Result<Self, Error> {
- let rexie = Rexie::builder("cdk")
- .version(DATABASE_VERSION)
- .add_object_store(
- ObjectStore::new(PROOFS)
- .add_index(Index::new("y", "y").unique(true))
- .add_index(Index::new("mint_url", "mint_url"))
- .add_index(Index::new("state", "state"))
- .add_index(Index::new("unit", "unit")),
- )
- .add_object_store(
- ObjectStore::new(MINTS).add_index(Index::new("mint_url", "mint_url").unique(true)),
- )
- .add_object_store(ObjectStore::new(MINT_KEYSETS))
- .add_object_store(
- ObjectStore::new(KEYSETS)
- .add_index(Index::new("keyset_id", "keyset_id").unique(true)),
- )
- .add_object_store(
- ObjectStore::new(MINT_KEYS)
- .add_index(Index::new("keyset_id", "keyset_id").unique(true)),
- )
- .add_object_store(ObjectStore::new(MINT_QUOTES))
- .add_object_store(ObjectStore::new(MELT_QUOTES))
- .add_object_store(ObjectStore::new(CONFIG))
- .add_object_store(ObjectStore::new(KEYSET_COUNTER))
- .add_object_store(ObjectStore::new(NOSTR_LAST_CHECKED))
- // Build the database
- .build()
- .await
- .unwrap();
- Ok(Self {
- db: Rc::new(Mutex::new(rexie)),
- })
- }
- async fn set_proof_states(
- &self,
- ys: Vec<PublicKey>,
- state: State,
- ) -> Result<(), cdk_database::Error> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[PROOFS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let proofs_store = transaction.store(PROOFS).map_err(Error::from)?;
- for y in ys {
- let y = serde_wasm_bindgen::to_value(&y).map_err(Error::from)?;
- let mut proof: ProofInfo = proofs_store
- .get(y.clone())
- .await
- .map_err(Error::from)?
- .and_then(|p| serde_wasm_bindgen::from_value(p).ok())
- .ok_or(Error::NotFound)?;
- proof.state = state;
- let proof = serde_wasm_bindgen::to_value(&proof).map_err(Error::from)?;
- proofs_store
- .put(&proof, Some(&y))
- .await
- .map_err(Error::from)?;
- }
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- }
- #[cfg_attr(target_arch = "wasm32", async_trait(?Send))]
- #[cfg_attr(not(target_arch = "wasm32"), async_trait)]
- impl WalletDatabase for WalletRexieDatabase {
- type Err = cdk::cdk_database::Error;
- async fn add_mint(
- &self,
- mint_url: MintUrl,
- mint_info: Option<MintInfo>,
- ) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINTS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let mints_store = transaction.store(MINTS).map_err(Error::from)?;
- let mint_url = serde_wasm_bindgen::to_value(&mint_url).map_err(Error::from)?;
- let mint_info = serde_wasm_bindgen::to_value(&mint_info).map_err(Error::from)?;
- mints_store
- .put(&mint_info, Some(&mint_url))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn remove_mint(&self, mint_url: MintUrl) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINTS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let mints_store = transaction.store(MINTS).map_err(Error::from)?;
- let mint_url = serde_wasm_bindgen::to_value(&mint_url).map_err(Error::from)?;
- mints_store.delete(mint_url).await.map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_mint(&self, mint_url: MintUrl) -> Result<Option<MintInfo>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINTS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let mints_store = transaction.store(MINTS).map_err(Error::from)?;
- let mint_url = serde_wasm_bindgen::to_value(&mint_url).map_err(Error::from)?;
- let mint_info = mints_store
- .get(mint_url)
- .await
- .map_err(Error::from)?
- .and_then(|m| serde_wasm_bindgen::from_value(m).ok());
- Ok(mint_info)
- }
- async fn get_mints(&self) -> Result<HashMap<MintUrl, Option<MintInfo>>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINTS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let mints_store = transaction.store(MINTS).map_err(Error::from)?;
- let mints = mints_store
- .scan(None, None, None, None)
- .await
- .map_err(Error::from)?;
- let mints: HashMap<MintUrl, Option<MintInfo>> = mints
- .into_iter()
- .map(|(url, info)| {
- (
- serde_wasm_bindgen::from_value(url).unwrap(),
- serde_wasm_bindgen::from_value(info).unwrap(),
- )
- })
- .collect();
- Ok(mints)
- }
- async fn update_mint_url(
- &self,
- old_mint_url: MintUrl,
- new_mint_url: MintUrl,
- ) -> Result<(), Self::Err> {
- let proofs = self
- .get_proofs(Some(old_mint_url), None, None, None)
- .await
- .map_err(Error::from)?;
- let updated_proofs: Vec<ProofInfo> = proofs
- .into_iter()
- .map(|mut p| {
- p.mint_url = new_mint_url.clone();
- p
- })
- .collect();
- if !updated_proofs.is_empty() {
- self.update_proofs(updated_proofs, vec![]).await?;
- }
- // Update mint quotes
- {
- let quotes = self.get_mint_quotes().await?;
- let unix_time = unix_time();
- let quotes: Vec<MintQuote> = quotes
- .into_iter()
- .filter_map(|mut q| {
- if q.expiry < unix_time {
- q.mint_url = new_mint_url.clone();
- Some(q)
- } else {
- None
- }
- })
- .collect();
- for quote in quotes {
- self.add_mint_quote(quote).await?;
- }
- }
- Ok(())
- }
- async fn add_mint_keysets(
- &self,
- mint_url: MintUrl,
- keysets: Vec<KeySetInfo>,
- ) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_KEYSETS, KEYSETS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let mint_keysets_store = transaction.store(MINT_KEYSETS).map_err(Error::from)?;
- let keysets_store = transaction.store(KEYSETS).map_err(Error::from)?;
- let mint_url = serde_wasm_bindgen::to_value(&mint_url).map_err(Error::from)?;
- let mut mint_keysets = mint_keysets_store
- .get(mint_url.clone())
- .await
- .map_err(Error::from)?
- .and_then(|m| serde_wasm_bindgen::from_value(m).ok());
- let new_keyset_ids: Vec<Id> = keysets.iter().map(|k| k.id).collect();
- mint_keysets
- .as_mut()
- .unwrap_or(&mut HashSet::new())
- .extend(new_keyset_ids);
- let mint_keysets = serde_wasm_bindgen::to_value(&mint_keysets).map_err(Error::from)?;
- mint_keysets_store
- .put(&mint_keysets, Some(&mint_url))
- .await
- .map_err(Error::from)?;
- for keyset in keysets {
- let id = serde_wasm_bindgen::to_value(&keyset.id).map_err(Error::from)?;
- let keyset = serde_wasm_bindgen::to_value(&keyset).map_err(Error::from)?;
- keysets_store
- .put(&keyset, Some(&id))
- .await
- .map_err(Error::from)?;
- }
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_mint_keysets(
- &self,
- mint_url: MintUrl,
- ) -> Result<Option<Vec<KeySetInfo>>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_KEYSETS, KEYSETS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let mints_store = transaction.store(MINT_KEYSETS).map_err(Error::from)?;
- let mint_url = serde_wasm_bindgen::to_value(&mint_url).map_err(Error::from)?;
- let mint_keysets: Option<HashSet<Id>> = mints_store
- .get(mint_url)
- .await
- .map_err(Error::from)?
- .and_then(|m| serde_wasm_bindgen::from_value(m).ok());
- let keysets_store = transaction.store(KEYSETS).map_err(Error::from)?;
- let keysets = match mint_keysets {
- Some(mint_keysets) => {
- let mut keysets = vec![];
- for mint_keyset in mint_keysets {
- let id = serde_wasm_bindgen::to_value(&mint_keyset).map_err(Error::from)?;
- let keyset = keysets_store
- .get(id)
- .await
- .map_err(Error::from)?
- .and_then(|k| serde_wasm_bindgen::from_value(k).ok());
- keysets.push(keyset);
- }
- let keysets = keysets.iter().flatten().cloned().collect();
- Some(keysets)
- }
- None => None,
- };
- transaction.done().await.map_err(Error::from)?;
- Ok(keysets)
- }
- async fn get_keyset_by_id(&self, keyset_id: &Id) -> Result<Option<KeySetInfo>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[KEYSETS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let keysets_store = transaction.store(KEYSETS).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(keyset_id).map_err(Error::from)?;
- let keyset = keysets_store
- .get(keyset_id)
- .await
- .map_err(Error::from)?
- .and_then(|k| serde_wasm_bindgen::from_value(k).ok());
- Ok(keyset)
- }
- async fn add_mint_quote(&self, quote: MintQuote) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_QUOTES], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MINT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e.id).map_err(Error::from)?;
- let quote = serde_wasm_bindgen::to_value("e).map_err(Error::from)?;
- quotes_store
- .put("e, Some("e_id))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_mint_quote(&self, quote_id: &str) -> Result<Option<MintQuote>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_QUOTES], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MINT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e_id).map_err(Error::from)?;
- let quote = quotes_store
- .get(quote_id)
- .await
- .map_err(Error::from)?
- .and_then(|q| serde_wasm_bindgen::from_value(q).ok());
- Ok(quote)
- }
- async fn get_mint_quotes(&self) -> Result<Vec<MintQuote>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_QUOTES], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MINT_QUOTES).map_err(Error::from)?;
- let quotes = quotes_store
- .scan(None, None, None, None)
- .await
- .map_err(Error::from)?;
- Ok(quotes
- .into_iter()
- .map(|(_id, q)| serde_wasm_bindgen::from_value(q))
- .collect::<Result<Vec<MintQuote>, serde_wasm_bindgen::Error>>()
- .map_err(<serde_wasm_bindgen::Error as Into<Error>>::into)?)
- }
- async fn remove_mint_quote(&self, quote_id: &str) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_QUOTES], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MINT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e_id).map_err(Error::from)?;
- quotes_store.delete(quote_id).await.map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn add_melt_quote(&self, quote: MeltQuote) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MELT_QUOTES], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MELT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e.id).map_err(Error::from)?;
- let quote = serde_wasm_bindgen::to_value("e).map_err(Error::from)?;
- quotes_store
- .put("e, Some("e_id))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_melt_quote(&self, quote_id: &str) -> Result<Option<MeltQuote>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MELT_QUOTES], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MELT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e_id).map_err(Error::from)?;
- let quote = quotes_store
- .get(quote_id)
- .await
- .map_err(Error::from)?
- .and_then(|q| serde_wasm_bindgen::from_value(q).ok());
- Ok(quote)
- }
- async fn remove_melt_quote(&self, quote_id: &str) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MELT_QUOTES], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let quotes_store = transaction.store(MELT_QUOTES).map_err(Error::from)?;
- let quote_id = serde_wasm_bindgen::to_value("e_id).map_err(Error::from)?;
- quotes_store.delete(quote_id).await.map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn add_keys(&self, keys: Keys) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_KEYS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let keys_store = transaction.store(MINT_KEYS).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(&Id::from(&keys)).map_err(Error::from)?;
- let keys = serde_wasm_bindgen::to_value(&keys).map_err(Error::from)?;
- keys_store
- .put(&keys, Some(&keyset_id))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_keys(&self, id: &Id) -> Result<Option<Keys>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_KEYS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let keys_store = transaction.store(MINT_KEYS).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(id).map_err(Error::from)?;
- let keys = keys_store
- .get(keyset_id)
- .await
- .map_err(Error::from)?
- .and_then(|k| serde_wasm_bindgen::from_value(k).ok());
- Ok(keys)
- }
- async fn remove_keys(&self, id: &Id) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[MINT_KEYS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let keys_store = transaction.store(MINT_KEYS).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(id).map_err(Error::from)?;
- keys_store.delete(keyset_id).await.map_err(Error::from)?;
- Ok(())
- }
- async fn update_proofs(
- &self,
- added: Vec<ProofInfo>,
- removed_ys: Vec<PublicKey>,
- ) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[PROOFS], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let proofs_store = transaction.store(PROOFS).map_err(Error::from)?;
- for proof in added {
- let y = serde_wasm_bindgen::to_value(&proof.y).map_err(Error::from)?;
- let proof = serde_wasm_bindgen::to_value(&proof).map_err(Error::from)?;
- proofs_store
- .put(&proof, Some(&y))
- .await
- .map_err(Error::from)?;
- }
- for y in removed_ys {
- let y = serde_wasm_bindgen::to_value(&y).map_err(Error::from)?;
- proofs_store.delete(y).await.map_err(Error::from)?;
- }
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn set_pending_proofs(&self, ys: Vec<PublicKey>) -> Result<(), Self::Err> {
- self.set_proof_states(ys, State::Pending).await
- }
- async fn reserve_proofs(&self, ys: Vec<PublicKey>) -> Result<(), Self::Err> {
- self.set_proof_states(ys, State::Reserved).await
- }
- async fn set_unspent_proofs(&self, ys: Vec<PublicKey>) -> Result<(), Self::Err> {
- self.set_proof_states(ys, State::Unspent).await
- }
- async fn get_proofs(
- &self,
- mint_url: Option<MintUrl>,
- unit: Option<CurrencyUnit>,
- state: Option<Vec<State>>,
- spending_conditions: Option<Vec<SpendingConditions>>,
- ) -> Result<Vec<ProofInfo>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[PROOFS], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let proofs_store = transaction.store(PROOFS).map_err(Error::from)?;
- let proofs = proofs_store
- .scan(None, None, None, None)
- .await
- .map_err(Error::from)?;
- let proofs: Vec<ProofInfo> = proofs
- .into_iter()
- .filter_map(|(_k, v)| {
- let mut proof = None;
- if let Ok(proof_info) = serde_wasm_bindgen::from_value::<ProofInfo>(v) {
- proof = match proof_info.matches_conditions(
- &mint_url,
- &unit,
- &state,
- &spending_conditions,
- ) {
- true => Some(proof_info),
- false => None,
- };
- }
- proof
- })
- .collect();
- transaction.done().await.map_err(Error::from)?;
- Ok(proofs)
- }
- async fn increment_keyset_counter(&self, keyset_id: &Id, count: u32) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[KEYSET_COUNTER], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let counter_store = transaction.store(KEYSET_COUNTER).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(keyset_id).map_err(Error::from)?;
- let current_count: Option<u32> = counter_store
- .get(keyset_id.clone())
- .await
- .map_err(Error::from)?
- .and_then(|c| serde_wasm_bindgen::from_value(c).ok());
- let new_count = current_count.unwrap_or_default() + count;
- let new_count = serde_wasm_bindgen::to_value(&new_count).map_err(Error::from)?;
- counter_store
- .put(&new_count, Some(&keyset_id))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_keyset_counter(&self, keyset_id: &Id) -> Result<Option<u32>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[KEYSET_COUNTER], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let counter_store = transaction.store(KEYSET_COUNTER).map_err(Error::from)?;
- let keyset_id = serde_wasm_bindgen::to_value(keyset_id).map_err(Error::from)?;
- let current_count = counter_store
- .get(keyset_id)
- .await
- .map_err(Error::from)?
- .and_then(|c| serde_wasm_bindgen::from_value(c).ok());
- Ok(current_count)
- }
- async fn add_nostr_last_checked(
- &self,
- verifying_key: PublicKey,
- last_checked: u32,
- ) -> Result<(), Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[NOSTR_LAST_CHECKED], TransactionMode::ReadWrite)
- .map_err(Error::from)?;
- let counter_store = transaction.store(NOSTR_LAST_CHECKED).map_err(Error::from)?;
- let verifying_key = serde_wasm_bindgen::to_value(&verifying_key).map_err(Error::from)?;
- let last_checked = serde_wasm_bindgen::to_value(&last_checked).map_err(Error::from)?;
- counter_store
- .put(&last_checked, Some(&verifying_key))
- .await
- .map_err(Error::from)?;
- transaction.done().await.map_err(Error::from)?;
- Ok(())
- }
- async fn get_nostr_last_checked(
- &self,
- verifying_key: &PublicKey,
- ) -> Result<Option<u32>, Self::Err> {
- let rexie = self.db.lock().await;
- let transaction = rexie
- .transaction(&[NOSTR_LAST_CHECKED], TransactionMode::ReadOnly)
- .map_err(Error::from)?;
- let nostr_last_check_store = transaction.store(NOSTR_LAST_CHECKED).map_err(Error::from)?;
- let verifying_key = serde_wasm_bindgen::to_value(verifying_key).map_err(Error::from)?;
- let last_checked = nostr_last_check_store
- .get(verifying_key)
- .await
- .map_err(Error::from)?
- .and_then(|l| serde_wasm_bindgen::from_value(l).ok());
- Ok(last_checked)
- }
- }
|