com.freeletics.mad.whetstone.codegen.common.NavDestinationModuleGenerator.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of whetstone-compiler Show documentation

Whetstone Compiler

There is a newer version: 0.14.1

package com.freeletics.mad.whetstone.codegen.common

import com.freeletics.mad.whetstone.CommonData
import com.freeletics.mad.whetstone.Navigation
import com.freeletics.mad.whetstone.codegen.Generator
import com.freeletics.mad.whetstone.codegen.compose.composableName
import com.freeletics.mad.whetstone.codegen.compose.fragmentName
import com.freeletics.mad.whetstone.codegen.util.contributesToAnnotation
import com.freeletics.mad.whetstone.codegen.util.intoSet
import com.freeletics.mad.whetstone.codegen.util.module
import com.freeletics.mad.whetstone.codegen.util.provides
import com.squareup.kotlinpoet.CodeBlock
import com.squareup.kotlinpoet.FunSpec
import com.squareup.kotlinpoet.TypeSpec

internal class NavDestinationModuleGenerator(
    override val data: CommonData,
) : Generator() {

    private val moduleClassName = ClassName("Whetstone${data.baseName}NavDestinationModule")

    internal fun generate(): TypeSpec {
        return TypeSpec.objectBuilder(moduleClassName)
            .addAnnotation(module)
            .addAnnotation(contributesToAnnotation(data.navigation!!.destinationScope))
            .addFunction(providesFunction())
            .build()
    }

    private fun providesFunction(): FunSpec {
        return FunSpec.builder("provideNavDestination")
            .addAnnotation(provides)
            .addAnnotation(intoSet)
            .returns(data.navigation!!.destinationClass)
            .addCode(providesCode())
            .build()
    }

    private fun providesCode(): CodeBlock {
        val navigation = data.navigation!!
        return when(data.navigation!!) {
             is Navigation.Compose -> {
                CodeBlock.builder()
                    .beginControlFlow("return %M<%T>",
                        navigation.destinationMethod, navigation.route)
                    .addStatement("%L(it)", composableName)
                    .endControlFlow()
                    .build()
            }
            is Navigation.Fragment -> {
                CodeBlock.of("return %M<%T, %T>()",
                    navigation.destinationMethod, navigation.route, ClassName(fragmentName))
            }
        }
    }
}