require "bundler/gem_tasks"
require "rspec/core/rake_task"
require "open-uri"

def yellow(str)
  "\e[33m#{str}\e[m"
end

ENV['VAGRANT_CWD'] = File.expand_path('spec/integration/vm')


desc 'Run unit and integration tests'
task :spec => ['spec:unit', 'spec:integration']

namespace :spec do
  RSpec::Core::RakeTask.new("unit") do |task|
    task.pattern = "./spec/unit{,/*/**}/*_spec.rb"
  end

  RSpec::Core::RakeTask.new("integration") do |task|
    task.pattern = "./spec/integration{,/*/**}/*_spec.rb"
  end

  namespace :integration do
    integration_dir = 'spec/integration'

    desc 'Clean'
    task :clean => ['destroy_vm', 'remove_berks'] do
    end

    desc 'Prepare'
    task :prepare => ['download_browsermob', 'start_vm'] do
    end

    task :download_browsermob do
      dir = File.join(integration_dir, 'vm/vendor/browsermob')
      unless Dir.exist?(dir)
        puts yellow("Downloading browsermob...")
        Dir.mktmpdir do |tmp|
          open('https://s3-us-west-1.amazonaws.com/lightbody-bmp/browsermob-proxy-2.0-beta-9-bin.zip') do |remote|
            open(File.join(tmp, 'browsermob-proxy.zip'), 'wb') do |f|
              f.write(remote.read)
            end
          end
          Dir.chdir(tmp) do
            system "unzip browsermob-proxy.zip"
          end
          FileUtils.mv(File.join(tmp, 'browsermob-proxy-2.0-beta-9'), dir)
        end
      end
    end

    task :berks_vendor do
      dir = File.join(integration_dir, 'vm/vendor/cookbooks')
      # Berkshelf
      if Dir.exist?(dir)
        puts yellow("'#{dir}' already exists. If you want update cookbooks, delete the directory and re-run rake command")
      else
        puts yellow('Installing cookbooks by berkshelf...')
        system "cd #{integration_dir}/vm && berks vendor vendor/cookbooks"
      end
    end

    task :start_vm => ['berks_vendor'] do
      puts yellow('Starting VM...')
      system '/usr/bin/vagrant up'
    end

    task :destroy_vm do
      puts yellow('Destroying VM...')
      system '/usr/bin/vagrant destroy -f'
    end

    task :remove_berks do
      dir = File.join(integration_dir, 'vm/vendor/cookbooks')
      FileUtils.rm_rf(dir)
    end
  end
end