Interpreter

lib/ahem.rb

@@ -1,5 +1,8 @@
 require 'ahem/ast'
 require 'ahem/cli'
+require 'ahem/core'
+require 'ahem/environment'
+require 'ahem/interpreter'
 require 'ahem/lexer'
 require 'ahem/parser'
 require 'ahem/token'

lib/ahem/ast/array.rb

@@ -8,4 +8,8 @@ class AST::Array
   def ==(other)
     other.is_a?(AST::Array) && other.elements == @elements
   end
+
+  def execute(env)
+    @elements.map { |el| el.execute(env) }
+  end
 end

lib/ahem/ast/binary.rb

@@ -11,4 +11,31 @@ class AST::Binary
     other.operation == @operation && other.left == @left &&
       other.right == @right
   end
+
+  def execute(env)
+    case @operation
+    when AST::Operators::ADD
+      @left.execute(env) + @right.execute(env)
+    when AST::Operators::SUBTRACT
+      @left.execute(env) - @right.execute(env)
+    when AST::Operators::MULTIPLY
+      @left.execute(env) * @right.execute(env)
+    when AST::Operators::DIVIDE
+      @left.execute(env) / @right.execute(env)
+    when AST::Operators::LESS_THAN
+      @left.execute(env) < @right.execute(env)
+    when AST::Operators::LESS_THAN_OR_EQUAL
+      @left.execute(env) <= @right.execute(env)
+    when AST::Operators::GREATER_THAN
+      @left.execute(env) > @right.execute(env)
+    when AST::Operators::GREATER_THAN_OR_EQUAL
+      @left.execute(env) >= @right.execute(env)
+    when AST::Operators::DOUBLE_EQUALS
+      @left.execute(env) == @right.execute(env)
+    when AST::Operators::OR
+      @left.execute(env) || @right.execute(env)
+    when AST::Operators::AND
+      @left.execute(env) && @right.execute(env)
+    end
+  end
 end

lib/ahem/ast/boolean.rb

@@ -8,4 +8,8 @@ class AST::Boolean
   def ==(other)
     other.is_a?(AST::Boolean) && other.value == @value
   end
+
+  def execute(env)
+    @value
+  end
 end

lib/ahem/ast/conditional.rb

@@ -8,4 +8,17 @@ class AST::Conditional
   def ==(other)
     other.is_a?(AST::Conditional) && other.branches == @branches
   end
+
+  def execute(env)
+    @branches.each do |branch|
+      if branch.condition.execute(env)
+        value = nil
+        branch.block.statements.each do |statement|
+          value = statement.execute(env)
+        end
+
+        return value
+      end
+    end
+  end
 end

lib/ahem/ast/function_call.rb

@@ -10,4 +10,29 @@ class AST::FunctionCall
     other.is_a?(AST::FunctionCall) && other.function == @function &&
       other.arguments == @arguments
   end
+
+  def execute(env)
+    function = env.get(@function.name)
+
+    if function.is_a?(AST::FunctionDefinition)
+      if @arguments.size != function.parameters.size
+        throw "Arity mismatch in call to function #{function.name}"
+      end
+
+      function.parameters.each_with_index do |param, i|
+        env.set(param.name, @arguments[i].execute(env))
+      end
+
+      value = nil
+
+      function.body.statements.each do |statement|
+        value = statement.execute(env)
+      end
+
+      value
+    elsif function.is_a?(Proc)
+      arguments = @arguments.map { |arg| arg.execute(env) }
+      function.call(*arguments)
+    end
+  end
 end

lib/ahem/ast/function_definition.rb

@@ -12,4 +12,8 @@ class AST::FunctionDefinition
       other.parameters == @parameters &&
       other.body == @body
   end
+
+  def execute(env)
+    env.set(@name.name, self)
+  end
 end

lib/ahem/ast/identifier.rb

@@ -8,4 +8,14 @@ class AST::Identifier
   def ==(other)
     other.is_a?(AST::Identifier) && other.name == @name
   end
+
+  def execute(env)
+    value = env.get(@name)
+
+    if value.respond_to?(:execute)
+      value.execute(env)
+    else
+      value
+    end
+  end
 end

lib/ahem/ast/null.rb

@@ -2,4 +2,8 @@ class AST::Null
   def ==(other)
     other.is_a?(AST::Null)
   end
+
+  def execute(env)
+    nil
+  end
 end

lib/ahem/ast/number.rb

@@ -8,4 +8,8 @@ class AST::Number
   def ==(other)
     other.is_a?(AST::Number) && other.value == @value
   end
+
+  def execute(env)
+    @value
+  end
 end

lib/ahem/ast/operators.rb

@@ -7,6 +7,7 @@ module AST::Operators
   GREATER_THAN = :>
   LESS_THAN_OR_EQUAL = :<=
   GREATER_THAN_OR_EQUAL = :>=
+  DOUBLE_EQUALS = :==
   OR = :or
   AND = :and
 end

lib/ahem/ast/string.rb

@@ -8,4 +8,8 @@ class AST::String
   def ==(other)
     other.is_a?(AST::String) && other.value == @value
   end
+
+  def execute(env)
+    @value
+  end
 end

lib/ahem/ast/variable_declaration.rb

@@ -10,4 +10,8 @@ class AST::VariableDeclaration
     other.is_a?(AST::VariableDeclaration) && other.name == @name &&
       other.value == @value
   end
+
+  def execute(env)
+    env.set(@name.name, @value)
+  end
 end

lib/ahem/cli.rb

@@ -1,9 +1,7 @@
-require 'pp'
-
 class CLI
   def self.run
     file = ARGV.first
     source = File.read(file).strip
-    Parser.new(Lexer.new(source)).parse.each { |node| pp node }
+    Interpreter.new(Parser.new(Lexer.new(source)).parse).interpret
   end
 end

lib/ahem/core.rb

@@ -0,0 +1,11 @@
+CORE = {
+  :print => -> *args do
+    args.each do |arg|
+      if arg.is_a?(Array)
+        puts "[#{arg.join(', ')}]"
+      else
+        puts arg
+      end
+    end
+  end,
+}

lib/ahem/environment.rb

@@ -0,0 +1,29 @@
+class Environment
+  attr_reader :data
+
+  def initialize(parent = nil)
+    @data = Hash.new
+
+    CORE.each do |k, v|
+      @data[k.to_s] = v
+    end
+
+    unless parent.nil?
+      parent.data.each do |k, v|
+        @data[k] = v
+      end
+    end
+  end
+
+  def get(name)
+    if @data.has_key?(name)
+      @data[name]
+    else
+      throw "Undefined variable #{name}"
+    end
+  end
+
+  def set(name, value)
+    @data[name] = value
+  end
+end

lib/ahem/interpreter.rb

@@ -0,0 +1,12 @@
+class Interpreter
+  def initialize(tree, env = Environment.new)
+    @tree = tree
+    @env = env
+  end
+
+  def interpret
+    @tree.each do |node|
+      node.execute(@env)
+    end
+  end
+end

lib/ahem/lexer.rb

@@ -75,6 +75,9 @@ class Lexer
     elsif source.match(/^\./)
       @position += 1
       Token.new(TokenKinds::DOT)
+    elsif source.match(/^==/)
+      @position += 2
+      Token.new(TokenKinds::OPERATOR, :==)      
     elsif source.match(/^\<=/)
       @position += 2
       Token.new(TokenKinds::OPERATOR, :<=)

lib/ahem/parser.rb

@@ -142,13 +142,7 @@ class Parser
   end
 
   def expression
-    expr = binary
-    if @current_token.type == TokenKinds::LPAREN
-      args = arguments
-      AST::FunctionCall.new(expr, args)
-    else
-      expr
-    end
+    binary
   end
 
   private
@@ -210,7 +204,7 @@ class Parser
 
   def primary
     token = @current_token
-    case token.type
+    expr = case token.type
     when TokenKinds::NULL
       advance
       AST::Null.new
@@ -230,6 +224,13 @@ class Parser
     else
       throw "Unexpected token #{token.type}"
     end
+
+    if @current_token.type == TokenKinds::LPAREN
+      args = arguments
+      AST::FunctionCall.new(expr, args)
+    else
+      expr
+    end
   end
 
   def identifier

spec/lexer_spec.rb

@@ -32,7 +32,7 @@ RSpec.describe Lexer do
   end
 
   it 'lexes operators' do
-    expect(Lexer.new('+-*/<<=>>= and or').scan_all).to eq(
+    expect(Lexer.new('+-*/<<=>>=== and or').scan_all).to eq(
       [
         Token.new(TokenKinds::OPERATOR, :+),
         Token.new(TokenKinds::OPERATOR, :-),
@@ -42,6 +42,7 @@ RSpec.describe Lexer do
         Token.new(TokenKinds::OPERATOR, :<=),
         Token.new(TokenKinds::OPERATOR, :>),
         Token.new(TokenKinds::OPERATOR, :>=),
+        Token.new(TokenKinds::OPERATOR, :==),
         Token.new(TokenKinds::OPERATOR, :and),
         Token.new(TokenKinds::OPERATOR, :or),
         Token.new(TokenKinds::EOF)

spec/parser_spec.rb

@@ -306,4 +306,21 @@ RSpec.describe Parser do
       ]
     )
   end
+
+  it 'gives function calls higher precedence than binary operations' do
+    expect(parse('x + func(y);')).to eq(
+      [
+          AST::Binary.new(
+            AST::Operators::ADD,
+            AST::Identifier.new('x'),
+            AST::FunctionCall.new(
+              AST::Identifier.new('func'),
+              [
+                AST::Identifier.new('y'),
+              ]
+            )
+          )
+      ]
+    )
+  end
 end