Class: Opal::Nodes::ArityCheckNode

Inherits:
Base
  • Object
show all
Defined in:
opal/lib/opal/nodes/args/arity_check.rb

Direct Known Subclasses

IterArityCheckNode

Instance Attribute Summary

Attributes inherited from Base

#compiler, #sexp, #type

Instance Method Summary collapse

Methods inherited from Base

#add_gvar, #add_ivar, #add_local, #add_temp, children, #children, #class_variable_owner, #class_variable_owner_nesting_level, #comments, #compile_to_fragments, #error, #expr, #expr?, #expr_or_nil, #fragment, handle, handlers, #has_rescue_else?, #helper, #in_ensure, #in_ensure?, #in_resbody, #in_resbody?, #in_rescue, #in_while?, #process, #push, #recv, #recv?, #s, #scope, #source_location, #stmt, #stmt?, #top_scope, truthy_optimize?, #unshift, #while_loop, #with_temp, #wrap

Methods included from Helpers

#conditional_send, #current_indent, #empty_line, #indent, #js_truthy, #js_truthy_optimize, #line, #mid_to_jsid, #property, #valid_name?

Constructor Details

#initialize ⇒ ArityCheckNode

Returns a new instance of ArityCheckNode.



12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# 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
  @kwnilarg  = arguments.kwnilarg
end

Instance Method Details

#all_args ⇒ Object 



44
45
46
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 44

def all_args
  @all_args ||= [*@args, *@optargs, @restarg, *@postargs, *kwargs].compact
end

#arity ⇒ Object 



76
77
78
79
80
81
82
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 76

def arity
  if @restarg || @optargs.any? || has_only_optional_kwargs?
    negative_arity
  else
    positive_arity
  end
end

#arity_checks ⇒ Object

Returns an array of JS conditions for raising and argument error caused by arity check



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 50

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

#compile ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 27

def compile
  scope.arity = arity

  return unless compiler.arity_check?

  unless arity_checks.empty?
    helper :ac
    meth = scope.mid.to_s.inspect
    line 'var $arity = arguments.length;'
    push " if (#{arity_checks.join(' || ')}) { $ac($arity, #{arity}, this, #{meth}); }"
  end
end

#has_only_optional_kwargs? ⇒ Boolean

Returns:

  • (Boolean) 


109
110
111
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 109

def has_only_optional_kwargs?
  kwargs.any? && kwargs.all? { |arg| %i[kwoptarg kwrestarg].include?(arg.type) }
end

#has_required_kwargs? ⇒ Boolean

Returns:

  • (Boolean) 


113
114
115
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 113

def has_required_kwargs?
  kwargs.any? { |arg| arg.type == :kwarg }
end

#kwargs ⇒ Object 



40
41
42
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 40

def kwargs
  [*@kwargs, *@kwoptargs, @kwrestarg].compact
end

#negative_arity ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 84

def negative_arity
  required_plain_args = all_args.select do |arg|
    %i[arg mlhs].include?(arg.type)
  end

  result = required_plain_args.size

  if has_required_kwargs?
    result += 1
  end

  result = -result - 1

  result
end

#positive_arity ⇒ Object 



100
101
102
103
104
105
106
107
 
# File 'opal/lib/opal/nodes/args/arity_check.rb', line 100

def positive_arity
  result = all_args.size

  result -= kwargs.size
  result += 1 if kwargs.any?

  result
end