thePR0M3TH3AN/seedPass
1
0
Fork 0
mirror of https://github.com/PR0M3TH3AN/SeedPass.git synced 2025-09-09 15:58:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add QR code submenu for entries

Browse Source
This commit is contained in:
thePR0M3TH3AN
2025-07-08 15:42:42 -04:00
parent 238091b790
commit 7a2c589643
1 changed files with 39 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
src/password_manager/manager.py
View File

@@ -1730,71 +1730,57 @@ class PasswordManager:
entry = self.entry_manager.retrieve_entry(index) or entry entry = self.entry_manager.retrieve_entry(index) or entry
def _entry_qr_menu(self, index: int, entry: dict) -> None: def _entry_qr_menu(self, index: int, entry: dict) -> None:
"""Sub-menu for displaying QR codes related to ``entry``.""" """Display QR codes for the given ``entry``."""
entry_type = entry.get("type", EntryType.PASSWORD.value)
while True: entry_type = entry.get("type")
print(colored("\n[+] QR Codes:", "green"))
options_available = False
if entry_type == EntryType.TOTP.value: try:
print(colored("O. Show otpauth URI", "cyan")) if entry_type in {EntryType.SEED.value, EntryType.MANAGED_ACCOUNT.value}:
options_available = True
elif entry_type == EntryType.NOSTR.value:
print(colored("N. Show npub", "cyan"))
print(colored("P. Show nsec", "cyan"))
options_available = True
elif entry_type in {EntryType.SEED.value, EntryType.MANAGED_ACCOUNT.value}:
print(colored("S. Show seed", "cyan"))
options_available = True
if not options_available:
print(colored("No QR codes available for this entry.", "yellow"))
pause()
return
choice = input("Select option or press Enter to go back: ").strip().lower()
if not choice:
break
if entry_type == EntryType.TOTP.value and choice == "o":
label = entry.get("label", "")
period = int(entry.get("period", 30))
digits = int(entry.get("digits", 6))
if "secret" in entry:
secret = entry["secret"]
else:
totp_index = int(entry.get("index", index))
secret = TotpManager.derive_secret(self.parent_seed, totp_index)
uri = TotpManager.make_otpauth_uri(label, secret, period, digits)
TotpManager.print_qr_code(uri)
elif entry_type == EntryType.NOSTR.value:
npub, nsec = self.entry_manager.get_nostr_key_pair(
index, self.parent_seed
)
if choice == "n":
TotpManager.print_qr_code(f"nostr:{npub}")
elif choice == "p":
TotpManager.print_qr_code(nsec)
else:
print(colored("Invalid choice.", "red"))
elif (
entry_type in {EntryType.SEED.value, EntryType.MANAGED_ACCOUNT.value}
and choice == "s"
):
if entry_type == EntryType.SEED.value: if entry_type == EntryType.SEED.value:
seed = self.entry_manager.get_seed_phrase(index, self.parent_seed) seed = self.entry_manager.get_seed_phrase(index, self.parent_seed)
else: else:
seed = self.entry_manager.get_managed_account_seed( seed = self.entry_manager.get_managed_account_seed(
index, self.parent_seed index, self.parent_seed
) )
print(color_text(seed, "deterministic"))
from password_manager.seedqr import encode_seedqr from password_manager.seedqr import encode_seedqr
TotpManager.print_qr_code(encode_seedqr(seed)) TotpManager.print_qr_code(encode_seedqr(seed))
else: return
print(colored("Invalid choice.", "red"))
entry = self.entry_manager.retrieve_entry(index) or entry if entry_type == EntryType.NOSTR.value:
while True:
print(colored("\n[+] QR Codes:", "green"))
print(colored("P. Public key", "cyan"))
print(colored("K. Private key", "cyan"))
choice = (
input("Select option or press Enter to return: ")
.strip()
.lower()
)
if not choice:
break
npub, nsec = self.entry_manager.get_nostr_key_pair(
index, self.parent_seed
)
if choice == "p":
print(colored(f"npub: {npub}", "cyan"))
TotpManager.print_qr_code(f"nostr:{npub}")
elif choice == "k":
print(color_text(f"nsec: {nsec}", "deterministic"))
TotpManager.print_qr_code(nsec)
else:
print(colored("Invalid choice.", "red"))
entry = self.entry_manager.retrieve_entry(index) or entry
return
print(colored("No QR codes available for this entry.", "yellow"))
except Exception as e: # pragma: no cover - best effort
logging.error(f"Error displaying QR menu: {e}", exc_info=True)
print(colored(f"Error: Failed to display QR codes: {e}", "red"))
def handle_retrieve_entry(self) -> None: def handle_retrieve_entry(self) -> None:
""" """