isolaatti-android/app/src/main/java/com/isolaatti/home/presentation/FeedViewModel.kt

package com.isolaatti.home.presentation

import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import com.isolaatti.auth.domain.AuthRepository
import com.isolaatti.posting.posts.domain.PostsRepository
import com.isolaatti.posting.posts.presentation.PostListingViewModelBase
import com.isolaatti.posting.posts.presentation.UpdateEvent
import com.isolaatti.profile.data.remote.UserProfileDto
import com.isolaatti.profile.domain.use_case.GetProfile
import com.isolaatti.utils.Resource
import dagger.hilt.android.lifecycle.HiltViewModel
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.flow.flowOn
import kotlinx.coroutines.flow.launchIn
import kotlinx.coroutines.flow.onEach
import kotlinx.coroutines.launch
import javax.inject.Inject

@HiltViewModel
class FeedViewModel @Inject constructor(
    private val getProfileUseCase: GetProfile,
    private val authRepository: AuthRepository,
    private val postsRepository: PostsRepository
) : PostListingViewModelBase() {

    override fun getFeed(refresh: Boolean) {
        viewModelScope.launch {
            if (refresh) {
                posts.value = null
            }
            postsRepository.getFeed(getLastId()).onEach { feedDtoResource ->
                when (feedDtoResource) {
                    is Resource.Success -> {
                        loadingPosts.postValue(false)
                        posts.postValue(Pair(posts.value?.first?.concatFeed(feedDtoResource.data) ?: feedDtoResource.data, UpdateEvent(UpdateEvent.UpdateType.PAGE_ADDED, null)))
                        noMoreContent.postValue(feedDtoResource.data?.moreContent == false)
                    }

                    is Resource.Loading -> {
                        loadingPosts.postValue(true)
                    }

                    is Resource.Error -> {
                        errorLoading.postValue(feedDtoResource.errorType)
                    }
                }
            }.flowOn(Dispatchers.IO).launchIn(this)
        }
    }

    // User profile
    private val _userProfile: MutableLiveData<UserProfileDto> = MutableLiveData()
    val userProfile: LiveData<UserProfileDto> get() = _userProfile

    fun getProfile() {
        viewModelScope.launch {
            authRepository.getUserId().onEach { userId ->
                userId?.let {
                    getProfileUseCase(userId).onEach { profile ->
                        if (profile is Resource.Success) {
                            profile.data?.let { _userProfile.postValue(it) }
                        }
                    }.flowOn(Dispatchers.IO).launchIn(this)
                }

            }.flowOn(Dispatchers.IO).launchIn(this)

        }

    }
}