import requests
import random
import time
import threading
from flask import Flask, request, Response

PROXY_LIST_URL = "https://proxies.typegpt.net/ips.txt"
proxies_cache = []
last_refresh = 0

app = Flask(__name__)

def fetch_proxies():
    """Fetch proxy list from remote URL."""
    try:
        resp = requests.get(PROXY_LIST_URL, timeout=10)
        resp.raise_for_status()
        proxies = [line.strip() for line in resp.text.splitlines() if line.strip()]
        return proxies
    except Exception as e:
        print(f"[ERROR] Failed to fetch proxies: {e}")
        return []

def get_random_proxy(proxies):
    """Pick a random proxy from the list."""
    if not proxies:
        return None
    return random.choice(proxies)

def refresh_proxies_loop():
    """Background thread: refresh proxy list every 5 minutes."""
    global proxies_cache, last_refresh
    while True:
        if time.time() - last_refresh > 300 or not proxies_cache:
            proxies_cache = fetch_proxies()
            last_refresh = time.time()
            print(f"[INFO] Refreshed {len(proxies_cache)} proxies.")
        time.sleep(60)

@app.route("/health")
def health():
    return "Healthy", 200

@app.route("/<path:target_url>")
def proxy_request(target_url):
    """Catch-all route: forward request to target_url using random proxy."""
    if not target_url.startswith("http"):
        target_url = "http://" + target_url  # allow shorthand

    proxy = get_random_proxy(proxies_cache)
    if not proxy:
        return {"error": "No proxies available"}, 500

    proxies = {"http": proxy, "https": proxy}
    try:
        print(f"[INFO] Forwarding to {target_url} via {proxy}")
        resp = requests.get(target_url, proxies=proxies, timeout=15)

        return Response(
            resp.content,
            status=resp.status_code,
            headers=dict(resp.headers),
        )
    except Exception as e:
        return {"error": f"Proxy failed: {proxy}", "details": str(e)}, 502

def main():
    # Start proxy refresher thread
    t = threading.Thread(target=refresh_proxies_loop, daemon=True)
    t.start()

    # Run Flask server
    app.run(host="0.0.0.0", port=5000)

if __name__ == "__main__":
    main()