reports_generator.rb in snapshot-0.9.0

- old
+ new

@@ -2,12 +2,15 @@
 require 'fastimage'
 
 module Snapshot
   class ReportsGenerator
     def generate
-      screens_path = SnapshotConfig.shared_instance.screenshots_path
 
+      config = SnapshotConfig.shared_instance
+      screens_path = config.screenshots_path
+
+      @title = config.html_title    
       @data = {}
 
       Dir["#{screens_path}/*"].sort.each do |language_path|
         language = File.basename(language_path)
         Dir[File.join(language_path, '*')].sort.each do |screenshot|
@@ -55,10 +58,11 @@
         {
           'iPhone6Plus' => "iPhone 6 Plus",
           'iPhone6' => "iPhone 6",
           'iPhone5' => "iPhone 5",
           'iPhone4' => "iPhone 4",
-          'iOS-iPad' => "iPad"
+          'iPad' => "iPad",
+          'Mac' => "Mac"
         }
       end
   end
 end
\ No newline at end of file