path: root/everest-bootstrap/bin/everest-bootstrap

#!/usr/bin/env ruby
require 'ostruct'
require 'everest-bootstrap'
require 'main'
require 'net/http'
require 'uri'
require "highline/import"
#http://www.mail-archive.com/capistrano@googlegroups.com/msg01822.html
HighLine.track_eof = false

include EverestBootstrap

Main {
  version Version::STRING

  def handle_exception e
    puts e.message
  end unless $DEBUG

  def say_red(text)
    say("<%= color(\"#{text}\", :red) %>") 
  end

  def get_input
    unless ENV['USER'] == 'root' || params["config-only"].given?
      raise "You must run this command as root unless " +
            "you specify the '--config-only' option" 
    end

    @facts = Hash.new

    @repo = ask("Enter your Everest repo machine name (leave off the '-repo'): ")
    @user = ask("Enter your kerberos username: ")

    say_red("Your machine name is the part right before the Everest type.")
    say_red("For example, 'sso1-build.usersys.redhat.com' has a machine name of 'sso1'.")
    say_red("If you do not specify a machine name it will default to your kerberos name.")

    if agree("Enter machine name? (y/n)", true)
      @machine_name = ask("Machine name: ")
    else 
      @machine_name = @user
    end

    # Once we know the repo we can fetch the supported machines
    @everest_repo = EverestRepo.new(@repo)
    say_red("To see a description of these machine types visit: " +
        "#{@everest_repo.machine_types_url}")

    @machine_type = choose do |menu|
      menu.prompt = "Select your machine type:  "
      menu.choices(*@everest_repo.machines.map{|m| m.name})
    end

    # This might be useful is we stop using DDNS
    @facts["everest_machine_type"] = @machine_type

    if agree("Configure this machine with Red Hat DDNS? (y/n)", true)
      # Our naming convention
      hostname = @machine_name + "-" + @machine_type
      @fqdn = hostname + ".usersys.redhat.com"

      if agree("Do you know the DDNS hash for this machine? (y/n)", true)
        @facts["rh_ddns_hash"] = ask("Enter Red Hat DDNS hash: ") 
      else 
        @password = ask("Enter your kerberos password to setup DDNS: ") { |q| q.echo = "*" }
        @facts["rh_ddns_hash"] = RedHatDDNS::DDNS.new(@user, @password, hostname).ddns_hash
      end
    else 
      @fqdn = ask("Enter fully qualified domain name for this machine: ")
    end

    @facts["cobbler_profile"] = choose do |menu|
      menu.prompt = "Select your cobbler profile (This determines the OS):  "
      menu.choices(*@everest_repo.cobbler_profiles)
    end

    say_red("Now you need to enter some parameters what will be used to configure this machine.")
    say_red("Most of the defaults will work out fine for development.")

    @everest_repo.facts_for(@machine_type, @machine_name).each do |f|
      say_red("Description: #{f.desc}") unless f.desc.empty?
      @facts[f.name] = ask("Enter value for #{f.name}: ") do |q|
        q.default = f.default
      end
    end

    if agree("Enter addition facts? (y/n)", true)
      begin
        name = ask("Enter fact name: ")
        value = ask("Enter fact value: ") 
        @facts[name] = value
      end while agree("Enter another? ", true)
    end

    unless params["config-only"].given?
      if agree("Use LVM for VM storage (recommended)? (y/n)", true)
        @virt_path = choose do |menu|
          menu.prompt = "Select the Volume Group to use: "
          menu.choices(*(`/usr/sbin/vgdisplay --short`.split("\n").map {|l| /"(.*)"/.match(l).captures[0]}))
        end
      else 
        @virt_path = ask("Enter the flat file to be used for guest: ") 
      end

      xen_conf = "/etc/xen/#{@fqdn.split(".")[0]}"
      if File.exist?(xen_conf)
        say_red("Previous Xen configuration detected at #{xen_conf}.")
        say_red("This will cause Koan to abort.")
        if agree("Clean up? (y/n)", true)
          File.delete(xen_conf)
        end
      end

      say_red("The Koan process is now starting.  After it finishes start the VM back up with " + 
          "'xm create <name>'.")
      say_red("HINT: 'xm list' will show you the running VMs.")
    end
  end

  option('config-only', 'c'){
    description "Only send the machine configuration to the Puppetmaster"
  }

  def run
    # Ask all the silly questions
    get_input

    classes = @everest_repo.classes_for(@machine_type)
    config = {"classes" => classes, "parameters" => @facts}

    e = EverestMachine.new(@machine_type,
                           @fqdn,
                           @repo,
                           config)
    e.post_config 
    e.add_system_to_cobbler
    e.koan(@virt_path) unless params["config-only"].given?   
  end

  mode 'advanced' do
    option('fqdn', 'f'){
      required
      description "Fully qualified domain name of machine to be provisioned"
      argument_required
    }

    option('repo', '-r'){
      required
      description "Fully qualified domain name for the Everest repo machine to use for provisioning"
      argument_required
    }

    option('yaml', 'y'){
      description "YAML configuration for this machine"
      argument_required
      validate {|f| File.exist?(f)}
    }

    option('virt-path', 'v'){
      description "Volume group or flat file for koan provisioning"
      argument_required
    }

    def run
      config = if params['yaml'].given? 
        YAML.load(File.read(params['yaml'].value))
      else
        YAML.load($stdin.read)
      end
  
      e = EverestMachine.new(config["parameters"]["everest_machine_type"],
                             params['fqdn'].value,
                             params['repo'].value,
                             config)
      e.post_config 
      e.add_system_to_cobbler
      e.koan(params['virt-path'].value) if params["virt-path"].given?   
    end
  end

  mode 'cloud' do
    def collect_input
      unless ENV['USER'] == 'root' || params["config-only"].given?
        raise "You must run this command as root unless " +
              "you specify the '--config-only' option" 
      end
  
      @facts = Hash.new
  
      @repo = ask("Enter your Everest repo machine name (leave off the '-repo'): ")
      @cloud = ask("Enter your cloud master (leave off the '-repo'): ")
      @user = ask("Enter your kerberos username: ")
  
      say_red("Your machine name is the part right before the Everest type.")
      say_red("For example, 'sso1-build.usersys.redhat.com' has a machine name of 'sso1'.")
      say_red("If you do not specify a machine name it will default to your kerberos name.")
  
      if agree("Enter machine name? (y/n)", true)
        @machine_name = ask("Machine name: ")
      else 
        @machine_name = @user
      end
  
      # Once we know the repo we can fetch the supported machines
      @everest_repo = EverestRepo.new(@repo)
      say_red("To see a description of these machine types visit: " +
          "#{@everest_repo.machine_types_url}")
  
      @machine_type = choose do |menu|
        menu.prompt = "Select your machine type:  "
        menu.choices(*@everest_repo.machines.map{|m| m.name})
      end
  
      # This might be useful is we stop using DDNS
      @facts["everest_machine_type"] = @machine_type
  
      if agree("Configure this machine with Red Hat DDNS? (y/n)", true)
        # Our naming convention
        hostname = @machine_name + "-" + @machine_type
        @fqdn = hostname + ".usersys.redhat.com"
  
        if agree("Do you know the DDNS hash for this machine? (y/n)", true)
          @facts["rh_ddns_hash"] = ask("Enter Red Hat DDNS hash: ") 
        else 
          @password = ask("Enter your kerberos password to setup DDNS: ") { |q| q.echo = "*" }
          @facts["rh_ddns_hash"] = RedHatDDNS::DDNS.new(@user, @password, hostname).ddns_hash
        end
      else 
        @fqdn = ask("Enter fully qualified domain name for this machine: ")
      end
  
      @facts["cobbler_profile"] = choose do |menu|
        menu.prompt = "Select your cobbler profile (This determines the OS):  "
        menu.choices(*@everest_repo.cobbler_profiles)
      end
  
      say_red("Now you need to enter some parameters what will be used to configure this machine.")
      say_red("Most of the defaults will work out fine for development.")
  
      @everest_repo.facts_for(@machine_type, @machine_name).each do |f|
        say_red("Description: #{f.desc}") unless f.desc.empty?
        @facts[f.name] = ask("Enter value for #{f.name}: ") do |q|
          q.default = f.default
        end
      end
  
      if agree("Enter addition facts? (y/n)", true)
        begin
          name = ask("Enter fact name: ")
          value = ask("Enter fact value: ") 
          @facts[name] = value
        end while agree("Enter another? ", true)
      end
  
      say_red("The Koan process will start on a machine in the cloud.")
    end

    def run
      collect_input

      classes = @everest_repo.classes_for(@machine_type)
      config = {"classes" => classes, "parameters" => @facts}

      @everest_repo.register_machine(@fqdn, config)
      cloud_repo = CloudRepo.new
      cloud_repo.create_machine(@fqdn, @cloud)
    end
  end

  mode 'cloud-advanced' do
    option('fqdn', 'f'){
      required
      description "Fully qualified domain name of machine to be provisioned"
      argument_required
    }

    option('repo', '-r'){
      required
      description "Fully qualified domain name for the Everest repo machine to use for provisioning"
      argument_required
    }

    option('cloud-controller', '-c'){
      description "Fully qualified domain name for the Everest repo machine to use for provisioning"
      argument_required
    }
    
    option('yaml', 'y'){
      description "YAML configuration for this machine"
      argument_required
      validate {|f| File.exist?(f)}
    }

    def run 
      config = if params['yaml'].given? 
        YAML.load(File.read(params['yaml'].value))
      else
        YAML.load($stdin.read)
      end

      cloud_controller = if params['cloud-controller'].given? 
        params['cloud-controller'].value
      else
        "everest-repo.usersys.redaht.com"
      end

      everest_repo = EverestRepo.new(params['repo'].value)
      everest_repo.register_machine(params['fqdn'].value, config)
      cloud_repo = CloudRepo.new(everest_repo.repo)
      cloud_repo.create_machine(params['fqdn'].value, everest_repo.repo)
    end

  end
}