-
Notifications
You must be signed in to change notification settings - Fork 7
/
replace_erb_stmt_with_expr_action.rb
41 lines (37 loc) · 1.26 KB
/
replace_erb_stmt_with_expr_action.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# frozen_string_literal: true
module Synvert::Core
# ReplaceErbStmtWithExprAction to replace erb stmt code to expr,
# e.g. <% form_for ... %> => <%= form_for ... %>.
class Rewriter::ReplaceErbStmtWithExprAction < Rewriter::Action
def initialize(instance, code = nil)
super
end
# Begin position of code to replace.
#
# @return [Integer] begin position.
def begin_pos
node_begin_pos = @node.loc.expression.begin_pos
while @node.loc.expression.source_buffer.source[node_begin_pos -= 1] == ' '
end
node_begin_pos - Engine::ERUBY_STMT_SPLITTER.length + 1
end
# End position of code to replace.
#
# @return [Integer] end position.
def end_pos
node_begin_pos = @node.loc.expression.begin_pos
node_begin_pos += @node.loc.expression.source.index 'do'
while @node.loc.expression.source_buffer.source[node_begin_pos += 1] != '@'
end
node_begin_pos
end
# The rewritten erb expr code.
#
# @return [String] rewritten code.
def rewritten_code
@node.loc.expression.source_buffer.source[begin_pos...end_pos]
.sub(Engine::ERUBY_STMT_SPLITTER, '@output_buffer.append= ')
.sub(Engine::ERUBY_STMT_SPLITTER, Engine::ERUBY_EXPR_SPLITTER)
end
end
end