aljaz/nutshell
1
0
Fork 0
mirror of https://github.com/aljazceru/nutshell.git synced 2025-12-22 11:24:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Mint] load the latest keyset from db based on the derivation path (#193)

Browse Source 
* load the latest keyset from db based on the derivation path and not the on-line generated keyset id (necessary because we changed the keyset id derivation algorithm)

* better comments

* generate keys for keyset from db
This commit is contained in:
callebtc
2023-05-07 17:47:38 +02:00
committed by GitHub
parent e9f33337db
commit c31f008fb4
1 changed files with 27 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
cashu/mint/ledger.py
View File

@@ -49,7 +49,7 @@ class Ledger:
self.proofs_used = set(proofs_used) self.proofs_used = set(proofs_used)
async def load_keyset(self, derivation_path, autosave=True): async def load_keyset(self, derivation_path, autosave=True):
"""Load current keyset keyset or generate new one. """Load the keyset for a derivation path if it already exists. If not generate new one and store in the db.
Args: Args:
derivation_path (_type_): Derivation path from which the keyset is generated. derivation_path (_type_): Derivation path from which the keyset is generated.
@@ -63,22 +63,37 @@ class Ledger:
derivation_path=derivation_path, derivation_path=derivation_path,
version=settings.version, version=settings.version,
) )
# check if current keyset is stored in db and store if not # load the keyest from db
logger.debug(f"Active keyset: {keyset.id}")
tmp_keyset_local: List[MintKeyset] = await self.crud.get_keyset( tmp_keyset_local: List[MintKeyset] = await self.crud.get_keyset(
id=keyset.id, db=self.db derivation_path=derivation_path, db=self.db
) )
if not len(tmp_keyset_local) and autosave: if tmp_keyset_local:
logger.debug(f"Storing new keyset {keyset.id}.") # we have a keyset for this derivation path
await self.crud.store_keyset(keyset=keyset, db=self.db) keyset = tmp_keyset_local[0]
# we need to initialize it
keyset.generate_keys(self.master_key)
else:
# no keyset for this derivation path yet
# we generate a new keyset
logger.debug(f"Generating new keyset {keyset.id}.")
keyset = MintKeyset(
seed=self.master_key,
derivation_path=derivation_path,
version=settings.version,
)
if autosave:
logger.debug(f"Storing new keyset {keyset.id}.")
await self.crud.store_keyset(keyset=keyset, db=self.db)
# store the new keyset in the current keysets # store the new keyset in the current keysets
if keyset.id: if keyset.id:
self.keysets.keysets[keyset.id] = keyset self.keysets.keysets[keyset.id] = keyset
logger.debug(f"Loaded keyset {keyset.id}.")
return keyset return keyset
async def init_keysets(self, autosave=True): async def init_keysets(self, autosave=True):
"""Initializes all keysets of the mint from the db. """Initializes all keysets of the mint from the db. Loads all past keysets and generate their keys. Then load the current keyset.
Args: Args:
autosave (bool, optional): Whether the current keyset should be saved if it is autosave (bool, optional): Whether the current keyset should be saved if it is
@@ -92,11 +107,7 @@ class Ledger:
if k.id and k.id not in self.keysets.keysets: if k.id and k.id not in self.keysets.keysets:
self.keysets.keysets[k.id] = k self.keysets.keysets[k.id] = k
logger.debug( # generate keys for all keysets in the database
f"We initialized {len(self.keysets.keysets)} previous keysets from the database."
)
# generate all private keys, public keys, and keyset id from the derivation path for all keysets that are not yet generated
for _, v in self.keysets.keysets.items(): for _, v in self.keysets.keysets.items():
# we already generated the keys for this keyset # we already generated the keys for this keyset
if v.id and v.public_keys and len(v.public_keys): if v.id and v.public_keys and len(v.public_keys):
@@ -104,6 +115,9 @@ class Ledger:
logger.debug(f"Generating keys for keyset {v.id}") logger.debug(f"Generating keys for keyset {v.id}")
v.generate_keys(self.master_key) v.generate_keys(self.master_key)
logger.debug(
f"Initialized {len(self.keysets.keysets)} keysets from the database."
)
# load the current keyset # load the current keyset
self.keyset = await self.load_keyset(self.derivation_path, autosave) self.keyset = await self.load_keyset(self.derivation_path, autosave)