aov-dec 1.0.0__py3-none-any.whl

aov_dec/__init__.py

@@ -0,0 +1,135 @@
+import os
+import zstandard as zstd
+from Crypto.Cipher import AES
+from Crypto.Util.Padding import pad, unpad
+
+DICT_PATH = "dict"
+ZSTD_DICT = open(DICT_PATH, "rb").read() if os.path.exists(DICT_PATH) else b""
+
+class JG:
+    MAGIC = bytes.fromhex("22 4A 67")
+
+    @staticmethod
+    def get_name(file_path):
+        base = os.path.basename(file_path)
+        return os.path.splitext(base)[0]
+
+    @staticmethod
+    def _key(name):
+        h = 0
+        for ch in name:
+            c = ord(ch)
+            if 97 <= c <= 122:
+                c -= 32
+            h = ((h * 31) + c) & 0xFFFFFFFF
+            
+        k = bytearray(bytes.fromhex("99 64 b1 b0 6b 03 8d 7f b7 7d b6 a7 54 90 8b 73"))
+        k0, k1, k2, k3 = (h & 0xFF, (h >> 8) & 0xFF, (h >> 16) & 0xFF, (h >> 24) & 0xFF)
+        for i in range(len(k)):
+            k[i] ^= (k0, k1, k2, k3)[i & 3]
+        return bytes(k)
+
+    @staticmethod
+    def decrypt(all_code, name):
+        key = JG._key(name)
+        ct = all_code[8:]
+        cipher = AES.new(key, AES.MODE_CBC, b"\x00" * 16)
+        try:
+            decrypted_data = cipher.decrypt(ct)
+            return unpad(decrypted_data, AES.block_size)
+        except (ValueError, KeyError):
+            raise ValueError("Giải mã mã hóa AES lỗi: Sai key hoặc dữ liệu hỏng.")
+
+    @staticmethod
+    def encrypt(all_code, name):
+        key = JG._key(name)
+        original_len = len(all_code)
+        pt_padded = pad(all_code, AES.block_size)
+        
+        cipher = AES.new(key, AES.MODE_CBC, b"\x00" * 16)
+        ct = cipher.encrypt(pt_padded)
+        hdr = JG.MAGIC + b"\x00" + original_len.to_bytes(4, "little")
+        return hdr + ct
+
+
+class ZSTD:
+    @staticmethod
+    def dec(all_code):
+        compressed_data = all_code[8:]
+        dctx = zstd.ZstdDecompressor(dict_data=zstd.ZstdCompressionDict(ZSTD_DICT))
+        return dctx.decompress(compressed_data)
+
+    @staticmethod
+    def com(all_code):
+        cctx = zstd.ZstdCompressor(level=17, dict_data=zstd.ZstdCompressionDict(ZSTD_DICT))
+        compressed = bytearray(cctx.compress(all_code))
+        compressed[0:0] = len(all_code).to_bytes(4, byteorder="little")
+        compressed[0:0] = b"\"J\x00\xef"
+        return compressed
+
+def auto(file_path):
+    try:
+        with open(file_path, "rb") as f:
+            all_code = f.read()
+            
+        if not all_code: 
+            return
+
+        if all_code[0:3] == JG.MAGIC:
+            new_code = JG.decrypt(all_code, JG.get_name(file_path))
+            with open(file_path, "wb") as f: f.write(new_code)
+            print("\33[1;32mDECRYPT AES\33[1;39m", file_path)
+        elif all_code[0:4] == b"\"J\x00\xef":
+            new_code = ZSTD.dec(all_code)
+            with open(file_path, "wb") as f: f.write(new_code)
+            print("\33[1;32mDECOMPRESS ZSTD\33[1;39m", file_path)
+        else:
+            new_code = ZSTD.com(all_code)
+            with open(file_path, "wb") as f: f.write(new_code)
+            print("\33[1;36mCOMPRESS ZSTD\33[1;39m", file_path)
+            
+    except Exception as e:
+        print(f"\33[1;31mERROR {file_path}: {e}\33[1;39m")
+
+def mode(file_path, mode):
+    try:
+        with open(file_path, "rb") as f:
+            all_code = f.read()
+            
+        if not all_code: 
+            return
+
+        if mode == "COM_ZSTD":
+            if all_code[0:4] == b"\"J\x00\xef":
+                raise ValueError("File đã được nén ZSTD")
+            new_code = ZSTD.com(all_code)
+            msg = "\33[1;36mCOMPRESS ZSTD\33[1;39m"
+
+        elif mode == "DEC_ZSTD":
+            if all_code[0:4] == b"\"J\x00\xef":
+                new_code = ZSTD.dec(all_code)
+                msg = "\33[1;32mDECOMPRESS ZSTD\33[1;39m"
+            else:
+                raise ValueError("File không phải là file ZSTD")
+
+        elif mode == "ENC_AES":
+            if all_code[0:3] == JG.MAGIC:
+                raise ValueError("File đã được mã hóa AES")
+            new_code = JG.encrypt(all_code, JG.get_name(file_path))
+            msg = "\33[1;36mENCRYPT AES\33[1;39m"
+
+        elif mode == "DEC_AES":
+            if all_code[0:3] == JG.MAGIC:
+                new_code = JG.decrypt(all_code, JG.get_name(file_path))
+                msg = "\33[1;32mDECRYPT AES\33[1;39m"
+            else:
+                raise ValueError("File không phải là file AES")
+        else:
+            return
+
+        with open(file_path, "wb") as f:
+            f.write(new_code)
+        print(msg, file_path)
+        
+    except Exception as e:
+        print(f"\33[1;31mERROR {file_path}: {e}\33[1;39m")

aov_dec-1.0.0.dist-info/METADATA

@@ -0,0 +1,47 @@
+Metadata-Version: 2.1
+Name: aov-dec
+Version: 1.0.0
+Summary: Compress/Decompress AOV File
+Author: Vinh LC
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: zstandard
+Requires-Dist: pycryptodome
+
+# 🛠️ File Processor (ZSTD & AES)
+
+Một công cụ Python nhỏ gọn hỗ trợ nén, giải nén dữ liệu bằng **ZSTD** và mã hóa, giải mã bảo mật bằng **AES**. 
+
+---
+
+## 🚀 Tính năng chính
+
+* ⚡ **Nén & Giải nén tốc độ cao:** Sử dụng thuật toán `ZSTD`.
+* 🔒 **Bảo mật dữ liệu:** Mã hóa và giải mã với chuẩn mã hóa nâng cao `AES`.
+* 🎨 **Giao diện dòng lệnh trực quan:** Hỗ trợ hiển thị màu sắc (ANSI escape codes) giúp dễ dàng theo dõi.
+
+---
+
+## 🛠️ Hướng dẫn Sử dụng
+
+Để xem hướng dẫn nhanh trong terminal, bạn có thể gọi hàm `help()` có sẵn trong công cụ.
+
+### 1. Chế độ Tự động (Auto Mode)
+Tự động phát hiện và xử lý file dựa trên định dạng hoặc cấu trúc sẵn có.
+```python
+auto(file_path)
+```
+
+### 2. Chế độ Thủ công (Mode)
+Xử lý file theo chế độ đã chọn
+
+| Tùy chọn Mode | Loại tác vụ | Mô tả chi tiết |
+| :--- | :--- | :--- |
+| **`COM_ZSTD`** | `Compression` | Nén tệp tin đích bằng thuật toán ZSTD để giảm dung lượng. |
+| **`DEC_ZSTD`** | `Decompression` | Giải nén tệp tin đã nén bằng ZSTD về trạng thái gốc ban đầu. |
+| **`ENC_AES`** | `Encryption` | Mã hóa bảo mật tệp tin bằng thuật toán mã hóa AES. |
+| **`DEC_AES`** | `Decryption` | Giải mã tệp tin đã được mã hóa bằng AES trở lại bình thường. |
+
+```python
+mode(file_path, mode)
+```

aov_dec-1.0.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+aov_dec/__init__.py,sha256=SApVFvyKdmmG20_j2_3CmU8ZNRydHQZ6VND07MUbp2g,4705
+aov_dec-1.0.0.dist-info/METADATA,sha256=8kp-Vx5oI9q3B9LLpP18iUDxIELlK8vjaG92EOisNfA,1744
+aov_dec-1.0.0.dist-info/WHEEL,sha256=BNRMDyzLkkcmlv0J8ppDQkk2VED33SesJDynr9ED1gc,91
+aov_dec-1.0.0.dist-info/top_level.txt,sha256=of9hsAfl6EJZ9QKVSiCOGWNn2iUp205Uj0Aq1YBqFeA,8
+aov_dec-1.0.0.dist-info/RECORD,,

aov_dec-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.3.4)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

aov_dec-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+aov_dec