module Travis::Tools::Completion

def cmp_file
  File.expand_path('travis.sh', config_path)
end

def compile
  commands = Travis::CLI.commands.sort_by { |c| c.command_name }
  template = Assets.read('travis.sh.erb')
  source   = ERB.new(template).result(binding).gsub(/^ +\n/, '')
  File.write(Assets['travis.sh'], source)
end

def completion_installed?
  source = 'source ' << config_path
  RCS.each do |file|
    next unless File.exist?(file) && File.writable?(file)
    return false unless File.read(file).include? source
  end
  true
end

def config_path
  ENV.fetch('TRAVIS_CONFIG_PATH') { File.expand_path('.travis', ENV['HOME']) }
end

def install_completion
  update_completion
  source = 'source ' << cmp_file
  RCS.each do |file|
    next unless File.exist?(file) && File.writable?(file)
    next if File.read(file).include? source
    File.open(file, 'a') { |f| f.puts('', '# added by travis gem', "[ ! -s #{cmp_file} ] || #{source}") }
  end
end

def update_completion
  FileUtils.mkdir_p(config_path)
  FileUtils.cp(Assets['travis.sh'], cmp_file)
end