From 755bd3bc11d1f47f6ad75e8a4011d7440f542282 Mon Sep 17 00:00:00 2001
From: Loic Nageleisen <loic.nageleisen@gmail.com>
Date: Tue, 6 Nov 2012 16:47:46 +0100
Subject: [PATCH] modulization

---
 tee.rb | 68 ++++++++++++++++++++++++++++++++--------------------------
 1 file changed, 37 insertions(+), 31 deletions(-)

diff --git a/tee.rb b/tee.rb
index 97c96de..db1ae3f 100644
--- a/tee.rb
+++ b/tee.rb
@@ -8,34 +8,44 @@ module Enumerable
 end
 
 class IO
-  def chunks(chunk_size=1024)
-    Enumerator.new { |y| y << read(chunk_size) until eof? }
-  end
-
-  def each_chunk(chunk_size=nil)
-    chunks.each { |*args| yield *args }
-  end
-
-  def digest_with(digest, chunk_size=nil)
-    chunks(chunk_size).each { |chunk| digest << chunk }
-    digest
-  end
-
-  def sha256(chunk_size=nil)
-    digest_with Digest::SHA2.new(256), chunk_size
-  end
-
-  def tee(*procs)
-    ios = procs.map { |proc| FiberChunkedIO.new &proc }
-
-    chunks.each do |chunk|
-      ios.each do |io|
-        io.write chunk
-      end
+  module Chunkable
+    def chunks(chunk_size=1024)
+      Enumerator.new { |y| y << read(chunk_size) until eof? }
+    end
+
+    def each_chunk(chunk_size=nil)
+      chunks.each { |*args| yield *args }
     end
-    ios.each { |io| io.close }
-    ios.map { |io| io.result }
   end
+
+  module Digestable
+    def digest_with(digest, chunk_size=nil)
+      chunks(chunk_size).each { |chunk| digest << chunk }
+      digest
+    end
+
+    def sha256(chunk_size=nil)
+      digest_with Digest::SHA2.new(256), chunk_size
+    end
+  end
+
+  module Utils
+    def tee(*procs)
+      ios = procs.map { |proc| FiberChunkedIO.new &proc }
+
+      chunks.each do |chunk|
+        ios.each do |io|
+          io.write chunk
+        end
+      end
+      ios.each { |io| io.close }
+      ios.map { |io| io.result }
+    end
+  end
+
+  include Chunkable
+  include Digestable
+  include Utils
 end
 
 class FiberChunkedIO
@@ -83,11 +93,7 @@ class FiberChunkedIO
     @chunks.shift
   end
 
-  def chunks(chunk_size=1024)
-    Enumerator.new do |y|
-      y << read(chunk_size) until eof?
-    end
-  end
+  include IO::Chunkable
 end
 
 File.open("test") do |f|