require 'yaml'
require 'rubygems'

def base_dir
  File.expand_path(File.dirname(__FILE__))
end

$LOAD_PATH <<  File.join(base_dir, 'lib')

require 'puppetlabs/os_tester'

include Puppetlabs::OsTester

def github_password
  YAML.load_file(File.join(base_dir, '.github_auth'))['password']
end

def github_admins
  YAML.load_file(File.join(base_dir, '.github_auth'))['admins'].to_a
end

def github_login
  YAML.load_file(File.join(base_dir, '.github_auth'))['login']
end

def log_file
  return @log_file if @log_file
  log_dir  = File.join(base_dir, 'logs')
  log_file = File.join(log_dir, "#{Time.now.to_i.to_s}.log")
  FileUtils.mkdir(log_dir) unless File.exists?(log_dir)
  FileUtils.touch(log_file)
  @log_file = log_file
end

namespace :openstack do

  desc 'get testable pull requests'
  task :testable_pull_requests do
    testable_pull_requests(
      ['cinder', 'nova', 'glance', 'openstack', 'keystone', 'horizon'],
      github_admins,
      github_login,
      github_password,
      'ready_for_testing'
    )
  end

  desc 'clone all required modules'
  task :setup do
    cmd_system('librarian-puppet install')
  end

  desc 'destroy all vms'
  task 'destroy' do
    destroy_all_vms
  end

  desc 'destroy all swift vms'
  task 'destroy_swift' do
    destroy_swift_vms
  end

  desc 'deploys the entire environment'
  task :deploy_two_node do
    deploy_two_node
  end

  desc 'deploy a swift cluster'
  task :deploy_swift do
    deploy_swift_cluster
  end

end


namespace :git do

  cwd = base_dir

  desc 'for all repos in the module directory, add a read/write remote'
  task :dev_setup do
    dev_setup(github_login)
  end

  desc 'pull the latest version of all code'
  task :pull_all do
    pull_all
  end

  desc 'shows the current state of code that has not been commited'
  task :status_all do
    status_all
  end

  desc 'make sure that the current version from the module file matches the last tagged version'
  task :check_tags , [:project_name] do |t, args|
    # I need to be able to return this as a data structure
    # when I start to do more complicated things like
    # automated releases, I will need this data
    check_tags(args.project_name)
  end

  desc 'make sure that the current version from the module file matches the last tagged version'
  task :check_all_tags do
    check_tags
  end

  task :check_sha_all do
    each_repo do |module_name|
      print module_name + ':'
      puts git_cmd('rev-parse HEAD --quiet')
    end
  end

  desc 'prints the total number of people that have contributed to all projects.'
  task :num_contributors do
    puts contributor_hash.size
  end

  desc 'print the names of all contributors (and what projects they contributed to'
  task :list_contributors do
    contributor_hash.each do |k, v|
      puts "#{k}:#{v[:repos].inspect}"
    end
  end


  desc 'total number of contributions for each user across all projects'
  task :user_contributions do
    contrib = {}
    contributor_hash.each do |k,v|
      contrib[k] = v[:repos].values.inject(0) {|result, x| result + x }
    end
    contrib.sort_by {|name, num| num }.reverse.each {|x| puts "#{x[0]}:#{x[1]}" }
  end
end


namespace :github do

  desc 'pick a single pull request to test. Accepts the project name and number of PR to test'
    # you can also specify the OPERATINGSYSTEM to test as an ENV variable
  task :checkout_pull_request, [:project_name, :number] do |t, args|
    checkout_pr(
      args.project_name,
      args.number,
      [github_login] + github_admins,
      'schedule_for_testing',
      {
        :login    => github_login,
        :password => github_password
      }
    )
  end

end

namespace :test do

  desc 'run openstack puppet module unit tests'
  task :unit do
    command = "export MODULEPATH=#{base_dir}/modules;export GEM_HOME=#{base_dir}/.vendor;"
    status = ['cinder', 'nova', 'glance', 'openstack', 'keystone', 'horizon', 'swift'].collect do |proj|
      Dir.chdir("modules/#{proj}") do
        local_command = command + "bundle exec rake spec_standalone"
        puts local_command
        system(local_command)
      end
    end.uniq
    exit 1 if status != [true]
  end

  desc 'reset test environment'
  task :reset do
    refresh_modules
    destroy_all_vms
  end

  desc 'Checkout fresh master environment and test a two node deployment'
  task 'openstack_master' do
    refresh_modules
    system "bash -c 'rspec spec/test_two_node.rb;echo $?' 2>&1 | tee #{log_file}"
  end

  desc 'Checkout fresh master environment and test the deployment of a swift cluster'
  task 'swift_master' do
    refresh_modules
    system "bash -c 'rspec spec/test_swift_cluster.rb;echo $?' 2>&1 | tee #{log_file}"
  end

  desc 'checkout a PR and test swift'
  task 'swift_pull_request', [:repo_name, :pull_request_number] do |t, args|
    test_pull_request(
      args.repo_name,
      args.pull_request_number,
      github_admins,
      github_login,
      github_password,
      'spec/test_swift_cluster.rb',
      log_file,
      'schedule_for_testing'
    )
  end

  desc 'checkout and test a pull request, publish the results'
  task 'pull_request', [:repo_name, :pull_request_number] do |t, args|
    test_pull_request(
      args.repo_name,
      args.pull_request_number,
      github_admins,
      github_login,
      github_password,
      'spec/test_two_node.rb',
      log_file,
      'schedule_for_testing'
    )
  end

  desc 'test openstack with basic test script on redhat and ubuntu'
  task 'two_node' do
    test_two_node(['redhat', 'ubuntu'])
  end

  desc 'test swift cluster'
  task 'swift_proxy' do
    test_swift
  end

  desc 'test all in one deployment on redhat/ubuntu (not yet implemented)'
  task 'all_in_one' do

  end

  desc 'test that openstack can boot an image from the vagrant box'
  task :controller do
    on_box('openstack_controller', 'sudo bash /tmp/test_nova.sh;exit $?')
  end
end