implementation("family.amma:keemun:1.2.3")

@Parcelize
data class State(
    val user: User?
) : Parcelable

data class ViewState(
    // Example
    // English - "Age: 30"
    // Russian - "Возраст: 30"
    // etc
    val localizedAge: String?
)

sealed class Msg {
    data class UserWasLoaded(val user: User) : Msg()
}

sealed class Effect {
    object LoadUser : Effect()
}

typealias SomeFeatureParams = FeatureParams<State, Msg, Effect>
typealias SomeFeatureEffectHandler = EffectHandler<Effect, Msg>

internal fun someFeatureParams(effectHandler: SomeFeatureEffectHandler): SomeFeatureParams =
    FeatureParams(
        init = init,
        update = updater,
        effectHandler = effectHandler
    )

val init = InitFeature<State, Effect> { previous ->
    val state = previous ?: State(user = null)
    state to setOf(Effect.LoadUser)
}

val updater = Update<State, Msg> { msg, state ->
    when (msg) {
        is Msg.UserWasLoaded -> state.copy(user = msg.user) to emptySet()
    }  
}

fun effectHandler(repo: UserRepository) = SomeFeatureEffectHandler { effect, dispatch ->
    when (effect) {
        Effect.LoadUser -> {
            val user = repo.loadUser()
            dispatch(Msg.UserWasLoaded(user))
        }
    }
}

// StateTransform<State, ViewState> = suspend (State) -> ViewState 
private fun stateTransform(getContext: () -> Context) = StateTransform<State, ViewState> { state ->
    ViewState(
        localizedAge = state.user?.age?.let { getContext().getString(R.string.age, it) }
    )
}

class SomeFragment(featureParams: () -> SomeFeatureParams) : Fragment(R.layout.fragment_some) {
    private val feature by androidConnectors(featureParams, getStateTransform = { stateTransform(::requireContext) })

    override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
        super.onViewCreated(view, savedInstanceState)
        feature.render(this) { viewState ->
            ageTextView.text = viewState.localizedAge
        }
        // OR 
        // feature.collectWithLifecycle(viewLifecycleOwner, Lifecycle.State.STARTED) { viewState -> }
    }
}

feature dispatch Msg.Foo
feature syncDispatch Msg.Bar