#!/usr/bin/python
"""Utility functions for IPMPV."""

import os
import re
import subprocess
import secrets

# Environment variables
is_wayland = "WAYLAND_DISPLAY" in os.environ
osd_corner_radius = os.environ.get("IPMPV_CORNER_RADIUS")
ipmpv_retroarch_cmd = os.environ.get("IPMPV_RETROARCH_CMD")
m3u_url = os.environ.get('IPMPV_M3U_URL')
hwdec = os.environ.get('IPMPV_HWDEC')
ao = os.environ.get('IPMPV_AO')

def setup_environment():
    """Set up environment variables."""
    os.environ["LC_ALL"] = "C"
    os.environ["LANG"] = "C"

def is_valid_url(url):
    """Check if a URL is valid."""
    return re.match(r"^(https?|rtmp|rtmps|udp|tcp):\/\/[\w\-]+(\.[\w\-]+)*(:\d+)?([\/?].*)?$", url) is not None

def get_current_resolution():
    """Get the current display resolution."""
    try:
        if is_wayland:
            wlr_randr_env = os.environ.copy()
            output = subprocess.check_output(["wlr-randr"], universal_newlines=True, env=wlr_randr_env)
            if "Composite-1" in output.split("\n")[0]:
                for line in output.split("\n"):
                    if "720x480" in line and "current" in line:
                        return "480i"
                    elif "720x240" in line and "current" in line:
                        return "240p"
                    elif "720x576" in line and "current" in line:
                        return "576i"
                    elif "720x288" in line and "current" in line:
                        return "288p"
        else:
            xrandr_env = os.environ.copy()
            output = subprocess.check_output(["xrandr"], universal_newlines=True, env=xrandr_env)
            for line in output.split("\n"):
                if "Composite-1" in line:
                    if "720x480" in line:
                        return "480i"
                    elif "720x240" in line:
                        return "240p"
                    elif "720x576" in line:
                        return "576i"
                    elif "720x288" in line:
                        return "288p"
    except subprocess.CalledProcessError:
        if is_wayland:
            print("Error: Cannot get display resolution. Is this a wl-roots compatible compositor?")
        else:
            print("Error: Cannot get display resolution. Is an X session running?")
    except FileNotFoundError:
        if is_wayland:
            print("Error: Could not find wlr-randr, resolution will be unknown")
        else:
            print("Error: Could not find xrandr, resolution will be unknown")
    return "UNK"

def change_resolution(current_resolution):
    """Change the display resolution."""
    new_res = ""
    if is_wayland:
        if current_resolution == "480i":
            new_res = "720x240"
        elif current_resolution == "240p":
            new_res = "720x480"
        elif current_resolution == "576i":
            new_res = "720x288"
        elif current_resolution == "288p":
            new_res = "720x576"
    else:
        if current_resolution == "480i":
            new_res = "720x240"
        elif current_resolution == "240p":
            new_res = "720x480i"
        elif current_resolution == "576i":
            new_res = "720x288"
        elif current_resolution == "288p":
            new_res = "720x576i"

    if new_res:
        if is_wayland:
            wlr_randr_env = os.environ.copy()
            wlr_randr_env["DISPLAY"] = ":0"
            subprocess.run(["wlr-randr", "--output", "Composite-1", "--mode", new_res], check=False, env=wlr_randr_env)
            return get_current_resolution()
        else:
            xrandr_env = os.environ.copy()
            xrandr_env["DISPLAY"] = ":0"
            subprocess.run(["xrandr", "--output", "Composite-1", "--mode", new_res], check=False, env=xrandr_env)
            return get_current_resolution()

    return current_resolution

def get_or_create_secret_key():
    """
    Get the secret key from a file or create a new one if it doesn't exist.

    Returns:
        str: The secret key
    """
    key_file = os.path.join(os.path.dirname(__file__), '.secret_key')

    # If the file exists, read the key
    if os.path.exists(key_file):
        with open(key_file, 'r') as f:
            return f.read().strip()

    # Otherwise, generate a new key and save it
    new_key = secrets.token_hex(16)  # 32-character hex string

    # Create the file with restricted permissions (readable only by the owner)
    try:
        # This works on Unix-like systems (Linux, macOS)
        with open(key_file, 'w') as f:
            f.write(new_key)
        os.chmod(key_file, 0o600)  # Owner can read/write, no one else can access
    except:
        # Simplified approach for Windows or if chmod fails
        with open(key_file, 'w') as f:
            f.write(new_key)

    return new_key