main.com.ramcosta.composedestinations.codegen.templates.NavArgsGetterTemplate.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of codegen Show documentation
Annotation processing library for type-safe Jetpack Compose navigation with no boilerplate.
There is a newer version: 2.1.0-beta12
package com.ramcosta.composedestinations.codegen.templates

import com.ramcosta.composedestinations.codegen.codeGenBasePackageName
import com.ramcosta.composedestinations.codegen.commons.CORE_PACKAGE_NAME
import com.ramcosta.composedestinations.codegen.templates.core.FileTemplate
import com.ramcosta.composedestinations.codegen.templates.core.setOfImportable

const val NAV_ARGS_METHOD_WHEN_CASES = "[NAV_ARGS_METHOD_WHEN_CASES]"
const val NAV_GRAPH_ARGS_METHOD_WHEN_CASES = "[NAV_GRAPH_ARGS_METHOD_WHEN_CASES]"
const val INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_START = "[INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_START]"
const val INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_END = "[INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_END]"
const val NAV_GRAPH_ARGS_METHODS_SECTION_START = "[NAV_GRAPH_ARGS_METHODS_SECTION_START]"
const val NAV_GRAPH_ARGS_METHODS_SECTION_END = "[NAV_GRAPH_ARGS_METHODS_SECTION_END]"
const val INLINE_DESTINATION_ARGS_METHODS_SECTION_START = "[INLINE_DESTINATION_ARGS_METHODS_SECTION_START]"
const val INLINE_DESTINATION_ARGS_METHODS_SECTION_END = "[INLINE_DESTINATION_ARGS_METHODS_SECTION_END]"
const val DESTINATION_ARGS_METHODS_SECTION_START = "[DESTINATION_ARGS_METHODS_SECTION_START]"
const val DESTINATION_ARGS_METHODS_SECTION_END = "[DESTINATION_ARGS_METHODS_SECTION_END]"
private const val CLASS_ESCAPED = "\${argsClass}"

val navArgsGettersTemplate = FileTemplate(
    packageStatement = "@file:Suppress(\"UNCHECKED_CAST\")\n\npackage $codeGenBasePackageName",
    imports = setOfImportable(
        "androidx.lifecycle.SavedStateHandle",
        "androidx.navigation.NavBackStackEntry",
        "$CORE_PACKAGE_NAME.spec.DestinationSpec",
        "$CORE_PACKAGE_NAME.spec.NavGraphSpec"
    ),
    sourceCode = """
$INLINE_DESTINATION_ARGS_METHODS_SECTION_START${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  SavedStateHandle.navArgs(): T {
    return navArgs(T::class.java, this)
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  NavBackStackEntry.navArgs(): T {
    return navArgs(T::class.java, this)
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public fun  navArgs(argsClass: Class, argsContainer: NavBackStackEntry): T {
    return destinationWithArgsType(argsClass).argsFrom(argsContainer) as T
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public fun  navArgs(argsClass: Class, argsContainer: SavedStateHandle): T {
    return destinationWithArgsType(argsClass).argsFrom(argsContainer) as T
}$INLINE_DESTINATION_ARGS_METHODS_SECTION_END

$INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_START${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  SavedStateHandle.navGraphArgs(): T? {
    return navGraphArgs(T::class.java, this)
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  NavBackStackEntry.navGraphArgs(): T? {
    return navGraphArgs(T::class.java, this)
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  SavedStateHandle.requireNavGraphArgs(): T {
    return requireNavGraphArgs(T::class.java, this)
}

${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}public inline fun  NavBackStackEntry.requireNavGraphArgs(): T {
    return requireNavGraphArgs(T::class.java, this)
}

public fun  navGraphArgs(argsClass: Class, argsContainer: SavedStateHandle): T? {
    return navGraphWithArgsType(argsClass).argsFrom(argsContainer) as T?
}

public fun  navGraphArgs(argsClass: Class, argsContainer: NavBackStackEntry): T? {
    return navGraphWithArgsType(argsClass).argsFrom(argsContainer) as T?
}

public fun  requireNavGraphArgs(argsClass: Class, argsContainer: SavedStateHandle): T {
    return navGraphWithArgsType(argsClass).requireGraphArgs(argsContainer) as T
}

public fun  requireNavGraphArgs(argsClass: Class, argsContainer: NavBackStackEntry): T {
    return navGraphWithArgsType(argsClass).requireGraphArgs(argsContainer) as T
}$INLINE_NAV_GRAPH_ARGS_METHODS_SECTION_END

$DESTINATION_ARGS_METHODS_SECTION_START${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}private fun  destinationWithArgsType(argsClass: Class): DestinationSpec {
    return when (argsClass) {
$NAV_ARGS_METHOD_WHEN_CASES
        else -> error("Class $CLASS_ESCAPED is not a destination arguments class known by this module!")
    }
}$DESTINATION_ARGS_METHODS_SECTION_END

${NAV_GRAPH_ARGS_METHODS_SECTION_START}${REQUIRE_OPT_IN_ANNOTATIONS_PLACEHOLDER}private fun  navGraphWithArgsType(argsClass: Class): NavGraphSpec {
    return when (argsClass) {
$NAV_GRAPH_ARGS_METHOD_WHEN_CASES
        else -> error("Class $CLASS_ESCAPED is not a navigation graph arguments class known by this module!")
    }
}$NAV_GRAPH_ARGS_METHODS_SECTION_END

""".trimIndent()
)