diff --git a/src/seedpass/core/backup.py b/src/seedpass/core/backup.py index 467d068..1cb3107 100644 --- a/src/seedpass/core/backup.py +++ b/src/seedpass/core/backup.py @@ -145,6 +145,28 @@ class BackupManager: ) ) + def restore_from_backup(self, backup_path: str) -> None: + """Restore the index file from a user-specified backup path.""" + try: + src = Path(backup_path) + if not src.exists(): + logger.error(f"Backup file '{src}' does not exist.") + print(colored(f"Error: Backup file '{src}' does not exist.", "red")) + return + shutil.copy2(src, self.index_file) + os.chmod(self.index_file, 0o600) + logger.info(f"Index file restored from backup '{src}'.") + print(colored(f"[+] Index file restored from backup '{src}'.", "green")) + except Exception as e: + logger.error( + f"Failed to restore from backup '{backup_path}': {e}", exc_info=True + ) + print( + colored( + f"Error: Failed to restore from backup '{backup_path}': {e}", "red" + ) + ) + def list_backups(self) -> None: try: backup_files = sorted( diff --git a/src/seedpass/core/manager.py b/src/seedpass/core/manager.py index 8a6644e..7e0e514 100644 --- a/src/seedpass/core/manager.py +++ b/src/seedpass/core/manager.py @@ -988,7 +988,8 @@ class PasswordManager: "2. Enter an existing seed one word at a time\n" "3. Generate a new seed\n" "4. Restore from Nostr\n" - "Enter choice (1/2/3/4): " + "5. Restore from local backup\n" + "Enter choice (1/2/3/4/5): " ).strip() if choice == "1": @@ -1001,6 +1002,15 @@ class PasswordManager: seed_phrase = masked_input("Enter your 12-word BIP-85 seed: ").strip() self.restore_from_nostr_with_guidance(seed_phrase) return + elif choice == "5": + backup_path = input("Enter backup file path: ").strip() + if not getattr(self, "fingerprint_manager", None): + self.initialize_fingerprint_manager() + seed_phrase = masked_input("Enter your 12-word BIP-85 seed: ").strip() + fp = self._finalize_existing_seed(seed_phrase) + if fp: + self.backup_manager.restore_from_backup(backup_path) + return else: print(colored("Invalid choice. Exiting.", "red")) sys.exit(1) diff --git a/src/tests/test_restore_from_nostr_setup.py b/src/tests/test_restore_from_nostr_setup.py index 96bc70e..9f30477 100644 --- a/src/tests/test_restore_from_nostr_setup.py +++ b/src/tests/test_restore_from_nostr_setup.py @@ -74,6 +74,61 @@ def test_handle_new_seed_setup_restore_from_nostr(monkeypatch, tmp_path, capsys) assert labels == ["site1"] +def test_handle_new_seed_setup_restore_from_local_backup(monkeypatch, tmp_path, capsys): + dir_a = tmp_path / "A" + dir_b = tmp_path / "B" + dir_a.mkdir() + dir_b.mkdir() + + pm_src = _init_pm(dir_a, None) + pm_src.notify = lambda *a, **k: None + pm_src.entry_manager.add_entry("site1", 12) + pm_src.backup_manager.create_backup() + backup_path = next( + pm_src.backup_manager.backup_dir.glob("entries_db_backup_*.json.enc") + ) + + pm_new = PasswordManager.__new__(PasswordManager) + pm_new.encryption_mode = EncryptionMode.SEED_ONLY + pm_new.notify = lambda *a, **k: None + + called = {"init": False} + + def init_fp_mgr(): + called["init"] = True + pm_new.fingerprint_manager = object() + + monkeypatch.setattr(pm_new, "initialize_fingerprint_manager", init_fp_mgr) + + def finalize(seed, *, password=None): + assert pm_new.fingerprint_manager is not None + vault, enc_mgr = create_vault(dir_b, seed, TEST_PASSWORD) + cfg_mgr = ConfigManager(vault, dir_b) + backup_mgr = BackupManager(dir_b, cfg_mgr) + entry_mgr = EntryManager(vault, backup_mgr) + pm_new.encryption_manager = enc_mgr + pm_new.vault = vault + pm_new.entry_manager = entry_mgr + pm_new.backup_manager = backup_mgr + pm_new.config_manager = cfg_mgr + pm_new.fingerprint_dir = dir_b + pm_new.current_fingerprint = "fp" + return "fp" + + monkeypatch.setattr(pm_new, "_finalize_existing_seed", finalize) + monkeypatch.setattr("seedpass.core.manager.masked_input", lambda *_: TEST_SEED) + + inputs = iter(["5", str(backup_path)]) + monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs)) + + pm_new.handle_new_seed_setup() + out = capsys.readouterr().out + assert "Index file restored from backup" in out + labels = [e[1] for e in pm_new.entry_manager.list_entries()] + assert labels == ["site1"] + assert called["init"] + + async def _no_snapshot(): return None