import os
import json
import re
import gradio as gr

# ───────────────────── 1. 기본 설정 ─────────────────────
BEST_FILE, PER_PAGE = "best_games.json", 9  # ❶ 페이지당 9개 유지


# ───────────────────── 2. BEST 데이터 ────────────────────
def _init_best():
    if not os.path.exists(BEST_FILE):
        json.dump([], open(BEST_FILE, "w"), ensure_ascii=False)


def _load_best():
    try:
        raw = json.load(open(BEST_FILE))
        # URL 리스트만 반환
        if isinstance(raw, list):
            return [u if isinstance(u, str) else u.get("url") for u in raw]
        return []
    except Exception as e:
        print("BEST 로드 오류:", e)
        return []


def _save_best(lst):  # URL 리스트 저장
    try:
        json.dump(lst, open(BEST_FILE, "w"), ensure_ascii=False, indent=2)
        return True
    except Exception as e:
        print("BEST 저장 오류:", e)
        return False


def to_hub_space_url(url: str) -> str:
    """
    *.hf.space URL을 https://huggingface.co/spaces/<owner>/<space> 형식으로 변환
    다른 URL은 그대로 반환
    """
    m = re.match(r"https?://([^-]+)-([^.]+)\.hf\.space(/.*)?", url)
    if m:
        owner, space, _ = m.groups()
        return f"https://huggingface.co/spaces/{owner}/{space}"
    return url


def add_url_to_best(url: str):
    # 저장할 때는 사용자가 준 원본을 그대로 둔다
    data = _load_best()
    if url in data:
        return False
    data.insert(0, url)
    return _save_best(data)


# ───────────────────── 3. 유틸 ──────────────────────────
def page(lst, pg):
    s, e = (pg - 1) * PER_PAGE, (pg - 1) * PER_PAGE + PER_PAGE
    total = (len(lst) + PER_PAGE - 1) // PER_PAGE
    return lst[s:e], total


def process_url_for_iframe(url):
    """
    반환: (iframe_url, extra_class, alternate_urls)
      extra_class : '' | 'huggingface' | 'hfspace'
    """
    # Hugging Face Spaces embed (Gradio/Streamlit)
    if "huggingface.co/spaces" in url:
        owner, name = url.rstrip("/").split("/spaces/")[1].split("/")[:2]
        return f"https://huggingface.co/spaces/{owner}/{name}/embed", "huggingface", []

    # *.hf.space (정적/static Space 포함)
    m = re.match(r"https?://([^/]+)\.hf\.space(/.*)?", url)
    if m:
        sub, rest = m.groups()
        static_url = f"https://{sub}.static.hf.space{rest or ''}"
        # alt_urls 에 원본 저장(실패 시 재시도 가능)
        return static_url, "hfspace", [url]

    return url, "", []


# ───────────────────── 6. HTML 그리드 ───────────────────
def html(cards, pg, total):
    if not cards:
        return (
            "<div style='text-align:center;padding:70px;color:#555;'>"
            "표시할 배포가 없습니다.</div>"
        )

    css = r"""
    <style>
    /* 파스텔 그라디에이션 배경 */
    body{
        margin:0;padding:0;font-family:Poppins,sans-serif;
        background:linear-gradient(135deg,#fdf4ff 0%,#f6fbff 50%,#fffaf4 100%);
        background-attachment:fixed;
        overflow-x:hidden;overflow-y:auto;
    }
    .container{width:100%;padding:10px 10px 70px;box-sizing:border-box;}
    .grid{display:grid;grid-template-columns:repeat(3,1fr);gap:12px;width:100%;}
    .card{
        background:#fff;border-radius:10px;overflow:hidden;box-shadow:0 4px 10px rgba(0,0,0,0.08);
        height:420px;display:flex;flex-direction:column;position:relative;
    }
    /* 게임 화면 축소 */
    .frame{flex:1;position:relative;overflow:hidden;}
    .frame iframe{
        position:absolute;top:0;left:0;
        width:166.667%;height:166.667%;
        transform:scale(0.6);transform-origin:top left;border:0;
    }
    /* Gradio/Streamlit Embed용 */
    .frame.huggingface iframe{
        width:100%!important;height:100%!important;
        transform:none!important;border:none!important;
    }
    /* hf.space 전체 페이지용 - 한 번 더 축소 */
    .frame.hfspace iframe{
        width:200%;height:200%;
        transform:scale(0.5);
        transform-origin:top left;border:0;
    }
    /* 하단 바로가기 */
    .foot{height:34px;display:flex;align-items:center;justify-content:center;background:#fafafa;border-top:1px solid #eee;}
    .foot a{font-size:0.85rem;font-weight:600;color:#4a6dd8;text-decoration:none;}
    .foot a:hover{text-decoration:underline;}
    /* 반응형 높이 */
    @media(min-width:1200px){.card{height:560px;}}
    @media(max-width:767px){
        .grid{grid-template-columns:1fr;}
        .card{height:480px;}
    }
    </style>"""

    js = """
    <script>
    /* 허깅페이스 iframe 로딩 오류 처리(생략 - 기존 스크립트 그대로) */
    </script>
    """

    h = css + js + '<div class="container"><div class="grid">'
    for idx, url in enumerate(cards):
        iframe_url, extra_cls, alt_urls = process_url_for_iframe(url)
        frame_class = f"frame {extra_cls}".strip()
        iframe_id = f"iframe-{idx}-{hash(url) % 10000}"
        alt_attr = f'data-alternate-urls="{",".join(alt_urls)}"' if alt_urls else ""
        safe_url = to_hub_space_url(url)  # 새 탭 링크용 변환 URL
        h += f"""
        <div class="card">
          <div class="{frame_class}">
            <iframe id="{iframe_id}" src="{iframe_url}" loading="lazy"
              sandbox="allow-forms allow-modals allow-popups allow-popups-to-escape-sandbox allow-same-origin allow-scripts allow-downloads"
              data-original-url="{url}" {alt_attr}></iframe>
          </div>
          <div class="foot">
            <a href="{safe_url}" target="_blank" rel="noopener noreferrer">
              ↗ Open in Full Screen (New Tab)
            </a>
          </div>
        </div>"""
    h += "</div></div>"
    h += f'<div class="page-info">Page {pg} / {total}</div>'
    return h


# ───────────────────── 5. Gradio UI ─────────────────────
def build():
    _init_best()

    header = """
    <style>
      .app-header{position:sticky;top:0;text-align:center;background:#fff;
                  padding:16px 0 8px;border-bottom:1px solid #eee;z-index:1100;}
      .badge-row{display:inline-flex;gap:8px;margin:8px 0;}
    </style>
    <div class="app-header">
        <h1 style="margin:0;font-size:28px;">🎮 Vibe Game Gallery</h1>
        <div class="badge-row">
          <a href="https://huggingface.co/spaces/openfree/Vibe-Game" target="_blank">
            <img src="https://img.shields.io/static/v1?label=huggingface&message=Vibe%20Game%20Craft&color=800080&labelColor=ffa500&logo=huggingface&logoColor=ffff00&style=for-the-badge">
          </a>
          <a href="https://huggingface.co/spaces/openfree/Game-Gallery" target="_blank">
            <img src="https://img.shields.io/static/v1?label=huggingface&message=Game%20Gallery&color=800080&labelColor=ffa500&logo=huggingface&logoColor=ffff00&style=for-the-badge">
          </a>
          <a href="https://discord.gg/openfreeai" target="_blank">
            <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=0000ff&labelColor=800080&logo=discord&logoColor=white&style=for-the-badge">
          </a>
        </div>
    </div>"""

    global_css = """
      footer{display:none !important;}
      .button-row{position:fixed;bottom:0;left:0;right:0;height:60px;
                  background:#f0f0f0;padding:10px;text-align:center;
                  box-shadow:0 -2px 10px rgba(0,0,0,.05);z-index:1000;}
      .button-row button{margin:0 10px;padding:10px 20px;font-size:16px;font-weight:bold;border-radius:50px;}
      #content-area{overflow-y:auto;height:calc(100vh - 60px - 120px);}
    """

    with gr.Blocks(title="Vibe Game Gallery", css=global_css) as demo:
        gr.HTML(header)
        out = gr.HTML(elem_id="content-area")
        with gr.Row(elem_classes="button-row"):
            b_prev = gr.Button("◀ 이전", size="lg")
            b_next = gr.Button("다음 ▶", size="lg")

        bp = gr.State(1)

        def render(p=1):
            data, tot = page(_load_best(), p)
            return html(data, p, tot), p

        b_prev.click(lambda p: render(max(1, p - 1)), inputs=bp, outputs=[out, bp])
        b_next.click(lambda p: render(p + 1), inputs=bp, outputs=[out, bp])

        demo.load(render, outputs=[out, bp])

    return demo


app = build()

if __name__ == "__main__":
    app.launch()