Class: CreateVMCommand

Inherits:

BaseCommand

• Object
• Clamp::Command
• BaseCommand
• CreateVMCommand

Defined in:

bin/esx

Instance Method Summary

• #download!(source_url, destination_file) ⇒ Object
• #execute ⇒ Object
• #report_progress(progress, total, show_parts = true) ⇒ Object

Instance Method Details

#download!(source_url, destination_file) ⇒ Object

# File 'bin/esx', line 376

def download!(source_url, destination_file)
  dst = File.open(destination_file, 'w')
  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

  res = 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
        dst.write(segment)
      end
    end
  end

  raise Exception.new("HTTP Error #{res}") if res.class != Net::HTTPOK
ensure
  dst.close
end

#execute ⇒ Object

# File 'bin/esx', line 299

def execute
  begin
    host = ESX::Host.connect address, user, password, true, {:free_license=>free_license?}
  rescue Exception => e
    $stderr.puts "Can't connect to the host #{address}."
    if debug?
      $stderr.puts e.message
    end
    exit 1
  end
  if disk_size and disk_file
    $stderr.puts "Both --disk-file and --disk-size specified. --disk-size will be ignored."
  end
  downloaded_file = nil
  if disk_file.nil?
    # if --disk-file nil? create the VM without disk
    vm = host.create_vm :vm_name => name,
                        :datastore => datastore, :disk_type => :flat, :memory => memory,
                        :disk_size => disk_size,
                        :guest_id => guest_id, :nics => [{:mac_address => mac_address, :network => vm_network}]
  else
    df = disk_file.dup
    if df.strip.chomp =~ /^http/
      begin
        downloaded_file = disk_file.dup
        tmpfile = "#{tmpdir}/#{Time.now.to_i}.esx"
        puts "Downloading file... (#{tmpfile})"
        download! downloaded_file, tmpfile
        puts
        df = tmpfile
      rescue Exception => e
        FileUtils.rm_f(tmpfile)
        $stderr.puts "Error downloading file from #{downloaded_file}."
        $stderr.puts e.message if debug?
        exit 1
      end
    end
    raise Exception.new("Invalid disk file") if not File.exist?(df)
    if not name
      $stderr.puts "Invalid VM name."
      $stderr.puts "Use --name option to specify the VM name"
      exit 1
    end
    host.remote_command "mkdir /vmfs/volumes/#{datastore}/#{name}"

    begin
      host.import_disk df, "/vmfs/volumes/#{datastore}/#{name}/#{name}.vmdk"
    rescue Exception => e
      $stderr.puts "Error uploading file to /vmfs/volumes/#{datastore}/#{name}/#{name}.vmdk"
      $stderr.puts e.message if debug?
      exit 1
    end

    if not downloaded_file.nil?
      puts "Deleting tmp file #{df}" if debug?
      FileUtils.rm_f(df)
    end
    vm = host.create_vm :vm_name => name,
                        :disk_file => "#{name}/#{name}.vmdk",
                        :datastore => datastore, :disk_type => :flat, :memory => memory,
                        :guest_id => guest_id, :nics => [{mac_address => mac_address, :network => vm_network}]
  end
  if poweron?
    vm.power_on
  end
end

#report_progress(progress, total, show_parts = true) ⇒ Object

# File 'bin/esx', line 366

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