beambridge 0.9.0

data/lib/beambridge/errors/beambridge_error.rb

@@ -0,0 +1,3 @@
+class BeambridgeError < StandardError
+
+end

data/lib/beambridge/errors/decode_error.rb

@@ -0,0 +1,3 @@
+class DecodeError < BeambridgeError
+
+end

data/lib/beambridge/errors/encode_error.rb

@@ -0,0 +1,3 @@
+class EncodeError < BeambridgeError
+
+end

data/lib/beambridge/matcher.rb

@@ -0,0 +1,21 @@
+module Beambridge
+  class Matcher
+    attr_accessor :condition, :block
+    attr_accessor :receiver
+
+    def initialize(parent, condition, block)
+      self.receiver = parent
+      @block = block
+      @condition = Condition.for(condition)
+    end
+
+    def run(arg)
+      args = @condition.binding_for(arg)
+      block.call(*args)
+    end
+
+    def matches?(arg)
+      @condition.satisfies?(arg)
+    end
+  end
+end

data/lib/beambridge/port.rb

@@ -0,0 +1,48 @@
+module Beambridge
+  class Port
+    attr_reader :input, :output
+    attr_reader :skipped
+    attr_reader :queue
+
+    def initialize(input=STDIN, output=STDOUT)
+      @input = input
+      @output = output
+
+      input.sync = true
+      output.sync = true
+
+      output.set_encoding("ASCII-8BIT")
+
+      @encoder = Beambridge::Encoder.new(nil)
+      @skipped = []
+      @queue = []
+    end
+
+    def receive
+      queue.empty? ? read_from_input : queue.shift
+    end
+
+    def send(term)
+      @encoder.out = StringIO.new('', 'w')
+      @encoder.write_any(term)
+      data = @encoder.out.string
+      output.write([data.length].pack("N"))
+      output.write(data)
+    end
+
+    def restore_skipped
+      @queue = self.skipped + self.queue
+    end
+
+    private
+
+    def read_from_input
+      raw = input.read(4)
+      return nil unless raw
+
+      packet_length = raw.unpack('N').first
+      data = input.read(packet_length)
+      Beambridge::Decoder.decode(data)
+    end
+  end
+end

data/lib/beambridge/receiver.rb

@@ -0,0 +1,69 @@
+module Beambridge
+  class Receiver
+    attr_accessor :port
+    attr_accessor :parent
+    attr_accessor :matchers
+
+    RECEIVE_LOOP = Object.new
+    NO_MATCH = Object.new
+
+    def initialize(port, parent = nil, &block)
+      @port = port
+      @parent = parent
+      @matchers = []
+      block.call(self) if block
+    end
+
+    def process(arg)
+      matcher = @matchers.find { |r| r.matches?(arg) }
+
+      if matcher
+        port.restore_skipped
+        matcher.run(arg)
+      else
+        NO_MATCH
+      end
+    end
+
+    def when(arg, &block)
+      condition = Condition.for(arg)
+      @matchers << Matcher.new(self, condition, block)
+    end
+
+    def run
+      loop do
+        msg = port.receive
+        return if msg.nil?
+
+        case result = process(msg)
+          when RECEIVE_LOOP then next
+          when NO_MATCH
+            port.skipped << msg
+            next
+          else
+            break result
+        end
+      end
+    end
+
+    def receive(&block)
+      Receiver.new(port, self, &block).run
+    end
+
+    def receive_loop
+      RECEIVE_LOOP
+    end
+
+    def send!(term)
+      port.send(term)
+    end
+  end
+end
+
+module Kernel
+  def receive(input = nil, output = nil, &block)
+    input ||= IO.new(3)
+    output ||= IO.new(4)
+    Beambridge::Receiver.new(Beambridge::Port.new(input, output), nil, &block).run
+  end
+end

data/lib/beambridge/types/function.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  Function = Struct.new :pid, :module, :index, :uniq, :free_vars
+end

data/lib/beambridge/types/list.rb

@@ -0,0 +1,3 @@
+class Beambridge::List < Array
+
+end

data/lib/beambridge/types/new_function.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  NewFunction = Struct.new :arity, :uniq, :index,:num_free, :module, :old_index, :old_uniq, :pid, :free_vars
+end

data/lib/beambridge/types/new_reference.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  NewReference = Struct.new :node, :creation, :id
+end

data/lib/beambridge/types/pid.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  Pid = Struct.new :node, :id, :serial, :creation
+end

data/lib/beambridge/types/reference.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  Reference = Struct.new :node, :id, :creator
+end

data/lib/beambridge/version.rb

@@ -0,0 +1,3 @@
+module Beambridge
+  VERSION = "0.9.0"
+end

data/spec/condition_spec.rb

@@ -0,0 +1,72 @@
+require "spec_helper"
+
+describe "Beambridge::StaticConditions" do
+  it "should satisfy on the same value" do
+    Beambridge::StaticCondition.new(:foo).satisfies?(:foo).should == true
+    Beambridge::StaticCondition.new([:foo]).satisfies?([:foo]).should == true
+    Beambridge::StaticCondition.new(3).satisfies?(3).should == true
+  end
+
+  it "should not satisfy on different values" do
+    Beambridge::StaticCondition.new(:foo).satisfies?("foo").should == false
+    Beambridge::StaticCondition.new([:foo]).satisfies?(:foo).should == false
+    Beambridge::StaticCondition.new(Object.new).satisfies?(Object.new).should == false
+    Beambridge::StaticCondition.new(3).satisfies?(3.0).should == false
+  end
+
+  it "should not produce any bindings" do
+    s = Beambridge::StaticCondition.new(:foo)
+    s.binding_for(:foo).should == nil
+  end
+end
+
+describe "Beambridge::TypeConditions" do
+  it "should be satisfied when the arg has the same class" do
+    Beambridge::TypeCondition.new(Symbol).satisfies?(:foo).should == true
+    Beambridge::TypeCondition.new(Symbol).satisfies?(:bar).should == true
+    Beambridge::TypeCondition.new(String).satisfies?("foo").should == true
+    Beambridge::TypeCondition.new(String).satisfies?("bar").should == true
+    Beambridge::TypeCondition.new(Array).satisfies?([]).should == true
+    Beambridge::TypeCondition.new(Fixnum).satisfies?(3).should == true
+  end
+
+  it "should be satisfied when the arg is of a descendent class" do
+    Beambridge::TypeCondition.new(Object).satisfies?(:foo).should == true
+    Beambridge::TypeCondition.new(Object).satisfies?("foo").should == true
+    Beambridge::TypeCondition.new(Object).satisfies?(3).should == true
+  end
+
+  it "should not be satisfied when the arg is of a different class" do
+    Beambridge::TypeCondition.new(String).satisfies?(:foo).should == false
+    Beambridge::TypeCondition.new(Symbol).satisfies?("foo").should == false
+    Beambridge::TypeCondition.new(Fixnum).satisfies?(3.0).should == false
+  end
+
+  it "should bind the arg with no transormations" do
+    s = Beambridge::TypeCondition.new(Symbol)
+    s.binding_for(:foo).should == :foo
+    s.binding_for(:bar).should == :bar
+  end
+end
+
+describe "Beambridge::HashConditions" do
+  it "should satisfy an args of the form [[key, value], [key, value]]" do
+    Beambridge::HashCondition.new.satisfies?([[:foo, 3], [:bar, Object.new]]).should == true
+    Beambridge::HashCondition.new.satisfies?([[:foo, 3]]).should == true
+  end
+
+  it "should satisfy on empty arrays" do
+    Beambridge::HashCondition.new.satisfies?([]).should == true
+  end
+
+  it "should nat satisfy other args" do
+     Beambridge::HashCondition.new.satisfies?(:foo).should == false
+     Beambridge::HashCondition.new.satisfies?("foo").should == false
+     Beambridge::HashCondition.new.satisfies?(3.0).should == false
+  end
+
+  it "should bind to a Hash" do
+    s = Beambridge::HashCondition.new()
+    s.binding_for([[:foo, 3], [:bar, [3,4,5]]]).should == {:foo => 3, :bar => [3,4,5] }
+  end
+end

data/spec/decode_spec.rb

@@ -0,0 +1,143 @@
+require "spec_helper"
+
+describe "When unpacking from a binary stream" do
+
+  it "an erlang atom should decode to a ruby symbol" do
+    get("haha").should == :haha
+  end
+
+  it "an erlang number encoded as a small_int (< 255) should decode to a fixnum" do
+    get("0").should == 0
+    get("255").should == 255
+  end
+
+  it "an erlang number encoded as a int (signed 27-bit number) should decode to a fixnum" do
+    get("256").should == 256
+    get("#{(1 << 27) -1}").should == (1 << 27) -1
+    get("-1").should == -1
+    get("#{-(1 << 27)}").should == -(1 << 27)
+  end
+
+  it "an erlang number encoded as a small bignum (1 byte length) should decode to fixnum if it can" do
+    get("#{(1 << 27)}").should == (1 << 27)
+    get("#{-(1 << 27) - 1}").should == -(1 << 27) - 1
+    get("#{(1 << word_length) - 1}").should == (1 << word_length) - 1
+    get("#{-(1 << word_length)}").should == -(1 << word_length)
+  end
+
+  it "an erlang number encoded as a small bignum (1 byte length) should decode to bignum if it can't be a fixnum" do
+    get("#{(1 << word_length)}").should == (1 << word_length)
+    get("#{-(1 << word_length) - 1}").should == -(1 << word_length) - 1
+    get("#{(1 << (255 * 8)) - 1}").should == (1 << (255 * 8)) - 1
+    get("#{-((1 << (255 * 8)) - 1)}").should == -((1 << (255 * 8)) - 1)
+  end
+
+  it "an erlang number encoded as a big bignum (4 byte length) should decode to bignum" do
+    get("#{(1 << (255 * 8)) }").should == (1 << (255 * 8))
+    get("#{-(1 << (255 * 8))}").should == -(1 << (255 * 8))
+    get("#{(1 << (512 * 8)) }").should == (1 << (512 * 8))
+    get("#{-(1 << (512 * 8))}").should == -(1 << (512 * 8))
+  end
+
+  it "an erlang float should decode to a Float" do
+    get("#{1.0}").should == 1.0
+    get("#{-1.0}").should == -1.0
+    get("#{123.456}").should == 123.456
+    get("#{123.456789012345}").should == 123.456789012345
+  end
+
+  it "an erlang reference should decode to a Reference object" do
+    ref = get("make_ref()")
+    ref.should be_an_instance_of Beambridge::NewReference
+    ref.node.should be_an_instance_of Symbol
+  end
+
+  it "an erlang pid should decode to a Pid object" do
+    pid = get("spawn(fun() -> 3 end)")
+    pid.should be_an_instance_of Beambridge::Pid
+    pid.node.should be_an_instance_of Symbol
+  end
+
+  it "an erlang tuple encoded as a small tuple (1-byte length) should decode to an array" do
+    ref = get("{3}")
+    ref.length.should == 1
+    ref.first.should == 3
+
+    ref = get("{3, a, make_ref()}")
+    ref.length.should == 3
+    ref[0].should == 3
+    ref[1].should == :a
+    ref[2].class.should == Beambridge::NewReference
+
+    tuple_meat = (['3'] * 255).join(', ')
+    ref = get("{#{tuple_meat}}")
+    ref.length.should == 255
+    ref.each{|r| r.should == 3}
+  end
+
+  it "an erlang tuple encoded as a large tuple (4-byte length) should decode to an array" do
+    tuple_meat = (['3'] * 256).join(', ')
+    ref = get("{#{tuple_meat}}")
+    ref.length.should == 256
+    ref.each{|r| r.should == 3}
+
+    tuple_meat = (['3'] * 512).join(', ')
+    ref = get("{#{tuple_meat}}")
+    ref.length.should == 512
+    ref.each{|r| r.should == 3}
+  end
+
+  it "an empty erlang list encoded as a nil should decode to an array" do
+    get("[]").class.should == Erl::List
+    get("[]").should == []
+  end
+
+  it "an erlang list encoded as a string should decode to an array of bytes (less than ideal, but consistent)" do
+    get("\"asdasd\"").class.should == Erl::List
+    get("\"asdasd\"").should == "asdasd".bytes.to_a
+    get("\"#{'a' * 65534}\"").should == "a".bytes.to_a * 65534
+  end
+
+  it "an erlang list encoded as a list should decode to an erl::list" do
+    get("[3,4,256]").class.should == Erl::List
+    get("[3,4,256]").should == [3,4,256]
+    get("\"#{'a' * 65535 }\"").should == [97] * 65535
+    get("[3,4, foo, {3,4,5,bar}, 256]").should == [3,4, :foo, [3,4,5,:bar], 256]
+  end
+
+  it "an erlang binary should decode to a string" do
+    get("<< 3,4,255 >>").should == "\003\004\377"
+    get("<< \"whatup\" >>").should == "whatup"
+    get("<< 99,0,99 >>").should == "c\000c"
+  end
+
+  it "the empty atom should decode to the empty symbol" do
+    empty_symbol = get("''")
+    empty_symbol.should be_an_instance_of Symbol
+    empty_symbol.to_s.should == ""
+  end
+
+  it "erlang atomic booleans should decode to ruby booleans" do
+    get("true").should == true
+    get("false").should == false
+    get("falsereio").should == :falsereio
+    get("t").should == :t
+    get("f").should == :f
+  end
+
+  it "massive binaries should not overflow the stack" do
+    bin = [131,109,0,128,0,0].pack('c*') + ('a' * (8 * 1024 * 1024))
+    Beambridge::Decoder.decode(bin).size.should eq(8 * 1024 * 1024)
+  end
+
+  it "a good thing should be awesome" do
+    get(%Q-[{options,{struct,[{test,<<"I'm chargin' mah lazer">>}]}},{passage,<<"Why doesn't this work?">>}]-).should ==
+    [[:options, [:struct, [[:test, "I'm chargin' mah lazer"]]]], [:passage, "Why doesn't this work?"]]
+  end
+
+  def get(str)
+    x = "term_to_binary(#{str.gsub(/"/, '\\\"')})"
+    bin = run_erl(x)
+    Beambridge::Decoder.decode(bin)
+  end
+end

data/spec/encode_spec.rb

@@ -0,0 +1,152 @@
+# coding: utf-8
+require "spec_helper"
+
+describe "When packing to a binary stream" do
+
+  let(:encoder) { Beambridge::Encoder.new(StringIO.new('', 'w')) }
+
+  it "A symbol should be encoded to an erlang atom" do
+    get{encoder.write_symbol :haha}.should == get_erl("haha")
+    write_any(:haha).should == get_erl_with_magic("haha")
+  end
+
+  it "A boolean should be encoded to an erlang atom" do
+    get{encoder.write_boolean true}.should == get_erl("true")
+    get{encoder.write_boolean false}.should == get_erl("false")
+    write_any(true).should == get_erl_with_magic("true")
+    write_any(false).should == get_erl_with_magic("false")
+  end
+
+  it "A number should be encoded as an erlang number would be" do
+    #SMALL_INTS
+    get{encoder.write_fixnum 0}.should == get_erl("0")
+    get{encoder.write_fixnum 255}.should == get_erl("255")
+    write_any(0).should == get_erl_with_magic("0")
+    write_any(255).should == get_erl_with_magic("255")
+
+    #INTS
+    get{encoder.write_fixnum 256}.should == get_erl("256")
+    get{encoder.write_fixnum((1 << 27) - 1)}.should == get_erl("#{(1 << 27) - 1}")
+    get{encoder.write_fixnum(-1)}.should == get_erl("-1")
+    get{encoder.write_fixnum(-(1 << 27))}.should == get_erl("#{-(1 << 27)}")
+    write_any(256).should == get_erl_with_magic("256")
+    write_any((1 << 27) - 1).should == get_erl_with_magic("#{(1 << 27) - 1}")
+    write_any(-1).should == get_erl_with_magic("-1")
+    write_any(-(1 << 27)).should == get_erl_with_magic("#{-(1 << 27)}")
+
+    # #SMALL_BIGNUMS
+    get{encoder.write_fixnum(10_000_000_000_000_000_000)}.should == get_erl("10000000000000000000")
+    # get{encoder.write_fixnum(1254976067)}.should == get_erl("1254976067")
+    # get{encoder.write_fixnum(-1254976067)}.should == get_erl("-1254976067")
+    # get{encoder.write_fixnum((1 << word_length))}.should == get_erl("#{(1 << word_length)}")
+    # get{encoder.write_fixnum(-(1 << word_length) - 1)}.should == get_erl("#{-(1 << word_length) - 1}")
+    # get{encoder.write_fixnum((1 << (255 * 8)) - 1)}.should == get_erl("#{(1 << (255 * 8)) - 1}")
+    # get{encoder.write_fixnum(-((1 << (255 * 8)) - 1))}.should == get_erl("#{-((1 << (255 * 8)) - 1)}")
+    #
+    # write_any((1 << word_length)).should == get_erl_with_magic("#{(1 << word_length)}")
+    # write_any(-(1 << word_length) - 1).should == get_erl_with_magic("#{-(1 << word_length) - 1}")
+    # write_any((1 << (255 * 8)) - 1).should == get_erl_with_magic("#{(1 << (255 * 8)) - 1}")
+    # write_any(-((1 << (255 * 8)) - 1)).should == get_erl_with_magic("#{-((1 << (255 * 8)) - 1)}")
+    #
+    # #LARGE_BIGNUMS
+    x = 1254976067 ** 256
+    get{encoder.write_fixnum(x)}.should == get_erl("#{x}")
+    get{encoder.write_fixnum(-x)}.should == get_erl("-#{x}")
+    # get{encoder.write_fixnum((1 << (255 * 8)))}.should == get_erl("#{(1 << (255 * 8))}")
+    # get{encoder.write_fixnum(-(1 << (255 * 8))}.should == get_erl("#{-(1 << (255 * 8)}")
+    # get{encoder.write_fixnum((1 << (512 * 8))}.should == get_erl("#{(1 << (512 * 8))}")
+    # get{encoder.write_fixnum(-((1 << (512 * 8)) - 1))}.should == get_erl("#{-((1 << (512 * 8)) - 1)}")
+    #
+    # write_any((1 << (255 * 8))).should == get_erl_with_magic("#{(1 << (255 * 8))}")
+    # write_any(-(1 << (255 * 8)).should == get_erl_with_magic("#{-(1 << (255 * 8)}")
+    # write_any((1 << (512 * 8))).should == get_erl_with_magic("#{(1 << (512 * 8))}")
+    # write_any(-((1 << (512 * 8)) - 1)).should == get_erl_with_magic("#{-((1 << (512 * 8)) - 1)}")
+  end
+
+  # it "A float (that is within the truncated precision of ruby compared to erlang) should encode as erlang does" do
+  #   get{encoder.write_float 1.0}.should == get_erl("1.0")
+  #   get{encoder.write_float -1.0}.should == get_erl("-1.0")
+  #   get{encoder.write_float 123.456}.should == get_erl("123.456")
+  #   get{encoder.write_float 123.456789012345}.should == get_erl("123.456789012345")
+  # end
+
+  it "An Erlectiricity::NewReference should encode back to its original form" do
+    ref_bin = run_erl("term_to_binary(make_ref())")
+    ruby_ref = Beambridge::Decoder.decode(ref_bin)
+
+    get{encoder.write_new_reference(ruby_ref)}.should == ref_bin[1..-1].bytes.to_a
+    write_any(ruby_ref).should == ref_bin.bytes.to_a
+  end
+
+  it "An Erlectiricity::Pid should encode back to its original form" do
+    pid_bin = run_erl("term_to_binary(spawn(fun() -> 3 end))")
+    ruby_pid = Beambridge::Decoder.decode(pid_bin)
+
+    get{encoder.write_pid(ruby_pid)}.should == pid_bin[1..-1].bytes.to_a
+    write_any(ruby_pid).should == pid_bin.bytes.to_a
+  end
+
+  it "An array written with write_tuple should encode as erlang would a tuple" do
+    get{encoder.write_tuple [1,2,3]}.should == get_erl("{1,2,3}")
+    get{encoder.write_tuple [3] * 255}.should == get_erl("{#{([3] * 255).join(',')}}")
+    get{encoder.write_tuple [3] * 256}.should == get_erl("{#{([3] * 256).join(',')}}")
+    get{encoder.write_tuple [3] * 512}.should == get_erl("{#{([3] * 512).join(',')}}")
+  end
+
+  it "An array should by default be written as a tuple" do
+    write_any([1,2,3]).should == get_erl_with_magic("{1,2,3}")
+    write_any([3] * 255).should == get_erl_with_magic("{#{([3] * 255).join(',')}}")
+    write_any([3] * 256).should == get_erl_with_magic("{#{([3] * 256).join(',')}}")
+    write_any([3] * 512).should == get_erl_with_magic("{#{([3] * 512).join(',')}}")
+  end
+
+  it "An Beambridge::List should by default be written as a list" do
+    write_any(Erl::List.new([1,2,300])).should == get_erl_with_magic("[1,2,300]")
+    write_any(Erl::List.new([300] * 255)).should == get_erl_with_magic("[#{([300] * 255).join(',')}]")
+    write_any(Erl::List.new([300] * 256)).should == get_erl_with_magic("[#{([300] * 256).join(',')}]")
+    write_any(Erl::List.new([300] * 512)).should == get_erl_with_magic("[#{([300] * 512).join(',')}]")
+  end
+
+  it "An array written with write_list should encode as erlang would a list" do
+    get{encoder.write_list [1,2,300]}.should == get_erl("[1,2,300]")
+    get{encoder.write_list [300] * 255}.should == get_erl("[#{([300] * 255).join(',')}]")
+    get{encoder.write_list [300] * 256}.should == get_erl("[#{([300] * 256).join(',')}]")
+    get{encoder.write_list [300] * 256}.should == get_erl("[#{([300] * 256).join(',')}]")
+  end
+
+  it "a string should be encoded as a erlang binary would be" do
+    get{encoder.write_binary "hey who"}.should == get_erl("<< \"hey who\" >>")
+    get{encoder.write_binary ""}.should == get_erl("<< \"\" >>")
+    get{encoder.write_binary "c\000c"}.should == get_erl("<< 99,0,99 >>")
+
+    write_any("hey who").should == get_erl_with_magic("<< \"hey who\" >>")
+    write_any("").should == get_erl_with_magic("<< \"\" >>")
+  end
+
+  it "encodes a string with unicode characters with the correct length" do
+    write_any([:hi, "😸", :bye]).should == [
+      131, 104, 3, 100, 0, 2, 104, 105, 109, 0, 0, 0, 4,
+      240, 159, 152, 184, 100, 0, 3, 98, 121, 101
+    ]
+  end
+
+  def get
+    encoder.out = StringIO.new('', 'w')
+    yield
+    encoder.out.string.bytes.to_a
+  end
+
+  def write_any(term)
+    encoder.out = StringIO.new('', 'w')
+    encoder.write_any term
+    encoder.out.string.bytes.to_a
+  end
+
+  def get_erl(str)
+    get_erl_with_magic(str)[1..-1] #[1..-1] to chop off the magic number
+  end
+
+  def get_erl_with_magic(str)
+    run_erl("term_to_binary(#{str.gsub(/"/, '\\\"')})").bytes.to_a
+  end
+end