wraith 3.0.4 → 3.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c409214f8f7834c0b3b2c6fa5f010686f092d9a4
-  data.tar.gz: a3542e23dbcbf0c3f8da7bb6c07cd92828912418
+  metadata.gz: 5da45ef7a6e2d91b95dfd97c78b2a67fbc213aed
+  data.tar.gz: f629294fbf9daf54fda68cd57ab46286161e9f8d
 SHA512:
-  metadata.gz: 479c53baad28975382e6080c8c0828bc7d63c89971c7d88a45903a2ae114940dc9b64b67e59852e226556d760bb073da40cb5d5e4e4f571d44183b9235bc8b10
-  data.tar.gz: 67c76a4f7a7638d08df5ff730aea5349cb10a5181f29df869cdacf439e9906dd317dde59e5376efe782920e1385a7ffcb360a96a8c2d90777ed954dedd138ba2
+  metadata.gz: f314042b1c3b46a96ed5b352ef41eb40ced18f64b2aef0c7498ba563d0d980399a9c38801ee0ce80dd77f198d3c02e92a21aa70ca819b11430d0dd44c1a39279
+  data.tar.gz: 3bc518f1a5ad518a9230f4c77edc859ba31d2a21ac989017692a4234e01373335560511e28a7ad4fe0aca1bd75be27636b4641d27ab6e8475e3ec4055209c6e6

data/.travis.yml

@@ -2,6 +2,9 @@ language: ruby
 rvm:
 - 1.9.3
 - 2.0.0
+before_install:
+  - gem update --system 2.4.5 # last known successful Travis build
+  - gem --version
 script: bundle exec rspec
 notifications:
   email:
@@ -11,3 +14,4 @@ notifications:
     on_success: never
   slack:
     secure: BgRAqwHabAtIBgtApDjyUiND2SNxd4sHMgq4ffnJ+EoMme6RSUAeK0G6LLyaGAk6YcpCeWRGOccEpzai87R3ckv6uycUVGxFcTvPmCEClakbUelWovVEyVT3hPLWznxJ8pz3EVB2+5aJnAsTg5M2ZnYtk3a5C1mrPS+WKceE/Ls=
+sudo: false

data/Gemfile

@@ -1,4 +1,4 @@
 source "https://rubygems.org"
 
-# Specify your gem's dependencies in ..gemspec
+# Specify your gem's dependencies in wraith.gemspec
 gemspec

data/lib/wraith/cli.rb

@@ -7,9 +7,14 @@ require "wraith/folder"
 require "wraith/thumbnails"
 require "wraith/compare_images"
 require "wraith/gallery"
+require "wraith/validate"
+require "wraith/version"
+require "wraith/helpers/logger"
+require "wraith/helpers/utilities"
 
 class Wraith::CLI < Thor
   include Thor::Actions
+  include Logging
 
   attr_accessor :config_name
 
@@ -17,127 +22,157 @@ class Wraith::CLI < Thor
     File.expand_path("../../../", __FILE__)
   end
 
+  # define internal methods which user should not be able to run directly
+  no_commands do
+    def within_acceptable_limits
+      yield
+    rescue CustomError => e
+      logger.error e.message
+      # other errors, such as SystemError, will not be caught nicely and will give a stack trace (which we'd need)
+    end
+
+    def check_for_paths(config_name)
+      spider = Wraith::Spidering.new(config_name)
+      spider.check_for_paths
+    end
+
+    def copy_old_shots(config_name)
+      create = Wraith::FolderManager.new(config_name)
+      create.copy_old_shots
+    end
+  end
+
+  desc "validate", "checks your configuration and validates that all required properties exist"
+  def validate(config_name)
+    within_acceptable_limits do
+      wraith = Wraith::Validate.new(config_name).validate
+    end
+  end
+
   desc "setup", "creates config folder and default config"
   def setup
-    directory("templates/configs", "configs")
-    directory("templates/javascript", "javascript")
+    within_acceptable_limits do
+      directory("templates/configs", "configs")
+      directory("templates/javascript", "javascript")
+    end
   end
 
   desc "reset_shots [config_name]", "removes all the files in the shots folder"
   def reset_shots(config_name)
-    reset = Wraith::FolderManager.new(config_name)
-    reset.clear_shots_folder
+    within_acceptable_limits do
+      reset = Wraith::FolderManager.new(config_name)
+      reset.clear_shots_folder
+    end
   end
 
   desc "setup_folders [config_name]", "create folders for images"
   def setup_folders(config_name)
-    create = Wraith::FolderManager.new(config_name)
-    create.create_folders
+    within_acceptable_limits do
+      create = Wraith::FolderManager.new(config_name)
+      create.create_folders
+    end
   end
 
   desc "copy_base_images [config_name]", "copies the required base images over for comparison with latest images"
   def copy_base_images(config_name)
-    copy = Wraith::FolderManager.new(config_name)
-    copy.copy_base_images
-  end
-
-  desc "make_sure_base_shots_exists [config_name]", "warns user if config is missing base shots"
-  def make_sure_base_shots_exists(config_name)
-    wraith = Wraith::Wraith.new(config_name)
-    if wraith.history_dir.nil?
-      puts "You need to specify a `history_dir` property at #{config_name} before you can run `wraith latest`!"
-      exit 1
-    end
-    if !File.directory?(wraith.history_dir)
-      puts "You need to run `wraith history` at least once before you can run `wraith latest`!"
-      exit 1
-    end
-  end
-
-  no_commands do
-    def check_for_paths(config_name)
-      spider = Wraith::Spidering.new(config_name)
-      spider.check_for_paths
-    end
-
-    def copy_old_shots(config_name)
-      create = Wraith::FolderManager.new(config_name)
-      create.copy_old_shots
+    within_acceptable_limits do
+      copy = Wraith::FolderManager.new(config_name)
+      copy.copy_base_images
     end
   end
 
   desc "save_images [config_name]", "captures screenshots"
   def save_images(config_name, history = false)
-    puts "SAVING IMAGES"
-    save_images = Wraith::SaveImages.new(config_name, history)
-    save_images.save_images
+    within_acceptable_limits do
+      logger.info "SAVING IMAGES"
+      save_images = Wraith::SaveImages.new(config_name, history)
+      save_images.save_images
+    end
   end
 
   desc "crop_images [config_name]", "crops images to the same height"
   def crop_images(config_name)
-    crop = Wraith::CropImages.new(config_name)
-    crop.crop_images
+    within_acceptable_limits do
+      crop = Wraith::CropImages.new(config_name)
+      crop.crop_images
+    end
   end
 
   desc "compare_images [config_name]", "compares images to generate diffs"
   def compare_images(config_name)
-    puts "COMPARING IMAGES"
-    compare = Wraith::CompareImages.new(config_name)
-    compare.compare_images
+    within_acceptable_limits do
+      logger.info "COMPARING IMAGES"
+      compare = Wraith::CompareImages.new(config_name)
+      compare.compare_images
+    end
   end
 
   desc "generate_thumbnails [config_name]", "create thumbnails for gallery"
   def generate_thumbnails(config_name)
-    puts "GENERATING THUMBNAILS"
-    thumbs = Wraith::Thumbnails.new(config_name)
-    thumbs.generate_thumbnails
+    within_acceptable_limits do
+      logger.info "GENERATING THUMBNAILS"
+      thumbs = Wraith::Thumbnails.new(config_name)
+      thumbs.generate_thumbnails
+    end
   end
 
   desc "generate_gallery [config_name]", "create page for viewing images"
   def generate_gallery(config_name, multi = false)
-    puts "GENERATING GALLERY"
-    gallery = Wraith::GalleryGenerator.new(config_name, multi)
-    gallery.generate_gallery
+    within_acceptable_limits do
+      logger.info "GENERATING GALLERY"
+      gallery = Wraith::GalleryGenerator.new(config_name, multi)
+      gallery.generate_gallery
+    end
   end
 
-  desc "capture [config_name]", "A full Wraith job"
+  desc "capture [config_name]", "Capture paths against two domains, compare them, generate gallery"
   def capture(config, multi = false)
-    reset_shots(config)
-    check_for_paths(config)
-    setup_folders(config)
-    save_images(config)
-    crop_images(config)
-    compare_images(config)
-    generate_thumbnails(config)
-    generate_gallery(config, multi)
+    within_acceptable_limits do
+      logger.info Wraith::Validate.new(config).validate('capture')
+      reset_shots(config)
+      check_for_paths(config)
+      setup_folders(config)
+      save_images(config)
+      crop_images(config)
+      compare_images(config)
+      generate_thumbnails(config)
+      generate_gallery(config, multi)
+    end
   end
 
   desc "multi_capture [filelist]", "A Batch of Wraith Jobs"
   def multi_capture(filelist)
-    config_array = IO.readlines(filelist)
-    config_array.each do |config|
-      capture(config.chomp, true)
+    within_acceptable_limits do
+      config_array = IO.readlines(filelist)
+      config_array.each do |config|
+        capture(config.chomp, true)
+      end
     end
   end
 
   desc "history [config_name]", "Setup a baseline set of shots"
   def history(config)
-    reset_shots(config)
-    check_for_paths(config)
-    setup_folders(config)
-    save_images(config)
-    copy_old_shots(config)
+    within_acceptable_limits do
+      logger.info Wraith::Validate.new(config).validate('history')
+      reset_shots(config)
+      check_for_paths(config)
+      setup_folders(config)
+      save_images(config)
+      copy_old_shots(config)
+    end
   end
 
   desc "latest [config_name]", "Capture new shots to compare with baseline"
   def latest(config)
-    make_sure_base_shots_exists(config)
-    reset_shots(config)
-    save_images(config, true)
-    copy_base_images(config)
-    crop_images(config)
-    compare_images(config)
-    generate_thumbnails(config)
-    generate_gallery(config)
-  end
-end
+    within_acceptable_limits do
+      logger.info Wraith::Validate.new(config).validate('latest')
+      reset_shots(config)
+      save_images(config, true)
+      copy_base_images(config)
+      crop_images(config)
+      compare_images(config)
+      generate_thumbnails(config)
+      generate_gallery(config)
+    end
+  end
+end

data/lib/wraith/compare_images.rb

@@ -1,10 +1,12 @@
 require "wraith"
+require "wraith/helpers/logger"
 require "image_size"
 require "open3"
 require "parallel"
 require "shellwords"
 
 class Wraith::CompareImages
+  include Logging
   attr_reader :wraith
 
   def initialize(config)
@@ -16,9 +18,9 @@ class Wraith::CompareImages
     Parallel.each(files.each_slice(2), :in_processes => Parallel.processor_count) do |base, compare|
       diff = base.gsub(/([a-zA-Z0-9]+).png$/, "diff.png")
       info = base.gsub(/([a-zA-Z0-9]+).png$/, "data.txt")
-      puts "Comparing #{base} and #{compare}"
+      logger.info "Comparing #{base} and #{compare}"
       compare_task(base, compare, diff, info)
-      puts "Saved diff"
+      logger.info "Saved diff"
     end
   end
 

data/lib/wraith/crop.rb

@@ -1,9 +1,11 @@
 require "wraith"
+require "wraith/helpers/logger"
 require "image_size"
 require "parallel"
 require "shellwords"
 
 class Wraith::CropImages
+  include Logging
   attr_reader :wraith
 
   def initialize(config)
@@ -14,7 +16,7 @@ class Wraith::CropImages
     files = Dir.glob("#{wraith.directory}/*/*.png").sort
 
     Parallel.each(files.each_slice(2), :in_processes => Parallel.processor_count) do |base, compare|
-      puts "cropping images"
+      logger.info "cropping images"
 
       width          = image_dimensions(base)[0]
       base_height    = image_dimensions(base)[1]

data/lib/wraith/folder.rb

@@ -1,6 +1,8 @@
 require "wraith"
+require "wraith/helpers/logger"
 
 class Wraith::FolderManager
+  include Logging
   attr_reader :wraith
 
   def initialize(config)
@@ -35,17 +37,17 @@ class Wraith::FolderManager
 
   def copy_old_shots
     if history_dir.nil?
-      abort 'Error: no `history_dir` attribute found in config. Cannot copy files.'
+      logger.error 'no `history_dir` attribute found in config. Cannot copy files.'
     else
       FileUtils.cp_r("#{dir}/.", "#{history_dir}/")
     end
   end
 
   def copy_base_images
-    puts "COPYING BASE IMAGES"
+    logger.info "COPYING BASE IMAGES"
     wraith.paths.each do |path|
       path = path[0]
-      puts "Copying #{history_dir}/#{path} to #{dir}"
+      logger.info "Copying #{history_dir}/#{path} to #{dir}"
       FileUtils.cp_r(Dir.glob("#{history_dir}/#{path}"), dir)
     end
   end
@@ -60,7 +62,7 @@ class Wraith::FolderManager
       FileUtils.mkdir_p("#{dir}/thumbnails/#{folder_label}")
       FileUtils.mkdir_p("#{dir}/#{folder_label}")
     end
-    puts "Creating Folders"
+    logger.info "Creating Folders"
   end
 
   def tidy_shots_folder(dirs)

data/lib/wraith/gallery.rb

@@ -2,8 +2,10 @@ require "erb"
 require "pp"
 require "fileutils"
 require "wraith/wraith"
+require "wraith/helpers/logger"
 
 class Wraith::GalleryGenerator
+  include Logging
   attr_reader :wraith
 
   MATCH_FILENAME = /(\S+)_(\S+)\.\S+/
@@ -35,9 +37,7 @@ class Wraith::GalleryGenerator
 
       Dir.foreach("#{dirname}/#{category}") do |filename|
         match = MATCH_FILENAME.match(filename)
-        unless match.nil?
-          matcher(match, filename, dirname, category)
-        end
+        matcher(match, filename, dirname, category) unless match.nil?
       end
     end
     @folder_manager.tidy_shots_folder(@dirs)
@@ -76,10 +76,7 @@ class Wraith::GalleryGenerator
   def get_group_from_match(match)
     group = match[2]
     dash = match[2].rindex('-')
-
-    if !dash.nil?
-      group = match[2][dash+1..-1]
-    end
+    group = match[2][dash+1..-1] unless dash.nil?
     group
   end
 
@@ -123,6 +120,16 @@ class Wraith::GalleryGenerator
     Hash[@sorted]
   end
 
+  def generate_gallery(with_path = "")
+    dest = "#{@location}/gallery.html"
+    directories = parse_directories(@location)
+
+    template = File.expand_path("gallery_template/#{wraith.gallery_template}.erb", File.dirname(__FILE__))
+    generate_html(@location, directories, template, dest, with_path)
+
+    report_gallery_status dest
+  end
+
   def generate_html(location, directories, template, destination, path)
     template = File.read(template)
     locals = {
@@ -130,7 +137,6 @@ class Wraith::GalleryGenerator
       :directories => directories,
       :path        => path,
       :threshold   => wraith.threshold
-
     }
     html = ERB.new(template).result(ErbBinding.new(locals).get_binding)
     File.open(destination, "w") do |outf|
@@ -138,15 +144,8 @@ class Wraith::GalleryGenerator
     end
   end
 
-  def generate_gallery(with_path = "")
-    dest = "#{@location}/gallery.html"
-    directories = parse_directories(@location)
-
-    slideshow_template = File.expand_path("gallery_template/#{wraith.gallery_template}.erb", File.dirname(__FILE__))
-
-    generate_html(@location, directories, slideshow_template, dest, with_path)
-
-    puts "Gallery generated"
+  def report_gallery_status(dest)
+    logger.info "Gallery generated"
     failed = check_failed_shots
     prompt_user_to_open_gallery dest
     exit 1 if failed
@@ -156,15 +155,15 @@ class Wraith::GalleryGenerator
     if @multi
       return false
     elsif @failed_shots == false
-      puts "Failures detected:"
+      logger.warn "Failures detected:"
 
       @dirs.each do |dir, sizes|
         sizes.to_a.sort.each do |size, files|
           file = dir.gsub('__', '/')
           if !files.include?(:diff)
-            puts "\t Unable to create a diff image for #{file}"
+            logger.warn "\t Unable to create a diff image for #{file}"
           elsif files[:data] > wraith.threshold
-            puts "\t #{file} failed at a resolution of #{size} (#{files[:data]}% diff)"
+            logger.warn "\t #{file} failed at a resolution of #{size} (#{files[:data]}% diff)"
           end
         end
       end
@@ -176,8 +175,8 @@ class Wraith::GalleryGenerator
   end
 
   def prompt_user_to_open_gallery(dest)
-    puts "\nView the gallery in your browser:"
-    puts "\t file://" + `pwd`.chomp + '/' + dest
+    logger.info "\nView the gallery in your browser:"
+    logger.info "\t file://" + Dir.pwd + '/' + dest
   end
 
   class ErbBinding < OpenStruct