Class: Opal::Nodes::ArityCheckNode

12
13
14
15
16
17
18
19
20
21
22
23
24

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 12
def initialize(*)
 super
 arguments = Rewriters ::Arguments .new (args_node.children)
 @args = arguments.args
 @optargs = arguments.optargs
 @restarg = arguments.restarg
 @postargs = arguments.postargs
 @kwargs = arguments.kwargs
 @kwoptargs = arguments.kwoptargs
 @kwrestarg = arguments.kwrestarg
end

42
43
44

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 42
def all_args
 @all_args ||= [*@args, *@optargs, @restarg, *@postargs, *kwargs].compact
end

74
75
76
77
78
79
80

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 74
def arity
 if @restarg || @optargs.any? || has_only_optional_kwargs?
 negative_arity
 else
 positive_arity
 end
end

48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 48
def arity_checks
 return @arity_checks if defined?(@arity_checks)
 arity = all_args.size
 arity -= @optargs.size
 arity -= 1 if @restarg
 arity -= kwargs.size
 arity = -arity - 1 if !@optargs.empty? || !kwargs.empty? || @restarg
 @arity_checks = []
 if arity < 0 # splat or opt args
 min_arity = -(arity + 1)
 max_arity = all_args.size
 @arity_checks << "$arity < #{min_arity}" if min_arity > 0
 @arity_checks << "$arity > #{max_arity}" unless @restarg
 else
 @arity_checks << "$arity !== #{arity}"
 end
 @arity_checks
end

26
27
28
29
30
31
32
33
34
35
36

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 26
def compile
 scope.arity = arity
 return unless compiler.arity_check?
 unless arity_checks.empty?
 meth = scope.mid.to_s.inspect
 line 'var $arity = arguments.length;'
 push " if (#{arity_checks.join(' || ')}) { Opal.ac($arity, #{arity}, this, #{meth}); }"
 end
end

107
108
109

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 107
def has_only_optional_kwargs?
 kwargs.any? && kwargs.all? { |arg| %i[kwoptargkwrestarg].include?(arg.type) }
end

111
112
113

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 111
def has_required_kwargs?
 kwargs.any? { |arg| arg.type == :kwarg }
end

38
39
40

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 38
def kwargs
 [*@kwargs, *@kwoptargs, @kwrestarg].compact
end

82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 82
def negative_arity
 required_plain_args = all_args.select do |arg|
 %i[argmlhs].include?(arg.type)
 end
 result = required_plain_args.size
 if has_required_kwargs?
 result += 1
 end
 result = -result - 1
 result
end

98
99
100
101
102
103
104
105

# File 'opal/lib/opal/nodes/args/arity_check.rb', line 98
def positive_arity
 result = all_args.size
 result -= kwargs.size
 result += 1 if kwargs.any?
 result
end