package content.annotations

import matchers.content.*
import org.jetbrains.dokka.Platform
import org.jetbrains.dokka.base.transformers.pages.annotations.SinceKotlinTransformer
import org.jetbrains.dokka.base.transformers.pages.annotations.SinceKotlinVersion
import org.jetbrains.dokka.model.DFunction
import org.jetbrains.dokka.model.dfs
import org.jetbrains.dokka.model.doc.CustomTagWrapper
import org.jetbrains.dokka.model.doc.Text
import org.jetbrains.dokka.pages.ContentPage
import org.junit.jupiter.api.AfterEach
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.Test
import signatures.AbstractRenderingTest
import utils.ParamAttributes
import utils.TestOutputWriterPlugin
import utils.assertNotNull
import utils.bareSignature
import kotlin.test.assertEquals


class SinceKotlinTest : AbstractRenderingTest() {

    val testConfiguration = dokkaConfiguration {
        sourceSets {
            sourceSet {
                sourceRoots = listOf("src/")
                analysisPlatform = "jvm"
            }
        }
    }

    @BeforeEach
    fun setSystemProperty() {
        System.setProperty(SinceKotlinTransformer.SHOULD_DISPLAY_SINCE_KOTLIN_SYS_PROP, "true")
    }
    @AfterEach
    fun clearSystemProperty() {
        System.clearProperty(SinceKotlinTransformer.SHOULD_DISPLAY_SINCE_KOTLIN_SYS_PROP)
    }

    @Test
    fun versionsComparing() {
        assert(SinceKotlinVersion("1.0").compareTo(SinceKotlinVersion("1.0")) == 0)
        assert(SinceKotlinVersion("1.0.0").compareTo(SinceKotlinVersion("1")) == 0)
        assert(SinceKotlinVersion("1.0") >= SinceKotlinVersion("1.0"))
        assert(SinceKotlinVersion("1.1") > SinceKotlinVersion("1"))
        assert(SinceKotlinVersion("1.0") < SinceKotlinVersion("2.0"))
        assert(SinceKotlinVersion("1.0") < SinceKotlinVersion("2.2"))
    }

    @Test
    fun `rendered SinceKotlin custom tag for typealias, extensions, functions, properties`() {
        val writerPlugin = TestOutputWriterPlugin()

        testInline(
            """
            |/src/main/kotlin/test/source.kt
            |package test
            |
            |@SinceKotlin("1.5")
            |fun ring(abc: String): String {
            |    return "My precious " + abc
            |}
            |@SinceKotlin("1.5")
            |fun String.extension(abc: String): String {
            |    return "My precious " + abc
            |}            
            |@SinceKotlin("1.5")
            |typealias Str = String
            |@SinceKotlin("1.5")
            |val str = "str"
        """.trimIndent(),
            testConfiguration,
            pluginOverrides = listOf(writerPlugin)
        ) {
            renderingStage = { _, _ ->
                val content = writerPlugin.renderedContent("root/test/index.html")
                assert(content.getElementsContainingOwnText("Since Kotlin").count() == 4)
            }
        }
    }

    @Test
    fun `should propagate SinceKotlin`() {
        testInline(
            """
            |/src/main/kotlin/test/source.kt
            |package test
            |
            |@SinceKotlin("1.5")
            |class A {
            |   fun ring(abc: String): String {
            |       return "My precious " + abc
            |   }
            |}
        """.trimIndent(), testConfiguration
        ) {
            documentablesTransformationStage = { module ->
                @Suppress("UNCHECKED_CAST") val funcs = module.children.single { it.name == "test" }
                    .children.single { it.name == "A" }
                    .children.filter { it.name == "ring" && it is DFunction } as List<DFunction>
                with(funcs) {
                    val sinceKotlin = mapOf(
                        Platform.jvm to SinceKotlinVersion("1.5"),
                    )

                    for(i in sinceKotlin) {
                        val tag =
                            find { it.sourceSets.first().analysisPlatform == i.key }?.documentation?.values?.first()
                                ?.dfs { it is CustomTagWrapper && it.name == "Since Kotlin" }
                                .assertNotNull("SinceKotlin[${i.key}]")
                        assertEquals((tag.children.first() as Text).body, i.value.toString())
                    }
                }
            }
        }
    }

    @Test
    fun `mpp fun without SinceKotlin annotation`() {
        val configuration =   dokkaConfiguration {
            sourceSets {
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "jvm"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "native"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "common"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "js"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "wasm"
                }
            }
        }
        testInline(
            """
            |/src/main/kotlin/test/source.kt
            |package test
            |
            |fun ring(abc: String): String {
            |    return "My precious " + abc
            |}
        """.trimIndent(), configuration
        ) {
            documentablesTransformationStage = { module ->
                @Suppress("UNCHECKED_CAST") val funcs = module.children.single { it.name == "test" }
                    .children.filter { it.name == "ring" && it is DFunction } as List<DFunction>
                with(funcs) {
                    val sinceKotlin = mapOf(
                        Platform.common to SinceKotlinVersion("1.0"),
                        Platform.jvm to SinceKotlinVersion("1.0"),
                        Platform.js to SinceKotlinVersion("1.1"),
                        Platform.native to SinceKotlinVersion("1.3"),
                        Platform.wasm to SinceKotlinVersion("1.8"),
                    )

                    for(i in sinceKotlin) {
                        val tag =
                            find { it.sourceSets.first().analysisPlatform == i.key }?.documentation?.values?.first()
                                ?.dfs { it is CustomTagWrapper && it.name == "Since Kotlin" }
                                .assertNotNull("SinceKotlin[${i.key}]")
                        assertEquals((tag.children.first() as Text).body, i.value.toString())
                    }
                }
            }
        }
    }

    @Test
    fun `mpp fun with SinceKotlin annotation`() {
        val configuration =   dokkaConfiguration {
            sourceSets {
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "jvm"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "native"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "common"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "js"
                }
                sourceSet {
                    sourceRoots = listOf("src/")
                    analysisPlatform = "wasm"
                }
            }
        }
        testInline(
            """
            |/src/main/kotlin/test/source.kt
            |package test
            |
            |/** dssdd */
            |@SinceKotlin("1.3")
            |fun ring(abc: String): String {
            |    return "My precious " + abc
            |}
        """.trimIndent(), configuration
        ) {
            documentablesTransformationStage = { module ->
                @Suppress("UNCHECKED_CAST") val funcs = module.children.single { it.name == "test" }
                    .children.filter { it.name == "ring" && it is DFunction } as List<DFunction>
                with(funcs) {
                    val sinceKotlin = mapOf(
                        Platform.common to SinceKotlinVersion("1.3"),
                        Platform.jvm to SinceKotlinVersion("1.3"),
                        Platform.js to SinceKotlinVersion("1.3"),
                        Platform.native to SinceKotlinVersion("1.3"),
                        Platform.wasm to SinceKotlinVersion("1.8"),
                    )

                    for(i in sinceKotlin) {
                        val tag =
                            find { it.sourceSets.first().analysisPlatform == i.key }?.documentation?.values?.first()
                                ?.dfs { it is CustomTagWrapper && it.name == "Since Kotlin" }
                                .assertNotNull("SinceKotlin[${i.key}]")
                        assertEquals((tag.children.first() as Text).body, i.value.toString())
                    }
                }
            }
        }
    }

    @Test
    fun `should do not render since kotlin tag when flag is unset`() {
        clearSystemProperty()
        testInline(
            """
            |/src/main/kotlin/test/source.kt
            |package test
            |
            |@SinceKotlin("1.3")
            |fun ring(abc: String): String {
            |    return "My precious " + abc
            |}
        """.trimIndent(), testConfiguration
        ) {
            pagesTransformationStage = { module ->
                val page = module.children.single { it.name == "test" }
                    .children.single { it.name == "ring" } as ContentPage
                page.content.assertNode {
                    group {
                        header(1) { +"ring" }
                    }
                    divergentGroup {
                        divergentInstance {
                            divergent {
                                bareSignature(
                                    emptyMap(),
                                    "",
                                    "",
                                    emptySet(),
                                    "ring",
                                    "String",
                                    "abc" to ParamAttributes(emptyMap(), emptySet(), "String")
                                )
                            }
                        }
                    }

                }
            }
        }
    }
}