artemis_framework 0.1.2__tar.gz → 0.1.3__tar.gz

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

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

artemis_framework-0.1.3/artemis_framework/asphodel/core/encrypt_decrypt.py

@@ -0,0 +1,160 @@
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+
+from artemis_framework.asphodel.core.gpg_context import build_gpg, check_result
+
+
+def encrypt_to_recipients(
+    gpg: "gnupg.GPG",   # type: ignore
+    plaintext: str | bytes,
+    recipients: list[str],
+    *,
+    armor: bool = True,
+    sign: str | None = None,
+    passphrase: str | None = None,
+    always_trust: bool = True,
+    extra_args: list[str] | None = None
+) -> str | bytes | None:
+    result = gpg.encrypt(
+        plaintext,
+        recipients,
+        armor=armor,
+        sign=sign,
+        passphrase=passphrase,
+        always_trust=always_trust,
+        extra_args=extra_args or []
+    )
+
+    if not check_result(result,  label=f"encrypt_to_recipients({recipients})"):
+        return None
+
+    return result.data.decode("ascii") if armor else result.data
+
+
+def encrypt_symmetric(
+    gpg: "gnupg.GPG",   # type: ignore
+    plaintext: str | bytes,
+    passphrase: str,
+    *,
+    armor: bool = True,
+    cipher_algo: str = "AES256"
+) -> str | bytes | None:
+    result = gpg.encrypt(
+        plaintext,
+        recipients=None,
+        symmetric=True,
+        passphrase=passphrase,
+        armor=armor,
+        extra_args=["--cipher-algo", cipher_algo, "--no-symkey-cache"]
+    )
+
+    if not check_result(result, label="encrypt_symmetric"):
+        return None
+
+    return result.data.decode("ascii") if armor else result.data
+
+
+def decrypt_data(
+    gpg: "gnupg.GPG",   # type: ignore
+    ciphertext: str | bytes,
+    *,
+    passphrase: str | None = None,
+    always_trust: bool = True
+) -> bytes | None:
+    result = gpg.decrypt(
+        ciphertext,
+        passphrase=passphrase,
+        always_trust=always_trust
+    )
+
+    if not check_result(result, label="decrypt_data"):
+        return None
+
+    return result.data
+
+
+def encrypt_file(
+    gpg: "gnupg.GPG",   # type: ignore
+    input_path: Path | str,
+    recipients: list[str],
+    *,
+    output_path: Path | str | None = None,
+    armor: bool = True,
+    sign : str | None = None,
+    passphrase: str | None = None,
+    always_trust: bool = True
+) -> Path | None:
+    src = Path(input_path)
+    if not src.exists():
+        print(f"[ERR] encrypt_file: {src} not found.")
+        return None
+
+    dst = Path(output_path) if output_path else src.with_suffix(src.suffix + (".asc" if armor else ".gpg"))
+
+    with open(src, "rb") as file:
+        result = gpg.encrypt_file(
+            file,
+            recipients=recipients,
+            armor=armor,
+            sign=sign,
+            passphrase=passphrase,
+            always_trust=always_trust,
+            output=str(dst)
+        )
+
+    if not check_result(result, label=f"encrypt_file({src.name})"):
+        return None
+
+    print(f"    output: {dst}")
+    return dst
+
+
+def decrypt_file(
+    gpg: "gnupg.GPG",   # type: ignore
+    input_path: Path | str,
+    *,
+    output_path: Path | str | None = None,
+    passphrase: str | None = None,
+    always_trust: bool = True
+) -> Path | None:
+    src = Path(input_path)
+    if not src.exists():
+        print(f"[ERR] decrypt_file: {src} not found.")
+        return None
+
+    if output_path:
+        dst = Path(output_path)
+    else:
+        dst = src.with_suffix("")
+
+    with open(src, "rb") as file:
+        result = gpg.decrypt_file(
+            file,
+            passphrase=passphrase,
+            always_trust=always_trust,
+            output=str(dst)
+        )
+
+    if not check_result(result, label=f"decrypt_file({src.name})"):
+        return None
+
+    print(f"    output: {dst}")
+    return dst
+
+
+def print_decrypt_signature_info(result) -> None:
+    sig_id = getattr(result, "signature_id", None)
+    key_id = getattr(result, "key_id", None)
+    uid = getattr(result, "username", None)
+    trust = getattr(result, "trust_text", None)
+
+    if sig_id or key_id:
+        print("\n  [Signature embedded in ciphertext]")
+        print(f"  signer UID    : {uid or '<unknown>'}")
+        print(f"  key ID        : {key_id or '<unknown>'}")
+        print(f"  trust         : {trust or '<unknown>'}")
+    else:
+        print("\n  [No embedded signature detected]")
+        

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

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