Squiggle VSCode extension comments break if function returns a call to another function.

[NunoSempere]

Here is a minimum viable example:

// Works
f(x) = {
    2*x
}

// Doesn't work
// i.e., comments after the definition don't get
// the correct syntax highlighting.

g(x) = {
    y = 1/x
    f(y)
}

// test comment which goes wrong.

h(x) = {
    y = f(x)
    y
}

// comments are displayed correctly

[NunoSempere]

This is mitigatable by assigning the return to a variable and then returning that variable.

[berekuk]

The entire grammar (squiggle.tmLanguage.yaml) should be rethought from scratch, I did the really minimal and incomplete implementation when I wrote it.

Unfortunately, new AST (#1107) won't help much here, TextMate grammars are different and regex-based.

[hickman-santini]

Related:

// Comment is good

mx(mx(1), 1 to 3, 2 to 10) 
// messed up comment

mx([1,2,3], [0.2, 0.2, 0.6])