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", 48
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 []

# ───────────────────── 3. 배포 보호 설정 관리 ─────────────────
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

# ───────────────────── 4. 배포 목록 가져오기 ─────────────────
def fetch_all(limit=200, auto_bypass=True):
    """
    Vercel API v6/deployments를 사용하여 모든 준비된 배포를 가져오고,
    필요에 따라 자동으로 보호 우회 설정을 적용합니다.
    """
    try:
        # 자동 보호 우회 적용 (선택 사항)
        if auto_bypass:
            print("모든 프로젝트에 대한 보호 설정 자동 우회를 시도합니다...")
            bypass_all_protection()
        
        # API 파라미터 설정
        params = {"limit": limit, "state": "READY"}
        if TEAM:
            params["teamId"] = TEAM

        print(f"Vercel API 호출 (배포): {API}/v6/deployments (params={params})")
        
        resp = requests.get(f"{API}/v6/deployments",
                            headers=HEAD, params=params, timeout=30)
        
        if resp.status_code != 200:
            print(f"API 응답 오류: {resp.status_code}, {resp.text[:200] if hasattr(resp, 'text') else ''}")
            return []
            
        data = resp.json()
        
        if "deployments" not in data:
            print(f"API 응답에 deployments 필드가 없습니다: {str(data)[:200]}...")
            return []
            
        deployments = data.get("deployments", [])
        print(f"{len(deployments)}개의 배포를 찾았습니다")
        
        # 도메인 패턴: vercel.app으로 끝나는지 확인
        domain_pat = re.compile(r"\.vercel\.app$", re.I)
        
        # 중복 방지를 위한 세트 (이미 처리한 URL)
        processed_urls = set()
        processed_projects = set()
        
        games = []
        for deployment in deployments:
            # URL 확인
            url = deployment.get("url", "")
            project_id = deployment.get("projectId")
            
            # 이미 처리한 URL은 건너뜀
            if url in processed_urls:
                continue
                
            processed_urls.add(url)
            
            # vercel.app 도메인인지 확인
            if url and domain_pat.search(url):
                # 프로젝트별 보호 우회 설정 (각 프로젝트당 한 번만)
                if auto_bypass and project_id and project_id not in processed_projects:
                    disable_protection(project_id)
                    processed_projects.add(project_id)
                
                # 타임스탬프 처리
                created_time = deployment.get("created")
                if created_time:
                    try:
                        ts = int(created_time / 1000)
                    except:
                        ts = int(time.time())
                else:
                    ts = int(time.time())
                
                # 이름에 projectId가 있으면 보기 좋게 변환
                name = deployment.get("name", "(제목 없음)")
                name = re.sub(r"-[a-f0-9]{9}$", "", name)  # projectId 부분 제거
                
                games.append({
                    "title": name,
                    "url": f"https://{url}",
                    "ts": ts,
                    "projectId": project_id,
                    "deploymentId": deployment.get("uid", "")
                })
                print(f"배포 추가: {name} - https://{url}")
        
        print(f"총 {len(games)}개의 유효한 배포를 필터링했습니다")
        return sorted(games, key=lambda x: x["ts"], reverse=True)

    except Exception as e:
        print(f"Vercel API 오류: {str(e)}")
        import traceback
        traceback.print_exc()
        return []

# ───────────────────── 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 20px 60px;}
    @media(max-width:1024px){.grid{grid-template-columns:repeat(2,1fr);} }
    @media(max-width:640px){ .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 40px;}
    </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;}") as demo:
        gr.Markdown("<h1 style='text-align:center;padding:32px 0 0;color:#333;'>🎮 Vibe Game Craft</h1>")
        with gr.Row():
            b_new  = gr.Button("NEW",  size="sm")
            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")

        tab = gr.State("new"); np = gr.State(1); bp = gr.State(1); out = gr.HTML()

        def show_new(p=1):  d,t=page(fetch_all(),p);  return html(d,p,t),"new",p
        def show_best(p=1): d,t=page(_load_best(),p); return html(d,p,t),"best",p
        
        def run_bypass():
            if bypass_all_protection():
                return "✅ 모든 프로젝트의 보호 설정이 비활성화되었습니다."
            else:
                return "❌ 일부 또는 모든 프로젝트의 보호 설정 비활성화에 실패했습니다."

        def prev(t,n,b):
            if t=="new": n=max(1,n-1); h,_,_=show_new(n);  return h,n,b
            b=max(1,b-1); h,_,_=show_best(b); return h,n,b
        def nxt(t,n,b):
            if t=="new":
                maxp=(len(fetch_all())+PER_PAGE-1)//PER_PAGE
                n=min(maxp,n+1); h,_,_=show_new(n); return h,n,b
            maxp=(len(_load_best())+PER_PAGE-1)//PER_PAGE
            b=min(maxp,b+1); h,_,_=show_best(b); return h,n,b

        b_new.click(show_new,  outputs=[out,tab,np])
        b_best.click(show_best,outputs=[out,tab,bp])
        b_prev.click(prev,     inputs=[tab,np,bp], outputs=[out,np,bp])
        b_next.click(nxt,      inputs=[tab,np,bp], outputs=[out,np,bp])
        b_ref.click(lambda t,n,b: show_new(n)[0] if t=="new" else show_best(b)[0],
                     inputs=[tab,np,bp], outputs=[out])
        b_bypass.click(run_bypass, outputs=[out])

        demo.load(show_new, outputs=[out,tab,np])
    return demo

app = build()

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