Initial commit

2026-07-12 14:06:45 +04:00
commit a75028d68f
148 changed files with 3275 additions and 0 deletions
@@ -0,0 +1,68 @@
+from __future__ import annotations
+
+from typing import Dict, Any
+
+from modes.keygen import KeygenMode
+from modes.encrypt import EncryptMode
+from modes.decrypt import DecryptMode
+
+
+class AutoMode:
+    """
+    Автоматический режим:
+    1. Генерация ключей (p, q, n, e, d, phi).
+    2. Шифрование текста или числа.
+    3. Расшифрование результата обратно.
+    """
+
+    def run(
+        self,
+        *,
+        text: str | None,
+        m: int | None,
+        p: int | None = None,
+        q: int | None = None,
+        min: int | None = None,
+        max: int | None = None,
+        bits: int | None = None,
+        seed: int | None = None,
+    ) -> Dict[str, Any]:
+        # 1. Генерация ключей
+        keygen = KeygenMode()
+        keys = keygen.run(
+            p=p,
+            q=q,
+            min=min,
+            max=max,
+            bits=bits,
+            seed=seed,
+        )
+
+        n = keys["n"]["dec"]
+        e = keys["e"]["dec"]
+        d = keys["d"]["dec"]
+
+        # 2. Шифрование
+        encrypt = EncryptMode()
+        encrypted = encrypt.run(
+            text=text,
+            m=m,
+            n=n,
+            e=e,
+        )
+
+        # 3. Расшифрование
+        decrypt = DecryptMode()
+        # В зависимости от формата encrypted["output"]
+        if encrypted["mode"] == "text":
+            ciphers = [item["cipher"] for item in encrypted["output"]]
+            decrypted = decrypt.run(c=ciphers, n=n, d=d)
+        else:
+            cipher = encrypted["output"]["cipher"]
+            decrypted = decrypt.run(c=cipher, n=n, d=d)
+
+        return {
+            "keys": keys,
+            "encrypted": encrypted,
+            "decrypted": decrypted,
+        }