from __future__ import annotations

from typing import Any


def build_wecom_oauth_url(corp_id: str, redirect_uri: str, state: str = "") -> str:
    params = {
        "appid": corp_id,
        "redirect_uri": redirect_uri,
        "response_type": "code",
        "scope": "snsapi_base",
    }
    query = "&".join(f"{k}={v}" for k, v in params.items())
    url = f"https://open.weixin.qq.com/connect/oauth2/authorize?{query}"
    if state:
        url += f"&state={state}"
    url += "#wechat_redirect"
    return url


def build_mp_oauth_url(app_id: str, redirect_uri: str, state: str = "", scope: str = "snsapi_userinfo") -> str:
    params = {
        "appid": app_id,
        "redirect_uri": redirect_uri,
        "response_type": "code",
        "scope": scope,
    }
    query = "&".join(f"{k}={v}" for k, v in params.items())
    url = f"https://open.weixin.qq.com/connect/oauth2/authorize?{query}"
    if state:
        url += f"&state={state}"
    url += "#wechat_redirect"
    return url


async def handle_wecom_oauth_callback(code: str, corp_id: str, corp_secret: str) -> dict[str, Any]:
    import httpx

    url = "https://qyapi.weixin.qq.com/cgi-bin/auth/getuserinfo"
    async with httpx.AsyncClient() as client:
        resp = await client.get(url, params={"access_token": "", "code": code})
        return resp.json()


async def handle_mp_oauth_callback(code: str, app_id: str, app_secret: str) -> dict[str, Any]:
    import httpx

    url = "https://api.weixin.qq.com/sns/oauth2/access_token"
    async with httpx.AsyncClient() as client:
        resp = await client.get(
            url,
            params={
                "appid": app_id,
                "secret": app_secret,
                "code": code,
                "grant_type": "authorization_code",
            },
        )
        return resp.json()