mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-08 07:18:47 +00:00
127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Generate a SeedPass test profile with realistic entries.
|
|
|
|
This script populates a profile directory with a variety of entry types.
|
|
If the profile does not exist, a new BIP-39 seed phrase is generated and
|
|
stored encrypted. A clear text copy is written to ``seed_phrase.txt`` so
|
|
it can be reused across devices.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import random
|
|
from pathlib import Path
|
|
|
|
from bip_utils import Bip39MnemonicGenerator, Bip39WordsNum, Bip39Languages
|
|
|
|
from constants import APP_DIR
|
|
from utils.key_derivation import derive_key_from_password, derive_index_key
|
|
from password_manager.encryption import EncryptionManager
|
|
from password_manager.vault import Vault
|
|
from password_manager.config_manager import ConfigManager
|
|
from password_manager.backup import BackupManager
|
|
from password_manager.entry_management import EntryManager
|
|
|
|
DEFAULT_PASSWORD = "testpassword"
|
|
|
|
|
|
def initialize_profile(profile_name: str) -> tuple[str, EntryManager, Path]:
|
|
"""Create or load a profile and return the seed phrase and manager."""
|
|
profile_dir = APP_DIR / profile_name
|
|
profile_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
seed_key = derive_key_from_password(DEFAULT_PASSWORD)
|
|
seed_mgr = EncryptionManager(seed_key, profile_dir)
|
|
seed_file = profile_dir / "parent_seed.enc"
|
|
|
|
if seed_file.exists():
|
|
seed_phrase = seed_mgr.decrypt_parent_seed()
|
|
else:
|
|
seed_phrase = (
|
|
Bip39MnemonicGenerator(Bip39Languages.ENGLISH)
|
|
.FromWordsNumber(Bip39WordsNum.WORDS_NUM_12)
|
|
.ToStr()
|
|
)
|
|
seed_mgr.encrypt_parent_seed(seed_phrase)
|
|
clear_path = profile_dir / "seed_phrase.txt"
|
|
clear_path.write_text(seed_phrase)
|
|
clear_path.chmod(0o600)
|
|
|
|
index_key = derive_index_key(seed_phrase)
|
|
enc_mgr = EncryptionManager(index_key, profile_dir)
|
|
vault = Vault(enc_mgr, profile_dir)
|
|
cfg_mgr = ConfigManager(vault, profile_dir)
|
|
backup_mgr = BackupManager(profile_dir, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
return seed_phrase, entry_mgr, profile_dir
|
|
|
|
|
|
def random_secret(length: int = 16) -> str:
|
|
alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567"
|
|
return "".join(random.choice(alphabet) for _ in range(length))
|
|
|
|
|
|
def populate(entry_mgr: EntryManager, seed: str, count: int) -> None:
|
|
"""Add ``count`` entries of varying types to the vault."""
|
|
start_index = entry_mgr.get_next_index()
|
|
for i in range(count):
|
|
idx = start_index + i
|
|
kind = idx % 7
|
|
if kind == 0:
|
|
entry_mgr.add_entry(
|
|
label=f"site-{idx}.example.com",
|
|
length=12,
|
|
username=f"user{idx}",
|
|
url=f"https://example{idx}.com",
|
|
notes=f"Website account {idx}",
|
|
custom_fields=[{"key": "id", "value": str(idx)}],
|
|
)
|
|
elif kind == 1:
|
|
entry_mgr.add_totp(f"totp-generated-{idx}", seed)
|
|
elif kind == 2:
|
|
entry_mgr.add_totp(f"totp-imported-{idx}", seed, secret=random_secret())
|
|
elif kind == 3:
|
|
entry_mgr.add_ssh_key(f"ssh-{idx}", seed, notes=f"SSH key for server {idx}")
|
|
elif kind == 4:
|
|
entry_mgr.add_seed(
|
|
f"derived-seed-{idx}", seed, words_num=24, notes=f"Seed {idx}"
|
|
)
|
|
elif kind == 5:
|
|
entry_mgr.add_nostr_key(f"nostr-{idx}", notes=f"Nostr key {idx}")
|
|
else:
|
|
entry_mgr.add_pgp_key(
|
|
f"pgp-{idx}",
|
|
seed,
|
|
user_id=f"user{idx}@example.com",
|
|
notes=f"PGP key {idx}",
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(
|
|
description="Create or extend a SeedPass test profile"
|
|
)
|
|
parser.add_argument(
|
|
"--profile",
|
|
default="test_profile",
|
|
help="profile name inside ~/.seedpass",
|
|
)
|
|
parser.add_argument(
|
|
"--count",
|
|
type=int,
|
|
default=100,
|
|
help="number of entries to add",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
seed, entry_mgr, dir_path = initialize_profile(args.profile)
|
|
print(f"Using profile directory: {dir_path}")
|
|
print(f"Parent seed: {seed}")
|
|
populate(entry_mgr, seed, args.count)
|
|
print(f"Added {args.count} entries.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|