#!/usr/bin/env python3

import random
from wordlists import get_wordlists


def generate_server_error() -> tuple[str, str]:
    wl = get_wordlists()
    server_errors = wl.server_errors

    if not server_errors:
        return ("500 Internal Server Error", "text/html")

    server_type = random.choice(list(server_errors.keys()))
    server_config = server_errors[server_type]

    error_codes = {
        400: "Bad Request",
        401: "Unauthorized",
        403: "Forbidden",
        404: "Not Found",
        500: "Internal Server Error",
        502: "Bad Gateway",
        503: "Service Unavailable",
    }

    code = random.choice(list(error_codes.keys()))
    message = error_codes[code]

    template = server_config.get("template", "")
    version = random.choice(server_config.get("versions", ["1.0"]))

    html = template.replace("{code}", str(code))
    html = html.replace("{message}", message)
    html = html.replace("{version}", version)

    if server_type == "apache":
        os = random.choice(server_config.get("os", ["Ubuntu"]))
        html = html.replace("{os}", os)
        html = html.replace("{host}", "localhost")

    return (html, "text/html")


def get_server_header(server_type: str = None) -> str:
    wl = get_wordlists()
    server_errors = wl.server_errors

    if not server_errors:
        return "nginx/1.18.0"

    if not server_type:
        server_type = random.choice(list(server_errors.keys()))

    server_config = server_errors.get(server_type, {})
    version = random.choice(server_config.get("versions", ["1.0"]))

    server_headers = {
        "nginx": f"nginx/{version}",
        "apache": f"Apache/{version}",
        "iis": f"Microsoft-IIS/{version}",
        "tomcat": f"Apache-Coyote/1.1",
    }

    return server_headers.get(server_type, "nginx/1.18.0")