app/src/main/java/se/leap/bitmaskclient/providersetup/fragments/viewmodel/ProviderSelectionViewModelFactory.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

package se.leap.bitmaskclient.providersetup.fragments.viewmodel;

import android.content.res.AssetManager;

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

import java.io.File;

public class ProviderSelectionViewModelFactory implements ViewModelProvider.Factory {
    private final AssetManager assetManager;
    private final File externalFilesDir;

    public ProviderSelectionViewModelFactory(AssetManager assetManager, File externalFilesDir) {
        this.assetManager = assetManager;
        this.externalFilesDir = externalFilesDir;
    }

    @NonNull
    @Override
    public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
        if (modelClass.isAssignableFrom(ProviderSelectionViewModel.class)) {
            return (T) new ProviderSelectionViewModel(assetManager, externalFilesDir);
        }
        throw new IllegalArgumentException("Unknown ViewModel class");
    }
}