DEV-MERGE-LEGACY-BACKEND-2026/internal/identity/handler/handler.go

// Package handler 身份域 HTTP 接入层（/auth/* 端点，架构规格 §3.2 三层之"Handler"）。
//
// 分层职责：
//   - 本包（Handler）：把请求体绑定/校验（validator）→ 调 service.AuthService（纯业务）→
//     把 *AppError 映射为统一响应信封（response.Fail / OK）。**不写业务逻辑、不直接碰 DB/Redis**。
//   - Service（identity/service）：两步登录 / me / refresh / logout 编排（见其包注释）。
//   - 验签中间件（identity/handler 的 auth.go，T4-5）：受保护路由前置，验签 + 作用域注入；
//     本包 Me/Logout 从 c.Locals 读取中间件写入的已验签 claims（见 ClaimsCtxKey）。
//
// 端点清单（路由挂载在 T4-6 main 完成；此处只定义方法）：
//   - POST /auth/login    两步登录①：验账密 → 一次性票据 + 用户 + 作用域列表（**公开**，无鉴权）
//   - POST /auth/issue    两步登录②：消费票据 + 选作用域 → access+refresh（**公开**，凭票据）
//   - POST /auth/refresh  refresh 轮换：旧 refresh 换新对（**公开**，凭 refresh）
//   - GET  /auth/wechat/qrcode、GET /auth/wechat/status  PC 微信扫码（state 状态机 + 复用两步②票据）
//   - GET  /auth/wechat/authorize                        移动端 App 内微信 webview 授权（snsapi_userinfo，整页跳）
//   - GET  /auth/wechat/callback                         微信重定向落点（code 换 openid，两 scope 共用）
//   - GET  /auth/me       登录态自查（**受保护**，需 Auth 中间件）
//   - POST /auth/logout   登出（**受保护**，恒 200 幂等；撤销尽力而为）
//
// 契约要点：
//   - HTTP 恒 200，真实状态在 body code（铁律 6；见 infra/response）。
//   - 校验失败由 validator 内部写 code=2001 并返回 false，Handler `return nil` 终止（铁律 7）。
//   - logout 对客户端**恒 200**（本地态必清）：撤销失败只记日志，不阻断（对齐前端 doLogout 幂等语义）。
package handler

import (
	"encoding/json"
	"time"

	"github.com/gofiber/fiber/v2"
	"go.uber.org/zap"

	"yuansiwei/unified-backend/internal/identity/model"
	"yuansiwei/unified-backend/internal/identity/service"
	"yuansiwei/unified-backend/internal/identity/token"
	apperrors "yuansiwei/unified-backend/internal/infra/errors"
	"yuansiwei/unified-backend/internal/infra/response"
	"yuansiwei/unified-backend/internal/infra/validator"
)

// ClaimsCtxKey 已验签 claims 在 fiber.Ctx 局部值（c.Locals）中的键。
//
// 由 T4-5 的 Auth 验签中间件在验签成功后写入（*token.Claims）；本包 Me/Logout
// 经 claimsFromCtx 读回。**中间件写入与 Handler 读取两端用同一常量**，避免字符串散落
// （对齐 infra/scope 的 CtxKey 纪律）。受保护路由若无此值 = 中间件链配置错误，按未登录拒绝。
const ClaimsCtxKey = "authClaims"

// Handler 身份域 HTTP 处理器（启动装配一次，全进程共享）。
//
// 字段（装配入口 New 注入）：
//   - svc：身份域业务服务（两步登录 / me / refresh / logout 编排，T4-3）——**必填**。
//   - wechat：微信扫码登录服务（二维码/状态轮询/回调编排）——可选（nil 时 /auth/wechat/* 端点
//     返回 4004"微信扫码未配置"；main 仅在 WeChatCfg 配齐 appid+回调地址时注入）。
//   - log：结构化日志器（logout 撤销失败的告警出口；nil 时静默不记，不阻断）。
type Handler struct {
	svc    *service.AuthService
	wechat *service.WechatService
	log    *zap.Logger
}

// Config Handler 装配参数（T4-6 main 按此构造）。
type Config struct {
	// Svc 身份域业务服务（必填）。
	Svc *service.AuthService
	// Wechat 微信扫码登录服务（可选；nil 时微信端点返回 4004 未配置）。
	Wechat *service.WechatService
	// Log 结构化日志器（可选；nil 时登出撤销失败不记日志，仍恒 200）。
	Log *zap.Logger
}

// New 构造身份域 HTTP 处理器。
//
// 入参：cfg Config：见 Config 字段说明。
// 出参：
//   - *Handler：就绪处理器。
//   - error：必填依赖缺失（Svc 为 nil，fail-fast——无业务服务不允许挂载端点）。
func New(cfg Config) (*Handler, error) {
	if cfg.Svc == nil {
		return nil, apperrors.New(apperrors.CodeInternal, "handler: AuthService 未注入")
	}
	return &Handler{svc: cfg.Svc, wechat: cfg.Wechat, log: cfg.Log}, nil
}

// ---- 请求体 DTO（入参；camelCase 与前端契约严格对齐，见 README 命名规范）----

// LoginReq 两步登录① 请求体（账号密码登录，账号 = 用户名或手机号，二选一）。
//
// 字段：
//   - Username：用户名（与 Phone 二选一；归一/校验在 Service，此处仅校验长度上限）。
//   - Phone：手机号（与 Username 二选一；11 位，Service 层 isValidPhone 做权威位数校验，此处 max=11 兜底）。
//   - Password：明文密码（6~128 位；**仅内存调用栈，禁止进日志/响应**）。
//
// 校验：Username 与 Phone 至少一个非空（required_without 交叉约束）；两者都给时 Handler 优先走 phone。
type LoginReq struct {
	Username string `json:"username" validate:"required_without=Phone,max=64"`
	Phone    string `json:"phone" validate:"required_without=Username,max=11"`
	Password string `json:"password" validate:"required,min=6,max=128"`
}

// IssueReq 两步登录② 请求体（选定作用域后换取 token）。
//
// 字段：
//   - LoginTicket：① 返回的一次性票据（32 字符 hex；已消费/过期/不存在 → 拒绝）。
//   - InstitutionID：选定机构（scope 根；0=平台级超管，故不标 required）。
//   - CampusID：选定校区（0=机构级/不限，故不标 required）。
type IssueReq struct {
	LoginTicket   string `json:"loginTicket" validate:"required,len=32"`
	InstitutionID int64  `json:"institutionId"`
	CampusID      int64  `json:"campusId"`
}

// RefreshReq refresh 轮换请求体。
type RefreshReq struct {
	// RefreshToken：旧 refresh 不透明串（64 字符 hex = 32 字节随机，Issue 返回的原样值；
	// 见 token.Issue——hex.EncodeToString(32 字节) 恒 64 位）。
	RefreshToken string `json:"refreshToken" validate:"required,len=64"`
}

// LogoutReq 登出请求体（refresh 可空——仅凭 access jti 也能拉黑当前 access）。
type LogoutReq struct {
	// RefreshToken：当前 refresh 串（删其白名单键；空则跳过，仅吊销 access）。
	RefreshToken string `json:"refreshToken"`
}

// ---- 端点 ----

// Login 两步登录**第一步**（POST /auth/login，公开）。
//
// 账号 = 用户名或手机号二选一（LoginReq 用 required_without 交叉约束）；两者都给时优先 phone。
// phone 走 model.IdentityTypePhone（Service 层 isValidPhone 权威校验 + 唯一键 (phone, 号, "") 命中），
// username 走 model.IdentityTypeUsername——两者共用同一套两步①②协议 + 统一 RS256 token。
//
// 出参（成功）：LoginStep1 → {loginTicket, user, memberships[]}（**不签 token**）。
// 出参（失败）：code 1001（统一文案防枚举）/ 2001（参数，含手机号位数不合法）/ 5002（DB·票据写失败）。
func (h *Handler) Login(c *fiber.Ctx) error {
	var req LoginReq
	if !validator.Validate(c, &req) { // 校验失败：body code=2001 已写出
		return nil
	}
	identityType, identifier := model.IdentityTypeUsername, req.Username
	if req.Phone != "" { // phone 优先（与 username 同传时按 phone 走）
		identityType, identifier = model.IdentityTypePhone, req.Phone
	}
	res, err := h.svc.Login(c.Context(), identityType, identifier, req.Password)
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// Issue 两步登录**第二步**（POST /auth/issue，公开，凭一次性票据）。
//
// 出参（成功）：TokenPair → {accessToken, refreshToken, expiresAt, user, roles, institutionId, campusId}。
// 出参（失败）：code 1001（票据失效）/ 1003（作用域越权/归属失效）/ 5002（签发写白名单失败）。
func (h *Handler) Issue(c *fiber.Ctx) error {
	var req IssueReq
	if !validator.Validate(c, &req) {
		return nil
	}
	res, err := h.svc.Issue(c.Context(), req.LoginTicket, req.InstitutionID, req.CampusID)
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// Me 登录态自查（GET /auth/me，**受保护**——需 T4-5 Auth 中间件前置验签并写入 claims）。
//
// 出参（成功）：MeResult → {user 字段 + roles, institutionId, campusId, memberships[]}（菜单刷新恢复）。
// 出参（失败）：code 1001（中间件未注入 claims = 未登录）/ 3001（用户不存在）/ 5002（DB 故障）。
func (h *Handler) Me(c *fiber.Ctx) error {
	claims := claimsFromCtx(c)
	if claims == nil { // 公开路由误配或中间件未执行：按未登录拒绝
		return response.Fail(c, apperrors.ErrUnauth)
	}
	res, err := h.svc.Me(c.Context(), *claims)
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// Refresh refresh 轮换（POST /auth/refresh，公开，凭旧 refresh）。
//
// 出参（成功）：TokenPair（新 access + 新 refresh；旧 refresh 立即失效）。
// 出参（失败）：code 1002（refresh 失效，前端跳重新登录）/ 5002（轮换写白名单/回读用户 DB 故障）。
func (h *Handler) Refresh(c *fiber.Ctx) error {
	var req RefreshReq
	if !validator.Validate(c, &req) {
		return nil
	}
	res, err := h.svc.Refresh(c.Context(), req.RefreshToken)
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// Logout 登出（POST /auth/logout，**受保护**，对客户端**恒 200** 幂等）。
//
// 语义：撤销"尽力而为"——删 refresh 白名单 + 拉黑当前 access jti；任一步失败仅记告警日志，
// 仍返回成功（前端本地态必清；access 30min 自然过期 + 黑名单兜底）。
//
// 出参（恒成功）：code 200，data 省略（无数据体）。
func (h *Handler) Logout(c *fiber.Ctx) error {
	var req LogoutReq
	// 尽力解析（refresh 可空；body 非 JSON 也不阻断——登出幂等，本地态必清）
	_ = json.Unmarshal(c.Body(), &req)

	claims := claimsFromCtx(c)
	if claims == nil {
		// 中间件未注入 claims（理论不该发生——logout 在受保护路由）：仍恒 200，仅记日志
		if h.log != nil {
			h.log.Warn("logout: 无已验签 claims（中间件未注入），仅清本地态")
		}
		return response.OK(c, nil)
	}
	remaining := time.Duration(0)
	if claims.ExpiresAt != nil {
		remaining = time.Until(claims.ExpiresAt.Time) // 黑名单键 TTL = access 剩余有效期
	}
	if err := h.svc.Logout(c.Context(), req.RefreshToken, claims.ID, remaining); err != nil {
		// 撤销失败不阻断登出（对齐前端 doLogout 幂等语义）：记告警供排障
		if h.log != nil {
			h.log.Warn("logout 撤销失败（已忽略，前端本地态必清）",
				zap.String("jti", claims.ID), zap.Error(err))
		}
	}
	return response.OK(c, nil)
}

// WechatQrcode 微信扫码登录——获取二维码（GET /auth/wechat/qrcode，**公开**）。
//
// 出参（成功）：QRCodeResult → {state, qrUrl, campusId, expiresIn}（前端渲染二维码 + 携 state 轮询）。
// 出参（失败）：code 4004（微信未配置——wechat 服务未注入 / appid·回调地址缺失）/ 5002（Redis 写 state 失败）。
func (h *Handler) WechatQrcode(c *fiber.Ctx) error {
	if h.wechat == nil {
		return response.Fail(c, apperrors.New(apperrors.CodeThirdNotConfigured, "微信扫码登录未配置"))
	}
	res, err := h.wechat.Qrcode(c.Context())
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// WechatStatus 微信扫码登录——轮询扫码状态（GET /auth/wechat/status?state=xxx，**公开**）。
//
// 出参（成功）：StatusResult → {status, campusId, loginTicket?, userId?, expired?}。
// 出参（失败）：code 4004（微信未配置）/ 5002（Redis 读 state 失败，fail-closed）。
func (h *Handler) WechatStatus(c *fiber.Ctx) error {
	if h.wechat == nil {
		return response.Fail(c, apperrors.New(apperrors.CodeThirdNotConfigured, "微信扫码登录未配置"))
	}
	state := c.Query("state")
	if state == "" {
		return response.Fail(c, apperrors.ErrBadParam)
	}
	res, err := h.wechat.Status(c.Context(), state)
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// WechatAuthorize 移动端 App 内微信授权——生成 webview 授权 URL（GET /auth/wechat/authorize，**公开**）。
//
// 与 WechatQrcode（PC 扫码）共用同一 state 状态机 / 同一 /auth/wechat/callback 换 openid /
// 同一套 ② /auth/issue 签 token；差异仅在前端拿到的 URL 形态：本端点返 **snsapi_userinfo** 的
// connect/oauth2/authorize（移动端 H5 跑微信内置浏览器，整页跳转），而 qrcode 返 snsapi_login 的
// connect/qrconnect（PC 渲染二维码，无 #wechat_redirect 尾段差异见 wechat.BuildAuthorizeURL）。
//
// 出参（成功）：AuthorizeResult → {state, authorizeUrl, campusId, expiresIn}（前端整页跳 authorizeUrl）。
// 出参（失败）：code 4004（微信未配置——wechat 服务未注入 / appid·回调地址缺失）/ 5002（Redis 写 state 失败）。
func (h *Handler) WechatAuthorize(c *fiber.Ctx) error {
	if h.wechat == nil {
		return response.Fail(c, apperrors.New(apperrors.CodeThirdNotConfigured, "微信授权登录未配置"))
	}
	res, err := h.wechat.Authorize(c.Context())
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// WechatCallback 微信扫码登录——微信重定向落点（GET /auth/wechat/callback?code=&state=，**公开**）。
//
// 行为：code 换 openid → 查 t_identity(wechat) → 已绑定发 loginTicket 写回 state=success；
// 未绑定写回 unbound；第三方失败写回 fail。前端拿到 success 的 loginTicket 后走**同一套** ② /auth/issue。
//
// 出参（成功）：StatusResult → {status, campusId, loginTicket?, userId?}。
// 出参（失败）：code 4004（微信未配置）/ 1001（state 失效/已终态，防重放）/ 2001（缺 code/state）/
//   4003（微信换 openid 失败）/ 5002（Redis/DB 故障）。
func (h *Handler) WechatCallback(c *fiber.Ctx) error {
	if h.wechat == nil {
		return response.Fail(c, apperrors.New(apperrors.CodeThirdNotConfigured, "微信扫码登录未配置"))
	}
	res, err := h.wechat.Callback(c.Context(), c.Query("code"), c.Query("state"))
	if err != nil {
		return response.Fail(c, err)
	}
	return response.OK(c, res)
}

// claimsFromCtx 从 fiber.Ctx 局部值读取已验签 claims（T4-5 中间件写入 *token.Claims）。
//
// 出参：*token.Claims；未写入或类型不符时返回 nil（调用方按未登录拒绝）。
func claimsFromCtx(c *fiber.Ctx) *token.Claims {
	if c == nil {
		return nil
	}
	if cl, ok := c.Locals(ClaimsCtxKey).(*token.Claims); ok {
		return cl
	}
	return nil
}