package wechatmp

import (
	"errors"
	"strings"
	"time"

	"github.com/beego/beego/v2/core/logs"

	"server/models"
)

// =============================================================
// 注册「账号注册」验证码：网页发起 → 展示带参二维码（scene）→
// 用户微信扫码关注 → 服务号被动回复 6 位验证码 → 网页填码核销。
// 复用 yz_platform_wechat_mp_verify_code 会话表（bind_type="register"），
// 扫码事件回复走通用 handleScanScene，无需新增事件处理。
// =============================================================

// WechatBindTypeRegister 注册验证码会话类型（yz_platform_wechat_mp_verify_code.bind_type）
const WechatBindTypeRegister = "register"

// ErrRegisterCodeInvalid 账号注册：验证码无效 / 已使用（引导语指向「注册」而非绑定用的「验证码」）
var ErrRegisterCodeInvalid = errors.New("验证码不正确或已过期：请在公众号内重新发送「注册」获取")

// RegisterQRDownNotice 带参二维码不可用（IP 白名单 40164 / 未认证 48001 等）时，
// 回显给注册页的用户级引导：详细原因见服务端日志，不暴露服务器出口 IP。
const RegisterQRDownNotice = "二维码暂不可用：请在公众号内直接发送「注册」获取验证码"

// StartRegisterVerify 创建注册验证码会话并生成带参二维码。
// 带参二维码生成失败（未认证公众号 48001 等）时回退：qrURL 为空、
// notice 返回引导（用户在公众号内发送「注册」获取验证码），scene 仍有效可轮询。
func StartRegisterVerify(ttl time.Duration) (scene, qrURL, notice string, expireAt time.Time, err error) {
	if ttl <= 0 {
		ttl = DefaultVerifyTTL
	}
	cfg, err := LoadEnabledConfig()
	if err != nil {
		return "", "", "", time.Time{}, err
	}
	scene = randomScene()
	code := randomCode6()
	now := time.Now()
	expireAt = now.Add(ttl)

	row := &models.WechatMpVerifyCode{
		Scene:      scene,
		Code:       code,
		Status:     models.WechatVerifyStatusWaiting,
		BindType:   WechatBindTypeRegister,
		ExpireAt:   &expireAt,
		CreateTime: now,
		UpdateTime: &now,
	}
	if _, err = models.Orm.Insert(row); err != nil {
		return "", "", "", time.Time{}, err
	}
	qrURL, err = CreateQRCode(cfg, scene, int(ttl.Seconds())+120)
	if err != nil {
		// 注册页匿名可访问：errcode / errmsg / 服务器出口 IP 等排查信息已在
		// CreateQRCode 内落到服务端日志（见 logAPIError），此处只回显用户级引导，
		// 避免向访客泄露服务器出口 IP 与内部错误细节；
		// 会话仍然有效，「公众号内发送注册」这条兜底路径可正常核销。
		logs.Warning("[wechat-mp] 注册二维码不可用，已回退公众号发送「注册」：%v", err)
		return scene, "", RegisterQRDownNotice, expireAt, nil
	}
	return scene, qrURL, "", expireAt, nil
}

// ConfirmAccountRegisterVerify 账号注册的验证码核销：
// 优先按 scene 核销二维码会话；该会话尚未扫码（用户没扫码，改为在公众号内发送
// 「注册」取码）、scene 失效或验证码不属于该会话时，直接按验证码认领会话。
//
// 账号注册不绑定手机号，服务号无法按手机号把「发送者」与网页会话关联起来，
// 因此不要求网页会话必须先被扫码：验证码有效即视为公众号内本人操作。
func ConfirmAccountRegisterVerify(scene, code string) (string, error) {
	code = strings.TrimSpace(code)
	if code == "" {
		return "", ErrCodeNotPending
	}
	if strings.TrimSpace(scene) != "" {
		if openID, err := ConfirmRegisterVerify(scene, code); err == nil {
			return openID, nil
		}
	}
	openID, err := ConfirmRegisterVerifyByCode(code)
	if err != nil {
		// 按码认领失败的提示要指向注册场景（发送「注册」），而非绑定场景的「验证码」
		if errors.Is(err, ErrCodeNotPending) || errors.Is(err, ErrCodeUsed) {
			return "", ErrRegisterCodeInvalid
		}
		return "", err
	}
	return openID, nil
}

// GetRegisterVerifyStatus 查询注册会话扫码状态（网页轮询展示「已扫码」）
func GetRegisterVerifyStatus(scene string) (scanned bool, expireAt time.Time, err error) {
	row, err := GetVerifyCode(scene)
	if err != nil {
		return false, time.Time{}, err
	}
	if row.ExpireAt != nil {
		expireAt = *row.ExpireAt
	}
	if row.BindType != WechatBindTypeRegister {
		return false, expireAt, ErrSceneInvalid
	}
	return row.Status == models.WechatVerifyStatusScanned, expireAt, nil
}

// ConfirmRegisterVerify 核销注册验证码：校验会话归属与状态，返回扫码人 openid。
// 要求会话已被扫码（openid 已写入）、验证码匹配，核销后一次性失效。
func ConfirmRegisterVerify(scene, code string) (openid string, err error) {
	row, err := GetVerifyCode(scene)
	if err != nil {
		return "", err
	}
	if row.BindType != WechatBindTypeRegister {
		return "", ErrSceneInvalid
	}
	switch row.Status {
	case models.WechatVerifyStatusWaiting:
		return "", ErrSceneNotScanned
	case models.WechatVerifyStatusUsed:
		return "", ErrCodeUsed
	case models.WechatVerifyStatusExpired:
		return "", ErrCodeExpired
	}
	if strings.TrimSpace(code) != row.Code {
		return "", ErrCodeNotPending
	}
	now := time.Now()
	if _, uerr := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
		Filter("id", row.ID).
		Filter("status", models.WechatVerifyStatusScanned).
		Update(map[string]interface{}{"status": models.WechatVerifyStatusUsed, "update_time": now}); uerr != nil {
		return "", uerr
	}
	return row.OpenID, nil
}

// CreateRegisterOpenIDVerify 公众号内发送「注册」时创建的注册会话：
// openid 已知（消息发送者），网页端按验证码核销（无带参二维码的兜底路径）。
func CreateRegisterOpenIDVerify(openid string, ttl time.Duration) (*models.WechatMpVerifyCode, error) {
	openid = strings.TrimSpace(openid)
	if openid == "" {
		return nil, errors.New("openid 为空")
	}
	if ttl <= 0 {
		ttl = DefaultVerifyTTL
	}
	now := time.Now()
	expireAt := now.Add(ttl)
	row := &models.WechatMpVerifyCode{
		Scene:      randomScene(),
		Code:       randomCode6(),
		OpenID:     openid,
		Status:     models.WechatVerifyStatusScanned, // 微信号已知，等同「已扫码待核销」
		BindType:   WechatBindTypeRegister,
		ExpireAt:   &expireAt,
		CreateTime: now,
		UpdateTime: &now,
	}
	if _, err := models.Orm.Insert(row); err != nil {
		return nil, err
	}
	return row, nil
}

// LatestPendingRegisterVerifyByOpenID 该微信号最近一条未核销的注册会话（重发验证码时复用）
func LatestPendingRegisterVerifyByOpenID(openid string) (*models.WechatMpVerifyCode, error) {
	var row models.WechatMpVerifyCode
	err := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
		Filter("openid", openid).
		Filter("bind_type", WechatBindTypeRegister).
		Filter("status", models.WechatVerifyStatusScanned).
		Filter("expire_at__gt", time.Now()).
		OrderBy("-id").
		One(&row)
	if err != nil {
		return nil, err
	}
	return &row, nil
}

// ConfirmRegisterVerifyByCode 无二维码兜底：按验证码认领「公众号发起」的注册会话。
// 只认领 bind_type=register、openid 非空、已扫码待核销的最近会话，CAS 核销防止重复使用。
func ConfirmRegisterVerifyByCode(code string) (string, error) {
	code = strings.TrimSpace(code)
	if code == "" {
		return "", ErrCodeNotPending
	}
	var row models.WechatMpVerifyCode
	err := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
		Filter("code", code).
		Filter("bind_type", WechatBindTypeRegister).
		Filter("openid__ne", "").
		Filter("status", models.WechatVerifyStatusScanned).
		Filter("expire_at__gt", time.Now()).
		OrderBy("-id").
		One(&row)
	if err != nil {
		return "", ErrCodeNotPending
	}
	now := time.Now()
	affected, uerr := models.Orm.QueryTable(new(models.WechatMpVerifyCode)).
		Filter("id", row.ID).
		Filter("status", models.WechatVerifyStatusScanned).
		Update(map[string]interface{}{"status": models.WechatVerifyStatusUsed, "update_time": now})
	if uerr != nil {
		return "", uerr
	}
	if affected == 0 {
		return "", ErrCodeUsed
	}
	return row.OpenID, nil
}