justMusic/Sources/justMUSIC/lib/view_model/UserViewModel.dart

import 'package:firebase_auth/firebase_auth.dart' as firebase_auth;
import 'package:justmusic/service/AuthService.dart';

import '../model/User.dart';
import '../model/mapper/UserMapper.dart';
import '../main.dart';

class UserViewModel {
  late User _userCurrent;
  final AuthService _authService = AuthService();

  User get userCurrent => _userCurrent;

  set userCurrent(User value) {
    _userCurrent = value;
  } // Constructor

  UserViewModel();

  // Methods
  Future<User?> getUser(String id) async {
    final user = await MyApp.db.collection("users").doc(id).get();
    return UserMapper.toModel(user, null);
  }

  login(String pseudo, String password) async {
    try {
      await _authService.login(pseudo, password);
      final user = await MyApp.db
          .collection("users")
          .doc(firebase_auth.FirebaseAuth.instance.currentUser?.uid)
          .get();
      User data = UserMapper.toModel(user, null);
      _userCurrent = data;
    } catch (e) {
      rethrow;
    }
  }

  register(String pseudo, String password, String email) async {
    _authService.register(pseudo, email, password);
    final user = await MyApp.db
        .collection("users")
        .doc(firebase_auth.FirebaseAuth.instance.currentUser?.uid)
        .get();
    User? data = UserMapper.toModel(user, null);
    _userCurrent = data;
  }

  logout() {
    _authService.signOut();
  }
}