package com.example.bryan.cryptotracker;

import android.app.Application;
import android.arch.lifecycle.ViewModel;
import android.arch.lifecycle.ViewModelProvider;
import android.support.annotation.NonNull;

public class CryptoTrackerViewModelFactory implements ViewModelProvider.Factory {

    private Application mApplication;
    private int mExtra;

    public CryptoTrackerViewModelFactory(Application application, int extra) {
        mApplication = application;
        mExtra = extra;
    }

    @NonNull
    @Override
    public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
        return (T) new CryptoTrackerViewModel(mApplication, mExtra);
    }
}