# requirements:
# Check kvm installed
# check kvm fullvirt available
# check if qemu-img is installed
# check distribution supported
#
if ARGV[0] == 'instant-deploy'
  ARGV.shift

  #
  # HTTPDownloader code based on code from http://www.vagrantup.com
  #
  require 'net/http'
  require 'net/https'
  require 'open-uri'
  require 'uri'
  require 'fileutils'

  class SystemCommands

    def self.kvm=(path)
      @@kvm = path
    end

    def self.kvm
      @@kvm
    end
    
  end

  class HTTPDownloader
    def self.match?(uri)
      # URI.parse barfs on '<drive letter>:\\files \on\ windows'
      extracted = URI.extract(uri).first
      extracted && extracted.include?(uri)
    end

    def report_progress(progress, total, show_parts=true)
      line_reset = "\r\e[0K" 
      percent = (progress.to_f / total.to_f) * 100
      line = "Progress: #{percent.to_i}%"
      line << " (#{progress} / #{total})" if show_parts
      line = "#{line_reset}#{line}"
      $stdout.sync = true
      $stdout.print line
    end

    def download!(source_url, destination_file)
      proxy_uri = URI.parse(ENV["http_proxy"] || "")
      uri = URI.parse(source_url)
      http = Net::HTTP.new(uri.host, uri.port, proxy_uri.host, proxy_uri.port, proxy_uri.user, proxy_uri.password)

      if uri.scheme == "https"
        http.use_ssl = true
        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
      end

      http.start do |h|
        h.request_get(uri.request_uri) do |response|
          total = response.content_length
          progress = 0
          segment_count = 0

          response.read_body do |segment|
            # Report the progress out
            progress += segment.length
            segment_count += 1

            # Progress reporting is limited to every 25 segments just so
            # we're not constantly updating
            if segment_count % 25 == 0
              report_progress(progress, total)
              segment_count = 0
            end


            # Store the segment
            destination_file.write(segment)
          end
        end
      end
    rescue SocketError
      raise Errors::DownloaderHTTPSocketError.new
    end
  end
  class InstantDeployCLI
    include Mixlib::CLI

    option :target_dir,
      :long => '--target-dir NAME',
      :description => 'Directory where the VM disk and ISO are created',
      :default => "abiquo-instant-deploy-#{Time.now.strftime "%s"}"
    
    option :vm_name,
      :long => '--vm-name NAME',
      :description => 'Virtual Machine name',
      :default => "instant-deploy"

    option :iso_url,
      :long => '--iso-url URL',
      :description => 'Abiquo ISO URL'

    option :mem,
      :long => '--mem MEM',
      :description => 'Virtual Machine memory (in bytes)',
      :default => '512'

    option :ssh_port,
      :long => '--ssh-port PORT',
      :description => 'Forwarded SSH port (Default 2300)',
      :default => '2300'

    option :tomcat_port,
      :long => '--tomcat-port PORT',
      :description => 'Forwarded Tomcat port (Default 8980)',
      :default => '8980'

    option :vnc,
      :long => '--vnc',
      :description => 'Use VNC instead of SDL/Graphical display',
      :default => false

    option :vnc_display,
      :long => '--vnc-display DISP',
      :description => 'Use VNC display DISP',
      :default => 0
    
    option :help,
      :short => "-h",
      :long => "--help",
      :description => ".\n\n",
      :on => :tail,
      :boolean => true,
      :show_options => true,
      :exit => 0
  end

  def pfcheck_ubuntu
    if (File.read('/etc/lsb-release') !~ /DISTRIB_ID=Ubuntu/)
      $stderr.puts "\nUbuntu not found. Your distribution is not supported.\n\n"
      exit
    end
    
    #
    # Check if KVM installed
    #
    if `which /usr/bin/kvm`.strip.chomp.empty?
      $stderr.puts "\nKVM not found. Install it first:\n\n"
      $stderr.puts "sudo apt-get install kvm\n\n"
      exit
    end
    
    SystemCommands.kvm = '/usr/bin/kvm'
  end

  def pfcheck_redhat
    if File.read('/etc/redhat-release') !~ /^(FrameOS|CentOS)/
      $stderr.puts "\nRHEL not found. Your distribution is not supported.\n\n"
      exit
    end
    
    #
    # Check if KVM installed
    #
    if `which /usr/libexec/qemu-kvm`.strip.chomp.empty?
      $stderr.puts "\nKVM not found. Install it first:\n\n"
      $stderr.puts "yum install kvm\n\n"
      exit
    end
    
    SystemCommands.kvm = '/usr/libexec/qemu-kvm'
  end

  def preflight_check
    #
    # Check if this is Ubuntu
    #
    if File.exist?('/etc/lsb-release')
      pfcheck_ubuntu
    elsif File.exist?('/etc/redhat-release')
      pfcheck_redhat
    else
      $stderr.puts "\nOnly Ubuntu and RHEL distributions are supported.\n\n"
      exit
    end

    #
    # Check if qemu-img installed
    #
    if `which /usr/bin/qemu-img`.strip.chomp.empty?
      $stderr.puts "\nqemu-img not found. Install it first:\n\n"
      $stderr.puts "sudo apt-get install kvm\n\n"
      exit
    end
  end

  def install_iso(params = {})
    target_dir = params[:target_dir] || "abiquo-instant-deploy-#{Time.now.strftime "%s"}"
    disk_file = params[:disk_file] || "#{target_dir}/abiquo.qcow2"
    iso_url = params[:iso_url]
    mem = params[:mem]
    tomcat_port = params[:tomcat_port]
    ssh_port = params[:ssh_port]
    graphics = ''
    vm_name = params[:vm_name]

    if params[:vnc]
      graphics = "--vnc :#{params[:vnc_display]}"
    end
    # Create target directory
    begin
      FileUtils.mkdir(target_dir)
    rescue Exception
      $stderr.puts "\nError creating directory #{target_dir}. Aborting.\n\n"
      exit 1
    end

    # Create qemu img
    if File.exist? disk_file 
      raise Exception.new("Image #{disk_file} already exists")
    end
    `/usr/bin/qemu-img create -f qcow2 #{disk_file} 20GB`
  

    if iso_url =~ /http:\/\//
      # Download the iso
      downloader = HTTPDownloader.new
      puts "Downloading Abiquo ISO..."
      cdrom = ""
      begin
        r = downloader.download! iso_url, File.new(target_dir + '/instant-deploy.iso', 'w')
        if r.class != Net::HTTPOK
          raise Exception
        end
        cdrom = target_dir + '/instant-deploy.iso'
      rescue Exception
        $stderr.puts "\nError downloading Abiquo ISO. Aborting."
        exit
      end
    else
      if iso_url =~ /file:/
        iso_url.gsub!('file://','')
      end
      if not File.exist? iso_url
        $stderr.puts "The ISO file specified does not exist."
        exit 1
      else
        cdrom = iso_url
      end
    end

    # Boot
    puts "\nAfter the install process, open the browser and type:\n"
    puts "\nhttp://127.0.0.1:#{tomcat_port}/client-premium\n\n"
    puts "To open the Abiquo Web Console."
    puts "\nTo SSH to the Abiquo VM type:"
    puts "ssh -p #{ssh_port} localhost\n\n"
    puts "\nBooting the Installer...\n\n"
    File.open(target_dir + '/run.sh', 'w') do |f|
      f.puts "#!/bin/sh"
      f.puts "MEM=#{mem}"
      f.puts "TAP=vtap0"
      f.puts ""
      f.puts "#{SystemCommands.kvm} -name #{vm_name} #{graphics} -m $MEM -drive file=#{File.basename(disk_file)} -net user,hostfwd=tcp:0.0.0.0:#{tomcat_port}-:80,hostfwd=tcp:0.0.0.0:#{ssh_port}-:22 -net nic -boot order=c > /dev/null"
      f.puts ""
      f.puts "#"
      f.puts "# Comment the above line and uncomment this to use bridged networking."
      f.puts "# You will need to have a working bridge setup in order to use this."
      f.puts "# Update TAP variable above to fill your needs."
      f.puts "#"
      f.puts "#sudo #{SystemCommands.kvm} -name #{vm_name} #{graphics} -m $MEM -drive file=#{File.basename(disk_file)} -net tap,ifname=$TAP -net nic -boot order=c > /dev/null 2>&1"
    end
    output = `#{SystemCommands.kvm} -name #{vm_name} #{graphics} -m 1024 -drive file=#{disk_file} -net user,hostfwd=tcp:0.0.0.0:#{tomcat_port}-:80,hostfwd=tcp:0.0.0.0:#{ssh_port}-:22 -net nic -drive file=#{cdrom},media=cdrom -boot order=cd -boot once=d 2>&1 `
    if $? != 0
      puts "Error booting the VM: #{output}"
    end
  end

  def distribution_version
    /DISTRIB_DESCRIPTION="(.*)"/.match File.read('/etc/lsb-release')
    version = $1.splitp[1] || '0'
    version
  end

  trap("INT") { puts "\n\nCleaning Environment..."; exit } 

  preflight_check

  cli = InstantDeployCLI.new
  cli.parse_options
  target_dir = cli.config[:target_dir]
  url = cli.config[:iso_url]
  if url.nil?
    $stderr.puts "\n--iso-url argument is mandatory.\n\n"
    puts cli.opt_parser.help
    exit
  end
  puts "\n"
  print "Abiquo Instant Deploy: ".bold
  puts "One Command Cloud Builder\n\n"
  puts "Building the cloud into #{target_dir.bold} directory..."
  install_iso(:target_dir => target_dir, :iso_url => url, :mem => cli.config[:mem],
              :tomcat_port => cli.config[:tomcat_port],
              :vm_name => cli.config[:vm_name],
              :ssh_port => cli.config[:ssh_port],
              :vnc => cli.config[:vnc],
              :vnc_display => cli.config[:vnc_display]
             )

end