package com.frontier_apps.dev.spradz.viewmodel.viewmodelfactory;

import android.content.Context;

import androidx.annotation.NonNull;
import androidx.lifecycle.ViewModel;
import androidx.lifecycle.ViewModelProvider;

import com.frontier_apps.dev.spradz.model.Repository.YoutubeRepository;
import com.frontier_apps.dev.spradz.view.helper.ProgressBar;
import com.frontier_apps.dev.spradz.viewmodel.YoutubeViewModel;

public class YoutubeViewModelFactory extends ViewModelProvider.NewInstanceFactory {
    private YoutubeRepository youtubeRepository;

    public YoutubeViewModelFactory(ProgressBar progressBar, Context context) {
        youtubeRepository = new YoutubeRepository(progressBar, context);
    }

    @NonNull
    @Override
    public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
        return (T) new YoutubeViewModel(youtubeRepository);

    }
}