kotlin {
    sourceSets {
        commonMain {
            dependencies {
                implementation("io.github.ellykits.litequest:litequest-library:1.0.0-alpha05")
            }
        }
    }
}

@Composable
fun MyQuestionnaireScreen() {
    val questionnaire = Questionnaire(
        id = "patient-intake",
        version = "1.0.0",
        title = "Patient Intake Form",
        items = listOf(
            Item(
                linkId = "name",
                type = ItemType.TEXT,
                text = "What is your full name?",
                required = true
            ),
            Item(
                linkId = "age",
                type = ItemType.INTEGER,
                text = "What is your age?",
                required = true
            )
        )
    )

    val evaluator = remember { LiteQuestEvaluator(questionnaire) }
    val manager = remember { QuestionnaireManager(questionnaire, evaluator) }
    val state by manager.state.collectAsState()
    var mode by remember { mutableStateOf(QuestionnaireMode.Edit) }

    QuestionnaireScreen(
        type = QuestionnaireType.Single(questionnaire),
        manager = manager,
        mode = mode,
        onModeChange = { newMode -> mode = newMode },
        onSubmit = { println("Form submitted: ${state.response}") },
        onDismiss = { /* Handle dismiss */ }
    )
}

val paginatedQuestionnaire = PaginatedQuestionnaire(
    id = "health-survey",
    title = "Health Survey",
    pages = listOf(
        QuestionnairePage(
            id = "demographics",
            title = "Demographics",
            order = 0,
            items = listOf(/* page 1 items */)
        ),
        QuestionnairePage(
            id = "health-history",
            title = "Health History",
            order = 1,
            items = listOf(/* page 2 items */)
        )
    )
)

val flatQuestionnaire = Questionnaire(
    id = paginatedQuestionnaire.id,
    title = paginatedQuestionnaire.title,
    version = paginatedQuestionnaire.version,
    items = paginatedQuestionnaire.pages.flatMap { it.items }
)

val evaluator = LiteQuestEvaluator(flatQuestionnaire)
val manager = QuestionnaireManager(flatQuestionnaire, evaluator)

QuestionnaireScreen(
    type = QuestionnaireType.Paginated(paginatedQuestionnaire),
    manager = manager,
    onModeChange = null,
    onSubmit = { /* Handle submission */ },
    onDismiss = { /* Handle dismiss */ }
)

// Simple row-scoped visibility
Item(
    linkId = "symptoms",
    text = "Please describe your symptoms",
    visibleIf = buildJsonObject {
        put("==", buildJsonArray {
            add(buildJsonObject { put("var", "has-symptoms") })
            add(JsonPrimitive(true))
        })
    }
)

// Qualified paths inside repeating groups
// If inside 'receivedItems', it correctly resolves to the current row
Item(
    linkId = "itemId",
    text = "Item ID",
    visibleIf = buildJsonObject {
        put("==", buildJsonArray {
            add(buildJsonObject { put("var", "receivedItems.method") })
            add(JsonPrimitive("SEARCH"))
        })
    }
)

// Logic with negation
Item(
    linkId = "additionalNote",
    text = "Additional Note",
    visibleIf = buildJsonObject {
        put("!", buildJsonObject { put("var", "skipNotes") })
    }
)

// BMI calculation
Item(
    linkId = "bmi",
    type = ItemType.DECIMAL,
    text = "Body Mass Index",
    readOnly = true,
    calculatedExpression = buildJsonObject {
        put("/", buildJsonArray {
            add(buildJsonObject { put("var", "weight") })
            add(buildJsonObject {
                put("*", buildJsonArray {
                    add(buildJsonObject { put("var", "height") })
                    add(buildJsonObject { put("var", "height") })
                })
            })
        })
    }
)

// String concatenation
Item(
    linkId = "fullName",
    type = ItemType.STRING,
    text = "Full Name",
    readOnly = true,
    calculatedExpression = buildJsonObject {
        put("cat", buildJsonArray {
            add(buildJsonObject { put("var", "firstName") })
            add(JsonPrimitive(" "))
            add(buildJsonObject { put("var", "lastName") })
        })
    }
)

class RatingWidget(override val item: Item) : ItemWidget {
    @Composable
    override fun Render(
        value: JsonElement?,
        onValueChange: (JsonElement, String?) -> Unit,
        errorMessage: String?
    ) {
        val rating = value?.jsonPrimitive?.intOrNull ?: 0
        Row(horizontalArrangement = Arrangement.spacedBy(4.dp)) {
            repeat(5) { index ->
                Icon(
                    imageVector = if (index < rating) Icons.Filled.Star else Icons.Outlined.Star,
                    contentDescription = "Star ${index + 1}",
                    modifier = Modifier
                        .size(32.dp)
                        .clickable { onValueChange(JsonPrimitive(index + 1), null) },
                    tint = if (index < rating) Color(0xFFFFB300) else Color.Gray
                )
            }
        }
    }
}

// 2. Register custom widget in the factory
val factory = DefaultWidgetFactory().apply {
    registerWidget(ItemType("RATING")) { RatingWidget(it) }
}

// 3. Pass factory to QuestionnaireManager
val manager = QuestionnaireManager(
    questionnaire = questionnaire,
    evaluator = evaluator,
    widgetFactory = factory
)

// 4. Use in QuestionnaireScreen
QuestionnaireScreen(
    type = QuestionnaireType.Single(questionnaire),
    manager = manager,
    onSubmit = { /* handle submit */ }
)

val evaluator = LiteQuestEvaluator(questionnaire, JsonLogicEvaluator())
val manager = QuestionnaireManager(questionnaire, evaluator)