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

Enhance CLI breadcrumbs and update menu color

Browse Source
This commit is contained in:
thePR0M3TH3AN
2025-07-06 09:25:45 -04:00
parent 3cfebf5680
commit c35c4d66a0
4 changed files with 101 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/main.py
View File

@@ -578,7 +578,8 @@ def handle_profiles_menu(password_manager: PasswordManager) -> None:
"""Submenu for managing seed profiles.""" """Submenu for managing seed profiles."""
while True: while True:
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None) getattr(password_manager, "current_fingerprint", None),
"Main Menu > Settings > Profiles",
) )
print(color_text("\nProfiles:", "menu")) print(color_text("\nProfiles:", "menu"))
print(color_text("1. Switch Seed Profile", "menu")) print(color_text("1. Switch Seed Profile", "menu"))
@@ -616,7 +617,8 @@ def handle_nostr_menu(password_manager: PasswordManager) -> None:
while True: while True:
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None) getattr(password_manager, "current_fingerprint", None),
"Main Menu > Settings > Nostr",
) )
print(color_text("\nNostr Settings:", "menu")) print(color_text("\nNostr Settings:", "menu"))
print(color_text("1. Backup to Nostr", "menu")) print(color_text("1. Backup to Nostr", "menu"))
@@ -652,7 +654,8 @@ def handle_settings(password_manager: PasswordManager) -> None:
"""Interactive settings menu with submenus for profiles and Nostr.""" """Interactive settings menu with submenus for profiles and Nostr."""
while True: while True:
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None) getattr(password_manager, "current_fingerprint", None),
"Main Menu > Settings",
) )
print(color_text("\nSettings:", "menu")) print(color_text("\nSettings:", "menu"))
print(color_text("1. Profiles", "menu")) print(color_text("1. Profiles", "menu"))
@@ -746,7 +749,8 @@ def display_menu(
pause() pause()
while True: while True:
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None) getattr(password_manager, "current_fingerprint", None),
"Main Menu",
) )
if time.time() - password_manager.last_activity > inactivity_timeout: if time.time() - password_manager.last_activity > inactivity_timeout:
print(colored("Session timed out. Vault locked.", "yellow")) print(colored("Session timed out. Vault locked.", "yellow"))
@@ -783,6 +787,10 @@ def display_menu(
sys.exit(0) sys.exit(0)
if choice == "1": if choice == "1":
while True: while True:
clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None),
"Main Menu > Add Entry",
)
print(color_text("\nAdd Entry:", "menu")) print(color_text("\nAdd Entry:", "menu"))
print(color_text("1. Password", "menu")) print(color_text("1. Password", "menu"))
print(color_text("2. 2FA (TOTP)", "menu")) print(color_text("2. 2FA (TOTP)", "menu"))
@@ -820,7 +828,8 @@ def display_menu(
password_manager.update_activity() password_manager.update_activity()
password_manager.handle_retrieve_entry() password_manager.handle_retrieve_entry()
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(password_manager, "current_fingerprint", None) getattr(password_manager, "current_fingerprint", None),
"Main Menu",
) )
elif choice == "3": elif choice == "3":
password_manager.update_activity() password_manager.update_activity()

97
src/password_manager/manager.py
View File

@@ -890,7 +890,10 @@ class PasswordManager:
def handle_add_password(self) -> None: def handle_add_password(self) -> None:
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > Password",
)
website_name = input("Enter the label or website name: ").strip() website_name = input("Enter the label or website name: ").strip()
if not website_name: if not website_name:
print(colored("Error: Label cannot be empty.", "red")) print(colored("Error: Label cannot be empty.", "red"))
@@ -977,7 +980,10 @@ class PasswordManager:
def handle_add_totp(self) -> None: def handle_add_totp(self) -> None:
"""Add a TOTP entry either derived from the seed or imported.""" """Add a TOTP entry either derived from the seed or imported."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > 2FA (TOTP)",
)
while True: while True:
print("\nAdd TOTP:") print("\nAdd TOTP:")
print("1. Make 2FA (derive from seed)") print("1. Make 2FA (derive from seed)")
@@ -1077,7 +1083,10 @@ class PasswordManager:
def handle_add_ssh_key(self) -> None: def handle_add_ssh_key(self) -> None:
"""Add an SSH key pair entry and display the derived keys.""" """Add an SSH key pair entry and display the derived keys."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > SSH Key",
)
label = input("Label: ").strip() label = input("Label: ").strip()
if not label: if not label:
print(colored("Error: Label cannot be empty.", "red")) print(colored("Error: Label cannot be empty.", "red"))
@@ -1119,7 +1128,10 @@ class PasswordManager:
def handle_add_seed(self) -> None: def handle_add_seed(self) -> None:
"""Add a derived BIP-39 seed phrase entry.""" """Add a derived BIP-39 seed phrase entry."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > Seed Phrase",
)
label = input("Label: ").strip() label = input("Label: ").strip()
if not label: if not label:
print(colored("Error: Label cannot be empty.", "red")) print(colored("Error: Label cannot be empty.", "red"))
@@ -1175,7 +1187,10 @@ class PasswordManager:
def handle_add_pgp(self) -> None: def handle_add_pgp(self) -> None:
"""Add a PGP key entry and display the generated key.""" """Add a PGP key entry and display the generated key."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > PGP Key",
)
label = input("Label: ").strip() label = input("Label: ").strip()
if not label: if not label:
print(colored("Error: Label cannot be empty.", "red")) print(colored("Error: Label cannot be empty.", "red"))
@@ -1228,7 +1243,10 @@ class PasswordManager:
def handle_add_nostr_key(self) -> None: def handle_add_nostr_key(self) -> None:
"""Add a Nostr key entry and display the derived keys.""" """Add a Nostr key entry and display the derived keys."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Add Entry > Nostr Key Pair",
)
label = input("Label: ").strip() label = input("Label: ").strip()
if not label: if not label:
print(colored("Error: Label cannot be empty.", "red")) print(colored("Error: Label cannot be empty.", "red"))
@@ -1295,7 +1313,10 @@ class PasswordManager:
and displaying the corresponding password and associated details. and displaying the corresponding password and associated details.
""" """
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Retrieve Entry",
)
index_input = input( index_input = input(
"Enter the index number of the entry to retrieve: " "Enter the index number of the entry to retrieve: "
).strip() ).strip()
@@ -1616,7 +1637,10 @@ class PasswordManager:
and new details to update. and new details to update.
""" """
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Modify Entry",
)
index_input = input( index_input = input(
"Enter the index number of the entry to modify: " "Enter the index number of the entry to modify: "
).strip() ).strip()
@@ -1854,7 +1878,10 @@ class PasswordManager:
def handle_search_entries(self) -> None: def handle_search_entries(self) -> None:
"""Prompt for a query, list matches and optionally show details.""" """Prompt for a query, list matches and optionally show details."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Search Entries",
)
query = input("Enter search string: ").strip() query = input("Enter search string: ").strip()
if not query: if not query:
print(colored("No search string provided.", "yellow")) print(colored("No search string provided.", "yellow"))
@@ -1868,7 +1895,10 @@ class PasswordManager:
return return
while True: while True:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Search Entries",
)
print(colored("\n[+] Search Results:\n", "green")) print(colored("\n[+] Search Results:\n", "green"))
for idx, label, username, _url, _b in results: for idx, label, username, _url, _b in results:
display_label = label display_label = label
@@ -1975,7 +2005,10 @@ class PasswordManager:
"""List entries and optionally show details.""" """List entries and optionally show details."""
try: try:
while True: while True:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > List Entries",
)
print(color_text("\nList Entries:", "menu")) print(color_text("\nList Entries:", "menu"))
print(color_text("1. All", "menu")) print(color_text("1. All", "menu"))
print(color_text("2. Passwords", "menu")) print(color_text("2. Passwords", "menu"))
@@ -2010,7 +2043,8 @@ class PasswordManager:
continue continue
while True: while True:
clear_and_print_fingerprint( clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None) getattr(self, "current_fingerprint", None),
"Main Menu > List Entries",
) )
print(colored("\n[+] Entries:\n", "green")) print(colored("\n[+] Entries:\n", "green"))
for idx, etype, label in summaries: for idx, etype, label in summaries:
@@ -2072,6 +2106,10 @@ class PasswordManager:
def handle_display_totp_codes(self) -> None: def handle_display_totp_codes(self) -> None:
"""Display all stored TOTP codes with a countdown progress bar.""" """Display all stored TOTP codes with a countdown progress bar."""
try: try:
clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > 2FA Codes",
)
data = self.entry_manager.vault.load_index() data = self.entry_manager.vault.load_index()
entries = data.get("entries", {}) entries = data.get("entries", {})
totp_list: list[tuple[str, int, int, bool]] = [] totp_list: list[tuple[str, int, int, bool]] = []
@@ -2091,7 +2129,10 @@ class PasswordManager:
totp_list.sort(key=lambda t: t[0].lower()) totp_list.sort(key=lambda t: t[0].lower())
print(colored("Press Enter to return to the menu.", "cyan")) print(colored("Press Enter to return to the menu.", "cyan"))
while True: while True:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > 2FA Codes",
)
print(colored("Press Enter to return to the menu.", "cyan")) print(colored("Press Enter to return to the menu.", "cyan"))
generated = [t for t in totp_list if not t[3]] generated = [t for t in totp_list if not t[3]]
imported_list = [t for t in totp_list if t[3]] imported_list = [t for t in totp_list if t[3]]
@@ -2146,7 +2187,10 @@ class PasswordManager:
Handles verifying the script's checksum against the stored checksum to ensure integrity. Handles verifying the script's checksum against the stored checksum to ensure integrity.
""" """
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Verify Script Checksum",
)
current_checksum = calculate_checksum(__file__) current_checksum = calculate_checksum(__file__)
try: try:
verified = verify_checksum(current_checksum, SCRIPT_CHECKSUM_FILE) verified = verify_checksum(current_checksum, SCRIPT_CHECKSUM_FILE)
@@ -2181,7 +2225,10 @@ class PasswordManager:
print(colored("Operation cancelled.", "yellow")) print(colored("Operation cancelled.", "yellow"))
return return
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Generate Script Checksum",
)
script_path = Path(__file__).resolve() script_path = Path(__file__).resolve()
if update_checksum_file(str(script_path), str(SCRIPT_CHECKSUM_FILE)): if update_checksum_file(str(script_path), str(SCRIPT_CHECKSUM_FILE)):
print( print(
@@ -2291,7 +2338,10 @@ class PasswordManager:
) -> Path | None: ) -> Path | None:
"""Export the current database to an encrypted portable file.""" """Export the current database to an encrypted portable file."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Export database",
)
path = export_backup( path = export_backup(
self.vault, self.vault,
self.backup_manager, self.backup_manager,
@@ -2308,7 +2358,10 @@ class PasswordManager:
def handle_import_database(self, src: Path) -> None: def handle_import_database(self, src: Path) -> None:
"""Import a portable database file, replacing the current index.""" """Import a portable database file, replacing the current index."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Import database",
)
import_backup( import_backup(
self.vault, self.vault,
self.backup_manager, self.backup_manager,
@@ -2323,7 +2376,10 @@ class PasswordManager:
def handle_export_totp_codes(self) -> Path | None: def handle_export_totp_codes(self) -> Path | None:
"""Export all 2FA codes to a JSON file for other authenticator apps.""" """Export all 2FA codes to a JSON file for other authenticator apps."""
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Export 2FA codes",
)
data = self.entry_manager.vault.load_index() data = self.entry_manager.vault.load_index()
entries = data.get("entries", {}) entries = data.get("entries", {})
@@ -2383,7 +2439,10 @@ class PasswordManager:
Handles the backup and reveal of the parent seed. Handles the backup and reveal of the parent seed.
""" """
try: try:
clear_and_print_fingerprint(getattr(self, "current_fingerprint", None)) clear_and_print_fingerprint(
getattr(self, "current_fingerprint", None),
"Main Menu > Settings > Backup Parent Seed",
)
print(colored("\n=== Backup Parent Seed ===", "yellow")) print(colored("\n=== Backup Parent Seed ===", "yellow"))
print( print(
colored( colored(

2
src/utils/color_scheme.py
View File

@@ -18,7 +18,7 @@ _COLOR_MAP = {
"deterministic": "red", "deterministic": "red",
"imported": "orange", "imported": "orange",
"index": "yellow", "index": "yellow",
"menu": "blue", "menu": "cyan",
"stats": "green", "stats": "green",
"default": "white", "default": "white",
} }

11
src/utils/terminal_utils.py
View File

@@ -11,11 +11,16 @@ def clear_screen() -> None:
print("\033c", end="") print("\033c", end="")
def clear_and_print_fingerprint(fingerprint: str | None) -> None: def clear_and_print_fingerprint(
"""Clear the screen and optionally display the current fingerprint.""" fingerprint: str | None, breadcrumb: str | None = None
) -> None:
"""Clear the screen and optionally display the current fingerprint and path."""
clear_screen() clear_screen()
if fingerprint: if fingerprint:
print(colored(f"Seed Profile: {fingerprint}", "green")) header = f"Seed Profile: {fingerprint}"
if breadcrumb:
header += f" > {breadcrumb}"
print(colored(header, "green"))
def pause(message: str = "Press Enter to continue...") -> None: def pause(message: str = "Press Enter to continue...") -> None: