Class: Melbourne::AST::When

Inherits:
Node
  • Object
show all
Defined in:
lib/melbourne/ast/control_flow.rb

Overview

A when statement in a case statement as in:

case a
  when 1 then
    # something
  when 2
    # something
end

Instance Attribute Summary collapse

Attributes inherited from Node

#line

Instance Method Summary collapse

Methods inherited from Node

#ascii_graph

Constructor Details

#initialize(line, conditions, body) ⇒ When

Returns a new instance of When.



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/melbourne/ast/control_flow.rb', line 83

def initialize(line, conditions, body)
  @line = line
  @body = body || Nil.new(line)
  @splat = nil
  @single = nil

  if conditions.kind_of? ArrayLiteral
    if conditions.body.last.kind_of? When
      last = conditions.body.pop
      if last.conditions.kind_of? ArrayLiteral
        conditions.body.concat last.conditions.body
      elsif last.single
        @splat = SplatWhen.new line, last.single
      else
        @splat = SplatWhen.new line, last.conditions
      end
    end

    if conditions.body.size == 1 and !@splat
      @single = conditions.body.first
    else
      @conditions = conditions
    end
  else
    @conditions = conditions
  end
end

Instance Attribute Details

#body ⇒ Object

The body for the when statement



73
74
75
 
# File 'lib/melbourne/ast/control_flow.rb', line 73

def body
  @body
end

#conditions ⇒ Object

The conditions for which the when statement is true or nil if only a single value is specified



69
70
71
 
# File 'lib/melbourne/ast/control_flow.rb', line 69

def conditions
  @conditions
end

#single ⇒ Object

The single value for the when statement if only a single value is specified, nil otherwise



77
78
79
 
# File 'lib/melbourne/ast/control_flow.rb', line 77

def single
  @single
end

#splat ⇒ Object

Any splat (+*something+) that is specified as a condition for the when statement or nil of no splat is specified



81
82
83
 
# File 'lib/melbourne/ast/control_flow.rb', line 81

def splat
  @splat
end