flea 0.1.0

data/lib/flea/standard_library/addition_operator.scm

@@ -0,0 +1,7 @@
+(define + 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      tmp = arguments.map {|item| interpreter.evaluate(item)}
+      tmp.inject {|sum, n| sum + n}
+    end
+  "))

data/lib/flea/standard_library/append.scm

@@ -0,0 +1,7 @@
+(define append
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      
+      arguments.map {|x| interpreter.evaluate(x) }.flatten
+    end
+  "))

data/lib/flea/standard_library/begin.scm

@@ -0,0 +1,11 @@
+(define begin 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      val = nil
+      arguments.each do |i|
+        val = interpreter.evaluate(i)
+      end
+      
+      val
+    end
+  "))

data/lib/flea/standard_library/car.scm

@@ -0,0 +1,7 @@
+(define car
+  (native_function "
+    Proc.new() do |argument, interpreter|
+      list = interpreter.evaluate(argument[0])
+      list[0]
+    end
+  "))

data/lib/flea/standard_library/cdr.scm

@@ -0,0 +1,7 @@
+(define cdr
+  (native_function "
+    Proc.new() do |argument, interpreter|
+      list = interpreter.evaluate(argument[0])
+      list.slice(1, list.length)
+    end
+  "))

data/lib/flea/standard_library/cons.scm

@@ -0,0 +1,13 @@
+(define cons
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      arg_1 = interpreter.evaluate(arguments[0])
+      arg_2 = interpreter.evaluate(arguments[1]).dup
+      
+      if arg_2.is_a? Array
+        arg_2.unshift arg_1
+      else
+        [arg_1, arg_2]
+      end
+    end
+  "))

data/lib/flea/standard_library/display.scm

@@ -0,0 +1,8 @@
+(define display 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      output = interpreter.evaluate(arguments[0])
+      print interpreter.parser.to_sexp(output)
+      output
+    end
+  "))

data/lib/flea/standard_library/division_operator.scm

@@ -0,0 +1,7 @@
+(define / 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      tmp = arguments.map {|item| interpreter.evaluate(item)}
+      tmp.inject {|sum, n| sum / n}
+    end
+  "))

data/lib/flea/standard_library/equality_operator.scm

@@ -0,0 +1,8 @@
+(define = 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      !(arguments.map{|x| interpreter.evaluate(x) == interpreter.evaluate(arguments[0])}).include?( false )
+    end
+  "))
+  
+(define equal? =)

data/lib/flea/standard_library/gets.scm

@@ -0,0 +1,6 @@
+(define gets
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      $stdin.gets
+    end
+  "))

data/lib/flea/standard_library/greater_than.scm

@@ -0,0 +1,8 @@
+(define greater-than? 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      arguments.slice(1, arguments.length).all? do |x|
+        interpreter.evaluate(arguments[0]) > interpreter.evaluate(x)
+      end
+    end
+  "))

data/lib/flea/standard_library/if.scm

@@ -0,0 +1,10 @@
+(define if 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      if(interpreter.evaluate(arguments[0]))
+        interpreter.evaluate(arguments[1])
+      else
+        interpreter.evaluate(arguments[2]) unless(arguments[2].nil?)
+      end
+    end
+  "))

data/lib/flea/standard_library/lambda.scm

@@ -0,0 +1,57 @@
+(define lambda 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      formals = arguments[0]
+      body = arguments.slice(1, arguments.length)
+      
+      if formals.is_a? Array
+        # detect if any formal names have been used more than once
+        error_message = 'Formal {FORMAL} declared more than once' 
+        formals.each_index do |x| 
+          tmp = formals.dup
+          tmp.delete_at(x)
+          raise(error_message.gsub('{FORMAL}', formals[x])) if tmp.include? formals[x] 
+        end
+      end
+      
+      sub_env = Flea::Environment.new(interpreter.current_environment)
+      
+      execute_body = Proc.new() do |body, environment, interpreter|
+        interpreter.current_environment = environment
+        result = nil
+        body.each do |expression|
+          result = interpreter.evaluate(expression)
+        end
+        interpreter.current_environment = environment.parent
+        result
+      end    
+      
+      if formals.is_a?(Array) && formals.include?(:'.')
+        Proc.new() do |arguments, interpreter|
+          args = arguments.dup
+          named_formals = formals.slice(0, formals.index(:'.'))
+          list_formal = formals[formals.index(:'.') + 1]
+          named_formals.each_index do |i|
+            sub_env.define(named_formals[i], interpreter.evaluate(args.shift))
+          end
+          sub_env.define(list_formal, args)
+          execute_body.call(body, sub_env, interpreter)
+        end
+        
+      elsif formals.is_a? Array
+        Proc.new() do |arguments, interpreter|
+          formals.each_index do |i|
+            sub_env.define(formals[i], interpreter.evaluate(arguments[i]))
+          end
+          execute_body.call(body, sub_env, interpreter)
+        end
+        
+      elsif formals.is_a? Symbol
+        Proc.new() do |arguments, interpreter|
+          arguments = arguments.map {|x| interpreter.evaluate(x) }
+          sub_env.define(formals, arguments)
+          execute_body.call(body, sub_env, interpreter)
+        end
+      end
+    end
+  "))

data/lib/flea/standard_library/less_than.scm

@@ -0,0 +1,8 @@
+(define less-than? 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      arguments.slice(1, arguments.length).all? do |x|
+        interpreter.evaluate(arguments[0]) < interpreter.evaluate(x)
+      end
+    end
+  "))

data/lib/flea/standard_library/list.scm

@@ -0,0 +1,8 @@
+(define list
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      arguments.map {|x|
+        interpreter.evaluate(x)
+      }
+    end
+  "))

data/lib/flea/standard_library/list_predicate.scm

@@ -0,0 +1,6 @@
+(define list?
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      interpreter.evaluate(arguments[0]).is_a? Array
+    end
+  "))

data/lib/flea/standard_library/list_tail.scm

@@ -0,0 +1,5 @@
+(define list-tail
+  (lambda (x k)
+    (if (equal? k 0)
+        x
+        (list-tail (cdr x) (- k 1)))))

data/lib/flea/standard_library/multiplication_operator.scm

@@ -0,0 +1,7 @@
+(define * 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      tmp = arguments.map {|item| interpreter.evaluate(item)}
+      tmp.inject {|sum, n| sum * n}
+    end
+  "))

data/lib/flea/standard_library/null.scm

@@ -0,0 +1,3 @@
+(define null?
+  (lambda (arg)
+    (equal? arg '())))

data/lib/flea/standard_library/quote.scm

@@ -0,0 +1,6 @@
+(define quote 
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      arguments[0]
+    end
+  "))

data/lib/flea/standard_library/rand.scm

@@ -0,0 +1,6 @@
+(define rand
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      rand(interpreter.evaluate(arguments[0]))
+    end
+  "))

data/lib/flea/standard_library/read.scm

@@ -0,0 +1,6 @@
+(define read
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      interpreter.parser.parse_string($stdin.gets)
+    end
+  "))

data/lib/flea/standard_library/set.scm

@@ -0,0 +1,9 @@
+(define set!
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      if( interpreter.current_environment.find(arguments[0]) == nil)
+        raise 'Cannot set unbound variable ' + arguments[0]
+      end
+      interpreter.current_environment.define(arguments[0], arguments[1])
+    end
+  "))

data/lib/flea/standard_library/string_to_num.scm

@@ -0,0 +1,6 @@
+(define string-to-num
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      interpreter.evaluate(arguments[0]).to_i
+    end
+  "))

data/lib/flea/standard_library/subtraction_operator.scm

@@ -0,0 +1,7 @@
+(define -
+  (native_function "
+    Proc.new() do |arguments, interpreter|
+      tmp = arguments.map {|item| interpreter.evaluate(item) }
+      tmp.inject {|sum, n| sum - n }
+    end
+  "))

data/spec/flea/environment_spec.rb

@@ -0,0 +1,114 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Flea" do
+  describe "::Environment" do
+    
+    include Flea
+    
+    describe ".new" do
+      it "should return an environment object" do
+        Environment.new.should be_an Environment
+      end
+      
+      it "should return an environment object with no parent" do
+        environment = Environment.new
+        environment.parent.should be_nil
+      end
+      
+      it "should return an environment object with the specified parent" do
+        parent_environment = mock("Environment")
+        environment = Environment.new(parent_environment)
+        environment.parent.should be parent_environment
+      end
+      
+      it "should add base variables for #t and #f" do
+        environment = Environment.new
+        environment.should have_variable :"#t"
+        environment.should have_variable :"#f"
+        environment.find(:"#t").should be_true
+        environment.find(:"#f").should be_false
+      end
+    end
+    
+    describe "#has_variable?" do
+      context "without a parent" do
+        before :each do
+          @environment = Environment.new
+        end
+        
+        it "should return false if the variable is not set in the current environment" do
+          @environment.should_not have_variable :test
+        end
+        
+        it "should return true if the variable is set in the current environment" do
+          @environment.define(:test, 1)
+          @environment.should have_variable :test
+        end
+      end
+      
+      context "with a parent" do
+        before :each do
+          @parent_environment = Environment.new
+          @environment = Environment.new(@parent_environment)
+        end
+        
+        it "should return false if the variable is not set in the parent environment" do
+          @environment.should_not have_variable :test
+        end
+        
+        it "should return true if the variable is set in the parent environment" do
+          @parent_environment.define(:test, 1)
+          @environment.should have_variable :test
+        end
+      end
+    end
+    
+    describe "#define" do
+      it "should set a variable to the supplied value" do
+        env = Environment.new
+        result = env.define(:test, 1)
+        env.should have_variable :test
+        env.find(:test).should == 1
+        result.should == 1
+      end
+    end
+    
+    describe "#find" do
+      context "without a parent" do
+        before :each do
+          @environment = Environment.new
+          @environment.define(:test, 1)
+        end
+        
+        it "should find a variable in the current environment" do
+          @environment.find(:test).should == 1
+        end
+        
+        it "should return nil when variable is not set" do
+          @environment.find(:fake).should be_nil
+        end
+      end
+      
+      context "with a parent" do
+        before :each do
+          @parent_environment = Environment.new
+          @environment = Environment.new(@parent_environment)
+          @parent_environment.define(:test, 1)
+        end
+        
+        it "should find a variable in the parent environment" do
+          @environment.find(:test).should == 1
+        end
+        
+        it "should return a variable from the current environment if it is set in both current and parent" do
+          @environment.define(:test, 5)
+          @environment.find(:test).should == 5
+        end
+        
+        it "should return nil when variable is not set in the current or parent environment" do
+          @environment.find(:fake).should be_nil
+        end
+      end
+    end
+  end
+end

data/spec/flea/interpreter_spec.rb

@@ -0,0 +1,85 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Flea" do
+  describe "::Interpreter" do
+    
+    include Flea
+   
+    describe "#new" do
+      it "should return an Interpreter" do
+        Interpreter.new.should be_an Interpreter
+      end
+      
+      it "should allow setting the base environment to use" do
+        environment = mock("Environment")
+        interpreter = Interpreter.new(
+          :base_environment => environment,
+          :load_standard_library => false
+        )
+        interpreter.base_environment.should be environment
+      end
+    end
+    
+    describe ".run" do
+      it "should run a program and return the last output from the program" do
+        interpreter = Interpreter.new
+        result = interpreter.run("(define test 1)")
+        result.should == 1
+        interpreter.base_environment.should have_variable :test
+      end
+    end
+    
+    describe ".parse" do
+      it "should return an abstract syntax tree representing the supplied program" do
+        ast = Interpreter.new.parse("(define test 1)")
+        ast.should == [[:define, :test, 1]]
+      end
+    end
+    
+    describe ".evaluate" do
+      before :each do
+        @environment = mock("Environment")
+        @interpreter = Interpreter.new(
+          :base_environment => @environment, 
+          :load_standard_library => false
+        )
+      end
+      
+      it "should return the value of a variable" do
+        @environment.should_receive(:find).with(:test).and_return(1)
+        result = @interpreter.evaluate(:test)
+        result.should == 1
+      end
+      
+      it "should define a variable in the current environment" do
+        @environment.should_receive(:define).with(:test, 1).and_return(1)
+        result = @interpreter.evaluate([:define, :test, 1])
+        result.should == 1
+      end
+      
+      it "should create a native function" do
+        result = @interpreter.evaluate([:native_function, "
+          Proc.new() do |arguments, interpreter|
+            1
+          end
+        "])
+        result.should be_a Proc
+        result.call.should == 1        
+      end
+      
+      it "should call a native function" do
+        @environment.should_receive(:find).with(:foo).and_return(Proc.new {|a,b| "bar"})
+        result = @interpreter.evaluate([:foo, 1, 2, 3])
+        result.should == "bar"
+      end
+      
+      [1, 1.0, "string"].each do |literal|
+        it "should return literal '#{literal}' as is" do
+          result = @interpreter.evaluate(literal)
+          result.should be literal
+        end
+      end
+    end
+    
+  end
+end