pangolin/client/lib/state/quota_provider.dart

// quota_provider.dart — 免费版每日额度状态(Riverpod)
//
// 设计约定(design/CLAUDE.md §7 / §2):免费额度权威在服务端,且**全账户共享**
// (非每设备)。总额度取自 me.quota_cap_min(= 套餐每日上限 + 看广告累加分钟),
// 今日剩余取自 me.quota_today_min(后端已算好 = 额度 − 今日已用)。
//
// 看广告加时(累加式):watchAd() 调 /v1/ads/unlock,服务端校验后 +N 分钟并回传最新
// 剩余,客户端随即刷新 me 让额度权威同步。占位广告 SDK 阶段用客户端生成的 ad_token,
// 服务端 DevVerifier 放行(nonce 仍防重放)。
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:uuid/uuid.dart';

import '../services/device_identity.dart';
import 'account_providers.dart';
import 'auth_provider.dart';

/// 免费额度快照。
class FreeQuotaState {
  const FreeQuotaState({
    this.totalMinutes = 10,
    this.remainingMinutes = 10,
  });

  /// 今日总额度(分钟)= 套餐每日上限 + 看广告累加。§7:免费版基础每日 10 分钟。
  final int totalMinutes;

  /// 今日剩余分钟(权威以服务端为准;连接期倒计时由连接状态机本地推算)。
  final int remainingMinutes;

  /// 进度(0–1),用于进度条宽度。
  double get progress =>
      totalMinutes == 0 ? 0 : (remainingMinutes / totalMinutes).clamp(0.0, 1.0);

  /// 是否进入低额度警示(≤3 分钟切 warning 色)。
  bool get isLow => remainingMinutes <= 3;

  /// 今日额度是否已耗尽(剩余 0):连接按钮据此灰化,点击弹广告/升级。
  bool get isExhausted => remainingMinutes <= 0;

  FreeQuotaState copyWith({int? totalMinutes, int? remainingMinutes}) => FreeQuotaState(
        totalMinutes: totalMinutes ?? this.totalMinutes,
        remainingMinutes: remainingMinutes ?? this.remainingMinutes,
      );
}

class QuotaController extends StateNotifier<FreeQuotaState> {
  QuotaController(this._ref) : super(const FreeQuotaState()) {
    _sync();
    // me / plans 任一就绪或变化都重算。
    _ref.listen(meProvider, (_, __) => _sync());
    _ref.listen(plansProvider, (_, __) => _sync());
  }

  final Ref _ref;

  void _sync() {
    final me = _ref.read(meProvider).valueOrNull;
    final plans = _ref.read(plansProvider).valueOrNull;
    var base = 10; // §7 默认免费基础 10 分钟,plans 就绪后以其为准
    if (plans != null) {
      for (final p in plans) {
        if (p.code == 'free' && p.dailyMinutes != null) base = p.dailyMinutes!;
      }
    }
    // 总额度优先取服务端 quota_cap_min(含看广告加时);缺省回退基础额度。
    final total = me?.quotaCapMin ?? base;
    final remaining = (me?.quotaTodayMin ?? total).clamp(0, total);
    state = FreeQuotaState(totalMinutes: total, remainingMinutes: remaining);
  }

  /// 连接期倒计时归零 → 本地立即置耗尽(按钮随即灰化);登录态下再拉 me 让服务端权威同步。
  void markExhausted() {
    state = state.copyWith(remainingMinutes: 0);
    if (_ref.read(authProvider).isLoggedIn) {
      _ref.read(meProvider.notifier).refresh();
    }
  }

  /// 看广告加时:调 /v1/ads/unlock 累加分钟,成功后刷新 me 拿最新额度。
  /// 返回本次加时分钟(null = 失败)。占位阶段用客户端生成的 ad_token。
  Future<int?> watchAd() async {
    try {
      final deviceId = await _ref.read(deviceIdentityProvider).deviceId();
      final res = await _ref.read(accountApiProvider).adUnlock(
            deviceId: deviceId,
            adToken: const Uuid().v4(), // 占位 ad_token(DevVerifier 放行)
          );
      // 乐观置位剩余,再拉 me 校准(账户共享,以服务端为准)。
      state = state.copyWith(remainingMinutes: res.minutesRemaining);
      await _ref.read(meProvider.notifier).refresh();
      return res.grantedMinutes;
    } catch (_) {
      return null;
    }
  }
}

final quotaProvider = StateNotifierProvider<QuotaController, FreeQuotaState>(
  (ref) => QuotaController(ref),
);