import tkinter as tk
import sys
import os
import webbrowser
import pyautogui
import time
import json
import getpass

# -------------------------------
# Configuration Setup (You run this once)
# -------------------------------
def setup_config():
    config = {}
    
    print("🎯 Prank Setup Menu")
    print("1. Enter YouTube URL to open")
    url = input("Enter video URL (e.g., https://www.youtube.com/watch?v=...): ").strip()
    
    print("\n2. Choose tab behavior:")
    print("   - 1: Open one tab")
    print("   - 2: Spam multiple tabs")
    choice = input("Enter choice (1 or 2): ").strip()
    
    if choice == "2":
        count = int(input("How many tabs? "))
        config['spam_count'] = count
    else:
        config['spam_count'] = 1
        
    config['url'] = url
    
    # Save config
    with open('prank_config.json', 'w') as f:
        json.dump(config, f)
    
    print("Configuration saved!")

# -------------------------------
# Install to Startup (hidden)
# -------------------------------
def install_to_startup():
    USER_NAME = getpass.getuser()
    startup_path = fr"C:\Users\{USER_NAME}\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Startup"
    bat_path = os.path.join(startup_path, "prank_launcher.bat")
    
    with open(bat_path, 'w') as bat_file:
        bat_file.write(f"@echo off\n")
        bat_file.write(f"pythonw \"{os.path.abspath('stealth_with_startup.py')}\"\n")
    
    print("Installed to startup successfully!")

# -------------------------------
# GUI Consent Window
# -------------------------------
def ask_consent():
    root = tk.Tk()
    root.title("System Verification")
    root.geometry("450x180")
    root.eval('tk::PlaceWindow . center')
    
    label = tk.Label(root, text="🔒 Administrator verification required.\n\nType 'I agree' to continue:", font=("Arial", 11))
    label.pack(pady=20)
    
    entry = tk.Entry(root, width=25, font=("Arial", 10))
    entry.pack(pady=5)
    
    def check_consent():
        if entry.get().strip().lower() == "i agree":
            root.destroy()
            time.sleep(1)
            perform_prank()
            # Remove from startup after first run
            remove_from_startup()
        else:
            sys.exit()
    
    btn = tk.Button(root, text="Submit", command=check_consent)
    btn.pack(pady=10)
    
    root.mainloop()

# -------------------------------
# Remove from Startup
# -------------------------------
def remove_from_startup():
    USER_NAME = getpass.getuser()
    startup_path = fr"C:\Users\{USER_NAME}\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Startup"
    bat_path = os.path.join(startup_path, "prank_launcher.bat")
    
    if os.path.exists(bat_path):
        os.remove(bat_path)

# -------------------------------
# Load Config & Perform Prank
# -------------------------------
def perform_prank():
    try:
        with open('prank_config.json', 'r') as f:
            config = json.load(f)
    except:
        config = {'url': 'https://www.youtube.com/watch?v=dQw4w9WgXcQ', 'spam_count': 3}
    
    url = config['url']
    spam_count = config['spam_count']
    
    time.sleep(2)
    
    # Spam tabs
    for _ in range(spam_count):
        webbrowser.open(url)
        time.sleep(0.5)
    
    # Optional: Add more pranks below
    # close_all_tabs() 
    # black_screen(5)

# -------------------------------
# Main Execution
# -------------------------------
if __name__ == "__main__":
    if len(sys.argv) > 1 and sys.argv[1] == "--setup":
        setup_config()
        install_to_startup()
        print("Prank deployed! It will run on next login.")
    else:
        ask_consent()
        
        cd C:\Users\iman\Documents
python stealth_with_startup.py --setup