90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
import argparse
|
|
import json
|
|
import logging
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from clan_cli.errors import ClanError
|
|
from clan_cli.git import commit_files
|
|
|
|
from . import sops
|
|
from .secrets import update_secrets
|
|
from .sops import default_admin_key_path, generate_private_key
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def extract_public_key(filepath: Path) -> str:
|
|
"""
|
|
Extracts the public key from a given text file.
|
|
"""
|
|
try:
|
|
with filepath.open() as file:
|
|
for line in file:
|
|
# Check if the line contains the public key
|
|
if line.startswith("# public key:"):
|
|
# Extract and return the public key part after the prefix
|
|
return line.strip().split(": ")[1]
|
|
except FileNotFoundError as e:
|
|
msg = f"The file at {filepath} was not found."
|
|
raise ClanError(msg) from e
|
|
except OSError as e:
|
|
msg = f"An error occurred while extracting the public key: {e}"
|
|
raise ClanError(msg) from e
|
|
|
|
msg = f"Could not find the public key in the file at {filepath}."
|
|
raise ClanError(msg)
|
|
|
|
|
|
def generate_key() -> str:
|
|
path = default_admin_key_path()
|
|
if path.exists():
|
|
log.info(f"Key already exists at {path}")
|
|
return extract_public_key(path)
|
|
priv_key, pub_key = generate_private_key(out_file=path)
|
|
log.info(
|
|
f"Generated age private key at '{default_admin_key_path()}' for your user. Please back it up on a secure location or you will lose access to your secrets."
|
|
)
|
|
return pub_key
|
|
|
|
|
|
def generate_command(args: argparse.Namespace) -> None:
|
|
pub_key = generate_key()
|
|
log.info(
|
|
f"Also add your age public key to the repository with: \nclan secrets users add <username> {pub_key}"
|
|
)
|
|
|
|
|
|
def show_command(args: argparse.Namespace) -> None:
|
|
key = sops.maybe_get_admin_public_key()
|
|
if not key:
|
|
msg = "No public key found"
|
|
raise ClanError(msg)
|
|
json.dump(key.as_dict(), sys.stdout, indent=2, sort_keys=True)
|
|
|
|
|
|
def update_command(args: argparse.Namespace) -> None:
|
|
flake_dir = args.flake.path
|
|
commit_files(update_secrets(flake_dir), flake_dir, "Updated secrets with new keys.")
|
|
|
|
|
|
def register_key_parser(parser: argparse.ArgumentParser) -> None:
|
|
subparser = parser.add_subparsers(
|
|
title="command",
|
|
description="the command to run",
|
|
help="the command to run",
|
|
required=True,
|
|
)
|
|
|
|
parser_generate = subparser.add_parser("generate", help="generate age key")
|
|
parser_generate.set_defaults(func=generate_command)
|
|
|
|
parser_show = subparser.add_parser("show", help="show public key")
|
|
parser_show.set_defaults(func=show_command)
|
|
|
|
parser_update = subparser.add_parser(
|
|
"update",
|
|
help="re-encrypt all secrets with current keys (useful when changing keys)",
|
|
)
|
|
parser_update.set_defaults(func=update_command)
|