import asyncio
import json
from playwright.async_api import async_playwright

async def export_cookies():
    async with async_playwright() as p:
        # 启动浏览器，headless=False 方便手动扫码
        browser = await p.chromium.launch(headless=False)
        context = await browser.new_context()
        page = await context.new_page()

        await page.goto("https://www.bilibili.com")
        
        print("请在浏览器中完成登录（扫码或账号密码）...")
        
        # 循环检查是否登录成功（根据头像或特定元素判断）
        # 这里我们等待用户手动在控制台回车确认登录已完成
        input("完成登录后，请按回车键继续导出 Cookie...")

        # 获取所有 Cookie
        cookies = await context.cookies()
        
        with open("bilibili.json", "w", encoding="utf-8") as f:
            json.dump(cookies, f, ensure_ascii=False, indent=4)
            
        print("Cookie 已成功保存至 bilibili.json")
        await browser.close()

if __name__ == "__main__":
    asyncio.run(export_cookies())