Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions lib/coffee-script/rewriter.js

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

15 changes: 15 additions & 0 deletions src/rewriter.coffee
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ class exports.Rewriter
@tagPostfixConditionals()
@addImplicitBracesAndParens()
@addLocationDataToGeneratedTokens()
@fixOutdentLocationData()
@tokens

# Rewrite the token stream, looking one token ahead and behind.
Expand Down Expand Up @@ -368,6 +369,20 @@ class exports.Rewriter
last_column: column
return 1

# OUTDENT tokens should always be positioned at the last character of the
# previous token, so that AST nodes ending in an OUTDENT token end up with a
# location corresponding to the last "real" token under the node.
fixOutdentLocationData: ->
@scanTokens (token, i, tokens) ->
return 1 unless token[0] is 'OUTDENT'
prevLocationData = tokens[i - 1][2]
token[2] =
first_line: prevLocationData.last_line
first_column: prevLocationData.last_column
last_line: prevLocationData.last_line
last_column: prevLocationData.last_column
return 1

# Because our grammar is LALR(1), it can't handle some single-line
# expressions that lack ending delimiters. The **Rewriter** adds the implicit
# blocks, so it doesn't need to. To keep the grammar clean and tidy, trailing
Expand Down
18 changes: 18 additions & 0 deletions test/location.coffee
Original file line number Diff line number Diff line change
Expand Up @@ -469,6 +469,24 @@ test "Verify tokens have locations that are in order", ->
ok token[2].first_column >= lastToken[2].last_column
lastToken = token

test "Verify OUTDENT tokens are located at the end of the previous token", ->
source = '''
SomeArr = [ ->
if something
lol =
count: 500
]
'''
tokens = CoffeeScript.tokens source
[..., number, curly, outdent1, outdent2, outdent3, bracket, terminator] = tokens
eq number[0], 'NUMBER'
for outdent in [outdent1, outdent2, outdent3]
eq outdent[0], 'OUTDENT'
eq outdent[2].first_line, number[2].last_line
eq outdent[2].first_column, number[2].last_column
eq outdent[2].last_line, number[2].last_line
eq outdent[2].last_column, number[2].last_column

test "Verify all tokens get a location", ->
doesNotThrow ->
tokens = CoffeeScript.tokens testScript
Expand Down