From 01a21f94dded482889cea476a73072a15b3696bc Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Sun, 7 Aug 2016 22:06:31 -0400
Subject: stub reader, add .read methods, use Error, better properties

---
 src/zip.cr | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 53 insertions(+), 12 deletions(-)

(limited to 'src')

diff --git a/src/zip.cr b/src/zip.cr
index 5b6f4c4..4761695 100644
--- a/src/zip.cr
+++ b/src/zip.cr
@@ -4,7 +4,7 @@ require "zlib"
 #
 # TODO:
 # [x] date/time
-# [x] reader
+# [ ] reader
 # [ ] documentation
 # [ ] full tests
 # [ ] zip64
@@ -182,10 +182,25 @@ module Zip
 
   # TODO
   class Reader
-    def initialize(path : String)
+    getter? :closed, :sync_close
+
+    def initialize(
+      @io         : IO,
+      @pos        : UInt32 = 0,
+      @sync_close : Bool = false,
+    )
+      @closed = false
     end
 
-    def initialize(io : IO)
+    private def assert_open
+      raise Error.new("already closed") if closed?
+    end
+
+    def close
+      assert_open
+
+      @io.close if @sync_close
+      @closed = true
     end
   end
 
@@ -387,9 +402,9 @@ module Zip
       path_len = path.bytesize
 
       # check file path
-      raise "empty file path" if path_len == 0
-      raise "file path too long" if path_len >= UInt16::MAX
-      raise "file path contains leading slash" if path[0] == '/'
+      raise Error.new("empty file path") if path_len == 0
+      raise Error.new("file path too long") if path_len >= UInt16::MAX
+      raise Error.new("file path contains leading slash") if path[0] == '/'
 
       # write magic (u32), version needed (u16), flags (u16), and
       # compression method (u16)
@@ -431,7 +446,7 @@ module Zip
       when CompressionMethod::DEFLATE
         compress_deflate(@io, dst_io)
       else
-        raise "unsupported compression method"
+        raise Error.new("unsupported compression method: #{@method}")
       end
     end
 
@@ -543,6 +558,8 @@ module Zip
   end
 
   class Writer
+    getter? :closed
+
     def initialize(
       @io       : IO,
       @pos      : UInt32 = 0,
@@ -554,12 +571,8 @@ module Zip
       @src_pos = @pos
     end
 
-    def closed?
-      @closed
-    end
-
     private def assert_open
-      raise "already closed" if closed?
+      raise Error.new("already closed") if closed?
     end
 
     def bytes_written : UInt32
@@ -699,6 +712,7 @@ module Zip
     &cb     : Writer -> \
   ) : UInt32
     r = 0_u32
+
     begin
       w = Writer.new(io, pos, comment, version)
       cb.call(w)
@@ -724,4 +738,31 @@ module Zip
       write(io, pos, comment, version, &cb)
     end
   end
+
+  def self.read(
+    io          : IO,
+    pos         : UInt32 = 0_u32,
+    sync_close  : Bool = false,
+    &cb         : Reader -> \
+  )
+    begin
+      r = Reader.new(io, pos, sync_close)
+      cb.call(r)
+    ensure
+      if r
+        r.close unless r.closed?
+      end
+    end
+
+    nil
+  end
+
+  def self.read(
+    path : String,
+    &cb  : Reader -> \
+  )
+    File.open(path, "rb") do |io|
+      read(io, 0_u32, true, &cb)
+    end
+  end
 end
-- 
cgit v1.2.3