Syntax highlighting issue when using plugin.lua

[eigenbom]

How are you using the lua-language-server?

Visual Studio Code Extension (sumneko.lua)

Which OS are you using?

Windows

What is the issue affecting?

Formatting, Plugins

Expected Behaviour

Syntax colouring should be preserved when using plugin.lua transformations

Actual Behaviour

I'm having an issue with syntax colouring when using plugin.lua, note the colour of the inline any annotation.

Reproduction steps

plugin.lua:

function OnSetText(uri, text)
    if uri == "file:///c%3A/dev/test/annotation_test/any.lua" then
        return {{
            start = 15,
            finish = 18,
            text = " --[[@type any]]"
        }}
 end

Source file (note the new line)

local anything:any
-- new line --

Additional Notes

No response

Log File

No response

[eigenbom]

And just for ref, here's my more complex example where I have mis-colourings in other places. The transformed code on the right is correct, but the syntax colouring on the left is like a mix of the various replacements.

[sumneko]

Could you please test if this patch works?
Please test latest version

[eigenbom]

Thx! Here's the result when using v3.6.18.

The variable type annotations seem correctly highlighted now, the function annotations still have some issues.

[sumneko]

Could you please provide a full example, include your code and plugin