import asyncio
from playwright.async_api import async_playwright

from playwright_sign import sign_with_playwright


async def main():
    """测试签名生成功能"""
    async with async_playwright() as p:
        # 启动浏览器
        browser = await p.chromium.launch(headless=True)
        context = await browser.new_context()
        page = await context.new_page()

        # 访问小红书页面以加载必要的 JavaScript 环境
        await page.goto('https://www.xiaohongshu.com')

        # 等待页面加载
        await page.wait_for_load_state('networkidle')

        # 测试签名生成
        uri = '/api/sns/web/v1/search/notes'
        data = {'keyword': 'test', 'page': 1}
        a1_value = 'example_a1_cookie_value'
        method = 'POST'

        signs = await sign_with_playwright(page, uri, data, a1_value, method)
        print(f"Generated signs: {signs}")

        # 关闭浏览器
        await browser.close()


if __name__ == '__main__':
    asyncio.run(main())