aboutsummaryrefslogblamecommitdiff
path: root/app/src/main/java/ml/docilealligator/infinityforreddit/message/MessageViewModel.java
blob: dbd497bf5377b41dbb2c0388d6e6c48bb5c56b05 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
13
                                                     











                                            
                                                         


                                                 





                                                                    










                                                                                                                   
                                                     












                                                                                                 
                                                       


                        
                                                               


                                      
                                                            


                                   
                                           


                                  
                           


                                                                     
                                    







                                                                              



                                         














                                                                                            
package ml.docilealligator.infinityforreddit.message;

import androidx.annotation.NonNull;
import androidx.lifecycle.LiveData;
import androidx.lifecycle.MutableLiveData;
import androidx.lifecycle.Transformations;
import androidx.lifecycle.ViewModel;
import androidx.lifecycle.ViewModelProvider;
import androidx.paging.LivePagedListBuilder;
import androidx.paging.PagedList;

import java.util.Locale;

import ml.docilealligator.infinityforreddit.NetworkState;
import retrofit2.Retrofit;

public class MessageViewModel extends ViewModel {
    private final MessageDataSourceFactory messageDataSourceFactory;
    private final LiveData<NetworkState> paginationNetworkState;
    private final LiveData<NetworkState> initialLoadingState;
    private final LiveData<Boolean> hasMessageLiveData;
    private final LiveData<PagedList<Message>> messages;
    private final MutableLiveData<String> whereLiveData;

    public MessageViewModel(Retrofit retrofit, Locale locale, String accessToken, String where) {
        messageDataSourceFactory = new MessageDataSourceFactory(retrofit, locale, accessToken, where);

        initialLoadingState = Transformations.switchMap(messageDataSourceFactory.getMessageDataSourceLiveData(),
                MessageDataSource::getInitialLoadStateLiveData);
        paginationNetworkState = Transformations.switchMap(messageDataSourceFactory.getMessageDataSourceLiveData(),
                MessageDataSource::getPaginationNetworkStateLiveData);
        hasMessageLiveData = Transformations.switchMap(messageDataSourceFactory.getMessageDataSourceLiveData(),
                MessageDataSource::hasPostLiveData);

        whereLiveData = new MutableLiveData<>(where);

        PagedList.Config pagedListConfig =
                (new PagedList.Config.Builder())
                        .setEnablePlaceholders(false)
                        .setPageSize(25)
                        .build();

        messages = Transformations.switchMap(whereLiveData, newWhere -> {
            messageDataSourceFactory.changeWhere(whereLiveData.getValue());
            return (new LivePagedListBuilder(messageDataSourceFactory, pagedListConfig)).build();
        });
    }

    public LiveData<PagedList<Message>> getMessages() {
        return messages;
    }

    public LiveData<NetworkState> getPaginationNetworkState() {
        return paginationNetworkState;
    }

    public LiveData<NetworkState> getInitialLoadingState() {
        return initialLoadingState;
    }

    public LiveData<Boolean> hasMessage() {
        return hasMessageLiveData;
    }

    public void refresh() {
        messageDataSourceFactory.getMessageDataSource().invalidate();
    }

    public void retryLoadingMore() {
        messageDataSourceFactory.getMessageDataSource().retryLoadingMore();
    }

    void changeWhere(String where) {
        whereLiveData.postValue(where);
    }

    public static class Factory extends ViewModelProvider.NewInstanceFactory {
        private final Retrofit retrofit;
        private final Locale locale;
        private final String accessToken;
        private final String where;

        public Factory(Retrofit retrofit, Locale locale, String accessToken, String where) {
            this.retrofit = retrofit;
            this.locale = locale;
            this.accessToken = accessToken;
            this.where = where;
        }

        @NonNull
        @Override
        public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
            return (T) new MessageViewModel(retrofit, locale, accessToken, where);
        }
    }
}