import urllib.parse
import os
import requests
import redis
import xml.etree.ElementTree as ET
from flask import request, Response


PROXY_URL = os.getenv("PROXY_URL")
REDIS_URL = os.getenv("REDIS_URL", "redis://localhost:6379/0")
CACHE_TTL = int(os.getenv("CACHE_TTL", 3600))

rdb = redis.from_url(REDIS_URL)


def init_proxy(app):
    @app.route("/proxy")
    def proxy():
        """Proxy RSS feed with per-item caching."""
        raw_query = request.query_string.decode()
        if raw_query.startswith("url="):
            url = urllib.parse.unquote(raw_query[4:])
        else:
            return "Missing URL", 400

        try:
            # Получаем ленту
            proxies = {"http": PROXY_URL, "https": PROXY_URL} if PROXY_URL else None
            r = requests.get(url, timeout=10, proxies=proxies)
            _encode = r.apparent_encoding.lower()
            r.encoding = _encode
            xml_data = r.text.replace(f'<?xml version="1.0" encoding="{_encode}"?>',
                                      '<?xml version="1.0" encoding="UTF-8"?>')

            # Разбираем XML
            root = ET.fromstring(xml_data)
            items = root.findall(".//item")

            cached_items = []
            new_items = []

            for item in items:
                guid = item.find("guid").text if item.find("guid") is not None else None
                if guid:
                    cache_key = f"rss:item:{guid}"
                    cached_item = rdb.get(cache_key)

                    if cached_item:
                        cached_items.append(cached_item.decode())
                    else:
                        item_str = ET.tostring(item, encoding="unicode")
                        rdb.setex(cache_key, CACHE_TTL, item_str)
                        new_items.append(item_str)

            # Собираем финальный RSS
            final_items = cached_items + new_items
            response_xml = f'<?xml version="1.0" encoding="UTF-8"?><rss><channel>{"".join(final_items)}</channel></rss>'

            return Response(response_xml, content_type="application/xml; charset=utf-8")

        except Exception as e:
            return f"Error: {e}", 500