totally_lazy 0.1.28 → 0.1.29

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MWZkMGRlODJjYzAwMDBmMmFiMTE3NDk0N2QzNTA4NTJkMTgxZTNkMw==
+    NTY1MTJjY2YzODk1NWFlZjI0Y2VjYzYyNzEyZWI0NTFlMjU5ZmI2NQ==
   data.tar.gz: !binary |-
-    OWM4YjJjMmIxYmUxZjE1N2RiYTI1Mjg0ZThhZmU4MzhlNGM2MWYxOQ==
+    NGE3YmJlNTg4NGRiY2I3YmFkYmUxODU3N2U2ODBiNzc3ZTlhN2Q2YQ==
 SHA512:
   metadata.gz: !binary |-
-    Y2FiN2M4YzIzN2M5OGIzNWFjNmFiM2UyYmYxYTE0MmI4OTBiNDZiZjE3MTdj
-    ZTY0YTkwYTYzM2FiY2FlNWFkMzlhNzhhMGRjZmQzYmE1OWE4NjE0M2UzN2I2
-    ZDBiNTE4NzFlMTY5OWExMjE5YTNlZGVjNzMxZjNkYzZiOWExYjY=
+    Njg1ZTY4MjM5ZTcwNzc0YTI4NzMzZDNkYWQ5MjY2NmJiMDZiZTRiYjZjOGUy
+    ZjQzMGM2YWY4NDZmN2ZhNWJmMGI2YzU2OGNhMTc4Njg1MTFmZDFjYjFmYWU0
+    NjEyMWVjMmI3NTRmNDliMzgxMTI4NmQyOTg1YzcyYjY0NDNjNTc=
   data.tar.gz: !binary |-
-    OTY2MDA1ZjI5MjAzOGIwOGI2NDJkZTQxNmZiZDk2MzJkODNlZTRjOWI1NWY5
-    MjU0YTAzZTYyMDdmYTdlYzY2ZDgzMDdiMjcwNTU1NzFjN2FiNzQ5NDUwMjk4
-    Yzk1MTJlNzk4MTgyYzg3OWFmODcyMTU5YTkwZTNjYmE0MTk2Yjk=
+    NTU2OTllZjkzZmRhZmI5NDZkNjg3ZmY5NWExYWVlMzk3MDJlNTdlODZlZDFh
+    M2Y5MGZmYTRjZGY4OGFkMDg2ODFlNGY5MWE0ZmExZDljMDQ1NGYyZjBhYTgy
+    MGQyOGFmNGRiNGJlYWZmMjZiMGY5MzkyOWNmOWI2ZWJhNzhlODY=

data/lib/totally_lazy/either.rb

@@ -1,5 +1,17 @@
 require_relative 'lambda_block'
 
+class Proc
+  def or_exception
+    -> (value) {
+      begin
+        right(self.(value))
+      rescue Exception => e
+        left(e)
+      end
+    }
+  end
+end
+
 module Eithers
   private
   def left(value)
@@ -54,6 +66,10 @@ class Left < Either
     left(block_given? ? block.call(@value) : fn.(@value))
   end
 
+  def flat_map(fn=nil, &block) # a function which returns an either
+    assert_funcs(fn, block_given?)
+    self
+end
   def <=>(other)
     @value <=> other.left_value
   end
@@ -94,6 +110,11 @@ class Right < Either
     self
   end
 
+  def flat_map(fn=nil, &block) # a function which returns an either
+    assert_funcs(fn, block_given?)
+    block_given? ? block.call(@value) : fn.(@value)
+  end
+
   def <=>(other)
     @value <=> other.right_value
   end

data/spec/totally_lazy/either_spec.rb

@@ -1,3 +1,5 @@
+# encoding: UTF-8
+
 require_relative '../spec_helper'
 
 describe 'Either' do
@@ -21,9 +23,24 @@ describe 'Either' do
     expect(left(3).map_left { |a| a+2 }).to eq(left(5))
   end
 
+  it 'should support flat_map' do
+    expect(right(4).flat_map(divide(2).or_exception)).to eq(right(2))
+    expect(right(4).flat_map { right(2) }).to eq(right(2))
+
+    result = right(4).flat_map(divide(0).or_exception)
+    expect(result.is_left?).to eq(true)
+    expect(result.left_value.class).to eq(ZeroDivisionError)
+
+    expect(left(2).flat_map(divide(2).or_exception)).to eq(left(2))
+    expect(left(2).flat_map { right(5) }).to eq(left(2))
+  end
+
   it 'should raise exception if you try to use both lambda and block' do
     expect { right(1).map(add(2)) { |a| a+2 } }.to raise_error(RuntimeError)
     expect { right(1).map_left(add(2)) { |a| a+2 } }.to raise_error(RuntimeError)
     expect { left(1).map_left(add(2)) { |a| a+2 } }.to raise_error(RuntimeError)
+    expect { right(10).flat_map(divide(2)) { |a| a/2 } }.to raise_error(RuntimeError)
+    expect { left(10).flat_map(divide(2)) { |a| a/2 } }.to raise_error(RuntimeError)
   end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: totally_lazy
 version: !ruby/object:Gem::Version
-  version: 0.1.28
+  version: 0.1.29
 platform: ruby
 authors:
 - Raymond Barlow