package com.topjohnwu.magisk.model.navigation

import android.os.Bundle
import androidx.annotation.AnimRes
import androidx.annotation.AnimatorRes
import androidx.fragment.app.Fragment
import com.skoumal.teanity.viewevents.NavigationDslMarker
import com.skoumal.teanity.viewevents.ViewEvent
import kotlin.reflect.KClass

class MagiskNavigationEvent(
    val navDirections: MagiskNavDirectionsBuilder,
    val navOptions: MagiskNavOptions,
    val animOptions: MagiskAnimBuilder
) : ViewEvent() {

    companion object {
        operator fun invoke(builder: Builder.() -> Unit) = Builder().apply(builder).build()
    }

    @NavigationDslMarker
    class Builder {

        private var animOptions: MagiskAnimBuilder = MagiskAnimBuilder()
        private var navOptions: MagiskNavOptions = MagiskNavOptions()
        private val directionsBuilder = MagiskNavDirectionsBuilder()

        fun args(builder: Bundle.() -> Unit) = directionsBuilder.args(builder)

        fun navAnim(builder: MagiskAnimBuilder.() -> Unit) {
            animOptions = MagiskAnimBuilder().apply(builder)
        }

        fun navOptions(builder: MagiskNavOptions.() -> Unit) {
            navOptions = MagiskNavOptions().apply(builder)
        }

        fun navDirections(builder: MagiskNavDirectionsBuilder.() -> Unit) {
            directionsBuilder.apply(builder)
        }

        internal fun build() = MagiskNavigationEvent(directionsBuilder, navOptions, animOptions)
    }
}

@NavigationDslMarker
class MagiskNavDirectionsBuilder {

    var destination: KClass<out Fragment>? = null
    var isActivity: Boolean = false
    val args: Bundle = Bundle()

    fun args(builder: Bundle.() -> Unit) = args.apply(builder)

}

@NavigationDslMarker
class MagiskNavOptions {
    var popUpTo: KClass<*>? = null
    var inclusive: Boolean = false
    var clearTask: Boolean = false
    var singleTop: Boolean = false
}

@NavigationDslMarker
class MagiskAnimBuilder {
    @AnimRes
    @AnimatorRes
    var enter = 0

    @AnimRes
    @AnimatorRes
    var exit = 0

    @AnimRes
    @AnimatorRes
    var popEnter = 0

    @AnimRes
    @AnimatorRes
    var popExit = 0
}