class Travis::CLI::RepoCommand

def branch(name)
  repository.branch(name)
end

def build(number_or_id)
  return super if number_or_id.is_a? Integer
  repository.build(number_or_id)
end

def detect_api_endpoint
  if explicit_api_endpoint? or enterprise?
    repo_config['endpoint'] = api_endpoint
  elsif ENV['TRAVIS_ENDPOINT']
    ENV['TRAVIS_ENDPOINT']
  elsif config['default_endpoint'] and config['default_endpoint'] !~ TRAVIS
    repo_config['endpoint'] ||= config['default_endpoint']
  else
    repo_config['endpoint'] ||= begin
      load_gh
      GH.head("/repos/#{slug}")
      Travis::Client::ORG_URI
    rescue GH::Error
      Travis::Client::PRO_URI
    end
  end
end

def detect_slug
  git_head    = `git name-rev --name-only HEAD 2>#{IO::NULL}`.chomp
  git_remote  = `git config --get branch.#{git_head}.remote 2>#{IO::NULL}`.chomp
  git_remote  = 'origin' if git_remote.empty?
  git_info    = `git config --get remote.#{git_remote}.url 2>#{IO::NULL}`.chomp
  if Addressable::URI.parse(git_info).path =~ GIT_REGEX
    detectected_slug = $1
    if interactive?
      if agree("Detected repository as #{color(detectected_slug, :info)}, is this correct? ") { |q| q.default = 'yes' }
        detectected_slug
      else
        ask("Repository slug: ") { |q| q.default = detectected_slug }
      end
    else
      info "detected repository as #{color(detectected_slug, :bold)}"
      detectected_slug
    end
  end
end

def detected_endpoint?
  !explicit_api_endpoint?
end

def find_slug
  load_slug || store_slug(detect_slug)
end

def job(number_or_id)
  return super if number_or_id.is_a? Integer
  repository.job(number_or_id)
end

def last_build
  repository.last_build or error("no build yet for #{slug}")
end

def load_slug
  stored = `git config --get travis.slug`.chomp
  stored unless stored.empty?
end

def repo_config
  config['repos'] ||= {}
  config['repos'][slug] ||= {}
end

def repository
  repo(slug)
rescue Travis::Client::NotFound
  error "repository not known to #{api_endpoint}: #{color(slug, :important)}"
end

def save_travis_config(file = travis_yaml)
  yaml = travis_config.to_yaml
  yaml.gsub! /^(\s+)('on'|true):/, "\\1on:"
  yaml.gsub! /\A---\s*\n/, ''
  File.write(file, yaml)
end

def setup
  setup_enterprise
  error "Can't figure out GitHub repo name. Ensure you're in the repo directory, or specify the repo name via the -r option (e.g. travis <command> -r <repo-name>)" unless self.slug ||= find_slug
  error "GitHub repo name is invalid, it should be on the form 'owner/repo'" unless self.slug.include?("/")
  self.api_endpoint = detect_api_endpoint
  super
  repository.load # makes sure we actually have access to the repo
end

def store_slug(value)
  `git config travis.slug #{value}` if value
  value
end

def travis_config
  @travis_config ||= begin
    payload = YAML.load_file(travis_yaml)
    payload.respond_to?(:to_hash) ? payload.to_hash : {}
  end
end

def travis_yaml(dir = Dir.pwd)
  path = File.expand_path('.travis.yml', dir)
  if File.exist? path
    path
  else
    parent = File.expand_path('..', dir)
    error "no .travis.yml found" if parent == dir
    travis_yaml(parent)
  end
end