module RailsUploads
  module Task

    def self.iterate_images
      model = ENV['MODEL'].classify.constantize
      presets = ENV['PRESETS'].split(',').map(&:to_sym)
      model.find_each do |record|
        model.attachments.each do |attr, options|
          if options[:type] == :image
            presets.each do |preset|
              image = record.send(attr)
              yield image, preset
            end
          end
        end
      end
    end

  end
end

namespace :uploads do

  task :migrate do
    files = Rails.root.join('public', 'uploads', 'files')
    images = Rails.root.join('public', 'uploads', 'images', 'original')
    ::FileUtils.mkdir_p files
    ::FileUtils.mkdir_p images
    Dir.glob(::File.join(Rails.root.join('public', 'uploads'), '', '*.*')) do |source|
      filename = ::File.basename(source)
      if filename =~ /-/
        ::FileUtils.rm source
      elsif filename =~ /\.(jpg|png|gif)$/
        ::FileUtils.mv source, ::File.join(images, filename)
      else filename =~ /\.!(htm|html|ico)$/
        ::FileUtils.mv source, ::File.join(files, filename)
      end
    end
  end

  namespace :presets do
    desc 'Refresh preset'
    task :refresh => :clean do
      RailsUploads::Task.iterate_images do |image, preset|
        puts "Generating preset #{image.url(preset)}."
        image.generate_preset preset
      end
      puts "Presets refreshed successfully."
    end
    desc 'Clean preset'
    task :clean => :environment do
      RailsUploads::Task.iterate_images do |image, preset|
        puts "Deleting preset #{image.url(preset)}."
        image.delete_preset preset
      end
      puts "Presets cleaned successfully."
    end
  end

  namespace :s3 do
    namespace :buckets do
      desc 'Create buckets'
      task :create do
        config = YAML.load_file(Rails.root.join('config', 's3.yml'))
        config.each do |env, options|
          require 'aws-sdk'
          AWS.config access_key_id: options['access_key_id'], secret_access_key: options['secret_access_key']
          bucket = options['bucket']
          begin
            AWS::S3.new.buckets.create bucket
          rescue AWS::S3::Errors::InvalidAccessKeyId
            puts "Invalid credentials in #{env} environment."
            next
          rescue AWS::S3::Errors::BucketAlreadyExists
            puts "Bucket #{bucket} already exists."
            next
          rescue AWS::S3::Errors::BucketAlreadyOwnedByYou
            puts "You already own bucket #{bucket}."
            next
          end
          puts "Bucket #{bucket} created successfully."
        end
      end
    end
  end

end