Update app.py

app.py

@@ -1,81 +1,48 @@
 import gradio as gr
 import subprocess
 import tempfile
-import shutil
 import os
-import zipfile
+import shutil
 
-def build_stub_jar(zip_file) -> tuple[str, str | None]:
-    if zip_file is None:
-        return "⚠️ Пожалуйста, загрузите .zip архив с .java файлами.", None
+def convert_apk_to_jar(file_obj) -> tuple[str, str | None]:
+    if file_obj is None:
+        return "⚠️ Пожалуйста, загрузите .apk или .dex файл.", None
 
-    workdir = tempfile.mkdtemp(prefix="stub_builder_")
+    workdir = tempfile.mkdtemp(prefix="dex2jar_")
     log_lines = []
     
     try:
-        src_dir = os.path.join(workdir, "src")
-        classes_dir = os.path.join(workdir, "classes")
-        os.makedirs(src_dir, exist_ok=True)
-        os.makedirs(classes_dir, exist_ok=True)
+        out_jar = os.path.join(workdir, "app_stubs.jar")
+        file_name = os.path.basename(file_obj.name)
         
-        log_lines.append("📦 Распаковка архива...")
-        with zipfile.ZipFile(zip_file.name, 'r') as zf:
-            zf.extractall(src_dir)
-            
-        # Ищем все .java файлы (включая подпапки)
-        java_files = []
-        for root, dirs, files in os.walk(src_dir):
-            for file in files:
-                if file.endswith(".java"):
-                    java_files.append(os.path.join(root, file))
-                    
-        if not java_files:
-            log_lines.append("❌ В архиве не найдено .java файлов!")
-            return "\n".join(log_lines), None
-            
-        log_lines.append(f"🔍 Найдено .java файлов: {len(java_files)}")
+        log_lines.append(f"📦 Файл загружен: {file_name}")
+        log_lines.append("⚙️ Запуск dex2jar (конвертация Dalvik -> JVM)...")
+        log_lines.append("⏳ Это может занять 1-3 минуты для больших приложений (например, Telegram).")
         
-        # Создаем файл со списком исходников (чтобы не упереться в лимит длины команды)
-        sources_list_path = os.path.join(workdir, "sources.txt")
-        with open(sources_list_path, "w", encoding="utf-8") as f:
-            for jf in java_files:
-                f.write(f'"{jf}"\n')
-
-        # ─── ШАГ 1: Компиляция (javac) ──────────────────────────────
-        log_lines.append("\n⚙️ Компиляция Java исходников...")
-        javac_cmd = [
-            "javac",
-            "-encoding", "UTF-8",
-            "-g:none", # Убираем отладочную инфу для уменьшения веса
-            "-nowarn", # Игнорируем варнинги
-            "-d", classes_dir,
-            f"@{sources_list_path}"
+        # Запускаем d2j-dex2jar
+        cmd = [
+            "d2j-dex2jar",
+            file_obj.name,
+            "-o", out_jar,
+            "--force"
         ]
         
-        result = subprocess.run(javac_cmd, capture_output=True, text=True)
+        # Увеличиваем лимит памяти для Java, так как Telegram огромный
+        env = os.environ.copy()
+        env["JAVA_OPTS"] = "-Xmx2048m"
+        
+        result = subprocess.run(cmd, capture_output=True, text=True, env=env)
+        
         if result.stdout: log_lines.append(result.stdout)
         if result.stderr: log_lines.append(result.stderr)
         
-        if result.returncode != 0:
-            log_lines.append(f"\n❌ Ошибка компиляции (код {result.returncode})")
+        if not os.path.exists(out_jar) or result.returncode != 0:
+            log_lines.append(f"\n❌ Ошибка конвертации (код {result.returncode})")
             return "\n".join(log_lines), None
 
-        log_lines.append("✅ Успешно скомпилировано в .class файлы")
-
-        # ─── ШАГ 2: Упаковка в JAR ─────────────────────────────────
-        out_jar = os.path.join(workdir, "custom_stubs.jar")
-        log_lines.append("\n📦 Упаковка в JAR...")
-        
-        jar_cmd = ["jar", "cf", out_jar, "-C", classes_dir, "."]
-        result2 = subprocess.run(jar_cmd, capture_output=True, text=True)
-        
-        if result2.returncode != 0:
-            log_lines.append(f"\n❌ Ошибка создания JAR (код {result2.returncode})")
-            if result2.stderr: log_lines.append(result2.stderr)
-            return "\n".join(log_lines), None
-            
         jar_size = os.path.getsize(out_jar)
-        log_lines.append(f"🎉 Готово! custom_stubs.jar ({jar_size:,} bytes)")
+        log_lines.append(f"\n✅ Готово! app_stubs.jar ({jar_size // 1024 // 1024} MB)")
+        log_lines.append("🚀 Теперь загрузи этот JAR во второе поле твоего Kotlin-компилятора!")
         
         return "\n".join(log_lines), out_jar
 
@@ -85,23 +52,23 @@ def build_stub_jar(zip_file) -> tuple[str, str | None]:
 
 CSS = """
 @import url('https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;600;700&family=Space+Grotesk:wght@400;600;700&display=swap');
-:root { --bg: #0d0e14; --surface: #13141e; --border: #22253a; --accent: #3ecf8e; --text: #e2e4f0; }
+:root { --bg: #0d0e14; --surface: #13141e; --border: #22253a; --accent: #ff4757; --text: #e2e4f0; }
 body, .gradio-container { background: var(--bg) !important; font-family: 'Space Grotesk', sans-serif !important; color: var(--text) !important; }
 .gr-panel, .gr-box, .gr-form { background: var(--surface) !important; border: 1px solid var(--border) !important; border-radius: 12px !important; }
 """
 
-with gr.Blocks(css=CSS, title="Java Stub Generator") as demo:
-    gr.HTML("<h1 style='text-align:center; padding:20px;'>☕ Java ZIP → Stub JAR</h1>")
+with gr.Blocks(css=CSS, title="APK to Stub Generator") as demo:
+    gr.HTML("<h1 style='text-align:center; padding:20px;'>🤖 APK/DEX → Stub JAR</h1><p style='text-align:center; color:#888;'>Генератор стабов для Xposed-модулей без декомпиляции в Java</p>")
     
     with gr.Row():
         with gr.Column():
-            zip_input = gr.File(label="Загрузите ZIP с .java файлами", file_types=[".zip"])
-            build_btn = gr.Button("⚙️ Сгенерировать JAR", variant="primary")
+            file_input = gr.File(label="Загрузите .apk или .dex файл", file_types=[".apk", ".dex"])
+            build_btn = gr.Button("⚙️ Создать JAR", variant="primary")
         with gr.Column():
-            log_output = gr.Textbox(label="Лог сборки", lines=10, interactive=False)
-            jar_output = gr.File(label="Скачать готовый .jar", interactive=False)
+            log_output = gr.Textbox(label="Лог конвертации", lines=12, interactive=False)
+            jar_output = gr.File(label="Скачать готовый .jar стаб", interactive=False)
 
-    build_btn.click(fn=build_stub_jar, inputs=[zip_input], outputs=[log_output, jar_output])
+    build_btn.click(fn=convert_apk_to_jar, inputs=[file_input], outputs=[log_output, jar_output])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)