utils/pkce.py

"""PKCE (Proof Key for Code Exchange) 工具函数"""
from __future__ import annotations

import base64
import hashlib
import secrets


def generate_pkce() -> tuple[str, str]:
    """生成 PKCE code_verifier 与对应的 code_challenge（S256）。

    Returns:
        (code_verifier, code_challenge) 元组
    """
    code_verifier = base64.urlsafe_b64encode(secrets.token_bytes(64)).rstrip(b"=").decode("ascii")
    code_challenge = base64.urlsafe_b64encode(
        hashlib.sha256(code_verifier.encode("ascii")).digest()
    ).rstrip(b"=").decode("ascii")
    return code_verifier, code_challenge