tddium 1.25.5

data/lib/tddium/cli/params_helper.rb

@@ -0,0 +1,36 @@
+# Copyright (c) 2014 Solano Labs All Rights Reserved
+
+module ParamsHelper
+  include TddiumConstant
+  
+  def load_params(defaults=true)
+    params = {}
+    if File.exists?(Default::PARAMS_PATH) then
+      File.open(Default::PARAMS_PATH, 'r') do |file|
+        params = JSON.parse file.read
+      end
+    elsif !defaults then
+      abort Text::Process::NOT_SAVED_OPTIONS
+    end
+    return params
+  end
+
+  def write_params options
+    begin
+      File.open(Default::PARAMS_PATH, File::CREAT|File::TRUNC|File::RDWR, 0600) do |file|
+        file.write options.to_json
+      end
+      say Text::Process::OPTIONS_SAVED
+    rescue Exception => e
+      say Text::Error::OPTIONS_NOT_SAVED
+    end
+  end
+
+  def display
+    store_params = load_params(false)
+    say 'Options:'
+    store_params.each do |k, v|
+      say "   #{k.capitalize}:\t#{v}"
+    end
+  end
+end

data/lib/tddium/cli/prompt.rb

@@ -0,0 +1,128 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    protected
+
+    def prompt(text, current_value, default_value, dont_prompt=false)
+      value = current_value || (dont_prompt ? nil : ask(text % default_value, :bold))
+      (value.nil? || value.empty?) ? default_value : value
+    end
+
+    def prompt_missing_ssh_key
+      keys = @tddium_api.get_keys
+      if keys.empty? then
+        say Text::Process::SSH_KEY_NEEDED
+        keydata = prompt_ssh_key(nil)
+        result = @tddium_api.set_keys({:keys => [keydata]})
+        return true
+      end
+      false
+    rescue TddiumError => e
+      exit_failure e.message
+    rescue TddiumClient::Error::API => e
+      exit_failure e.explanation
+    end
+
+    def prompt_ssh_key(current, name='default')
+      # Prompt for ssh-key file
+      ssh_file = prompt(Text::Prompt::SSH_KEY, current, Default::SSH_FILE)
+      Tddium::Ssh.load_ssh_key(ssh_file, name)
+    end
+
+    def prompt_suite_params(options, params, current={})
+      say Text::Process::DETECTED_BRANCH % params[:branch] if params[:branch]
+      params[:ruby_version] ||= tool_version(:ruby)
+      params[:bundler_version] ||= normalize_bundler_version(tool_version(:bundle))
+      params[:rubygems_version] ||= tool_version(:gem)
+
+      ask_or_update = lambda do |key, text, default|
+        params[key] = prompt(text, options[key], current.fetch(key.to_s, default), options[:non_interactive])
+      end
+
+      account_announced = false
+
+      # If we already have a suite, it already has an account, so no need to
+      # figure it out here.
+      unless current['account_id']
+        # Find an account id. Strategy:
+        # 1. Use a command line option, if specified.
+        # 2. If the user has only one account, use that.
+        # 3. If the user has existing suites with the same repo, and they are
+        # all in the same account, prompt with that as a default.
+        # 4. Prompt.
+        # IF we're not allowed to prompt and have no default, fail.
+        accounts = user_details["participating_accounts"]
+        account_name = if options[:account]
+          account_announced = true
+          say Text::Process::USING_ACCOUNT_FROM_FLAG % options[:account]
+          options[:account]
+        elsif accounts.length == 1
+          account_announced = true
+          say Text::Process::USING_ACCOUNT % accounts.first["account"]
+          accounts.first["account"]
+        else
+          # Get all of this user's suites with this repo.
+          repo_suites = @tddium_api.get_suites(:repo_url => params[:repo_url])
+          acct_ids = repo_suites.map{|s| s['account']}.uniq
+          default = acct_ids.length == 1 ? acct_ids.first : nil
+
+          if not options[:non_interactive] or default.nil?
+            say "You are a member of these organizations:"
+            accounts.each do |account|
+              say "  " + account['account']
+            end
+          end
+
+          msg = default.nil? ? Text::Prompt::ACCOUNT : Text::Prompt::ACCOUNT_DEFAULT
+          prompt(msg, nil, default, options[:non_interactive])
+        end
+
+        if account_name.nil?
+          exit_failure (options[:non_interactive] ?
+                        Text::Error::MISSING_ACCOUNT_OPTION :
+                        Text::Error::MISSING_ACCOUNT)
+        end
+        account = accounts.select{|a| a['account'] == account_name}.first
+        if account.nil?
+          exit_failure Text::Error::NOT_IN_ACCOUNT % account_name
+        end
+
+        if !account_announced then
+          say Text::Process::USING_ACCOUNT % account_name
+        end
+        params[:account_id] = account["account_id"].to_s
+      end
+
+      pattern = configured_test_pattern
+      cfn = @repo_config.config_filename
+
+      if pattern.is_a?(Array)
+        say Text::Process::CONFIGURED_PATTERN % [cfn, pattern.map{|p| " - #{p}"}.join("\n"), cfn]
+        params[:test_pattern] = pattern.join(",")
+      elsif pattern
+        exit_failure Text::Error::INVALID_CONFIGURED_PATTERN % [cfn, cfn, pattern.inspect, cfn]
+      else
+        say Text::Process::TEST_PATTERN_INSTRUCTIONS unless options[:non_interactive]
+        ask_or_update.call(:test_pattern, Text::Prompt::TEST_PATTERN, Default::SUITE_TEST_PATTERN)
+      end
+
+      exclude_pattern = configured_test_exclude_pattern
+      cfn = @repo_config.config_filename
+
+      if exclude_pattern.is_a?(Array)
+        say Text::Process::CONFIGURED_EXCLUDE_PATTERN % [cfn, exclude_pattern.map{|p| " - #{p}"}.join("\n"), cfn]
+        params[:test_exclude_pattern] = exclude_pattern.join(",")
+      elsif exclude_pattern
+        exit_failure Text::Error::INVALID_CONFIGURED_PATTERN % [cfn, cfn, exclude_pattern.inspect, cfn]
+      end
+
+      unless options[:non_interactive]
+        say(Text::Process::SETUP_CI)
+      end
+
+      ask_or_update.call(:ci_pull_url, Text::Prompt::CI_PULL_URL, @scm.origin_url) 
+      ask_or_update.call(:ci_push_url, Text::Prompt::CI_PUSH_URL, nil)
+    end
+  end
+end

data/lib/tddium/cli/show.rb

@@ -0,0 +1,122 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    protected
+
+    def show_config_details(scope, config)
+      if !config || config.length == 0
+        say Text::Process::NO_CONFIG
+      else
+        say Text::Status::CONFIG_DETAILS % scope
+        config.each do |k,v| 
+          say "#{k}=#{v}"
+        end
+      end
+      say Text::Process::CONFIG_EDIT_COMMANDS
+    end
+
+    def show_attributes(names_to_display, attributes)
+      names_to_display.each do |attr|
+        say Text::Status::ATTRIBUTE_DETAIL % [attr.gsub("_", " ").capitalize, attributes[attr]] if attributes[attr]
+      end
+    end
+
+    def show_keys_details(keys)
+      say Text::Status::KEYS_DETAILS
+      if keys.length == 0
+        say Text::Process::NO_KEYS
+      else
+        keys.each do |k| 
+          if k["fingerprint"]
+            say((" %-18.18s %s" % [k["name"], k["fingerprint"]]).rstrip)
+          elsif k["pub"]
+            fingerprint = ssh_key_fingerprint(k["pub"])
+            if fingerprint then
+              say((" %-18.18s %s" % [k["name"], fingerprint]).rstrip)
+            else
+              say((" %-18.18s" % k["name"]).rstrip)
+            end
+          else
+            say((" %-18.18s" % k["name"]).rstrip)
+          end
+        end
+      end
+      say Text::Process::KEYS_EDIT_COMMANDS
+    end
+
+    def show_third_party_keys_details(user)
+      say ERB.new(Text::Status::USER_THIRD_PARTY_KEY_DETAILS).result(binding)
+    end
+
+    def show_ssh_config(dir=nil)
+      dir ||= ENV['TDDIUM_GEM_KEY_DIR']
+      dir ||= Default::SSH_OUTPUT_DIR
+
+      path = File.expand_path(File.join(dir, "identity.tddium.*"))
+
+      Dir[path].reject{|fn| fn =~ /.pub$/}.each do |fn|
+        say Text::Process::SSH_CONFIG % {:scm_host=>"git.solanolabs.com", :file=>fn}
+      end
+    end
+
+    def format_usage(usage)
+      "All tests: %.2f worker-hours  ($%.2f)" % [
+        usage["hours"] || 0, usage["charge"] || 0]
+    end
+
+    def show_user_details(user)
+      current_suites = @tddium_api.get_suites
+      memberships = @tddium_api.get_memberships
+      account_usage = @tddium_api.get_usage
+
+      # Given the user is logged in, he should be able to
+      # use "tddium account" to display information about his account:
+      # Email address
+      # Account creation date
+      say ERB.new(Text::Status::USER_DETAILS).result(binding)
+
+      # Use "all_accounts" here instead of "participating_accounts" -- these
+      # are the accounts the user can administer.
+      user["all_accounts"].each do |acct|
+        id = acct['account_id'].to_i
+
+        say ERB.new(Text::Status::ACCOUNT_DETAILS).result(binding)
+
+        acct_suites = current_suites.select{|s| s['account_id'].to_i == id}
+        if acct_suites.empty? then
+          say '  ' + Text::Status::NO_SUITE
+        else
+          say '  ' + Text::Status::ALL_SUITES
+          suites = acct_suites.sort_by{|s| "#{s['org_name']}/#{s['repo_name']}"}
+          print_table suites.map {|suite|
+            repo_name = suite['repo_name']
+            if suite['org_name'] && suite['org_name'] != 'unknown'
+              repo_name = suite['org_name'] + '/' + repo_name
+            end
+            [repo_name, suite['branch'], suite['repo_url'] || '']
+          }, :indent => 4
+        end
+
+        # Uugh, json converts the keys to strings.
+        usage = account_usage[id.to_s]
+        if usage
+          say "\n  Usage:"
+          say "    Current month:  " + format_usage(usage["current_month"])
+          say "    Last month:     " + format_usage(usage["last_month"])
+        end
+
+        acct_members = memberships.select{|m| m['account_id'].to_i == id}
+        if acct_members.length > 1
+          say "\n  " + Text::Status::ACCOUNT_MEMBERS
+          print_table acct_members.map {|ar|
+            [ar['user_handle'], ar['user_email'], ar['role']]
+          }, :indent => 4
+        end
+      end
+
+    rescue TddiumClient::Error::Base => e
+      exit_failure e.message
+    end
+  end
+end

data/lib/tddium/cli/suite.rb

@@ -0,0 +1,179 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    protected
+
+    def update_suite(suite, options)
+      params = {}
+      prompt_suite_params(options, params, suite)
+
+      ask_or_update = lambda do |key, text, default|
+        params[key] = prompt(text, options[key], suite.fetch(key.to_s, default), options[:non_interactive])
+      end
+
+      ask_or_update.call(:campfire_room, Text::Prompt::CAMPFIRE_ROOM, '')
+      ask_or_update.call(:hipchat_room, Text::Prompt::HIPCHAT_ROOM, '')
+
+      @tddium_api.update_suite(suite['id'], params)
+      say Text::Process::UPDATED_SUITE
+    end
+
+    def suite_auto_configure
+      # Did the user set a configuration option on the command line?
+      # If so, auto-configure a new suite and re-configure an existing one
+      user_config = options.member?(:tool)
+
+      current_suite_id = @tddium_api.current_suite_id
+      if current_suite_id && !user_config then
+        current_suite = @tddium_api.get_suite_by_id(current_suite_id)
+      else
+        params = Hash.new
+        params[:branch] = @scm.current_branch
+        params[:repo_url] = @scm.origin_url
+        params[:repo_name] = @scm.repo_name
+        params[:scm] = @scm.scm_name
+        if options[:account] && !params.member?(:account_id) then
+          account_id = @tddium_api.get_account_id(options[:account])
+          params[:account_id] = account_id if account_id
+        end
+
+        tool_cli_populate(options, params)
+        defaults = {}
+
+        prompt_suite_params(options.merge({:non_interactive => true}), params, defaults)
+
+        # Create new suite if it does not exist yet
+        say Text::Process::CREATING_SUITE % [params[:repo_name], params[:branch]]
+
+        current_suite = @tddium_api.create_suite(params)
+
+        # Save the created suite
+        @api_config.set_suite(current_suite)
+        @api_config.write_config
+      end
+      return current_suite
+    end
+
+    def format_suite_details(suite)
+      # Given an API response containing a "suite" key, compose a string with
+      # important information about the suite
+      tddium_deploy_key_file_name = @api_config.tddium_deploy_key_file_name
+      details = ERB.new(Text::Status::SUITE_DETAILS).result(binding)
+      details
+    end
+
+    def suite_for_current_branch?
+      return true if @tddium_api.current_suite_id
+      say Text::Error::NO_SUITE_EXISTS % @scm.current_branch
+      false
+    end
+
+    def suite_for_default_branch?
+      return true if @tddium_api.default_suite_id
+      say Text::Error::NO_SUITE_EXISTS % @scm.default_branch
+      false
+    end
+
+    # Update the suite parameters from solano.yml
+    def update_suite_parameters!(current_suite, session_id=nil)
+      update_params = {}
+
+      update_params[:session_id] = session_id if session_id
+
+      pattern = configured_test_pattern
+      if pattern.is_a?(Array)
+        pattern = pattern.join(",")
+      end
+      if pattern && current_suite["test_pattern"] != pattern then
+        update_params[:test_pattern] = pattern
+      end
+
+      exclude_pattern = configured_test_exclude_pattern
+      if exclude_pattern.is_a?(Array)
+        exclude_pattern = exclude_pattern.join(",")
+      end
+      if exclude_pattern && current_suite["test_exclude_pattern"] != exclude_pattern then
+        update_params[:test_exclude_pattern] = exclude_pattern
+      end
+
+      ruby_version = sniff_ruby_version
+      if ruby_version && ruby_version != current_suite["ruby_version"] then
+        update_params[:ruby_version] = ruby_version
+      end
+
+      bundler_version = @repo_config["bundler_version"]
+      if bundler_version && bundler_version != current_suite["bundler_version"] then
+        update_params[:bundler_version] = bundler_version
+      end
+
+      test_configs = @repo_config["tests"] || []
+      if test_configs != (current_suite['test_configs'] || []) then
+        if test_configs != 'disable' && !test_configs.is_a?(Array) then
+          warn(Text::Warning::TEST_CONFIGS_MUST_BE_LIST)
+          test_configs = []
+        end
+        update_params[:test_configs] = test_configs
+      end
+
+      %w(golang java leiningen nodejs php python scala).each do |lang|
+        config_name = "#{lang}_config"
+        lang_config = @repo_config[lang] || {}
+        current_lang_config = current_suite[config_name] || {}
+        if lang_config != (current_suite[config_name] || {}) then
+          update_params[lang.to_sym] = lang_config
+        end
+      end
+
+      if !update_params.empty? then
+        @tddium_api.update_suite(@tddium_api.current_suite_id, update_params)
+        if update_params[:test_pattern]
+          say Text::Process::UPDATED_TEST_PATTERN % pattern
+        end
+        if update_params[:test_exclude_pattern]
+          say Text::Process::UPDATED_TEST_EXCLUDE_PATTERN % exclude_pattern
+        end
+        if update_params[:ruby_version]
+          say Text::Process::UPDATED_RUBY_VERSION % ruby_version
+        end
+        if update_params[:bundler_version]
+          say Text::Process::UPDATED_BUNDLER_VERSION % bundler_version
+        end
+        if update_params[:test_configs]
+          say Text::Process::UPDATED_TEST_CONFIGS % YAML.dump(test_configs)
+          say "(was:\n#{YAML.dump(current_suite['test_configs'])})\n"
+        end
+        if update_params[:python_config]
+          say Text::Process::UPDATED_PYTHON_CONFIG % YAML.dump(python_config)
+        end
+        if update_params[:golang_config]
+          say Text::Process::UPDATED_PYTHON_CONFIG % YAML.dump(golang_config)
+        end
+        if update_params[:java_config]
+          say Text::Process::UPDATED_PYTHON_CONFIG % YAML.dump(java_config)
+        end
+      end
+    end
+
+    def suite_remembered_option(options, key, default, &block)
+      remembered = false
+      if options[key] != default
+        result = options[key]
+      elsif remembered = current_suite_options[key.to_s]
+        result = remembered
+        remembered = true
+      else
+        result = default
+      end
+
+      if result then
+        msg = Text::Process::USING_SPEC_OPTION[key] % result
+        msg +=  Text::Process::REMEMBERED if remembered
+        msg += "\n"
+        say msg
+        yield result if block_given?
+      end
+      result
+    end
+  end
+end