app.py

import os, re, time, json, datetime, requests, gradio as gr

# ───────────────────── 1. Vercel API ─────────────────────
TOKEN = os.getenv("SVR_TOKEN")
TEAM  = os.getenv("VERCEL_TEAM_ID")
if not TOKEN:
    raise EnvironmentError("SVR_TOKEN 환경변수를 설정하세요.")
API   = "https://api.vercel.com"
HEAD  = {"Authorization": f"Bearer {TOKEN}"}

print(f"API 토큰: {TOKEN[:4]}... (마스킹됨)")
print(f"팀 ID: {TEAM if TEAM else '없음'}")

# ───────────────────── 2. BEST 데이터 ────────────────────
BEST_FILE, PER_PAGE = "best_games.json", 9
def _init_best():
    if not os.path.exists(BEST_FILE):
        json.dump([], open(BEST_FILE, "w"))
def _load_best():
    try:
        data = json.load(open(BEST_FILE))
        for it in data:
            if "ts" not in it:
                it["ts"] = int(it.get("timestamp", time.time()))
        return data
    except Exception as e:
        print(f"BEST 데이터 로드 오류: {e}")
        return []

def _save_best(data):
    try:
        json.dump(data, open(BEST_FILE, "w"))
        return True
    except Exception as e:
        print(f"BEST 데이터 저장 오류: {e}")
        return False

# ───────────────────── 3. URL 추가 기능 ─────────────────────
def add_url_to_best(title, url):
    """사용자가 제공한 URL을 BEST 목록에 추가합니다."""
    try:
        # 현재 BEST 데이터 로드
        data = _load_best()
        
        # URL이 이미 존재하는지 확인
        for item in data:
            if item.get("url") == url:
                print(f"URL이 이미 존재합니다: {url}")
                return False
        
        # 새 항목 추가
        new_item = {
            "title": title,
            "url": url,
            "ts": int(time.time()),
            "projectId": "",  # 사용자가 직접 추가하므로 projectId 없음
            "deploymentId": ""  # 사용자가 직접 추가하므로 deploymentId 없음
        }
        
        data.append(new_item)
        
        # 시간순으로 정렬
        data = sorted(data, key=lambda x: x["ts"], reverse=True)
        
        # 저장
        if _save_best(data):
            print(f"URL이 성공적으로 추가되었습니다: {url}")
            return True
        return False
    except Exception as e:
        print(f"URL 추가 오류: {str(e)}")
        return False

# ───────────────────── 4. 배포 보호 설정 관리 ─────────────────
def get_projects():
    """Vercel 프로젝트 목록을 가져옵니다."""
    try:
        params = {"limit": 100}  # 최대 100개 프로젝트
        if TEAM:
            params["teamId"] = TEAM
            
        print(f"프로젝트 목록 가져오는 중...")
        resp = requests.get(f"{API}/v10/projects", headers=HEAD, params=params)
        
        if resp.status_code != 200:
            print(f"프로젝트 목록 가져오기 실패: {resp.status_code}")
            return []
            
        projects = resp.json().get("projects", [])
        print(f"{len(projects)}개의 프로젝트를 찾았습니다.")
        return projects
    except Exception as e:
        print(f"프로젝트 목록 가져오기 오류: {e}")
        return []

def disable_protection(project_id):
    """특정 프로젝트의 배포 보호를 비활성화합니다."""
    try:
        url = f"{API}/v9/projects/{project_id}/protection"
        payload = {
            "protection": {
                "preview": {
                    "enabled": False
                },
                "production": {
                    "enabled": False
                }
            }
        }
        
        print(f"프로젝트 {project_id}의 보호 설정 비활성화 중...")
        resp = requests.put(url, headers=HEAD, json=payload)
        
        if resp.status_code in [200, 201, 204]:
            print(f"프로젝트 {project_id}의 보호 설정이 성공적으로 비활성화되었습니다.")
            return True
        else:
            print(f"보호 설정 변경 실패: {resp.status_code}, {resp.text[:200] if hasattr(resp, 'text') else ''}")
            return False
            
    except Exception as e:
        print(f"보호 설정 변경 오류: {str(e)}")
        return False

def add_exception_domain(project_id, domain):
    """특정 프로젝트에 보호 예외 도메인을 추가합니다."""
    try:
        url = f"{API}/v9/projects/{project_id}/protection"
        
        # 먼저 현재 보호 설정 가져오기
        get_resp = requests.get(url, headers=HEAD)
        if get_resp.status_code != 200:
            print(f"보호 설정 가져오기 실패: {get_resp.status_code}")
            return False
            
        current = get_resp.json()
        
        # 예외 도메인 목록에 추가
        exceptions = current.get("protection", {}).get("preview", {}).get("exceptions", [])
        if domain not in exceptions:
            exceptions.append(domain)
        
        # 업데이트된 설정 적용
        payload = {
            "protection": {
                "preview": {
                    "enabled": True,
                    "exceptions": exceptions
                }
            }
        }
        
        print(f"프로젝트 {project_id}에 예외 도메인 {domain} 추가 중...")
        resp = requests.put(url, headers=HEAD, json=payload)
        
        if resp.status_code in [200, 201, 204]:
            print(f"프로젝트 {project_id}에 예외 도메인이 성공적으로 추가되었습니다.")
            return True
        else:
            print(f"예외 도메인 추가 실패: {resp.status_code}, {resp.text[:200] if hasattr(resp, 'text') else ''}")
            return False
            
    except Exception as e:
        print(f"예외 도메인 추가 오류: {str(e)}")
        return False

def bypass_all_protection():
    """모든 프로젝트의 배포 보호를 비활성화합니다."""
    projects = get_projects()
    if not projects:
        print("프로젝트가 없거나 가져오지 못했습니다.")
        return False
        
    success_count = 0
    for project in projects:
        project_id = project.get("id")
        name = project.get("name", "알 수 없음")
        if project_id:
            print(f"프로젝트 처리 중: {name} ({project_id})")
            if disable_protection(project_id):
                success_count += 1
                
    print(f"총 {len(projects)}개 중 {success_count}개 프로젝트의 보호 설정을 비활성화했습니다.")
    return success_count > 0

# ───────────────────── 5. 페이지네이션 ───────────────────
def page(lst, pg):
    s=(pg-1)*PER_PAGE; e=s+PER_PAGE
    total=(len(lst)+PER_PAGE-1)//PER_PAGE
    return lst[s:e], total

# ───────────────────── 6. HTML 3-열 그리드 ───────────────
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;font-family:Poppins,sans-serif;background:linear-gradient(135deg,#C5E8FF 0%,#FFD6E0 100%);}
    .grid{display:grid;grid-template-columns:repeat(3,1fr);gap:28px 24px;margin:0 auto;max-width:1200px;padding:20px;}
    @media(max-width:1024px){.grid{grid-template-columns:repeat(3,1fr);} }
    @media(max-width:768px){.grid{grid-template-columns:repeat(2,1fr);} }
    @media(max-width:500px){ .grid{grid-template-columns:1fr;} }
    .card{background:#fff;border-radius:18px;overflow:hidden;box-shadow:0 10px 25px rgba(0,0,0,.08);transition:.3s}
    .card:hover{transform:translateY(-6px);box-shadow:0 16px 40px rgba(0,0,0,.12)}
    .hdr{padding:20px 24px;background:rgba(255,255,255,.75);backdrop-filter:blur(8px);border-bottom:1px solid #eee;}
    .ttl{margin:0;font-size:1.15rem;font-weight:700;color:#333;white-space:nowrap;overflow:hidden;text-overflow:ellipsis;}
    .date{margin-top:4px;font-size:.85rem;color:#777;}
    .frame{position:relative;width:100%;padding-top:60%;overflow:hidden;}
    .frame iframe{position:absolute;top:0;left:0;width:142.857%;height:142.857%;
                  transform:scale(.7);transform-origin:top left;border:0;}
    .foot{padding:14px 24px;background:rgba(255,255,255,.85);backdrop-filter:blur(8px);text-align:right;}
    .link{font-size:.85rem;font-weight:600;color:#4a6dd8;text-decoration:none;}
    .cnt{text-align:center;font-size:.85rem;color:#555;margin:10px 0 20px;}
    </style>"""
    
    h=css+"<div class='grid'>"
    for c in cards:
        date=datetime.datetime.fromtimestamp(int(c["ts"])).strftime("%Y-%m-%d")
        h+=f"""
        <div class='card'>
          <div class='hdr'><p class='ttl'>{c['title']}</p><p class='date'>{date}</p></div>
          <div class='frame'><iframe src="{c['url']}" loading="lazy"
               allow="accelerometer; camera; encrypted-media; gyroscope;"></iframe></div>
          <div class='foot'><a class='link' href="{c['url']}" target="_blank">원본↗</a></div>
        </div>"""
    h+="</div><p class='cnt'>Page "+str(pg)+" / "+str(total)+"</p>"
    return h

# ───────────────────── 7. Gradio Blocks UI ───────────────
def build():
    _init_best()
    with gr.Blocks(title="Vibe Game Craft", css="body{overflow-x:hidden;} .gradio-container{margin:0 auto; max-width:1280px;}") as demo:
        gr.Markdown("<h1 style='text-align:center;padding:32px 0 0;color:#333;'>🎮 Vibe Game Craft</h1>")
        
        # URL 추가 인터페이스
        with gr.Row():
            title_input = gr.Textbox(label="제목", placeholder="게임 제목을 입력하세요")
            url_input = gr.Textbox(label="URL", placeholder="https://...")
            add_btn = gr.Button("URL 추가", variant="primary")
        
        # 페이지 네비게이션
        with gr.Row():
            b_best = gr.Button("BEST", size="sm")
            b_prev = gr.Button("⬅️ Prev", size="sm")
            b_next = gr.Button("Next ➡️", size="sm")
            b_ref  = gr.Button("🔄 Reload", size="sm")
            b_bypass = gr.Button("🔓 보호 해제", size="sm", variant="secondary")

        # 상태 및 출력
        bp = gr.State(1) 
        out = gr.HTML()
        status_msg = gr.Textbox(label="상태 메시지", interactive=False)

        def show_best(p=1): 
            d, t = page(_load_best(), p)
            return html(d, p, t), p
        
        def prev(b):
            b = max(1, b-1)
            h, _ = show_best(b)
            return h, b
        
        def nxt(b):
            maxp = (len(_load_best()) + PER_PAGE - 1) // PER_PAGE
            b = min(maxp, b+1)
            h, _ = show_best(b)
            return h, b

        # 페이지 네비게이션 버튼 (하단에 배치)
        with gr.Row(elem_classes="navigation-buttons"):
            b_prev = gr.Button("⬅️ 이전 페이지", size="sm")
            b_next = gr.Button("다음 페이지 ➡️", size="sm")
            
        # 이벤트 연결
        b_prev.click(prev, inputs=[bp], outputs=[out, bp])
        b_next.click(nxt, inputs=[bp], outputs=[out, bp])

        # CSS 스타일 추가
        css = """
        .navigation-buttons {
            display: flex;
            justify-content: center;
            margin-bottom: 40px;
        }
        .navigation-buttons button {
            margin: 0 10px;
            min-width: 120px;
        }
        """
        demo.load(show_best, outputs=[out, bp])
    return demo

app = build()

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