AI-Health/health_app/lib/providers/auth_provider.dart

import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:dio/dio.dart';
import '../core/api_client.dart';
import '../core/local_database.dart';

class UserInfo {
  final String id;
  final String phone;
  final String role;
  final String? name;
  final String? avatarUrl;

  UserInfo({required this.id, required this.phone, this.role = 'User', this.name, this.avatarUrl});
}

class AuthState {
  final UserInfo? user;
  final bool isLoggedIn;
  final bool isLoading;

  const AuthState({this.user, this.isLoggedIn = false, this.isLoading = true});
}

final authProvider = NotifierProvider<AuthNotifier, AuthState>(AuthNotifier.new);

final localDbProvider = Provider<LocalDatabase>((ref) => LocalDatabase.instance);

final apiClientProvider = Provider<ApiClient>((ref) {
  return ApiClient(db: ref.watch(localDbProvider));
});

class AuthNotifier extends Notifier<AuthState> {
  @override
  AuthState build() {
    _checkAuth();
    return const AuthState(isLoading: true);
  }

  Future<void> _checkAuth() async {
    final db = ref.read(localDbProvider);
    final refresh = await db.read('refresh_token');
    if (refresh == null) {
      state = const AuthState(isLoggedIn: false, isLoading: false);
      return;
    }

    try {
      final response = await Dio(BaseOptions(baseUrl: baseUrl))
          .post('/api/auth/refresh', data: {'refreshToken': refresh});
      final data = response.data['data'];
      if (data != null) {
        await db.write('access_token', data['accessToken']);
        await db.write('refresh_token', data['refreshToken']);
        final u = data['user'] as Map<String, dynamic>?;
        state = AuthState(isLoggedIn: true, isLoading: false,
          user: UserInfo(id: '', phone: '', role: u?['role'] ?? 'User'),
        );
        _loadProfile();
      } else {
        state = const AuthState(isLoggedIn: false, isLoading: false);
      }
    } catch (_) {
      state = const AuthState(isLoggedIn: false, isLoading: false);
    }
  }

  Future<void> _loadProfile() async {
    try {
      final api = ref.read(apiClientProvider);
      final response = await api.get('/api/user/profile');
      final user = response.data['data'];
      if (user != null) {
        state = AuthState(isLoggedIn: true, isLoading: false,
          user: UserInfo(
            id: user['id'] ?? '', phone: user['phone'] ?? '',
            role: user['role'] ?? state.user?.role ?? 'User',
            name: user['name'], avatarUrl: user['avatarUrl'],
          ),
        );
      }
    } catch (e) {
      print('[Auth] loadProfile: $e');
    }
  }

  /// 发送验证码
  Future<({String? error, String? devCode})> sendSms(String phone) async {
    try {
      final api = ref.read(apiClientProvider);
      final response = await api.post('/api/auth/send-sms', data: {'phone': phone});
      return (error: null, devCode: response.data['data']?['devCode'] as String?);
    } catch (e) {
      return (error: '发送失败: $e', devCode: null);
    }
  }

  /// 注册（新用户，需选身份）
  Future<String?> register(String phone, String code, String role, {String? inviteCode}) async {
    try {
      final api = ref.read(apiClientProvider);
      final response = await api.post('/api/auth/register', data: {
        'phone': phone, 'smsCode': code, 'role': role,
        if (inviteCode != null) 'inviteCode': inviteCode,
      });
      final data = response.data['data'];
      if (data == null) return response.data['message'] ?? '注册失败';

      await api.saveTokens(data['accessToken'], data['refreshToken']);
      final user = data['user'];
      state = AuthState(isLoggedIn: true, isLoading: false,
        user: UserInfo(id: user['id'] ?? '', phone: user['phone'] ?? '', role: role),
      );
      return null;
    } catch (e) {
      return '注册失败: $e';
    }
  }

  /// 登录（已有账号）
  Future<String?> login(String phone, String code) async {
    try {
      final api = ref.read(apiClientProvider);
      final response = await api.post('/api/auth/login', data: {'phone': phone, 'smsCode': code});
      final data = response.data['data'];
      if (data == null) return response.data['message'] ?? '登录失败';

      await api.saveTokens(data['accessToken'], data['refreshToken']);
      final user = data['user'];
      state = AuthState(isLoggedIn: true, isLoading: false,
        user: UserInfo(
          id: user['id'] ?? '', phone: user['phone'] ?? '',
          role: user['role'] ?? 'User', name: user['name'],
          avatarUrl: user['avatarUrl'],
        ),
      );
      return null;
    } catch (e) {
      return '登录失败: $e';
    }
  }

  /// 登出
  Future<void> logout() async {
    final api = ref.read(apiClientProvider);
    final db = ref.read(localDbProvider);
    final refresh = await db.read('refresh_token');
    if (refresh != null) {
      try { await api.post('/api/auth/logout', data: {'refreshToken': refresh}); } catch (e) { print('[Auth] logout: $e'); }
    }
    await api.clearTokens();
    state = const AuthState(isLoggedIn: false, isLoading: false);
  }
}

/// 便捷：当前用户角色
final userRoleProvider = Provider<String>((ref) {
  return ref.watch(authProvider).user?.role ?? 'User';
});