mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-10 08:19:23 +00:00
test: cover legacy migration prompt and sync
This commit is contained in:
@@ -1154,6 +1154,15 @@ class PasswordManager:
|
||||
fingerprint_dir=self.fingerprint_dir,
|
||||
config_manager=self.config_manager,
|
||||
)
|
||||
|
||||
migrated = False
|
||||
try:
|
||||
self.vault.load_index()
|
||||
migrated = getattr(self.vault, "migrated_from_legacy", False)
|
||||
except RuntimeError as exc:
|
||||
print(colored(str(exc), "red"))
|
||||
sys.exit(1)
|
||||
|
||||
self.entry_manager = EntryManager(
|
||||
vault=self.vault,
|
||||
backup_manager=self.backup_manager,
|
||||
@@ -1213,6 +1222,9 @@ class PasswordManager:
|
||||
delta_since=self.delta_since or None,
|
||||
)
|
||||
|
||||
if migrated and not self.offline_mode:
|
||||
self.start_background_vault_sync()
|
||||
|
||||
logger.debug("Managers re-initialized for the new fingerprint.")
|
||||
|
||||
except Exception as e:
|
||||
|
@@ -3,6 +3,9 @@
|
||||
from pathlib import Path
|
||||
from typing import Optional, Union
|
||||
from os import PathLike
|
||||
import shutil
|
||||
|
||||
from termcolor import colored
|
||||
|
||||
from .encryption import EncryptionManager
|
||||
|
||||
@@ -29,17 +32,47 @@ class Vault:
|
||||
|
||||
# ----- Password index helpers -----
|
||||
def load_index(self) -> dict:
|
||||
"""Return decrypted password index data as a dict, applying migrations."""
|
||||
"""Return decrypted password index data as a dict, applying migrations.
|
||||
|
||||
If a legacy ``seedpass_passwords_db.json.enc`` file is detected, the
|
||||
user is prompted to migrate it. A backup copy of the legacy file (and
|
||||
its checksum) is saved under ``legacy_backups`` within the fingerprint
|
||||
directory before renaming to the new filename.
|
||||
"""
|
||||
|
||||
legacy_file = self.fingerprint_dir / "seedpass_passwords_db.json.enc"
|
||||
self.migrated_from_legacy = False
|
||||
if legacy_file.exists() and not self.index_file.exists():
|
||||
print(colored("Legacy index detected.", "yellow"))
|
||||
resp = (
|
||||
input("Would you like to migrate this to the new index format? [y/N]: ")
|
||||
.strip()
|
||||
.lower()
|
||||
)
|
||||
if resp != "y":
|
||||
raise RuntimeError("Migration declined by user")
|
||||
|
||||
legacy_checksum = (
|
||||
self.fingerprint_dir / "seedpass_passwords_db_checksum.txt"
|
||||
)
|
||||
backup_dir = self.fingerprint_dir / "legacy_backups"
|
||||
backup_dir.mkdir(exist_ok=True)
|
||||
shutil.copy2(legacy_file, backup_dir / legacy_file.name)
|
||||
if legacy_checksum.exists():
|
||||
shutil.copy2(legacy_checksum, backup_dir / legacy_checksum.name)
|
||||
|
||||
legacy_file.rename(self.index_file)
|
||||
if legacy_checksum.exists():
|
||||
legacy_checksum.rename(
|
||||
self.fingerprint_dir / "seedpass_entries_db_checksum.txt"
|
||||
)
|
||||
self.migrated_from_legacy = True
|
||||
print(
|
||||
colored(
|
||||
"Migration complete. Original index backed up to 'legacy_backups'",
|
||||
"green",
|
||||
)
|
||||
)
|
||||
|
||||
data = self.encryption_manager.load_json_data(self.index_file)
|
||||
from .migrations import apply_migrations, LATEST_VERSION
|
||||
|
Reference in New Issue
Block a user