lib/rupert/rpm.rb in rupert-0.0.1 vs lib/rupert/rpm.rb in rupert-0.0.2

@@ -1,9 +1,10 @@
 require 'rupert/errors'
 require 'rupert/parser'
 require 'rupert/rpm/lead'
 require 'rupert/rpm/signature'
+require 'rupert/rpm/header'
 
 require 'base64'
 
 module Rupert
   class RPM
@@ -24,95 +25,123 @@
       end
 
       # Tells whether given filename points to a valid RPM or not.
       #
       # @param filename [String] filename to inspect
-      # @return `true` if file starts with the correct magic header
+      # @return +true+ if file starts with the correct magic header
       def rpm?(filename)
         Lead.new(File.open(filename, 'r')).rpm?
       end
     end
 
     # Initialize the RPM object, given its components.
     #
     # This method is not intended to be used to instantiate RPM objects
-    # directly. Instead, use Rupert::RPM::load for a more straightforward
+    # directly. Instead, use {Rupert::RPM.load} for a more straightforward
     # alternative.
     #
     # @param lead [Rupert::RPM::Lead] RPM lead section
-    # @param signature [Rupert::RPM::Signature] RPM signature section
+    # @param signature [Rupert::RPM::Signature] RPM signature information
     # @param content [String] Raw content found after the signature structure
-    def initialize(lead, signature, content)
+    # @param header [Rupert::RPM::Header] RPM header holding package metadata
+    def initialize(lead, signature, content, header)
       @lead = lead
       @signature = signature
       @content = content
+      @header = header
     end
 
     # RPM version used to encode the package.
     #
-    # @return [String] the RPM version in `<major>.<minor>` format
+    # @return [String] the RPM version in +<major>.<minor>+ format
     def rpm_version
       @lead.rpm_version
     end
 
-    # @return `true` if the RPM is of type binary, `false` otherwise
+    # @return +true+ if the RPM is of type binary, +false+ otherwise
     def binary?
       @lead.binary_type?
     end
 
-    # @return `true` if the RPM is of type source, `false` otherwise
+    # @return +true+ if the RPM is of type source, +false+ otherwise
     def source?
       @lead.source_type?
     end
 
-    # Which architecture the package was built for, e.g. `i386/x86_64` or
-    # `arm`
+    # Which architecture the package was built for, e.g. +i386/x86_64+ or
+    # +arm+
     #
     # @return [String] package architecture name
     def rpm_arch
      @lead.arch
     end
 
     # Full package name
     #
-    # @return [String] package name in the form <name>-<version>-<rev>.<suffix>
+    # @return [String] package name in the form +<name>-<version>-<rev>.<suffix>+
     def name
-      @lead.name
+      @header.name
     end
 
     # OS for which the package was built
     #
-    # @return [String] as defined in /usr/lib/rpm/rpmrc under the canonical OS
+    # @return [String] as defined in _/usr/lib/rpm/rpmrc_ under the canonical OS
     #         names section
     def os
       @lead.os
     end
 
-    # @return `true` if the package is signed, `false` otherwise
+    # @return +true+ if the package is signed, +false+ otherwise
     def signed?
       @lead.signed?
     end
 
     # MD5 checksum stored in the package (base64 encoded). To be used to check
     # package integrity.
     #
-    # NOTE: This is not the MD5 of the whole package; rather, the digest is
+    # *NOTE*: This is not the MD5 of the whole package; rather, the digest is
     # calculated over the header and payload, leaving out the lead and the
-    # signature header. I.e., running `md5sum <myrpm>` won't held the same
-    # result as `Rupert::RPM.load('<myrpm>').md5`.
+    # signature header. I.e., running +md5sum <myrpm>+ won't held the same
+    # result as +Rupert::RPM.load('<myrpm>').md5+.
     #
     # @return [String] Base64-encoded MD5 checksum of package's header and
     #         payload, stored in the RPM itself
     def md5
       Base64.strict_encode64(@signature.md5)
     end
 
     # Verifies package integrity. Compares MD5 checksum stored in the package
     # with checksum calculated over header(s) and payload (archive).
     #
-    # @return `true` if package is intact, `false` if package (either stored MD5 or
+    # @return +true+ if package is intact, +false+ if package (either stored MD5 or
     # payload) is corrupted
     def intact?
-      @signature.verify_checksum(@content)      
+      @signature.md5 == Digest::MD5.digest(@content)
+    end
+
+    # Package uncompressed size.
+    #
+    # This is the size (in bytes) of the uncompressed archive, or if you
+    # prefer, package's installed size.
+    #
+    # *NOTE*: if reading a package built with native +rpmbuild+, this number
+    # (which is stored in the RPM itself) might not be precise, as
+    # this[http://rpm5.org/community/rpm-devel/2689.html] thread explains. 
+    #
+    # @return [Fixnum] package uncompressed size (bytes)
+    def uncompressed_size
+      @header.uncompressed_size
+    end
+
+    # List of installed files (full paths).
+    #
+    # @return [Array] array of +String+, with entries corresponding to 
+    #         absolute filenames
+    def filenames
+      @header.dirindexes.map { |idx|
+        @header.dirnames[idx]
+      }.zip(@header.basenames).map { |dir, name|
+        File.join(dir, name)
+      }
     end
   end
 end