require 'date' require 'rake/testtask' require 'tmpdir'

namespace :bootdisk do

def outputdir
  if ENV['OUTPUT']
    File.directory?(ENV['OUTPUT']) ? ENV['OUTPUT'] : File.dirname(ENV['OUTPUT'])
  elsif File.writable?(Dir.pwd)
    Dir.pwd
  else
    Dir.tmpdir
  end
end

namespace :generate do
  desc 'Generate a static boot disk for a specific host.  NAME=fqdn, OUTPUT path'
  task host: :environment do
    User.as_anonymous_admin do
      host = Host::Base.unscoped.find_by(name: ENV['NAME']) || raise("cannot find host '#{ENV['NAME']}', specify NAME=fqdn")
      tmpl = host.bootdisk_template_render

      ForemanBootdisk::ISOGenerator.generate(ipxe: tmpl, dir: outputdir) do |image|
        output = ENV['OUTPUT'] || File.join(outputdir, "#{host.name}.iso")
        FileUtils.mv image, output
        puts "Wrote #{output}"
      end
    end
  end

  desc 'Generate a full boot disk for a specific host with the OS bootloader included.  NAME=fqdn, OUTPUT path'
  task full_host: :environment do
    User.as_anonymous_admin do
      host = Host::Base.unscoped.find_by(name: ENV['NAME']) || raise("cannot find host '#{ENV['NAME']}', specify NAME=fqdn")
      ForemanBootdisk::ISOGenerator.generate_full_host(host, dir: outputdir) do |image|
        output = ENV['OUTPUT'] || File.join(outputdir, "#{host.name}_#{Date.today.strftime('%Y%m%d')}.iso")
        FileUtils.cp image, output
        puts "Wrote #{output}"
      end
    end
  end

  desc 'Generate a generic boot disk.  OUTPUT=path'
  task generic: :environment do
    User.as_anonymous_admin do
      tmpl = ForemanBootdisk::Renderer.new.generic_template_render

      ForemanBootdisk::ISOGenerator.generate(ipxe: tmpl, dir: outputdir) do |image|
        output = ENV['OUTPUT'] || File.join(outputdir, "bootdisk_#{URI.parse(Setting[:foreman_url]).host}.iso")
        FileUtils.cp image, output
        puts "Wrote #{output}"
      end
    end
  end

  desc 'Generate a subnet disk for a specific subnet. NAME=subnet, OUTPUT=path'
  task subnet: :environment do
    User.as_anonymous_admin do
      subnet = Subnet.unscoped.find_by(name: ENV['NAME']) || raise("cannot find subnet '#{ENV['NAME']}', specify NAME=subnet")
      subnet.tftp || raise(::Foreman::Exception.new(N_('TFTP feature not enabled for subnet %s'), subnet.name))
      tmpl = ForemanBootdisk::Renderer.new.generic_template_render(subnet)
      ForemanBootdisk::ISOGenerator.generate(ipxe: tmpl, dir: outputdir) do |image|
        output = ENV['OUTPUT'] || File.join(outputdir, "bootdisk_subnet_#{subnet.name}.iso")
        FileUtils.cp image, output
        puts "Wrote #{output}"
      end
    end
  end
end

end

# Tests namespace :test do

desc 'Test foreman_bootdisk'
Rake::TestTask.new(:foreman_bootdisk) do |t|
  test_dir = File.join(File.dirname(__FILE__), '../..', 'test')
  t.libs << ['test', test_dir]
  t.pattern = "#{test_dir}     _test.rb"
  t.verbose = true
  t.warning = false
end

end

Rake::Task.enhance ['test:foreman_bootdisk']

namespace :foreman_bootdisk do

task :rubocop do
  begin
    require 'rubocop/rake_task'
    RuboCop::RakeTask.new(:rubocop_foreman_bootdisk) do |task|
      task.patterns = ["#{ForemanBootdisk::Engine.root}/app/   *.rb",
                       "#{ForemanBootdisk::Engine.root}/lib/   *.rb",
                       "#{ForemanBootdisk::Engine.root}/test/**/*.rb"]
    end
  rescue StandardError
    puts 'Rubocop not loaded.'
  end

  Rake::Task['rubocop_foreman_bootdisk'].invoke
end

end

load 'tasks/jenkins.rake' Rake::Task.enhance ['test:foreman_bootdisk', 'foreman_bootdisk:rubocop'] if Rake::Task.task_defined?(:'jenkins:unit')