Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import zipfile
 import tempfile
 import json
+import re
 import subprocess
 import UnityPy
 from fastapi import FastAPI, UploadFile, File
@@ -21,7 +22,6 @@ app.add_middleware(
 
 @app.post("/generate_dll")
 async def generate_dll(binary_file: UploadFile = File(...), metadata_file: UploadFile = File(...)):
-    # Bu adımı değiştirmedik, DummyDll üretmek isteyenler için sağlam duruyor
     temp_dir = tempfile.mkdtemp()
     binary_path = os.path.join(temp_dir, binary_file.filename)
     metadata_path = os.path.join(temp_dir, metadata_file.filename)
@@ -46,112 +46,103 @@ async def extract_dialogues(file: UploadFile = File(...), assembly_file: Optiona
     env = UnityPy.load(asset_path)
     zip_filename = os.path.join(temp_dir, f"{file.filename}_dialogues.zip")
     extracted_count = 0
-    log_content = ["=== UNITYPY İKİLİ (BINARY) İŞLEM RAPORU ==="]
     
     with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
         for obj in env.objects:
             if obj.type.name in ["TextAsset", "MonoBehaviour"]:
-                # Dosya adını güvenli şekilde al
+                # Dosya ismini geri paketlerken tanıyabilmemiz için ID'yi garanti altına alıyoruz
                 obj_name = f"{obj.type.name}_{obj.path_id}"
-                try:
-                    data = obj.read()
-                    if hasattr(data, "name") and data.name: obj_name = data.name
-                except: pass
                 
-                log_content.append(f"\nİnceleniyor: {obj.type.name} (ID: {obj.path_id})")
-                
-                if obj.type.name == "TextAsset":
-                    try:
-                        content = obj.read().text.encode('utf-8')
-                        zipf.writestr(f"{obj_name}.txt", content)
-                        extracted_count += 1
-                        log_content.append(f"  -> Başarılı: {obj_name}.txt oluşturuldu.")
-                    except: pass
-                    
-                elif obj.type.name == "MonoBehaviour":
-                    # TİP AĞACI ÇÖKERSE DOĞRUDAN HAM BELLEK TARAMASI YAP
+                if obj.type.name == "MonoBehaviour":
                     raw_data = obj.get_raw_data()
                     found_strings = {}
-                    i = 28 # MonoBehaviour ana başlığını atla
+                    i = 28
                     
                     while i <= len(raw_data) - 4:
                         length = int.from_bytes(raw_data[i:i+4], 'little')
-                        # Mantıklı bir metin uzunluğu arıyoruz (2 ile 5000 karakter arası)
                         if 2 <= length <= 5000 and (i + 4 + length) <= len(raw_data):
                             str_bytes = raw_data[i+4 : i+4+length]
                             pad_len = (4 - (length % 4)) % 4
                             
-                            # Padding (Boşluk) kontrolü
                             if (i + 4 + length + pad_len) <= len(raw_data):
                                 padding = raw_data[i+4+length : i+4+length+pad_len]
                                 if all(b == 0 for b in padding):
                                     try:
                                         text = str_bytes.decode('utf-8')
-                                        # Metin sadece sayılardan veya bozuk sembollerden oluşmuyorsa al
                                         if len(text.strip()) > 1 and any(c.isalpha() for c in text):
                                             found_strings[text] = text
                                         i += 4 + length + pad_len
                                         continue
                                     except UnicodeDecodeError:
                                         pass
-                        i += 4 # Unity bayt dizilimleri her zaman 4'ün katıdır
+                        i += 4
                     
                     if found_strings:
                         zipf.writestr(f"{obj_name}_raw.json", json.dumps(found_strings, indent=4, ensure_ascii=False).encode('utf-8'))
                         extracted_count += 1
-                        log_content.append(f"  -> BİNARY BAŞARISI: {obj_name}_raw.json ({len(found_strings)} metin bulundu)")
-                    else:
-                        log_content.append(f"  -> Atlandı: Bellekte metin bulunamadı.")
-
-        if extracted_count == 0: log_content.append("\nSONUÇ: Hedef dosyada metin bulunamadı.")
-        zipf.writestr("islem_raporu.txt", "\n".join(log_content).encode('utf-8'))
+                        
+                elif obj.type.name == "TextAsset":
+                    try:
+                        content = obj.read().text.encode('utf-8')
+                        zipf.writestr(f"{obj_name}.txt", content)
+                        extracted_count += 1
+                    except: pass
 
     return FileResponse(zip_filename, media_type="application/zip", filename=f"{file.filename}_dialogues.zip")
-# app.py'nin son kısmındaki repack fonksiyonunu bu şekilde güncelle:
+
 @app.post("/repack")
 async def repack_dialogues(asset_file: UploadFile = File(...), mod_zip: UploadFile = File(...)):
-    # Geçici dosyaları tanımla
     temp_dir = tempfile.mkdtemp()
     asset_path = os.path.join(temp_dir, asset_file.filename)
     zip_path = os.path.join(temp_dir, mod_zip.filename)
-    
     with open(asset_path, "wb") as f: f.write(await asset_file.read())
     with open(zip_path, "wb") as f: f.write(await mod_zip.read())
 
-    # Paketi yükle
+    # DİNAMİK PAKETLEME VE IŞIK HIZINDA O(1) ALGORİTMASI
     env = UnityPy.load(asset_path)
     
+    # 1. Zip'in içindeki çevirileri ID'lerine göre hafızaya al (Hız Aşırtma)
+    path_id_to_translations = {}
     with zipfile.ZipFile(zip_path, 'r') as zipf:
         for file_info in zipf.infolist():
             filename = file_info.filename
-            content = zipf.read(filename)
+            if filename.endswith(".json"):
+                # Regex ile dosya isminden ID'yi güvenle kopar
+                match = re.search(r'_(\d+)(_raw)?\.json$', filename)
+                if match:
+                    path_id = int(match.group(1))
+                    content = zipf.read(filename)
+                    path_id_to_translations[path_id] = json.loads(content.decode('utf-8'))
+
+    # 2. Oyun dosyasını sadece BİR KERE tara (Zaman Aşımını Kırar)
+    for obj in env.objects:
+        if obj.path_id in path_id_to_translations:
+            translations = path_id_to_translations[obj.path_id]
             
-            for obj in env.objects:
-                obj_name = f"{obj.type.name}_{obj.path_id}"
-                try:
-                    if hasattr(obj.read(), "name") and obj.read().name: obj_name = obj.read().name
-                except: pass
+            if obj.type.name == "MonoBehaviour":
+                raw_data = obj.get_raw_data()
+                patched_data = bytearray(raw_data)
                 
-                # Sadece _raw.json olanları işle (Binary Patching)
-                if obj.type.name == "MonoBehaviour" and filename == f"{obj_name}_raw.json":
-                    translations = json.loads(content.decode('utf-8'))
-                    raw_data = obj.get_raw_data()
-                    patched_data = bytearray(raw_data)
+                for orig_str, trans_str in translations.items():
+                    if orig_str == trans_str: continue
                     
-                    for orig_str, trans_str in translations.items():
-                        if orig_str == trans_str: continue
-                        orig_bytes = orig_str.encode('utf-8')
-                        trans_bytes = trans_str.encode('utf-8')
-                        
-                        # Basit bayt değiştirme (Kapsayıcıyı bozmadan yama)
-                        # Eğer çeviri daha uzunsa dosyayı bozabilir, bu yüzden sadece aynı uzunlukta ise değiştir
-                        if len(trans_bytes) <= len(orig_bytes):
-                            patched_data = patched_data.replace(orig_bytes, trans_bytes.ljust(len(orig_bytes), b'\x00'))
+                    # Orijinal metnin tam Binary (İkili) haritasını çıkar
+                    orig_bytes = orig_str.encode('utf-8')
+                    orig_pad = (4 - (len(orig_bytes) % 4)) % 4
+                    orig_block = len(orig_bytes).to_bytes(4, 'little') + orig_bytes + (b'\x00' * orig_pad)
                     
-                    obj.set_raw_data(bytes(patched_data))
-                    break
+                    # Çevrilmiş YENİ metnin (uzunluğu fark etmeksizin) Binary haritasını çıkar
+                    trans_bytes = trans_str.encode('utf-8')
+                    trans_pad = (4 - (len(trans_bytes) % 4)) % 4
+                    trans_block = len(trans_bytes).to_bytes(4, 'little') + trans_bytes + (b'\x00' * trans_pad)
+                    
+                    # Bloğu değiştir (Eski sınırlar kalktı!)
+                    patched_data = patched_data.replace(orig_block, trans_block)
+                
+                # Değişmiş veriyi kaydet (UnityPy burada POINTER REBUILD yapar!)
+                obj.set_raw_data(bytes(patched_data))
 
-    # Paketle ve kaydet (Daha hızlı yazma için optimize edilmiş)
+    # 3. Dosyayı Kaydet
     output_path = os.path.join(temp_dir, f"repacked_{asset_file.filename}")
     with open(output_path, "wb") as f: f.write(env.file.save())