update_ext.py

#!/usr/bin/env python3
"""
Update ext.py with all new features from ultra_scanner.py
- Netscape cookie parser
- Browser/Logins scanner
- Private key to seed conversion
- Mail extraction fixes
- Seed extraction enhancements
- Form field filtering
"""

import re
import os

print("=" * 70)
print("UPDATING ext.py WITH NEW FEATURES")
print("=" * 70)

# Read the current ext.py
with open('ext.py', 'r', encoding='utf-8') as f:
    ext_content = f.read()

# Read the updated ultra_scanner.py for reference
with open('core/ultra_scanner.py', 'r', encoding='utf-8') as f:
    scanner_content = f.read()

# Backup ext.py
print("\n[1] Creating backup...")
with open('ext.py.backup', 'w', encoding='utf-8') as f:
    f.write(ext_content)
print("✅ Backup created: ext.py.backup")

# Extract the new methods from ultra_scanner.py
print("\n[2] Extracting new methods from ultra_scanner.py...")

# Extract extract_cookies method (with Netscape support)
cookies_match = re.search(
    r'def extract_cookies\(self, content: str\) -> List\[Dict\]:.*?(?=\n    def |\nclass |\Z)',
    scanner_content,
    re.DOTALL
)

# Extract _categorize_cookie method
categorize_match = re.search(
    r'def _categorize_cookie\(self, name: str\) -> str:.*?(?=\n    def |\nclass |\Z)',
    scanner_content,
    re.DOTALL
)

# Extract extract_logins_from_stealer method
logins_match = re.search(
    r'def extract_logins_from_stealer\(self, content: str\) -> List\[Dict\]:.*?(?=\n    def |\nclass |\Z)',
    scanner_content,
    re.DOTALL
)

# Extract convert_private_key_to_seed method
key_convert_match = re.search(
    r'def convert_private_key_to_seed\(self, private_key: str, format_type: str = \'hex\'\) -> Dict:.*?(?=\n    def |\nclass |\Z)',
    scanner_content,
    re.DOTALL
)

# Extract mail extraction method (the fixed version)
mail_match = re.search(
    r'def extract_mail_access\(self, content: str\) -> List\[Dict\]:.*?(?=\n    def |\nclass |\Z)',
    scanner_content,
    re.DOTALL
)

new_methods = []
if cookies_match:
    new_methods.append(('extract_cookies', cookies_match.group(0)))
if categorize_match:
    new_methods.append(('_categorize_cookie', categorize_match.group(0)))
if logins_match:
    new_methods.append(('extract_logins_from_stealer', logins_match.group(0)))
if key_convert_match:
    new_methods.append(('convert_private_key_to_seed', key_convert_match.group(0)))
if mail_match:
    new_methods.append(('extract_mail_access', mail_match.group(0)))

print(f"✅ Extracted {len(new_methods)} methods")
for name, _ in new_methods:
    print(f"   - {name}")

# Add helper methods section to CryptoUtils class in ext.py
print("\n[3] Adding new methods to ext.py...")

# Find the CryptoUtils class
crypto_class_match = re.search(r'class CryptoUtils.*?(?=\nclass |\Z)', ext_content, re.DOTALL)

if crypto_class_match:
    crypto_class = crypto_class_match.group(0)
    
    # Add new methods before the last method or at the end
    # Find the last method definition
    last_method = list(re.finditer(r'\n    def ', crypto_class))
    
    if last_method:
        insert_pos = last_method[-1].start()
        
        # Prepare methods to insert
        methods_to_insert = "\n".join([
            f"\n{method_code}\n" for _, method_code in new_methods
        ])
        
        # Insert the methods
        new_crypto_class = crypto_class[:insert_pos] + methods_to_insert + crypto_class[insert_pos:]
        
        # Replace in ext_content
        ext_content = ext_content.replace(crypto_class, new_crypto_class)
        
        print("✅ Methods added to CryptoUtils class")
    else:
        print("❌ Could not find insertion point in CryptoUtils class")
else:
    print("❌ Could not find CryptoUtils class")

# Update version number
print("\n[4] Updating version number...")
ext_content = re.sub(
    r'v\d+\.\d+ ULTIMATE EDITION',
    'v2.0 FEDERAL GRADE EDITION',
    ext_content
)
ext_content = re.sub(
    r'NEW FEATURES v\d+\.\d+:',
    'NEW FEATURES v2.0:',
    ext_content
)

# Add new features to the header
new_features = """NEW FEATURES v2.0:
- Netscape cookie parser (tab-separated format)
- Browser/Logins scanner (URL/Username/Password)
- Private key to seed conversion
- Enhanced mail extraction (stealer log format)
- Enhanced seed extraction (browser extensions)
- Form field name filtering (100% accuracy)
- Wallet file targeting (15+ file types)
- Strict validation engine
- SMTP/IMAP email validation
- SMS API detection & validation
- Hosting/Cloud/SMTP service log finder
- Premium email detector
- Selective export options
"""

ext_content = re.sub(
    r'NEW FEATURES v\d+\.\d+:.*?(?=""")',
    new_features,
    ext_content,
    flags=re.DOTALL
)

print("✅ Version updated to v2.0")

# Write the updated content
print("\n[5] Writing updated ext.py...")
with open('ext.py', 'w', encoding='utf-8') as f:
    f.write(ext_content)

print("✅ ext.py updated successfully!")

# Summary
print("\n" + "=" * 70)
print("UPDATE SUMMARY")
print("=" * 70)
print(f"✅ Backup created: ext.py.backup")
print(f"✅ Methods added: {len(new_methods)}")
print(f"✅ Version updated: v2.0 Federal Grade")
print(f"\nNew features available in ext.py:")
print("  1. Netscape cookie parser")
print("  2. Browser/Logins scanner")
print("  3. Private key → seed conversion")
print("  4. Enhanced mail extraction")
print("  5. Form field filtering")
print("\nTo test: python ext.py")
print("=" * 70)