mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
Merge pull request #461 from PR0M3TH3AN/codex/fix-fingerprint-command-password-prompt
Fix CLI fingerprint handling
This commit is contained in:
15
src/main.py
15
src/main.py
@@ -919,8 +919,16 @@ def display_menu(
|
|||||||
print(colored("Invalid choice. Please select a valid option.", "red"))
|
print(colored("Invalid choice. Please select a valid option.", "red"))
|
||||||
|
|
||||||
|
|
||||||
def main(argv: list[str] | None = None) -> int:
|
def main(argv: list[str] | None = None, *, fingerprint: str | None = None) -> int:
|
||||||
"""Entry point for the SeedPass CLI."""
|
"""Entry point for the SeedPass CLI.
|
||||||
|
|
||||||
|
Parameters
|
||||||
|
----------
|
||||||
|
argv:
|
||||||
|
Command line arguments.
|
||||||
|
fingerprint:
|
||||||
|
Optional seed profile fingerprint to select automatically.
|
||||||
|
"""
|
||||||
configure_logging()
|
configure_logging()
|
||||||
initialize_app()
|
initialize_app()
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@@ -928,6 +936,7 @@ def main(argv: list[str] | None = None) -> int:
|
|||||||
|
|
||||||
load_global_config()
|
load_global_config()
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument("--fingerprint")
|
||||||
sub = parser.add_subparsers(dest="command")
|
sub = parser.add_subparsers(dest="command")
|
||||||
|
|
||||||
exp = sub.add_parser("export")
|
exp = sub.add_parser("export")
|
||||||
@@ -948,7 +957,7 @@ def main(argv: list[str] | None = None) -> int:
|
|||||||
args = parser.parse_args(argv)
|
args = parser.parse_args(argv)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
password_manager = PasswordManager()
|
password_manager = PasswordManager(fingerprint=args.fingerprint or fingerprint)
|
||||||
logger.info("PasswordManager initialized successfully.")
|
logger.info("PasswordManager initialized successfully.")
|
||||||
except (PasswordPromptError, Bip85Error) as e:
|
except (PasswordPromptError, Bip85Error) as e:
|
||||||
logger.error(f"Failed to initialize PasswordManager: {e}", exc_info=True)
|
logger.error(f"Failed to initialize PasswordManager: {e}", exc_info=True)
|
||||||
|
@@ -61,7 +61,7 @@ def main(ctx: typer.Context, fingerprint: Optional[str] = fingerprint_option) ->
|
|||||||
ctx.obj = {"fingerprint": fingerprint}
|
ctx.obj = {"fingerprint": fingerprint}
|
||||||
if ctx.invoked_subcommand is None:
|
if ctx.invoked_subcommand is None:
|
||||||
tui = importlib.import_module("main")
|
tui = importlib.import_module("main")
|
||||||
raise typer.Exit(tui.main())
|
raise typer.Exit(tui.main(fingerprint=fingerprint))
|
||||||
|
|
||||||
|
|
||||||
@entry_app.command("list")
|
@entry_app.command("list")
|
||||||
|
@@ -45,7 +45,7 @@ def test_cli_export_creates_file(monkeypatch, tmp_path):
|
|||||||
}
|
}
|
||||||
vault.save_index(data)
|
vault.save_index(data)
|
||||||
|
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -83,7 +83,7 @@ def test_cli_import_round_trip(monkeypatch, tmp_path):
|
|||||||
|
|
||||||
vault.save_index({"schema_version": 4, "entries": {}})
|
vault.save_index({"schema_version": 4, "entries": {}})
|
||||||
|
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
|
@@ -28,7 +28,7 @@ def make_pm(search_results, entry=None, totp_code="123456"):
|
|||||||
|
|
||||||
def test_search_command(monkeypatch, capsys):
|
def test_search_command(monkeypatch, capsys):
|
||||||
pm = make_pm([(0, "Example", "user", "", False)])
|
pm = make_pm([(0, "Example", "user", "", False)])
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -41,7 +41,7 @@ def test_search_command(monkeypatch, capsys):
|
|||||||
def test_get_command(monkeypatch, capsys):
|
def test_get_command(monkeypatch, capsys):
|
||||||
entry = {"type": EntryType.PASSWORD.value, "length": 8}
|
entry = {"type": EntryType.PASSWORD.value, "length": 8}
|
||||||
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -55,7 +55,7 @@ def test_totp_command(monkeypatch, capsys):
|
|||||||
entry = {"type": EntryType.TOTP.value, "period": 30, "index": 0}
|
entry = {"type": EntryType.TOTP.value, "period": 30, "index": 0}
|
||||||
pm = make_pm([(0, "Example", None, None, False)], entry=entry)
|
pm = make_pm([(0, "Example", None, None, False)], entry=entry)
|
||||||
called = {}
|
called = {}
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -72,7 +72,7 @@ def test_totp_command(monkeypatch, capsys):
|
|||||||
|
|
||||||
def test_search_command_no_results(monkeypatch, capsys):
|
def test_search_command_no_results(monkeypatch, capsys):
|
||||||
pm = make_pm([])
|
pm = make_pm([])
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -85,7 +85,7 @@ def test_search_command_no_results(monkeypatch, capsys):
|
|||||||
def test_get_command_multiple_matches(monkeypatch, capsys):
|
def test_get_command_multiple_matches(monkeypatch, capsys):
|
||||||
matches = [(0, "Example", "user", "", False), (1, "Ex2", "bob", "", False)]
|
matches = [(0, "Example", "user", "", False), (1, "Ex2", "bob", "", False)]
|
||||||
pm = make_pm(matches)
|
pm = make_pm(matches)
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -98,7 +98,7 @@ def test_get_command_multiple_matches(monkeypatch, capsys):
|
|||||||
def test_get_command_wrong_type(monkeypatch, capsys):
|
def test_get_command_wrong_type(monkeypatch, capsys):
|
||||||
entry = {"type": EntryType.TOTP.value}
|
entry = {"type": EntryType.TOTP.value}
|
||||||
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -111,7 +111,7 @@ def test_get_command_wrong_type(monkeypatch, capsys):
|
|||||||
def test_totp_command_multiple_matches(monkeypatch, capsys):
|
def test_totp_command_multiple_matches(monkeypatch, capsys):
|
||||||
matches = [(0, "GH", None, None, False), (1, "Git", None, None, False)]
|
matches = [(0, "GH", None, None, False), (1, "Git", None, None, False)]
|
||||||
pm = make_pm(matches)
|
pm = make_pm(matches)
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -124,7 +124,7 @@ def test_totp_command_multiple_matches(monkeypatch, capsys):
|
|||||||
def test_totp_command_wrong_type(monkeypatch, capsys):
|
def test_totp_command_wrong_type(monkeypatch, capsys):
|
||||||
entry = {"type": EntryType.PASSWORD.value, "length": 8}
|
entry = {"type": EntryType.PASSWORD.value, "length": 8}
|
||||||
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
pm = make_pm([(0, "Example", "user", "", False)], entry=entry)
|
||||||
monkeypatch.setattr(main, "PasswordManager", lambda: pm)
|
monkeypatch.setattr(main, "PasswordManager", lambda *a, **k: pm)
|
||||||
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
@@ -132,3 +132,21 @@ def test_totp_command_wrong_type(monkeypatch, capsys):
|
|||||||
assert rc == 1
|
assert rc == 1
|
||||||
out = capsys.readouterr().out
|
out = capsys.readouterr().out
|
||||||
assert "Entry is not a TOTP entry" in out
|
assert "Entry is not a TOTP entry" in out
|
||||||
|
|
||||||
|
|
||||||
|
def test_main_fingerprint_option(monkeypatch):
|
||||||
|
"""Ensure the argparse CLI forwards the fingerprint to PasswordManager."""
|
||||||
|
called = {}
|
||||||
|
|
||||||
|
def fake_pm(fingerprint=None):
|
||||||
|
called["fp"] = fingerprint
|
||||||
|
return make_pm([])
|
||||||
|
|
||||||
|
monkeypatch.setattr(main, "PasswordManager", fake_pm)
|
||||||
|
monkeypatch.setattr(main, "configure_logging", lambda: None)
|
||||||
|
monkeypatch.setattr(main, "initialize_app", lambda: None)
|
||||||
|
monkeypatch.setattr(main.signal, "signal", lambda *a, **k: None)
|
||||||
|
|
||||||
|
rc = main.main(["--fingerprint", "abc", "search", "q"])
|
||||||
|
assert rc == 0
|
||||||
|
assert called.get("fp") == "abc"
|
||||||
|
@@ -307,6 +307,21 @@ def test_api_start_passes_fingerprint(monkeypatch):
|
|||||||
assert called.get("fp") == "abc"
|
assert called.get("fp") == "abc"
|
||||||
|
|
||||||
|
|
||||||
|
def test_entry_list_passes_fingerprint(monkeypatch):
|
||||||
|
"""Ensure entry commands receive the fingerprint."""
|
||||||
|
called = {}
|
||||||
|
|
||||||
|
class PM:
|
||||||
|
def __init__(self, fingerprint=None):
|
||||||
|
called["fp"] = fingerprint
|
||||||
|
self.entry_manager = SimpleNamespace(list_entries=lambda *a, **k: [])
|
||||||
|
|
||||||
|
monkeypatch.setattr(cli, "PasswordManager", PM)
|
||||||
|
result = runner.invoke(app, ["--fingerprint", "abc", "entry", "list"])
|
||||||
|
assert result.exit_code == 0
|
||||||
|
assert called.get("fp") == "abc"
|
||||||
|
|
||||||
|
|
||||||
def test_entry_add(monkeypatch):
|
def test_entry_add(monkeypatch):
|
||||||
called = {}
|
called = {}
|
||||||
|
|
||||||
@@ -447,3 +462,21 @@ def test_update_checksum_command(monkeypatch):
|
|||||||
result = runner.invoke(app, ["util", "update-checksum"])
|
result = runner.invoke(app, ["util", "update-checksum"])
|
||||||
assert result.exit_code == 0
|
assert result.exit_code == 0
|
||||||
assert called.get("called") is True
|
assert called.get("called") is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_tui_forward_fingerprint(monkeypatch):
|
||||||
|
"""Ensure --fingerprint is forwarded when launching the TUI."""
|
||||||
|
called = {}
|
||||||
|
|
||||||
|
def fake_main(*, fingerprint=None):
|
||||||
|
called["fp"] = fingerprint
|
||||||
|
return 0
|
||||||
|
|
||||||
|
fake_mod = SimpleNamespace(main=fake_main)
|
||||||
|
monkeypatch.setattr(
|
||||||
|
cli, "importlib", SimpleNamespace(import_module=lambda n: fake_mod)
|
||||||
|
)
|
||||||
|
|
||||||
|
result = runner.invoke(app, ["--fingerprint", "abc"])
|
||||||
|
assert result.exit_code == 0
|
||||||
|
assert called.get("fp") == "abc"
|
||||||
|
Reference in New Issue
Block a user