#!/usr/bin/env python3

import json
import sys
import subprocess
import string
import secrets
from init_paths import *
from paths import *

def restart_hysteria():
    """Restart the Hysteria2 service using the CLI script."""
    try:
        subprocess.run(["python3", CLI_PATH, "restart-hysteria2"], 
                       stdout=subprocess.DEVNULL, 
                       stderr=subprocess.DEVNULL)
    except Exception as e:
        print(f"⚠️ Failed to restart Hysteria2: {e}")

def remove_obfs():
    """Remove the 'obfs' section from the config."""
    try:
        with open(CONFIG_FILE, 'r') as f:
            config = json.load(f)

        if 'obfs' in config:
            del config['obfs']
            with open(CONFIG_FILE, 'w') as f:
                json.dump(config, f, indent=2)
            print("✅ Successfully removed 'obfs' from config.json.")
        else:
            print("ℹ️ 'obfs' section not found in config.json.")

        restart_hysteria()

    except FileNotFoundError:
        print(f"❌ Config file not found: {CONFIG_FILE}")
    except Exception as e:
        print(f"❌ Error removing 'obfs': {e}")

def generate_obfs():
    """Generate and add an 'obfs' section with a random password."""
    try:
        with open(CONFIG_FILE, 'r') as f:
            config = json.load(f)

        if 'obfs' in config:
            print("ℹ️ 'obfs' section already exists. Replacing it.")
            del config['obfs']

        password = ''.join(secrets.choice(string.ascii_letters + string.digits) for _ in range(32))

        config['obfs'] = {
            "type": "salamander",
            "salamander": {
                "password": password
            }
        }

        with open(CONFIG_FILE, 'w') as f:
            json.dump(config, f, indent=2)

        print(f"✅ Successfully added 'obfs' to config.json with password: {password}")

        restart_hysteria()

    except FileNotFoundError:
        print(f"❌ Config file not found: {CONFIG_FILE}")
    except Exception as e:
        print(f"❌ Error generating 'obfs': {e}")

def main():
    if len(sys.argv) != 2:
        print("Usage: python3 obfs_manager.py --remove|-r | --generate|-g")
        sys.exit(1)

    option = sys.argv[1]
    if option in ("--remove", "-r"):
        print("Removing 'obfs' from config.json...")
        remove_obfs()
    elif option in ("--generate", "-g"):
        print("Generating 'obfs' in config.json...")
        generate_obfs()
    else:
        print("Invalid option. Use --remove|-r or --generate|-g")
        sys.exit(1)

if __name__ == "__main__":
    main()