ViewModel Kotlin Android Studio Template

ViewModel.kt

#if (${PACKAGE_NAME} && ${PACKAGE_NAME} != "")package ${PACKAGE_NAME}#end

import com.mstv.presentation.scenes.base.view.ContentState
import com.mstv.presentation.scenes.base.view.LoadingState

data class ${NAME}(val loadingState: LoadingState = LoadingState.NONE,
                   val contentState: ContentState = ContentState.NONE,
                   val data: ${DataType}? = null,
                   val errorMessage: String? = null,
                   val snackMessage: String? = null) {

    companion object {
        fun createLoading() = ${NAME}(loadingState = LoadingState.LOADING, contentState = ContentState.CONTENT)

        fun createRetryLoading() = ${NAME}(loadingState = LoadingState.RETRY, contentState = ContentState.ERROR)

        fun createData(data: ${DataType}) = ${NAME}(contentState = ContentState.CONTENT, data = data)

        fun createError(error: String) = ${NAME}(contentState = ContentState.ERROR, errorMessage = error)

        fun createSnack(snackMessage: String) = ${NAME}(contentState = ContentState.CONTENT, snackMessage = snackMessage)
    }

}