66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
|
"""auth_oauth 身份源插件 — 第三方授权登录示例（基于 chongming-auth 共享契约）。
|
|||
|
|
|
|||
|
|
演示 auth 域拆分后的身份源插件标准形态：
|
|||
|
|
- 准入判断归插件（GitHub OAuth 授权码模式 / 手机验证码模式）
|
|||
|
|
- 平台凭证签发归 chongming_auth.issue_token（claims 契约与内置登录完全一致）
|
|||
|
|
- 账号落地到共享 users 表，第三方身份与账号的映射存插件自有绑定表
|
|||
|
|
（identity_bindings，建在同一 auth 库，不改动 chongming-auth 拥有的 users schema）
|
|||
|
|
|
|||
|
|
演示模式说明：
|
|||
|
|
- 手机验证码默认直接随响应返回 dev_code（无短信通道的本地演示用），
|
|||
|
|
生产必须接真实短信通道并关闭 return_otp_in_response
|
|||
|
|
- GitHub OAuth 需在 [workers.config.github] 配置 client_id/secret/redirect_uri
|
|||
|
|
"""
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from chongming_worker.plugin import Plugin
|
|||
|
|
|
|||
|
|
from .router import * # noqa: F401,F403 触发 @rpc_method 注册
|
|||
|
|
|
|||
|
|
# 模块加载时即注册 SQLModel 元数据（identity_bindings 表与 users 同库，
|
|||
|
|
# 由 chongming_auth.open_database 的 create_all 一并创建）
|
|||
|
|
from .models import IdentityBinding # noqa: F401
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AuthOAuthPlugin(Plugin):
|
|||
|
|
"""AuthOAuth 插件 — 身份源示例，统一生命周期。"""
|
|||
|
|
|
|||
|
|
type = "auth_oauth"
|
|||
|
|
# 不用基类建库：与网关内置登录共用同一 auth 库（chongming_auth.open_database
|
|||
|
|
# 按 [workers.config] 的 data_dir / database 解析，指向 users + identity_bindings）
|
|||
|
|
db_name = None
|
|||
|
|
auto_bind = False # 请求/响应模型由端点函数的类型注解直接提供，无需 gen_models
|
|||
|
|
|
|||
|
|
async def on_init(self):
|
|||
|
|
"""首次 create_workers 时调用一次：打开共享 auth 库 + 缓存。"""
|
|||
|
|
from chongming_auth import open_database
|
|||
|
|
from chongming_cache import Cache
|
|||
|
|
from chongming_config import get_config
|
|||
|
|
|
|||
|
|
db = await open_database(self.config)
|
|||
|
|
cache = Cache.from_config(get_config()["cache"])
|
|||
|
|
|
|||
|
|
# 写入全局变量供 router 使用（cleanup 时统一释放）
|
|||
|
|
self.set_var("db", db)
|
|||
|
|
self.set_var("cache", cache)
|
|||
|
|
self.set_var("plugin_config", dict(self.config or {}))
|
|||
|
|
logger.info("[auth_oauth] 初始化完成（共享 auth 库 + 缓存就绪）")
|
|||
|
|
|
|||
|
|
async def on_cleanup(self):
|
|||
|
|
"""cleanup 时调用：释放自有资源（基类不再托管本插件的数据库）。"""
|
|||
|
|
db = self.get_var("db")
|
|||
|
|
cache = self.get_var("cache")
|
|||
|
|
if db is not None:
|
|||
|
|
await db.dispose()
|
|||
|
|
if cache is not None:
|
|||
|
|
cache.close()
|
|||
|
|
logger.info("[auth_oauth] 清理完成")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 暴露给 app lifespan 的模块级入口（bound method，签名与旧写法完全一致）
|
|||
|
|
plugin = AuthOAuthPlugin()
|
|||
|
|
create_workers = plugin.create_workers
|
|||
|
|
cleanup = plugin.cleanup
|