Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
package org.mozilla.tv.firefox.architecture
import androidx.lifecycle.ViewModelProvider
import androidx.lifecycle.ViewModelProviders
import androidx.fragment.app.Fragment
import androidx.fragment.app.FragmentActivity
import org.mozilla.tv.firefox.ext.serviceLocator
/**
* Returns an instance of the view model for the given application building block (fragment, activity)
* for the given lifecycle. This is a wrapper around the framework's [ViewModelProvider] using the application's
* [ViewModelFactory] to reduce code duplication.
*/
object FirefoxViewModelProviders {
fun of(activity: FragmentActivity): ViewModelProvider {
return ViewModelProviders.of(activity, activity.serviceLocator.viewModelFactory)
}
fun of(fragment: Fragment): ViewModelProvider {
// If we're attempting to retrieve a view model, we should be attached to a context already.
return ViewModelProviders.of(fragment, fragment.context!!.serviceLocator.viewModelFactory)
}
}