bosh_cli 0.16

data/lib/cli/commands/misc.rb

@@ -0,0 +1,240 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Cli::Command
+  class Misc < Base
+    DEFAULT_STATUS_TIMEOUT = 3 # seconds
+
+    def version
+      say("BOSH %s" % [Bosh::Cli::VERSION])
+    end
+
+    def status
+      if config.target && options[:director_checks]
+        say("Updating director data...", " ")
+
+        begin
+          timeout(config.status_timeout || DEFAULT_STATUS_TIMEOUT) do
+            director = Bosh::Cli::Director.new(config.target)
+            status = director.get_status
+
+            config.target_name = status["name"]
+            config.target_version = status["version"]
+            config.target_uuid = status["uuid"]
+            config.save
+            say("done".green)
+          end
+        rescue TimeoutError
+          say("timed out".red)
+        rescue => e
+          say("error".red)
+        end
+        nl
+      end
+
+      target_name = full_target_name ? full_target_name.green : "not set".red
+      target_uuid = config.target_uuid ? config.target_uuid.green : "n/a".red
+      user = logged_in? ? username.green : "not set".red
+      deployment = config.deployment ? config.deployment.green : "not set".red
+
+      say("Target".ljust(15) + target_name)
+      say("UUID".ljust(15) + target_uuid)
+      say("User".ljust(15) + user)
+      say("Deployment".ljust(15) + deployment)
+
+      if in_release_dir?
+        header("You are in release directory")
+
+        dev_name = release.dev_name
+        dev_version = Bosh::Cli::VersionsIndex.new(
+            File.join(work_dir, "dev_releases")).latest_version
+
+        final_name = release.final_name
+        final_version = Bosh::Cli::VersionsIndex.new(
+            File.join(work_dir, "releases")).latest_version
+
+        say("Dev name:      %s" % [dev_name ? dev_name.green : "not set".red])
+        say("Dev version:   %s" % [dev_version && dev_version > 0 ?
+                                       dev_version.to_s.green : "
+                                       no versions yet".red])
+        say("\n")
+        say("Final name:    %s" % [final_name ?
+                                       final_name.green :
+                                       "not set".red])
+        say("Final version: %s" % [final_version && final_version > 0 ?
+                                       final_version.to_s.green :
+                                       "no versions yet".red])
+
+        say("\n")
+        say("Packages")
+        print_specs("package", "packages")
+
+        say("\n")
+        say("Jobs")
+        print_specs("job", "jobs")
+      end
+    end
+
+    def login(username = nil, password = nil)
+      target_required
+
+      unless options[:non_interactive]
+        username = ask("Your username: ").to_s if username.blank?
+
+        password_retries = 0
+        while password.blank? && password_retries < 3
+          password = ask("Enter password: ") { |q| q.echo = "*" }.to_s
+          password_retries += 1
+        end
+      end
+
+      if username.blank? || password.blank?
+        err("Please provide username and password")
+      end
+      logged_in = false
+
+      if options[:director_checks]
+        director = Bosh::Cli::Director.new(target, username, password)
+
+        if director.authenticated?
+          say("Logged in as '#{username}'")
+          logged_in = true
+        else
+          say("Cannot log in as '#{username}', please try again")
+          unless options[:non_interactive]
+            redirect(:misc, :login, username, nil)
+          end
+        end
+      end
+
+      if logged_in || !options[:director_checks]
+        config.set_credentials(target, username, password)
+        config.save
+      end
+    end
+
+    def logout
+      target_required
+      config.set_credentials(target, nil, nil)
+      config.save
+      say("You are no longer logged in to '#{target}'")
+    end
+
+    def purge_cache
+      if cache.cache_dir != Bosh::Cli::DEFAULT_CACHE_DIR
+        say("Cache directory '#{@cache.cache_dir}' differs from default, " +
+            "please remove manually")
+      else
+        FileUtils.rm_rf(cache.cache_dir)
+        say("Purged cache")
+      end
+    end
+
+    def show_target
+      say(target ?
+              "Current target is '#{full_target_name.green}'" :
+              "Target not set")
+    end
+
+    def set_target(director_url, name = nil)
+      if name.nil?
+        director_url = config.resolve_alias(:target, director_url) ||
+            director_url
+      end
+
+      if director_url.blank?
+        err("Target name cannot be blank")
+      end
+
+      director_url = normalize_url(director_url)
+      if director_url == target
+        say("Target already set to '#{full_target_name.green}'")
+        return
+      end
+
+      director = Bosh::Cli::Director.new(director_url)
+
+      if options[:director_checks]
+        begin
+          status = director.get_status
+        rescue Bosh::Cli::AuthError
+          status = {}
+        rescue Bosh::Cli::DirectorError
+          err("Cannot talk to director at '#{director_url}', " +
+              "please set correct target")
+        end
+      else
+        status = { "name" => "Unknown Director", "version" => "n/a" }
+      end
+
+      config.target = director_url
+      config.target_name = status["name"]
+      config.target_version = status["version"]
+      config.target_uuid = status["uuid"]
+
+      unless name.blank?
+        config.set_alias(:target, name, director_url)
+      end
+
+      unless status["uuid"].blank?
+        config.set_alias(:target, status["uuid"], director_url)
+      end
+
+      config.save
+      say("Target set to '#{full_target_name.green}'")
+
+      if interactive? && (config.username.blank? || config.password.blank?)
+        redirect(:misc, :login)
+      end
+    end
+
+    def set_alias(name, value)
+      config.set_alias(:cli, name, value.to_s.strip)
+      config.save
+      say("Alias `#{name.green}' created for command `#{value.green}'")
+    end
+
+    private
+
+    def print_specs(entity, dir)
+      specs = Dir[File.join(work_dir, dir, "*", "spec")]
+
+      if specs.empty?
+        say("No #{entity} specs found")
+      end
+
+      t = table ["Name", "Dev", "Final"]
+
+      specs.each do |spec_file|
+        if spec_file.is_a?(String) && File.file?(spec_file)
+          spec = load_yaml_file(spec_file)
+          name = spec["name"]
+
+          unless name.bosh_valid_id?
+            err("`#{name}' is an invalid #{entity} name, " +
+                "please fix before proceeding")
+          end
+
+          begin
+            dev_index   = Bosh::Cli::VersionsIndex.new(
+                File.join(work_dir, ".dev_builds", dir, name))
+            final_index = Bosh::Cli::VersionsIndex.new(
+                File.join(work_dir, ".final_builds", dir, name))
+
+            dev_version   = dev_index.latest_version || "n/a"
+            final_version = final_index.latest_version || "n/a"
+
+            t << [name, dev_version.gsub(/\-dev$/, "").rjust(8),
+                  final_version.to_s.rjust(8)]
+          rescue Bosh::Cli::InvalidIndex => e
+            say("Problem with #{entity} index for `#{name}': #{e.message}".red)
+          end
+        else
+          say("Spec file `#{spec_file}' is invalid")
+        end
+      end
+
+      say(t) unless t.rows.empty?
+    end
+
+  end
+end

data/lib/cli/commands/package.rb

@@ -0,0 +1,112 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Cli::Command
+  class Package < Base
+
+    def create_all
+      specs_glob = File.join(work_dir, "packages", "*", "spec")
+
+      Dir[specs_glob].each do |spec|
+        create(spec)
+      end
+    end
+
+    def create(name_or_path)
+      if name_or_path == "--all"
+        redirect(:package, :create_all)
+      end
+
+      spec = read_spec(name_or_path)
+
+      unless spec.is_a?(Hash) && spec.has_key?("name") && spec.has_key?("files")
+        err("Sorry, '#{name_or_path}' doesn't look like a valid package spec")
+      end
+
+      package_name = spec["name"]
+      header("Found '#{package_name}' spec")
+      print_spec(spec)
+      header("Building package...")
+
+      builder = Bosh::Cli::PackageBuilder.new(spec, work_dir,
+                                              false, release.blobstore)
+      builder.build
+      builder
+    end
+
+    def generate(name)
+      check_if_release_dir
+
+      unless name.bosh_valid_id?
+        err("`#{name}' is not a vaild BOSH id")
+      end
+
+      package_dir = File.join("packages", name)
+
+      if File.exists?(package_dir)
+        err("Package `#{name}' already exists, please pick another name")
+      end
+
+      say("create\t#{package_dir}")
+      FileUtils.mkdir_p(package_dir)
+
+      generate_file(package_dir, "packaging") do
+        "# abort script on any command that exit with a non zero value\nset -e\n"
+      end
+
+      generate_file(package_dir, "pre_packaging") do
+        "# abort script on any command that exit with a non zero value\nset -e\n"
+      end
+
+      generate_file(package_dir, "spec") do
+        "---\nname: #{name}\n\ndependencies:\n\nfiles:\n"
+      end
+
+      say("\nGenerated skeleton for `#{name}' package in `#{package_dir}'")
+    end
+
+    private
+
+    def generate_file(dir, file)
+      path = File.join(dir, file)
+      say("create\t#{path}")
+      FileUtils.touch(path)
+      File.open(path, "w") do |f|
+        f.write(yield)
+      end
+    end
+
+    def print_spec(spec)
+      say("Package name: #{spec["name"]}")
+      say("Files:")
+      for file in spec["files"]
+        say("  - #{file}")
+      end
+    end
+
+    def read_spec(name)
+      load_yaml_file(find_spec(name))
+    end
+
+    def find_spec(name)
+      if File.directory?(name)
+        spec_path = File.join(name, "spec")
+        if File.exists?(spec_path)
+          spec_path
+        else
+          err("Cannot find spec file in '#{name}' directory")
+        end
+      elsif File.file?(name)
+        name
+      else
+        package_dir = File.join(work_dir, "packages", name)
+        if File.directory?(package_dir)
+          find_spec(package_dir)
+        else
+          err("Cannot find package '#{name}' (tried '#{package_dir}')")
+        end
+
+      end
+    end
+
+  end
+end

data/lib/cli/commands/property_management.rb

@@ -0,0 +1,125 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Cli::Command
+  class PropertyManagement < Base
+    include Bosh::Cli::DeploymentHelper
+
+    def set(name, value)
+      prepare
+      show_header
+
+      status, body = director.get_property(@deployment_name, name)
+      existing_property = status == 200
+
+      if existing_property
+        say("Current `#{name.green}' value is " +
+            "`#{format_property(body["value"]).green}'")
+      else
+        say("This will be a new property")
+      end
+
+      prompt = "Are you sure you want to set property" +
+          " `#{name.green}' to `#{format_property(value).green}'? " +
+          "(type yes to proceed): "
+
+      if interactive? && ask(prompt) != "yes"
+        err("Canceled")
+      end
+
+      if existing_property
+        status, body = director.update_property(@deployment_name, name, value)
+      else
+        status, body = director.create_property(@deployment_name, name, value)
+      end
+
+      if status == 204
+        say("Property `#{name.green}' set to `#{value.green}'")
+      else
+        err(director.parse_error_message(status, body))
+      end
+    end
+
+    def unset(name)
+      prepare
+      show_header
+
+      prompt = "Are you sure you want to unset property " +
+          "`#{name.green}'? (type yes to proceed): "
+
+      if interactive? && ask(prompt) != "yes"
+        err("Canceled")
+      end
+
+      status, body = director.delete_property(@deployment_name, name)
+
+      if status == 204
+        say("Property `#{name.green}' has been unset")
+      else
+        err(director.parse_error_message(status, body))
+      end
+    end
+
+    def get(name)
+      prepare
+      show_header
+
+      status, body = director.get_property(@deployment_name, name)
+      if status == 200
+        say("Property `#{name.green}' value is " +
+            "`#{format_property(body["value"]).green}'")
+      else
+        err(director.parse_error_message(status, body))
+      end
+    end
+
+    def list(*args)
+      prepare
+      terse = args.include?("--terse")
+      show_header unless terse
+
+      properties = director.list_properties(@deployment_name)
+      unless properties.kind_of?(Enumerable)
+        err("Invalid properties format, please check your director")
+      end
+
+      output = properties.sort do |p1, p2|
+        p1["name"] <=> p2["name"]
+      end.map do |property|
+        [property["name"], format_property(property["value"])]
+      end
+
+      if terse
+        output.each { |row| say("#{row[0]}\t#{row[1]}") }
+      else
+        if output.size > 0
+          properties_table = table do |t|
+            t.headings = ["Name", "Value"]
+            output.each { |row| t << [row[0], row[1].truncate(40)] }
+          end
+          say(properties_table)
+        else
+          say("No properties found")
+        end
+      end
+    end
+
+    private
+
+    def prepare
+      auth_required
+      manifest = prepare_deployment_manifest
+      @deployment_name = manifest["name"]
+    end
+
+    def show_header
+      say("Target #{target_name.green}")
+      say("Deployment #{@deployment_name.green}")
+      nl
+    end
+
+    def format_property(value)
+      value.gsub("\n", "\\n")
+    end
+
+  end
+end