import os

def search_keywords_in_files(keywords, directory):
    # Membuat dictionary untuk menyimpan hasil berdasarkan keyword
    results = {keyword: [] for keyword in keywords}

    # Mendapatkan semua file .txt dalam direktori
    txt_files = [f for f in os.listdir(directory) if f.endswith('.txt')]
    
    for filename in txt_files:
        file_path = os.path.join(directory, filename)
        try:
            with open(file_path, 'r', encoding='utf-8', errors='replace') as file:
                for line in file:
                    for keyword in keywords:
                        if keyword in line:
                            results[keyword].append(line.strip())
        except Exception as e:
            print(f"❌ ERROR saat membaca file '{filename}': {e}")

    # Menyimpan hasil ke file terpisah
    for keyword, lines in results.items():
        if lines:  # Hanya simpan jika ada hasil
            # Ganti karakter yang tidak valid dengan '_'
            safe_keyword = keyword.replace('/', '_').replace(':', '_')
            output_filename = f"{safe_keyword}.txt"
            output_path = os.path.join(directory, output_filename)
            with open(output_path, 'w', encoding='utf-8') as output_file:
                output_file.write("\n".join(lines))
            print(f"✅ Hasil untuk '{keyword}' disimpan di '{output_filename}'")

if __name__ == "__main__":
    # Daftar keyword yang ingin dicari
    keywords = [
        "//103.24", "//210.57", "//43.252", "//103.241", "//142.250",
        "//172.217", "//103.242", "//101.50", "//103.94", "//202.46",
        "//103.112", "//142.250", "//103.29", "//103.69", "//147.93",
        "//103.163", "//203.114", "//157.53", "//199.180", "//185.151",
        "//109.106", "siakadcloud.com", ".ac.id", "sevima.com", ":2083/", ":2082/", "/phpmyadmin", "/adminer.php", "website/auto-installer", "wordpress/install", "wordpress/dashboard", "enter-wordpress-details", "login_up.php", "CMD_LOGIN", ":root", "/administrator", "/evo/"
    ]
    
    # Input direktori
    directory = input("📁 Nama direktori: ").strip()
    if os.path.isdir(directory):
        search_keywords_in_files(keywords, directory)
    else:
        print(f"❌ Direktori '{directory}' tidak ditemukan.")