dependencies {
    // For the core business logic
    implementation("com.n-apos.stepper:stepper-core:0.3.0")

    // For the Compose UI components
    implementation("com.n-apos.stepper:stepper-compose:0.3.0")
}

// 1. Define your data classes that hold the data for each step.
// These should implement MilestoneData.
@Serializable
data class FirstData(val name: String) : MilestoneData()
// ... define SecondData, ThirdData, etc.

// 2. Define your Milestones by extending the Milestone class.
class FirstMilestone(
    override var data: FirstData? = null,
    override var next: Milestone<*>? = null,
    override var previous: Milestone<*>? = null,
) : Milestone<FirstData>()
// ... define SecondMilestone, ThirdMilestone, etc.

// 3. Create a Roadmap using the DSL.
val roadmap = Roadmap {
    // The roadmap can be configured with a kotlinx.serialization Json instance
    // if you have polymorphic data classes.
    // configuration = json

    // Add instances of your milestone classes in the desired order.
    milestones += listOf(
        FirstMilestone(),
        SecondMilestone(),
        ThirdMilestone()
    )
}

// 4. As the user progresses, you fill the milestones with data.
roadmap.milestones.first().fill(FirstData("John"))

// 5. You can then aggregate all the data from the milestones into a single data class.
@Serializable
data class AggregatedResult(
    val first: FirstData,
    val second: SecondData,
    val third: ThirdData
)
val result = roadmap.aggregate<AggregatedResult>()

// 1. Create a Roadmap instance (see Core Module example).
val roadmap: Roadmap = { } // ... same as above

// 2. Create a class that implements MilestoneScreenProvider.
// This class is responsible for mapping a given milestone to its Composable screen.
class MyScreenProvider : MilestoneScreenProvider() {
    override fun provide(milestone: Milestone<*>): MilestoneScreen<*, *> {
        return when (milestone) {
            is FirstMilestone -> FirstScreen(milestone)
            is SecondMilestone -> SecondScreen(milestone)
            is ThirdMilestone -> ThirdScreen(milestone)
            else -> error("Milestone not supported: ${milestone::class.simpleName}")
        }
    }
}

// FirstScreen, SecondScreen, etc. are implementations of MilestoneScreen
// which provides the actual @Composable content for that step.
// For example:
class FirstScreen(
    override val milestone: FirstMilestone, 
    override val viewModel: MilestoneViewModel
) : MilestoneScreen<FirstMilestone, MilestoneViewModel> {
    
    @Composable
    override fun title(): String = "First Step"
    
    @Composable
    override fun render() {
        // Build your UI for the "First" step here.
        Text("This is the UI for the first step: ${title()}")
    }
}

// 3. In your UI, instantiate your provider and use the Stepper composable.
val myScreenProvider = remember { MyScreenProvider() }

Stepper(
    roadmap = roadmap,
    provider = myScreenProvider,
    onSubmit = {
        // This is called when the user clicks the final submit button.
        val result = roadmap.aggregate<AggregatedResult>()
        println("All steps complete! $result")
    },
)