Skip to content

More whitequark/parser translation #312

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Feb 19, 2023
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
11 changes: 10 additions & 1 deletion lib/syntax_tree/parser.rb
Original file line number Diff line number Diff line change
Expand Up @@ -1559,7 +1559,14 @@ def on_elsif(predicate, statements, consequent)
beginning = consume_keyword(:elsif)
ending = consequent || consume_keyword(:end)

start_char = find_next_statement_start(predicate.location.end_char)
delimiter =
find_keyword_between(:then, predicate, statements) ||
find_token_between(Semicolon, predicate, statements)

tokens.delete(delimiter) if delimiter
start_char =
find_next_statement_start((delimiter || predicate).location.end_char)

statements.bind(
self,
start_char,
Expand Down Expand Up @@ -2045,6 +2052,7 @@ def on_if(predicate, statements, consequent)

start_char =
find_next_statement_start((keyword || predicate).location.end_char)

statements.bind(
self,
start_char,
Expand Down Expand Up @@ -3805,6 +3813,7 @@ def on_unless(predicate, statements, consequent)

start_char =
find_next_statement_start((keyword || predicate).location.end_char)

statements.bind(
self,
start_char,
Expand Down
80 changes: 70 additions & 10 deletions lib/syntax_tree/translation/parser.rb
Original file line number Diff line number Diff line change
Expand Up @@ -336,8 +336,8 @@ def visit_assign(node)
# Visit an Assoc node.
def visit_assoc(node)
if node.value.nil?
# { foo: }
expression = srange(node.start_char, node.end_char - 1)

type, location =
if node.key.value.start_with?(/[A-Z]/)
[:const, smap_constant(nil, expression, expression)]
Expand All @@ -356,13 +356,38 @@ def visit_assoc(node)
srange_node(node)
)
)
else
elsif node.key.is_a?(Label)
# { foo: 1 }
s(
:pair,
[visit(node.key), visit(node.value)],
smap_operator(
srange_search_between(node.key, node.value, "=>") ||
srange_length(node.key.end_char, -1),
srange_length(node.key.end_char, -1),
srange_node(node)
)
)
elsif (operator = srange_search_between(node.key, node.value, "=>"))
# { :foo => 1 }
s(
:pair,
[visit(node.key), visit(node.value)],
smap_operator(operator, srange_node(node))
)
else
# { "foo": 1 }
key = visit(node.key)
key_location =
smap_collection(
key.location.begin,
srange_length(node.key.end_char - 2, 1),
srange(node.key.start_char, node.key.end_char - 1)
)

s(
:pair,
[s(key.type, key.children, key_location), visit(node.value)],
smap_operator(
srange_length(node.key.end_char, -1),
srange_node(node)
)
)
Expand Down Expand Up @@ -769,7 +794,11 @@ def visit_command_call(node)

srange(node.start_char, end_char)
elsif node.block
srange_node(node.message)
if node.receiver
srange(node.receiver.start_char, node.message.end_char)
else
srange_node(node.message)
end
else
srange_node(node)
end
Expand Down Expand Up @@ -1010,6 +1039,21 @@ def visit_else(node)

# Visit an Elsif node.
def visit_elsif(node)
begin_start = node.predicate.end_char
begin_end =
if node.statements.empty?
node.statements.end_char
else
node.statements.body.first.start_char
end

begin_token =
if buffer.source[begin_start...begin_end].include?("then")
srange_find(begin_start, begin_end, "then")
elsif buffer.source[begin_start...begin_end].include?(";")
srange_find(begin_start, begin_end, ";")
end

else_token =
case node.consequent
when Elsif
Expand All @@ -1029,7 +1073,7 @@ def visit_elsif(node)
],
smap_condition(
srange_length(node.start_char, 5),
nil,
begin_token,
else_token,
nil,
expression
Expand Down Expand Up @@ -1529,12 +1573,14 @@ def visit_lambda_var(node)
location =
if node.start_char == node.end_char
smap_collection_bare(nil)
else
elsif buffer.source[node.start_char - 1] == "("
smap_collection(
srange_length(node.start_char, 1),
srange_length(node.end_char, -1),
srange_node(node)
)
else
smap_collection_bare(srange_node(node))
end

s(:args, visit(node.params).children + shadowargs, location)
Expand Down Expand Up @@ -1565,7 +1611,7 @@ def visit_method_add_block(node)
srange_node(node.block.opening),
srange_length(
node.block.end_char,
node.block.opening.is_a?(Kw) ? -3 : -1
node.block.keywords? ? -3 : -1
),
srange_node(node)
)
Expand Down Expand Up @@ -2244,7 +2290,16 @@ def visit_super(node)
)
)
when ArgsForward
s(:super, [visit(node.arguments.arguments)], nil)
s(
:super,
[visit(node.arguments.arguments)],
smap_keyword(
srange_length(node.start_char, 5),
srange_find(node.start_char + 5, node.end_char, "("),
srange_length(node.end_char, -1),
srange_node(node)
)
)
else
s(
:super,
Expand Down Expand Up @@ -2442,10 +2497,15 @@ def visit_unless(node)
srange_find(begin_start, begin_end, ";")
end

else_token =
if node.consequent
srange_length(node.consequent.start_char, 4)
end

smap_condition(
srange_length(node.start_char, 6),
begin_token,
nil,
else_token,
srange_length(node.end_char, -3),
srange_node(node)
)
Expand Down