mini_tarball 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 186593f6df9a85bfd1247d7ac48d2e2516d21d3dc7cd92e97774667ecc0768be
-  data.tar.gz: 7b4c0465dddd0c67822fca7f7d44c5e134dea5dae4e839f307975d42aacf5728
+  metadata.gz: bb6115ef818ae567ea4efad8646977a26cbb62f5164e656bd8d403dd1775ff38
+  data.tar.gz: 5364a6eece1aa623c4bb900eaf50de383017dcd4b9285f5f7b3b52423638d9b2
 SHA512:
-  metadata.gz: 76239123207c12d071fa25ac28a7ee9e1d1796e3a68f15a5d124eeaf695590900130dff61cfe3e8af8c92f6b48ffa41e9be6e311ab0e93c42da5b0fb7efa1da8
-  data.tar.gz: e8bfcc562c8569b8f9168757406dfe162f31417a5bf7fb3b658b99a76de6896fa9b8dd3b5d73f8e1c33b434499dadc332a9c5dd6659185359a0d278d41b872c5
+  metadata.gz: d8978adda8f400f47d37d037fc439d6d803538910aaa97a78cee3ec111b34616c0b3acd298a6a5b406a02a8e0d698e25743a5a4ff32a2e1b827835892621d186
+  data.tar.gz: 543abee46d1e1bd3df2abdd0b0164f2e72363d65ec63a7ac8c120302c94a0fc247315fcb7c4541774ab08ef00178e4de1df355cb2699e0056f0d984e53f532bc

data/lib/mini_tarball/header.rb

@@ -11,7 +11,7 @@ module MiniTarball
     # rubocop:disable Layout/HashAlignment
     FIELDS = {
       name:     { length: 100, type: :chars },
-      mode:     { length:   8, type: :number },
+      mode:     { length:   8, type: :mode },
       uid:      { length:   8, type: :number },
       gid:      { length:   8, type: :number },
       size:     { length:  12, type: :number },

data/lib/mini_tarball/header_formatter.rb

@@ -4,6 +4,8 @@ module MiniTarball
   ValueTooLargeError = Class.new(StandardError)
 
   class HeaderFormatter
+    PERMISSION_BITMASK = 0007777
+
     # @param value [Integer]
     # @param length [Integer]
     def self.format_number(value, length)
@@ -38,5 +40,12 @@ module MiniTarball
 
       encoded.pack("C#{length}")
     end
+
+    # Removes file type bitfields and returns file permissions as formatted number
+    # @param value [Integer]
+    # @param length [Integer]
+    def self.format_permissions(value, length)
+      format_number(value & PERMISSION_BITMASK, length)
+    end
   end
 end

data/lib/mini_tarball/header_writer.rb

@@ -22,6 +22,8 @@ module MiniTarball
         case field[:type]
         when :number
           values_by_field[name] = HeaderFormatter.format_number(value, field[:length])
+        when :mode
+          values_by_field[name] = HeaderFormatter.format_permissions(value, field[:length])
         when :checksum
           values_by_field[name] = " " * field[:length]
         end

data/lib/mini_tarball/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MiniTarball
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/mini_tarball/writer.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'etc'
+
 module MiniTarball
   class Writer
     END_OF_TAR_BLOCK_SIZE = 1024
@@ -27,7 +29,7 @@ module MiniTarball
     end
 
     def initialize(io)
-      check_io!(io)
+      ensure_valid_io!(io)
 
       @io = io
       @write_only_io = WriteOnlyStream.new(@io)
@@ -35,8 +37,30 @@ module MiniTarball
       @closed = false
     end
 
-    def add_file(name:, mode: 0644, uname: "nobody", gname: "nogroup", uid: nil, gid: nil, mtime: nil)
-      check_closed!
+    def add_file(name:, source_file_path:, mode: nil, uname: nil, gname: nil, uid: nil, gid: nil, mtime: nil)
+      ensure_not_closed!
+
+      file = File.open(source_file_path, "rb")
+      stat = File.stat(source_file_path)
+
+      @header_writer.write(Header.new(
+        name: name,
+        size: stat.size,
+        mode: mode || stat.mode,
+        uid: uid || stat.uid,
+        gid: gid || stat.gid,
+        uname: uname || Etc.getpwuid(stat.uid).name,
+        gname: gname || Etc.getgrgid(stat.gid).name,
+        mtime: mtime || stat.mtime
+      ))
+
+      IO.copy_stream(file, @write_only_io)
+      write_padding
+      nil
+    end
+
+    def add_file_from_stream(name:, mode: 0644, uname: "nobody", gname: "nogroup", uid: nil, gid: nil, mtime: nil)
+      ensure_not_closed!
 
       header_start_position = @io.pos
       @header_writer.write(Header.new(name: name))
@@ -59,6 +83,7 @@ module MiniTarball
       ))
 
       @io.seek(0, IO::SEEK_END)
+      nil
     end
 
     def closed?
@@ -66,7 +91,7 @@ module MiniTarball
     end
 
     def close
-      check_closed!
+      ensure_not_closed!
 
       @io.write("\0" * END_OF_TAR_BLOCK_SIZE)
       @io.close
@@ -75,15 +100,12 @@ module MiniTarball
 
     private
 
-    def check_io!(io)
+    def ensure_valid_io!(io)
       raise "No IO object given" unless io.respond_to?(:pos) &&
-        io.respond_to?(:seek) && io.respond_to?(:write) && io.respond_to?(:close)
-
-      io.seek(0, IO::SEEK_END)
-      raise "Stream must be empty" unless io.pos == 0
+        io.respond_to?(:write) && io.respond_to?(:close)
     end
 
-    def check_closed!
+    def ensure_not_closed!
       raise IOError.new("#{self.class} is closed") if closed?
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mini_tarball
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Discourse
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-02-22 00:00:00.000000000 Z
+date: 2021-03-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -129,7 +129,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.1.4
 signing_key: 
 specification_version: 4
 summary: A minimal implementation of the GNU Tar format.