Class: Parser::AST::Node

Inherits:
Object
  • Object
show all
Defined in:
lib/ruby_friendly_error/ast.rb

Constant Summary collapse

CLASS_VARIABLE_ASSIGNED_TYPE = 
:cvasgn
CLASS_VARIABLE_TYPE = 
:cvar
INSTANCE_VARIABLE_ASSIGNED_TYPE = 
:ivasgn
INSTANCE_VARIABLE_TYPE = 
:ivar
LOCAL_VARIABLE_ASSIGNED_TYPE = 
:lvasgn
VARIABLE_TYPES = 
[
  CLASS_VARIABLE_ASSIGNED_TYPE,
  CLASS_VARIABLE_TYPE,
  INSTANCE_VARIABLE_ASSIGNED_TYPE,
  INSTANCE_VARIABLE_TYPE,
  LOCAL_VARIABLE_ASSIGNED_TYPE,
].freeze
ARG_TYPE = 
:arg
OPTION_ARG_TYPE = 
:optarg
REST_ARG_TYPE = 
:restarg
BLOCK_ARG_TYPE = 
:blockarg
KEYWORD_ARG_TYPE = 
:kwarg
KEYWORD_ARG_OPTION_TYPE = 
:kwoptarg
KEYWORD_REST_ARG_TYPE = 
:kwrestarg
ARG_TYPES = 
[
  ARG_TYPE,
  OPTION_ARG_TYPE,
  REST_ARG_TYPE,
  BLOCK_ARG_TYPE,
  KEYWORD_ARG_TYPE,
  KEYWORD_ARG_OPTION_TYPE,
  KEYWORD_REST_ARG_TYPE,
].freeze
REQURED_ARG_TYPES = 
[
  ARG_TYPE,
].freeze
OPTIONAL_ARG_TYPES = 
[
  OPTION_ARG_TYPE,
  KEYWORD_ARG_OPTION_TYPE,
].freeze
DEF_TYPE = 
:def
SEND_TYPE = 
:send
NOT_HAVE_LINE_TYPES = 
%i[args].freeze

Instance Method Summary collapse

Instance Method Details

#def_args_num_range ⇒ Object 



88
89
90
91
92
93
94
95
 
# File 'lib/ruby_friendly_error/ast.rb', line 88

def def_args_num_range
  return nil unless type == DEF_TYPE

  required_arg_num  = to_a[1].to_a.select { |n| REQURED_ARG_TYPES.include? n.type }.size
  optionarl_arg_num = to_a[1].to_a.select { |n| OPTIONAL_ARG_TYPES.include? n.type }.size

  (required_arg_num..(required_arg_num + optionarl_arg_num))
end

#find_by_line_number(line_number) ⇒ Object 



65
66
67
68
69
70
71
72
73
74
 
# File 'lib/ruby_friendly_error/ast.rb', line 65

def find_by_line_number line_number
  to_a.each do |node|
    next unless node.is_a? Parser::AST::Node
    next if NOT_HAVE_LINE_TYPES.include? node.type
    return node if node.location.line == line_number
    result = node.find_by_line_number line_number
    return result if result
  end
  nil
end

#find_by_variable_name(variable_name) ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/ruby_friendly_error/ast.rb', line 52

def find_by_variable_name variable_name
  to_a.each do |node|
    next unless node.is_a? Parser::AST::Node
    if VARIABLE_TYPES.include?(node.type) || ARG_TYPES.include?(node.type)
      return node if node.to_a[0].to_s.sub(/^@*/, '') == variable_name.to_s
    end

    result = node.find_by_variable_name variable_name
    return result if result
  end
  nil
end

#find_send_by_def_name(def_name) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/ruby_friendly_error/ast.rb', line 76

def find_send_by_def_name def_name
  to_a.each do |node|
    next unless node.is_a? Parser::AST::Node

    return node if node.type == SEND_TYPE && node.to_a[1].to_s == def_name.to_s

    result = node.find_send_by_def_name def_name
    return result if result
  end
  nil
end

#send_args_num ⇒ Object 



97
98
99
100
101
 
# File 'lib/ruby_friendly_error/ast.rb', line 97

def send_args_num
  return nil unless type == SEND_TYPE

  to_a.size - 2
end