tddium 1.25.5

data/lib/tddium/cli/commands/status.rb

@@ -0,0 +1,107 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+require 'stringio'
+
+module Tddium
+  class TddiumCli < Thor
+    desc "status", "Display information about this suite, and any open dev sessions"
+    method_option :json, :type => :boolean, :default => false
+    def status
+      tddium_setup
+
+      begin
+        # tddium_setup asserts that we're in a git repo
+        origin_url = @scm.origin_url
+        repo_params = {
+          :active => true, 
+          :repo_url => origin_url
+        }
+
+        if suite_for_current_branch? then
+          status_branch = @tddium_api.current_branch
+          suite_params = {
+            :suite_id => @tddium_api.current_suite_id,
+            :active => false,
+            :limit => 10
+          }
+        elsif suite_for_default_branch? then
+          status_branch = @tddium_api.default_branch
+          say Text::Error::TRY_DEFAULT_BRANCH % status_branch
+          suite_params = {
+            :suite_id => @tddium_api.default_suite_id,
+            :active => false,
+            :limit => 10
+          }
+        end
+
+        if options[:json] 
+          res = {}
+          res[:running] = { origin_url => @tddium_api.get_sessions(repo_params) }          
+          res[:history] = { 
+            status_branch => @tddium_api.get_sessions(suite_params)
+          } if suite_params
+          puts JSON.pretty_generate(res)
+        else
+          show_session_details(
+            status_branch,
+            repo_params, 
+            Text::Status::NO_ACTIVE_SESSION, 
+            Text::Status::ACTIVE_SESSIONS,
+            true
+          )
+          show_session_details(
+            status_branch,
+            suite_params, 
+            Text::Status::NO_INACTIVE_SESSION, 
+            Text::Status::INACTIVE_SESSIONS,
+            false
+          ) if suite_params
+          say Text::Process::RERUN_SESSION
+        end
+
+      rescue TddiumClient::Error::Base => e
+        exit_failure e.message
+      end
+    end
+
+    private 
+
+    def show_session_details(status_branch, params, no_session_prompt, all_session_prompt, include_branch)
+      current_sessions = @tddium_api.get_sessions(params)
+
+      say ""
+      if current_sessions.empty? then
+        say no_session_prompt
+      else
+        commit_size = 0...7
+        head = @scm.current_commit[commit_size]
+
+        say all_session_prompt % (params[:suite_id] ? status_branch : "")
+        say ""
+        header = ["Session #", "Commit", ("Branch" if include_branch), "Status", "Duration", "Started"].compact
+        table = [header, header.map { |t| "-" * t.size }] + current_sessions.map do |session|
+          duration = "%ds" % session['duration']
+          start_timeago = "%s ago" % Tddium::TimeFormat.seconds_to_human_time(Time.now - Time.parse(session["start_time"]))
+
+          [
+            session["id"].to_s,
+            session["commit"] ? session['commit'][commit_size] : '-      ',
+            (session["branch"] if include_branch),
+            session["status"],
+            duration,
+            start_timeago
+          ].compact
+        end
+        say(capture_stdout { print_table table }.gsub(head, "\e[7m#{head}\e[0m"))
+      end
+    end
+
+    def capture_stdout
+      old, $stdout = $stdout, StringIO.new
+      yield
+      $stdout.string
+    ensure
+      $stdout = old
+    end
+  end
+end  

data/lib/tddium/cli/commands/stop.rb

@@ -0,0 +1,19 @@
+# Copyright (c) 2011-2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    desc "stop [SESSION]", "Stop session by id"
+    def stop(ls_id=nil)
+      tddium_setup({:scm => false})
+      if ls_id
+        begin
+          say "Stoping session #{ls_id} ..."
+          say @tddium_api.stop_session(ls_id)['notice']
+        rescue 
+        end
+      else
+        say 'Stop requires a session id -- e.g. `tddium stop 7869764`'
+      end
+    end
+  end
+end

data/lib/tddium/cli/commands/suite.rb

@@ -0,0 +1,110 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    desc "suite", "Register the current repo/branch, view/edit CI repos & deploy keys"
+    method_option :edit, :type => :boolean, :default => false
+    method_option :account, :type => :string, :default => nil,
+      :aliases => %w(--org --organization)
+    method_option :name, :type => :string, :default => nil
+    method_option :repo_url, :type => :string, :default => nil
+    method_option :ci_pull_url, :type => :string, :default => nil
+    method_option :ci_push_url, :type => :string, :default => nil
+    method_option :test_pattern, :type => :string, :default => nil
+    method_option :campfire_room, :type => :string, :default => nil
+    method_option :hipchat_room, :type => :string, :default => nil
+    method_option :non_interactive, :type => :boolean, :default => false
+    method_option :tool, :type => :hash, :default => {}
+    method_option :delete, :type => :boolean, :default => false
+    def suite(*argv)
+      tddium_setup({:repo => true})
+
+      params = {}
+
+      # Load tool options into params
+      tool_cli_populate(options, params)
+
+      begin
+
+        if options[:delete]
+          # Deleting works differently (for now) because api_config can't handle
+          # multiple suites with the same branch name in two different accounts.
+
+          repo_url = @scm.origin_url
+
+          if argv.is_a?(Array) && argv.size > 0
+            branch = argv[0]
+          else
+            branch = @tddium_api.current_branch
+          end
+
+          suites = @tddium_api.get_suites(:repo_url => repo_url, :branch => branch)
+          if suites.count == 0
+            exit_failure Text::Error::CANT_FIND_SUITE % [repo_url, branch]
+          elsif suites.count > 1
+            say Text::Process::SUITE_IN_MULTIPLE_ACCOUNTS % [repo_url, branch]
+            suites.each { |s| say '  ' + s['account'] }
+            account = ask Text::Process::SUITE_IN_MULTIPLE_ACCOUNTS_PROMPT
+            suites = suites.select { |s| s['account'] == account }
+            if suites.count == 0
+              exit_failure Text::Error::INVALID_ACCOUNT_NAME
+            end
+          end
+
+          suite = suites.first
+
+          unless options[:non_interactive]
+            yn = ask Text::Process::CONFIRM_DELETE_SUITE % [suite['repo_url'], suite['branch'], suite['account']]
+            unless yn.downcase == Text::Prompt::Response::YES
+              exit_failure Text::Process::ABORTING
+            end
+          end
+
+          @tddium_api.permanent_destroy_suite(suite['id'])
+          @api_config.delete_suite(suite['branch'])
+          @api_config.write_config
+
+        elsif @tddium_api.current_suite_id then
+          # Suite ID set in tddium config file
+          current_suite = @tddium_api.get_suite_by_id(@tddium_api.current_suite_id)
+          if options[:edit] then
+            update_suite(current_suite, options)
+          else
+            say Text::Process::EXISTING_SUITE, :bold
+            say format_suite_details(current_suite)
+          end
+
+          @api_config.set_suite(current_suite)
+          @api_config.write_config
+        else
+          # Need to find or construct the suite (and repo)
+          params[:branch] = @scm.current_branch
+          params[:repo_url] = @scm.origin_url
+          params[:repo_name] = @scm.repo_name
+          params[:scm] = @scm.scm_name
+
+          say Text::Process::NO_CONFIGURED_SUITE % [params[:repo_name], params[:branch]]
+
+          prompt_suite_params(options, params)
+
+          params.each do |k,v|
+            params.delete(k) if v == 'disable'
+          end
+
+          # Create new suite if it does not exist yet
+          say Text::Process::CREATING_SUITE % [params[:repo_name], params[:branch]]
+          new_suite = @tddium_api.create_suite(params)
+
+          # Save the created suite
+          @api_config.set_suite(new_suite)
+          @api_config.write_config
+
+          say Text::Process::CREATED_SUITE, :bold
+          say format_suite_details(new_suite)
+        end
+      rescue TddiumClient::Error::Base => e
+        exit_failure(e.explanation)
+      end
+    end
+  end
+end

data/lib/tddium/cli/commands/web.rb

@@ -0,0 +1,22 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+module Tddium
+  class TddiumCli < Thor
+    desc "web [SESSION_ID]", "Open build report in web browser"
+    def web(*args)
+      tddium_setup({:login => false, :scm => false})
+
+      session_id = args.first
+      if session_id then
+        fragment = "1/reports/#{session_id.to_i}" if session_id =~ /^[0-9]+$/
+      end
+      fragment ||= 'latest'
+
+      begin
+        Launchy.open("#{options[:proto]}://#{options[:host]}/#{fragment}")
+      rescue Launchy::Error => e
+        exit_failure e.message
+      end
+    end
+  end
+end

data/lib/tddium/cli/config.rb

@@ -0,0 +1,245 @@
+# Copyright (c) 2011, 2012, 2013, 2014 Solano Labs All Rights Reserved
+
+require 'yaml'
+
+module ConfigHelper
+  def hash_stringify_keys(h)
+    case h
+    when Hash
+      Hash[ h.map { |k, v| [ k.to_s, hash_stringify_keys(v) ] } ]
+    when Enumerable
+      h.map { |v| hash_stringify_keys(v) }
+    else
+      h
+    end
+  end
+end
+
+module Tddium
+  class RepoConfig
+    include TddiumConstant
+    include ConfigHelper
+
+    def initialize
+      @scm = Tddium::SCM.configure
+      @config = load_config
+    end
+
+    def [](key)
+      return @config[key.to_s]
+    end
+
+    def config_filename
+      @config_filename
+    end
+
+    def load_config
+      config = nil
+
+      root = @scm.root
+      cfgfile_pair = pick_config_pair(root, Config::CONFIG_PATHS)
+      cfgfile_pair_depr = pick_config_pair(root, Config::CONFIG_PATHS_DEPRECATED)
+
+      if cfgfile_pair && cfgfile_pair_depr then
+        abort Text::Error::CONFIG_PATHS_COLLISION % [cfgfile_pair, cfgfile_pair_depr]
+      end
+
+      cfgfile_pair = cfgfile_pair_depr if cfgfile_pair.nil?
+
+      if cfgfile_pair && cfgfile_pair.first then
+        cfgfile = cfgfile_pair.first
+        @config_filename = cfgfile_pair[1]
+        begin
+          rawconfig = File.read(cfgfile)
+          if rawconfig && rawconfig !~ /\A\s*\z/ then
+            config = YAML.load(rawconfig)
+            config = hash_stringify_keys(config)
+            config = config['solano'] || config['tddium'] || config
+          end
+        rescue Exception => e
+          warn(Text::Warning::YAML_PARSE_FAILED % cfgfile)
+        end
+      end
+
+      config ||= Hash.new
+      return config
+    end
+
+    private
+
+    def pick_config_pair(root, config_paths)
+      files = config_paths.map { |fn| [ File.join(root, fn), fn ] }
+      files.select { |p| File.exists?(p.first) }.first
+    end
+  end
+
+  class ApiConfig
+    include TddiumConstant
+
+    # BOTCH: should be a state object rather than entire CLI object
+    def initialize(tddium_client, host, cli_options)
+      @scm = Tddium::SCM.configure
+      @tddium_client = tddium_client
+      @config = Hash.new
+      @host = host
+      @cli_options = cli_options
+    end
+
+    # BOTCH: fugly
+    def set_api(tddium_api)
+      @tddium_api = tddium_api
+    end
+
+    def logout
+      remove_tddium_files
+    end
+
+    def populate_branches(branch)
+      suites = @tddium_api.get_suites(:repo_url => @scm.origin_url, :branch=>branch)
+      suites.each do |ste|
+        set_suite(ste)
+      end
+    end
+
+    def get_branch(branch, var, options={})
+      if options['account'].nil? && @cli_options[:account] then
+        options['account'] = @cli_options[:account]
+      end
+
+      val = fetch_branch(branch, var, options)
+      return val unless val.nil?
+
+      populate_branches(branch)
+
+      return fetch_branch(branch, var, options)
+    end
+
+    def get_api_key(options = {})
+      options.any? ? load_config(options)['api_key'] : @config['api_key']
+    end
+
+    def set_api_key(api_key, user)
+      @config['api_key'] = api_key
+    end
+
+    def scm_ready_sleep
+      s = ENV["TDDIUM_SCM_READY_SLEEP"] || Default::SCM_READY_SLEEP
+      s.to_f
+    end
+
+    def set_suite(suite)
+      id = suite['id']
+      branch = suite['branch']
+      return if id.nil? || branch.nil? || branch.empty?
+
+      keys = %w(id branch account repo_id ci_ssh_pubkey)
+      metadata = keys.inject({}) { |h, v| h[v] = suite[v]; h }
+
+      branches = @config["branches"] || {}
+      branches.merge!({id => metadata})
+      @config.merge!({"branches" => branches})
+    end
+
+    def delete_suite(branch, account=nil)
+      branches = @config["branches"] || {}
+      branches.delete_if do |k, v|
+        v['branch'] == branch && (account.nil? || v['account'] == account)
+      end
+    end
+
+    def load_config(options = {})
+      global_config = load_config_from_file(:global)
+      return global_config if options[:global]
+
+      repo_config = load_config_from_file
+      return repo_config if options[:repo]
+
+      @config = global_config.merge(repo_config)
+    end
+
+    def write_config
+      path = tddium_file_name(:global)
+      File.open(path, File::CREAT|File::TRUNC|File::RDWR, 0600) do |file|
+        config = Hash.new
+        config['api_key'] = @config['api_key'] if @config.member?('api_key')
+        file.write(config.to_json)
+      end
+
+      path = tddium_file_name(:repo)
+      File.open(path, File::CREAT|File::TRUNC|File::RDWR, 0600) do |file|
+        file.write(@config.to_json)
+      end
+
+      if @scm.repo? then
+        branch = @scm.current_branch
+        id = get_branch(branch, 'id', {})
+        suite = @config['branches'][id] rescue nil
+
+        if suite then
+          path = tddium_deploy_key_file_name
+          File.open(path, File::CREAT|File::TRUNC|File::RDWR, 0644) do |file|
+            file.write(suite["ci_ssh_pubkey"])
+          end
+        end
+        write_scm_ignore		# BOTCH: no need to write every time
+      end
+    end
+
+    def write_scm_ignore
+      path = @scm.ignore_path
+      content = File.exists?(path) ? File.read(path) : ''
+      unless content.include?(".tddium*\n")
+        File.open(path, File::CREAT|File::APPEND|File::RDWR, 0644) do |file|
+          file.write(".tddium*\n")
+        end
+      end
+    end
+
+    def tddium_file_name(scope=:repo, kind='', root=nil)
+      ext = (@host == 'api.tddium.com' || @host == 'ci.solanolabs.com') ? '' : ".#{@host}"
+
+      case scope
+      when :repo
+        root ||= @scm.repo? ? @scm.root : Dir.pwd
+
+      when :global
+        root = ENV['HOME']
+      end
+
+      return File.join(root, ".tddium#{kind}#{ext}")
+    end
+
+    def tddium_deploy_key_file_name
+      return tddium_file_name(:repo, '-deploy-key')
+    end
+
+    protected
+
+    def fetch_branch(branch, var, options)
+      h = @config['branches']
+      return nil unless h.is_a?(Hash)
+      h.each_pair do |id, data|
+        next unless data.is_a?(Hash)
+        branch_name = data['branch']
+        next unless branch_name == branch
+        if options.keys.all? { |k| data.member?(k) && data[k] == options[k] }
+          return data[var]
+        end
+      end
+      return nil
+    end
+
+    private
+
+    def remove_tddium_files
+      [tddium_file_name, tddium_file_name(:global)].each do |tddium_file_path|
+        File.delete(tddium_file_path) if File.exists?(tddium_file_path)
+      end
+    end
+
+    def load_config_from_file(tddium_file_type = :repo)
+      path = tddium_file_name(tddium_file_type)
+      File.exists?(path) ? (JSON.parse(File.read(path)) rescue {}) : {}
+    end
+  end
+end