inimit 0.0.1 → 0.0.2

data/Gemfile

@@ -4,6 +4,7 @@ source "http://rubygems.org"
 #   gem "activesupport", ">= 2.3.5"
 
 gem 'paint', '~> 0.8.5'
+#gem 'iopts', '~> 0.1.0'
 
 # Add dependencies to develop your gem here.
 # Include everything needed to run rake, tests, features, etc.

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.2

data/bin/inimit

@@ -1,4 +1,4 @@
 #!/usr/bin/env ruby
 require File.join(File.dirname(__FILE__), '..', 'lib', 'inimit')
 
-Inimit::compare *ARGV
+Inimit::run *ARGV

data/inimit.gemspec

@@ -0,0 +1,79 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "inimit"
+  s.version = "0.0.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["AlphaHydrae"]
+  s.date = "2012-09-12"
+  s.description = "Compares files in two directories using SHA1."
+  s.email = "hydrae.alpha@gmail.com"
+  s.executables = ["inimit"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.md"
+  ]
+  s.files = [
+    ".rspec",
+    ".rvmrc",
+    ".screenrc",
+    ".travis.yml",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.md",
+    "Rakefile",
+    "VERSION",
+    "bin/inimit",
+    "inimit.gemspec",
+    "lib/inimit.rb",
+    "spec/helper.rb",
+    "spec/version_spec.rb"
+  ]
+  s.homepage = "http://github.com/AlphaHydrae/inimit"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Recursive file comparison command."
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<paint>, ["~> 0.8.5"])
+      s.add_development_dependency(%q<bundler>, [">= 0"])
+      s.add_development_dependency(%q<rake>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, [">= 0"])
+      s.add_development_dependency(%q<jeweler>, [">= 0"])
+      s.add_development_dependency(%q<gemcutter>, [">= 0"])
+      s.add_development_dependency(%q<gem-release>, [">= 0"])
+      s.add_development_dependency(%q<rake-version>, [">= 0"])
+      s.add_development_dependency(%q<simplecov>, [">= 0"])
+    else
+      s.add_dependency(%q<paint>, ["~> 0.8.5"])
+      s.add_dependency(%q<bundler>, [">= 0"])
+      s.add_dependency(%q<rake>, [">= 0"])
+      s.add_dependency(%q<rspec>, [">= 0"])
+      s.add_dependency(%q<jeweler>, [">= 0"])
+      s.add_dependency(%q<gemcutter>, [">= 0"])
+      s.add_dependency(%q<gem-release>, [">= 0"])
+      s.add_dependency(%q<rake-version>, [">= 0"])
+      s.add_dependency(%q<simplecov>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<paint>, ["~> 0.8.5"])
+    s.add_dependency(%q<bundler>, [">= 0"])
+    s.add_dependency(%q<rake>, [">= 0"])
+    s.add_dependency(%q<rspec>, [">= 0"])
+    s.add_dependency(%q<jeweler>, [">= 0"])
+    s.add_dependency(%q<gemcutter>, [">= 0"])
+    s.add_dependency(%q<gem-release>, [">= 0"])
+    s.add_dependency(%q<rake-version>, [">= 0"])
+    s.add_dependency(%q<simplecov>, [">= 0"])
+  end
+end
+

data/lib/inimit.rb

@@ -1,83 +1,171 @@
 require 'find'
-require 'digest/sha1'
+require 'digest'
 require 'paint'
+require 'optparse'
+
+# TODO: lenient option to only fail on mismatch
 
 module Inimit
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 
-  def self.compare *args
+  class Opts < OptionParser
+    def to_s
+      "#{banner}\nOptions:\n#{summarize ''}"
+    end
+  end
 
-    self.abort "A source and target directory are required (got #{args.length} arguments)" unless args.length == 2
+  def self.run *args
+
+    options = {
+      :verbose => 0,
+      :mismatch_only => false
+    }
+    Opts.new do |opts|
+      opts.banner = "#{Paint[opts.program_name, :bold]} recursively compares files.\n\nUsage:\n    inimit [OPTION]... [SOURCE] TARGET\n"
+      opts.version = VERSION
+      opts.on('-d', '--digest ALGORITHM', 'Use ALGORITHM to hash file contents (available: md5, sha1, sha256, sha384, sha512)') do |digest|
+        options[:digest] = digest
+      end
+      opts.on('-m', '--mismatch', 'Only show files when missing or the content does not match') do
+        options[:mismatch_only] = true
+      end
+      opts.on('-v', '--verbose', 'Increase verbosity (once: show hash prefixes, twice: show full hashes)') do
+        options[:verbose] += 1
+      end
+      opts.on('-h', '--help', 'Show this help and exit'){ puts opts; exit 0 }
+      opts.on('-u', '--usage', 'Show this help and exit'){ puts opts; exit 0 }
+    end.parse! args
+
+    # check number of args
+    self.abort "A target directory must be given" if args.length < 1
+    self.abort "Only one source and target must be given (got #{args.length} arguments)" if args.length > 2
+
+    source, target = case args.length
+    when 1
+      [ '.', args.shift ]
+    when 2
+      [ args.shift, args.shift ]
+    end
 
-    source, target = args.shift, args.shift
-    self.abort "Source #{source} is not a directory" unless File.directory? source
-    self.abort "Target #{target} is not a directory" unless File.directory? target
+    if !File.exists?(source)
+      self.abort "Unknown file or directory #{source}"
+    elsif !File.file?(source) and !File.directory?(source)
+      self.abort "Source #{source} is neither a file nor a directory"
+    elsif File.file?(source) and !File.file?(target)
+      self.abort "Source #{source} is a file, target #{target} is not"
+    elsif File.directory?(source) and !File.directory?(target)
+      self.abort "Source #{source} is a directory, target #{target} is not"
+    end
+
+    if File.directory? source
+      self.compare source, target, options
+    else
+      self.compare_files source, target, options
+    end
+  end
+
+  def self.compare source, target, options = {}
 
     source, target = File.expand_path(source), File.expand_path(target)
 
     puts
-    puts "Comparing files in #{source} to #{target}:"
-    puts
+    puts "Comparing files in #{source} to #{target}."
+    puts unless options[:mismatch_only]
 
     Dir.chdir source
-    unknown = []
-    different = []
+    total, match, unknown, mismatch = 0, 0, 0, 0
 
     Find.find '.' do |file|
 
       # skip directories
       next unless File.file? file
-
-      print "- #{file.sub /^\.\//, ''}..."
+      total += 1
 
       source_file = File.join source, file
       target_file = File.join target, file
 
       if !File.file? target_file
-        unknown << file
+        unknown += 1
         puts Paint[" not found", :yellow]
         next
       end
 
-      source_sha1 = self.hash source_file
-      target_sha1 = self.hash target_file
+      source_sha1 = self.hash source_file, options
+      target_sha1 = self.hash target_file, options
 
       if source_sha1 == target_sha1
-        puts Paint[" #{source_sha1[0,7]} == #{target_sha1[0,7]}", :green]
+        match += 1
+        unless options[:mismatch_only]
+          print Paint[self.hash_notice(source_sha1, target_sha1, options), :green]
+          puts " #{file.sub /^\.\//, ''}..."
+        end
       else
-        puts Paint[" #{source_sha1[0,7]} != #{target_sha1[0,7]}", :red]
-        different << { :file => file, :source => source_sha1, :target => target_sha1 }
+        mismatch += 1
+        print Paint[self.hash_notice(source_sha1, target_sha1, options), :red]
+        puts " #{file.sub /^\.\//, ''}..."
       end
     end
 
-    if different.empty? and unknown.empty?
+    puts
+
+    notice = "Checked #{total} files: "
+    notice << Paint["#{match} identical", :green]
+    notice << ", "
+    notice << Paint["#{unknown} not found", :yellow]
+    notice << ", "
+    notice << Paint["#{mismatch} different", :red]
+
+    puts notice
+    puts
+
+    exit 1 if unknown >= 1 or mismatch >= 1
+  end
+
+  def self.compare_files source, target, options = {}
+
+    source_sha1, target_sha1 = self.hash(source, options), self.hash(target, options)
+
+    puts
+
+    if source_sha1 == target_sha1
+      msg = "Files are identical"
+      msg << " (#{self.hash_notice source_sha1, target_sha1, options})" if options[:verbose] >= 1
+      msg << "."
+      puts Paint[msg, :green]
       puts
-      puts Paint["All files in #{source} are the same in #{target}.", :green, :bold]
+    else
+      msg = "The contents of the files do not match"
+      msg << " (#{self.hash_notice source_sha1, target_sha1, options})" if options[:verbose] >= 1
+      msg << "."
+      puts Paint[msg, :red]
       puts
-      return
+      exit 1
     end
+  end
 
-    if unknown.any?
-      puts
-      puts Paint["The following files in #{source} were not found in #{target}:", :yellow, :bold]
-      unknown.each do |file|
-        puts "- #{file}"
+  def self.hash_notice h1, h2, options = {}
+    if h1 == h2
+      case options[:verbose]
+      when 1; "#{h1[0,7]} == #{h2[0,7]}"
+      when 2; "#{h1} == #{h2}"
+      else; "identical"
       end
-    end
-
-    if different.any?
-      puts
-      puts Paint["The following files in #{source} do not have the same content in #{target}:", :red, :bold]
-      different.each do |h|
-        puts "- #{h[:file]}"
+    else
+      case options[:verbose]
+      when 1..2; "#{h1} != #{h2}"
+      else; "different"
       end
     end
-
-    puts
   end
 
-  def self.hash file
-    dig = Digest::SHA1.new
+  def self.hash file, options = {}
+    dig = case options[:digest]
+    when 'md5'; Digest::MD5.new
+    when 'sha256'; Digest::SHA256.new
+    when 'sha384'; Digest::SHA384.new
+    when 'sha512'; Digest::SHA512.new
+    else; Digest::SHA1.new
+    end
     File.open file, 'rb' do |io|
       dig.update io.readpartial(4096) while !io.eof
     end
@@ -85,7 +173,7 @@ module Inimit
   end
 
   def self.abort msg
-    warn msg
-    exit 1
+    warn Paint[msg, :yellow]
+    exit 2
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: inimit
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-09-11 00:00:00.000000000 Z
+date: 2012-09-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: paint
@@ -175,6 +175,7 @@ files:
 - Rakefile
 - VERSION
 - bin/inimit
+- inimit.gemspec
 - lib/inimit.rb
 - spec/helper.rb
 - spec/version_spec.rb