artemis_framework 0.1.1__tar.gz → 0.1.2__tar.gz

{artemis_framework-0.1.1 → artemis_framework-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: artemis_framework
-Version: 0.1.1
+Version: 0.1.2
 Summary: A package for doing great things!
 License: MIT
 License-File: LICENSE

artemis_framework-0.1.2/artemis_framework/asphodel/core/key_management.py

@@ -0,0 +1,193 @@
+from __future__ import annotations
+
+import sys
+import textwrap
+from pathlib import Path
+
+from artemis_framework.asphodel.core.gpg_context import check_result
+
+
+RSA_KEY_PARAMS = textwrap.dedent("""\
+    Key-Type: RSA
+    Key-Length: 4096
+    Subkey-Type: RSA
+    Subkey-Length: 4096
+    Name-Real: {name}
+    Name-Email: {email}
+    Name-Comment: {comment}
+    Expire-Date: {expire}
+    %no-protection
+""")
+
+ECC_KEY_PARAMS = textwrap.dedent("""\
+    Key-Type: EDDSA
+    Key-Curve: ed25519
+    Subkey-Type: ECDH
+    Subkey-Curve: cv25519
+    Name-Real: {name}
+    Name-Email: {email}
+    Name-Comment: {comment}
+    Expire-Date: {expire}
+    %no-protection
+""")
+
+
+def generate_key(
+    gpg: "gnupg.GPG",    # type: ignore[name-defined]
+    name: str,
+    email: str,
+    *,
+    comment: str = "",
+    expire: str = "2y",
+    algorithm: str = "rsa",
+    passphrase: str | None = None
+) -> str | None:
+    template = RSA_KEY_PARAMS if algorithm.lower() == "rsa" else ECC_KEY_PARAMS
+
+    params = template.format(
+        name=name,
+        email=email,
+        comment=comment,
+        expire=expire
+    )
+
+    if passphrase:
+        params = params.replace(
+            "%no-protection",
+            f"Passphrase: {passphrase}"
+        )
+
+    print(f"Generating {algorithm.upper()} key for <{email}> … (this may take a moment)")
+    result = gpg.gen_key(params)
+
+    if check_result(result, label=f"gen_key({email})"):
+        return result.fingerprint
+    return None
+
+
+def list_keys_verbose(gpg: "gnupg.GPG", *, secret: bool = False) -> list[dict]:  # type: ignore[name-defined]
+    keys = gpg.list_keys(secret)
+    kind = "secret" if secret else "public"
+
+    print(f"\n{'='*60}")
+    print(f"{kind.upper()} KEYS   ({len(keys)} total)")
+    print(f"{'='*60}")
+
+    for key in keys:
+        uid_str = "; ".join(key.get("uids", ["<no UID>"]))
+        print("\n")
+
+        print(f"Fingerprint     : {key['fingerprint']}")
+        print(f"Key ID          : {key['keyid']}")
+        print(f"UID(s)          : {uid_str}")
+        print(f"Type/Length     : {key.get('type', '?')} / {key.get('length', '?')}")
+        print(f"Trust           : {key.get('trust', '?')}")
+        print(f"Expires         : {key.get('expires', 'never') or 'never'}")
+        print(f"Subkeys         : {len(key.get('subkeys', []))}")
+
+    return list(keys)
+
+
+def find_key(gpg: "gnupg.GPG", identifier: str, *, secret: bool = False) -> dict | None:    # type: ignore[name-defined]
+    keys = gpg.list_keys(secret)
+    ident_lower = identifier.lower()
+
+    for key in keys:
+        if ident_lower in key.get("fingerprint", "").lower():
+            return dict(key)
+        for uid in key.get("uids", []):
+            if ident_lower in uid.lower():
+                return dict(key)
+    return None
+
+
+def export_public_key(
+        gpg: "gnupg.GPG",   # type: ignore[name-defined]
+        fingerprint: str,
+        *,
+        armor: bool = True,
+        output_path: Path | None = None
+) -> str | bytes:
+    data = gpg.export_keys(fingerprint, armor=armor)
+
+    if not data:
+        print(f"[ERR] export_public_key: no data returned for {fingerprint}")
+        return b"" if not armor else ""
+
+    if output_path:
+        Path(output_path).write_text(data) if armor else Path(output_path).write_bytes(data.encode())
+        print(f"[OK]  Public key written to » {output_path}")
+    else:
+        print(f"[OK]  export_public_key({fingerprint[:16]}…)")
+
+    return data
+
+
+def export_secret_key(
+    gpg: "gnupg.GPG",    # type: ignore[name-defined]
+    fingerprint: str,
+    *,
+    armor: bool = True,
+    passphrase: str | None = None,
+    output_path: Path | None = None
+) -> str | bytes:
+    data = gpg.export_keys(
+        fingerprint,
+        secret=True,
+        armor=armor,
+        passphrase=passphrase
+    )
+
+    if not data:
+        print(f"[ERR] export_secret_key: no data returned for {fingerprint}")
+        return b"" if not armor else ""
+
+    if output_path:
+        Path(output_path).write_text(data) if armor else Path(output_path).write_bytes(data.encode())
+        print(f"[OK]  Secret key written to » {output_path}")
+    else:
+        print(f"[OK]  secret_key_export({fingerprint[:16]}…)")
+
+    return data
+
+
+def import_key_data(
+    gpg: "gnupg.GPG",   # type: ignore[name-defined],
+    key_data: str | bytes,
+    *,
+    label: str = "import"
+) -> list[str]:
+    result = gpg.import_keys(key_data)
+    check_result(result, label=label)
+
+    fps = result.fingerprints
+    print(f"    imported : {len(fps)} key(s)")
+    for fp in fps:
+        print(f"        {fp}")
+
+    return fps
+
+
+def import_key_file(
+    gpg: "gnupg.GPG",   # type: ignore[name-defined],
+    path: Path | str,
+    *,
+    label: str | None = None
+) -> list[str]:
+    p = Path(path)
+    label = label or f"import_key_file({p.name})"
+    key_data = p.read_text() if p.suffix in (".asc", ".txt") else p.read_bytes()
+    return import_key_data(gpg, key_data, label=label)
+
+
+def delete_key(
+        gpg: "gnupg.GPG",   # type: ignore
+        fingerprint: str,
+        *,
+        secret: bool = False,
+        passphrase: str | None = None
+) -> bool:
+    kind = "secret" if secret else "public"
+    result = gpg.delete_keys(fingerprint, secret=secret, passphrase=passphrase)
+    return check_result(result, label=f"delete_key({kind}, {fingerprint[:16]}…)")
+

{artemis_framework-0.1.1 → artemis_framework-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "artemis_framework"
-version = "0.1.1"
+version = "0.1.2"
 description = "A package for doing great things!"
 authors = ["Dylan Garrett"]
 license = "MIT"