Class: Dentaku::Parser

Inherits:
Object
  • Object
show all
Defined in:
lib/dentaku/parser.rb

Constant Summary collapse

AST_OPERATIONS = 
{
  add:      AST::Addition,
  subtract: AST::Subtraction,
  multiply: AST::Multiplication,
  divide:   AST::Division,
  pow:      AST::Exponentiation,
  negate:   AST::Negation,
  mod:      AST::Modulo,

  bitor:    AST::BitwiseOr,
  bitand:   AST::BitwiseAnd,
  bitshiftleft:  AST::BitwiseShiftLeft,
  bitshiftright: AST::BitwiseShiftRight,

  lt:       AST::LessThan,
  gt:       AST::GreaterThan,
  le:       AST::LessThanOrEqual,
  ge:       AST::GreaterThanOrEqual,
  ne:       AST::NotEqual,
  eq:       AST::Equal,

  and:      AST::And,
  or:       AST::Or,
  xor:      AST::Xor,
}.freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(tokens, options = {}) ⇒ Parser

Returns a new instance of Parser.



33
34
35
36
37
38
39
40
41
 
# File 'lib/dentaku/parser.rb', line 33

def initialize(tokens, options = {})
  @input             = tokens.dup
  @output            = []
  @operations        = options.fetch(:operations, [])
  @arities           = options.fetch(:arities, [])
  @function_registry = options.fetch(:function_registry, nil)
  @case_sensitive    = options.fetch(:case_sensitive, false)
  @skip_indices      = []
end

Instance Attribute Details

#arities ⇒ Object (readonly)

Returns the value of attribute arities.



31
32
33
 
# File 'lib/dentaku/parser.rb', line 31

def arities
  @arities
end

#case_sensitive ⇒ Object (readonly)

Returns the value of attribute case_sensitive.



31
32
33
 
# File 'lib/dentaku/parser.rb', line 31

def case_sensitive
  @case_sensitive
end

#input ⇒ Object (readonly)

Returns the value of attribute input.



31
32
33
 
# File 'lib/dentaku/parser.rb', line 31

def input
  @input
end

#operations ⇒ Object (readonly)

Returns the value of attribute operations.



31
32
33
 
# File 'lib/dentaku/parser.rb', line 31

def operations
  @operations
end

#output ⇒ Object (readonly)

Returns the value of attribute output.



31
32
33
 
# File 'lib/dentaku/parser.rb', line 31

def output
  @output
end

Instance Method Details

#consume(count = 2) ⇒ Object 



43
44
45
46
47
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
73
74
75
76
77
78
79
80
 
# File 'lib/dentaku/parser.rb', line 43

def consume(count = 2)
  operation = operations.pop
  fail! :invalid_statement if operation.nil?

  output_size = output.length
  args_size = operation.arity || count
  min_size = operation.arity || operation.min_param_count || count
  max_size = operation.arity || operation.max_param_count || count

  if output_size < min_size || args_size < min_size
    expected = min_size == max_size ? min_size : min_size..max_size
    fail! :too_few_operands, operation: operation, expected: expected, actual: output_size
  end

  if (output_size > max_size && operations.empty?) || args_size > max_size
    expected = min_size == max_size ? min_size : min_size..max_size
    fail! :too_many_operands, operation: operation, expected: expected, actual: output_size
  end

  args = []
  if operation == AST::Array && output.empty?
    # special case: empty array literal '{}'
    output.push(operation.new)
  else
    fail! :invalid_statement if output_size < args_size
    args = Array.new(args_size) { output.pop }.reverse
    output.push operation.new(*args)
  end

  if operation.respond_to?(:callback) && !operation.callback.nil?
    operation.callback.call(args)
  end
rescue ::ArgumentError => e
  raise if e.is_a?(Dentaku::ArgumentError)
  raise Dentaku::ArgumentError.for(:invalid_value), e.message
rescue NodeError => e
  raise ParseError.for(:node_invalid, operation: operation, operand: e.operand, expected: e.expected, actual: e.actual), e.message
end

#function(token) ⇒ Object 



109
110
111
 
# File 'lib/dentaku/parser.rb', line 109

def function(token)
  function_registry.get(token.value)
end

#function_registry ⇒ Object 



113
114
115
 
# File 'lib/dentaku/parser.rb', line 113

def function_registry
  @function_registry ||= Dentaku::AST::FunctionRegistry.new
end

#operation(token) ⇒ Object 



105
106
107
 
# File 'lib/dentaku/parser.rb', line 105

def operation(token)
  AST_OPERATIONS.fetch(token.value)
end

#parse ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/dentaku/parser.rb', line 82

def parse
  return AST::Nil.new if input.empty?

  i = 0
  while i < input.length
    if @skip_indices.include?(i)
      i += 1
      next
    end

    token = input[i]
    lookahead = input[i + 1]
    process_token(token, lookahead, i)
    i += 1
  end

  consume while operations.any?

  fail! :invalid_statement unless output.count == 1

  output.first
end