CodeAnalyzerTest.kt

package dev.snipme.highlights.internal

import dev.snipme.highlights.Highlights
import dev.snipme.highlights.model.ColorHighlight
import dev.snipme.highlights.model.PhraseLocation
import dev.snipme.highlights.model.SyntaxLanguage
import kotlin.test.Test
import kotlin.test.assertEquals

internal class CodeAnalyzerTest {

    @Test
    fun `Returns structure of code analyzed first time`() {
        val testCode = """
            /** a */
            // b
            class C extends {}
            "d";
            @E
            ...
            123.00f
        """.trimIndent()

        val result = CodeAnalyzer.analyze(testCode)

        assertEquals(
            setOf(
                PhraseLocation(30, 31),
                PhraseLocation(31, 32)
            ),
            result.marks
        )

        assertEquals(
            setOf(
                PhraseLocation(36, 37),
                PhraseLocation(41, 42),
                PhraseLocation(42, 43),
                PhraseLocation(43, 44),
                PhraseLocation(48, 49),
            ),
            result.punctuations
        )

        assertEquals(
            setOf(
                PhraseLocation(14, 19),
                PhraseLocation(22, 29)
            ),
            result.keywords
        )

        assertEquals(
            setOf(
                PhraseLocation(33, 36),
            ),
            result.strings
        )

        assertEquals(
            setOf(
                PhraseLocation(45, 52),
            ),
            result.literals
        )

        assertEquals(
            setOf(
                PhraseLocation(9, 13),
            ),
            result.comments
        )

        assertEquals(
            setOf(
                PhraseLocation(0, 8),
            ),
            result.multilineComments
        )

        assertEquals(
            setOf(
                PhraseLocation(38, 40),
            ),
            result.annotations
        )

        assertEquals(false, result.incremental)
    }

    @Test
    fun `Returns incremental structure of code analyzed second time`() {
        val testCode = """
            /** a */
            // b
            class C extends {}
            "d";
            @E
            ...
            123.00f
        """.trimIndent()

        val firstResult = CodeAnalyzer.analyze(testCode)
        assertEquals(false, firstResult.incremental)

        val secondTestCode = """
            /** a */
            // b
            class C extends {}
            "d";
            @E
            ...
            123.00f
            
            class
        """.trimIndent()

        val snapshot = CodeSnapshot(testCode, firstResult, SyntaxLanguage.DEFAULT)

        val result = CodeAnalyzer.analyze(secondTestCode, snapshot = snapshot)
        assertEquals(true, result.incremental)

        assertEquals(
            setOf(
                PhraseLocation(30, 31),
                PhraseLocation(31, 32),
            ),
            result.marks
        )

        assertEquals(
            setOf(
                PhraseLocation(36, 37),
                PhraseLocation(41, 42),
                PhraseLocation(42, 43),
                PhraseLocation(43, 44),
                PhraseLocation(48, 49),
            ),
            result.punctuations
        )

        assertEquals(
            setOf(
                PhraseLocation(14, 19),
                PhraseLocation(22, 29),
            ),
            result.keywords
        )

        assertEquals(
            setOf(
                PhraseLocation(33, 36),
            ),
            result.strings
        )

        assertEquals(
            setOf(
                PhraseLocation(45, 52),
            ),
            result.literals
        )

        assertEquals(
            setOf(
                PhraseLocation(9, 13),
            ),
            result.comments
        )

        assertEquals(
            setOf(
                PhraseLocation(0, 8),
            ),
            result.multilineComments
        )

        assertEquals(
            setOf(
                PhraseLocation(38, 40),
            ),
            result.annotations
        )
    }

    @Test
    fun `Returns incremental structure of decreased code analyzed second time`() {
        val testCode = """
            /** a */
            // b
            class C extends {}
            "d";
            @E
            ...
            123.00f
        """.trimIndent()

        val firstResult = CodeAnalyzer.analyze(testCode)
        assertEquals(false, firstResult.incremental)

        val secondTestCode = """
            /** a */
            // b
            class 
        """.trimIndent()

        val snapshot = CodeSnapshot(testCode, firstResult, SyntaxLanguage.DEFAULT)

        val result = CodeAnalyzer.analyze(secondTestCode, snapshot = snapshot)
        assertEquals(true, result.incremental)

        assertEquals(
            emptySet(),
            result.marks
        )

        assertEquals(
            emptySet(),
            result.punctuations
        )

        assertEquals(
            setOf(
                PhraseLocation(14, 19),
            ),
            result.keywords
        )

        assertEquals(
            emptySet(),
            result.strings
        )

        assertEquals(
            emptySet(),
            result.literals
        )

        assertEquals(
            setOf(
                PhraseLocation(9, 13),
            ),
            result.comments
        )

        assertEquals(
            setOf(
                PhraseLocation(0, 8),
            ),
            result.multilineComments
        )

        assertEquals(
            emptySet(),
            result.annotations
        )
    }


    @Test
    fun basic_getHighlights_location() {
        val highlighter = Highlights.Builder()
            .language(SyntaxLanguage.JAVASCRIPT)
            .build()

        val code1 = "const foo = 'bar';";

        highlighter.setCode(code1)
        val highlights1 = highlighter.getHighlights().filterIsInstance<ColorHighlight>().sortedBy { it.location.start }
        highlights1.size shouldEqual 4
        highlights1[0].location shouldEqual PhraseLocation(start=0, end=5)
        highlights1[1].location shouldEqual PhraseLocation(start=10, end=11)
        highlights1[2].location shouldEqual PhraseLocation(start=12, end=17)
        highlights1[3].location shouldEqual PhraseLocation(start=17, end=18)

        highlights1.map { it.location }.printResults(code1)

        val code2 = "const foo = 'barrr';"

        highlighter.setCode(code2)
        val highlights2 = highlighter.getHighlights().filterIsInstance<ColorHighlight>().sortedBy { it.location.start }
        highlights2.map { it.location }.printResults(code2)
        highlights2.size shouldEqual 4
        highlights2[0].location shouldEqual PhraseLocation(start=0, end=5)
        highlights2[1].location shouldEqual PhraseLocation(start=10, end=11)
        // FAILS HERE:
        highlights2[2].location shouldEqual PhraseLocation(start=12, end=19)
        highlights2[3].location shouldEqual PhraseLocation(start=19, end=20)

    }

    @Test
    fun basic_getHighlights_location_alt() {
        var highlighter = Highlights.Builder()
            .language(SyntaxLanguage.JAVASCRIPT)
            .build()

        highlighter = highlighter
            .getBuilder()
            .code("const foo = 'bar';")
            .build()

        val highlights1 = highlighter.getHighlights().filterIsInstance<ColorHighlight>().sortedBy { it.location.start }
        highlights1.size shouldEqual 4
        highlights1[0].location shouldEqual PhraseLocation(start=0, end=5)
        highlights1[1].location shouldEqual PhraseLocation(start=10, end=11)
        highlights1[2].location shouldEqual PhraseLocation(start=12, end=17)
        highlights1[3].location shouldEqual PhraseLocation(start=17, end=18)

        highlighter = highlighter
            .getBuilder()
            .code("const foo = 'barrr';")
            .build()
        val highlights2 = highlighter.getHighlights().filterIsInstance<ColorHighlight>().sortedBy { it.location.start }
        highlights2.size shouldEqual 4
        highlights2[0].location shouldEqual PhraseLocation(start=0, end=5)
        highlights2[1].location shouldEqual PhraseLocation(start=10, end=11)
        highlights2[2].location shouldEqual PhraseLocation(start=12, end=19)
        highlights2[3].location shouldEqual PhraseLocation(start=19, end=20)
    }

    private infix fun Any?.shouldEqual(expected: Any?) {
        assertEquals(expected, this)
    }

}