SDPepe/AppArt

package ch.epfl.sdp.appart.ad;

import android.util.Log;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.CompletableFuture;

import javax.inject.Inject;

import androidx.lifecycle.MutableLiveData;
import androidx.lifecycle.ViewModel;
import ch.epfl.sdp.appart.database.DatabaseService;
import ch.epfl.sdp.appart.login.LoginService;
import ch.epfl.sdp.appart.scrolling.card.Card;
import ch.epfl.sdp.appart.user.User;
import dagger.hilt.android.lifecycle.HiltViewModel;

@HiltViewModel
public class UserAdsViewModel extends ViewModel {
    private final MutableLiveData<List<Card>> lUserAds = new MutableLiveData<>();
    final DatabaseService database;
    final LoginService loginService;

    @Inject
    public UserAdsViewModel(DatabaseService database, LoginService loginService) {
        this.database = database;
        this.loginService = loginService;
    }

    public void initHome() {
        User currUser = loginService.getCurrentUser();
        if (currUser == null) {
            lUserAds.setValue(new ArrayList<>());
            Log.d("USER_ADS", "Current user is null!");
            return;
        }

        CompletableFuture<User> user = database.getUser(currUser.getUserId());
        user.exceptionally(e -> {
            Log.d("EXCEPTION_DB", e.getMessage());
            return null;
        });
        user.thenAccept(u -> {
            CompletableFuture<List<Card>> cards = database.getCards();
            cards.exceptionally(e -> {
                Log.d("EXCEPTION_DB", e.getMessage());
                return null;
            });
            cards.thenAccept(cs -> {
                List<Card> filteredCards = new LinkedList<>();
                for (Card c : cs) {
                    if (c.getUserId().equals(u.getUserId()))
                        filteredCards.add(c);
                }
                Log.d("user_ads", "number of added cards : " + filteredCards.size());
                lUserAds.setValue(filteredCards);
            });
        });
    }


    /**
     * Getter for the LiveData of the list of favorite cards
     */
    public MutableLiveData<List<Card>> getUserAds() {
        return lUserAds;
    }
}