OpenFlare/backend/plugins/domain/auth/cap_service.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package auth

import (
	"Wavelet/plugins/domain/auth/pow"
	"context"
	"crypto/sha256"
	"encoding/hex"
	"strconv"
	"strings"
	"sync"
	"time"
)

const (
	redeemTokenIDLength  = 8  // 兑换 Token ID 字节长度
	redeemVerTokenLength = 15 // 兑换验证 Token 字节长度
	tokenPartsCount      = 2  // 兑换 Token 由两部分组成
	valuePartsCount      = 2  // 存储值由 scope 和过期时间组成
)

// CaptchaManager orchestrates challenge generation and solution validation.
type CaptchaManager struct {
	secret []byte
	store  pow.Store
}

// NewCaptchaManager creates a new CAPTCHA Manager.
func NewCaptchaManager(secret []byte, store pow.Store) *CaptchaManager {
	return &CaptchaManager{
		secret: secret,
		store:  store,
	}
}

// Generate creates a challenge response.
func (m *CaptchaManager) Generate(ctx context.Context, scope string) (*pow.ChallengeResponse, error) {
	settings, err := CurrentCapSettings(ctx)
	if err != nil {
		return nil, err
	}

	challengeConfig := pow.ChallengeConfig{
		Count:      settings.ChallengeCount,
		Size:       settings.ChallengeSize,
		Difficulty: settings.ChallengeDifficulty,
		Expires:    settings.ChallengeTTL,
	}
	return pow.GenerateChallenge(m.secret, challengeConfig, scope)
}

// Redeem verifies PoW solutions and returns a one-time redeem token.
func (m *CaptchaManager) Redeem(ctx context.Context, token string, solutions []int, scope string) (*RedeemResponse, error) {
	sigHex := pow.JwtSigHex(token)
	if sigHex == "" {
		return &RedeemResponse{Success: false, Error: redeemErrInvalidToken}, nil
	}

	nonceKey := "cap:nonce:" + sigHex

	payload, err := pow.VerifyChallengeSolutions(token, solutions, m.secret, scope)
	if err != nil {
		return &RedeemResponse{Success: false, Error: err.Error()}, nil //nolint:nilerr // validation errors are returned as response, not system errors
	}

	now := time.Now().UnixNano() / int64(time.Millisecond)
	nonceTTL := time.Duration(payload.Expires-now) * time.Millisecond
	if nonceTTL < time.Second {
		nonceTTL = time.Second
	}

	set, err := m.store.SetNX(ctx, nonceKey, "1", nonceTTL)
	if err != nil {
		return &RedeemResponse{Success: false, Error: redeemErrNonceStoreFailed}, err
	}
	if !set {
		return &RedeemResponse{Success: false, Error: redeemErrAlreadyRedeemed}, nil
	}

	settings, err := CurrentCapSettings(ctx)
	if err != nil {
		return &RedeemResponse{Success: false, Error: redeemErrSettingsLoad}, err
	}

	id := pow.RandomHex(redeemTokenIDLength)
	verToken := pow.RandomHex(redeemVerTokenLength)
	verHashBytes := sha256.Sum256([]byte(verToken))
	verHashHex := hex.EncodeToString(verHashBytes[:])

	tokenKey := "cap:token:" + id + ":" + verHashHex
	tokenExpires := time.Now().Add(settings.TokenTTL)
	storeVal := strconv.FormatInt(tokenExpires.UnixNano(), 10) + "|" + scope

	if err := m.store.Set(ctx, tokenKey, storeVal, settings.TokenTTL); err != nil {
		return &RedeemResponse{Success: false, Error: redeemErrTokenStoreFailed}, err
	}

	return &RedeemResponse{
		Success: true,
		Token:   id + ":" + verToken,
		Expires: tokenExpires.UnixNano() / int64(time.Millisecond),
	}, nil
}

// VerifyToken validates and consumes the redeem token (single-use).
func (m *CaptchaManager) VerifyToken(ctx context.Context, token, expectedScope string) (bool, error) {
	if token == "" {
		return false, nil
	}
	parts := strings.Split(token, ":")
	if len(parts) != tokenPartsCount {
		return false, nil
	}
	id := parts[0]
	verToken := parts[1]

	verHashBytes := sha256.Sum256([]byte(verToken))
	verHashHex := hex.EncodeToString(verHashBytes[:])

	tokenKey := "cap:token:" + id + ":" + verHashHex

	val, exists, err := sGetAndDelete(ctx, m.store, tokenKey)
	if err != nil {
		return false, err
	}
	if !exists {
		return false, nil
	}

	valParts := strings.Split(val, "|")
	if len(valParts) != valuePartsCount {
		return false, nil
	}

	expNano, err := strconv.ParseInt(valParts[0], 10, 64)
	if err != nil {
		return false, nil //nolint:nilerr // invalid format is treated as validation failure
	}
	tokenScope := valParts[1]

	if expectedScope != "" && tokenScope != expectedScope {
		return false, nil
	}

	if time.Now().UnixNano() > expNano {
		return false, nil
	}

	return true, nil
}

func sGetAndDelete(ctx context.Context, store pow.Store, key string) (string, bool, error) {
	if store == nil {
		return "", false, nil
	}
	return store.GetAndDelete(ctx, key)
}

var (
	defaultCapManagerMu sync.RWMutex
	defaultCapManager   *CaptchaManager
)

// SetCapSecret sets the shared secret used by the default CAPTCHA manager.
func SetCapSecret(secret []byte) {
	defaultCapManagerMu.Lock()
	defer defaultCapManagerMu.Unlock()
	if len(secret) > 0 {
		store := pow.NewMemoryStore(1 * time.Minute)
		defaultCapManager = NewCaptchaManager(secret, store)
	}
}

// GetDefaultCapManager yields the global singleton CAPTCHA manager.
func GetDefaultCapManager() *CaptchaManager {
	defaultCapManagerMu.RLock()
	defer defaultCapManagerMu.RUnlock()
	return defaultCapManager
}

type captchaService struct{}

func (captchaService) VerifyMiddleware(scope string) any {
	return VerifyCaptchaMiddleware(GetDefaultCapManager(), scope)
}

func (captchaService) ChallengeHandler() any { return Challenge }

func (captchaService) RedeemHandler() any { return Redeem }