require 'erb'
require "fastimage"

module Snapshot
  class ReportsGenerator
    def generate
      screens_path = SnapshotConfig.shared_instance.screenshots_path

      @data = {}

      Dir["#{screens_path}/*"].sort.each do |language_path|
        language = language_path.split('/').last
        Dir[[language_path, '*'].join('/')].sort.each do |screenshot|

          available_devices.each do |key_name, output_name|

            if screenshot.split('/').last.include?key_name
              # This screenshot it from this device
              @data[language] ||= {}
              @data[language][output_name] ||= []
              @data[language][output_name] << screenshot
              break # to not include iPhone 6 and 6 Plus
            end
          end
        end
      end

      html_path = [lib_path, "snapshot/page.html.erb"].join('/')
      html = ERB.new(File.read(html_path)).result(binding) # http://www.rrn.dk/rubys-erb-templating-system

      
      export_path = SnapshotConfig.shared_instance.html_path
      File.write(export_path, html)

      Helper.log.info "Successfully created HTML file with all the screenshots: #{export_path}".green
    end

    private
      def lib_path
        "./lib/"
      end

      def available_devices
        # The order IS important, since those names are used to check for include?
        # and the iPhone 6 is inlucded in the iPhone 6 Plus
        {
          'iPhone6Plus' => "iPhone 6 Plus",
          'iPhone6' => "iPhone 6",
          'iPhone5' => "iPhone 5",
          'iPhone4' => "iPhone 4",
        }
      end
  end
end