dssh

#!/usr/bin/env ruby

require 'net/http'
require 'json'

raise "Invalid arguments: dssh [container-name] [command=/bin/bash]" if ARGV.length < 1
service_name = "#{ARGV[0]}"
ARGV[1] ||= "/bin/bash"
command = ARGV[1..-1].join(' ')

uri = URI.parse("https://dashboard.tutum.co")
http = Net::HTTP.new(uri.host, uri.port)
http.use_ssl = true
headers = { "Authorization" => ENV['TUTUM_AUTH'] }

# Get list of containers that match
container = "#{service_name}-1"
response = http.get("/api/v1/container/?name=#{container}", headers)

if response.code == "200"
  container_info = JSON.load(response.body)["objects"].first
  if container_info
    dns = container_info["public_dns"]
    docker_id = container_info["docker_id"]
    exec "ssh -t #{dns} sudo docker exec -i -t #{docker_id} '#{command}'"
  else
    puts "No conatiner found"
    exit 1
  end
else
  puts "Error accessing tutum: #{response.body}"
  exit 1
end