agile-alliance-brazil/submissions

#!/usr/bin/env ruby
# frozen_string_literal: true

require 'json'
require 'net/https'
require 'uri'
require 'English'
require 'dotenv'

APP_NAME = 'submissoes'

Dotenv.load
unless ENV['TOKEN']
  puts "Ensure you've set the Digital ocean token using \"export TOKEN='your_token'\" or added it to your .env file"
  exit 1
end

TOKEN = ENV['TOKEN']
TYPE = if !ARGV.empty? && ARGV[0] == 'production'
         :production
       else
         :staging
       end

staging = TYPE != :production
SSH = staging ? '36:18:0e:5c:aa:0c:58:9e:d2:72:5b:f7:f8:e7:f2:5d' : 'ba:49:c2:40:4e:18:dd:cb:bb:cd:9c:f6:99:11:67:db'
POSTFIX = staging ? '-staging' : ''
ROOT = File.expand_path(File.join(File.dirname(__FILE__), '../../'))

def get_json(uri)
  uri = URI.parse(uri)

  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true
  request = Net::HTTP::Get.new(uri.request_uri,
                               'Content-Type' => 'application/json',
                               'Authorization' => "Bearer #{TOKEN}")
  http.request(request)
end

def post_json(uri, body)
  uri = URI.parse(uri)

  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true
  request = Net::HTTP::Post.new(uri.request_uri,
                                'Content-Type' => 'application/json',
                                'Authorization' => "Bearer #{TOKEN}")
  request.body = body
  http.request(request)
end

def link_files(ip_address, file_name)
  "rm -f #{ip_address}_#{file_name} && ln -s #{TYPE}_#{file_name} #{ip_address}_#{file_name}"
end

droplets = get_json('https://api.digitalocean.com/v2/droplets')
machine_id = (JSON.parse(droplets.body)['droplets'].count do |d|
  d['name'].match(/#{APP_NAME}-\d\d#{POSTFIX}\./)
end + 1)

bootstrap_info = File.read(File.join(ROOT, 'puppet/script/server_bootstrap.sh'))
body = {
  names: ["#{APP_NAME}-#{format('%<machine_id>02d', machine_id: machine_id)}#{POSTFIX}.agilebrazil.com"],
  region: 'nyc3',
  size: '1gb',
  image: 'ubuntu-14-04-x64',
  ssh_keys: [SSH],
  backups: false,
  ipv6: true,
  user_data: bootstrap_info,
  private_networking: nil
}

def setup_droplet(droplet)
  setup = "cd #{ROOT}/config && #{link_files(droplet[:ipv4], 'config.yml')} &&\
    #{link_files(droplet[:ipv4], 'database.yml')} &&\
    #{link_files(droplet[:ipv4], 'newrelic.yml')} &&\
    cd #{ROOT}/certs && #{link_files(droplet[:ipv4], 'server.crt')} &&\
    #{link_files(droplet[:ipv4], 'server_key.pem')} &&\
    #{link_files(droplet[:ipv4], 'intermediate.crt')}"
  result = `#{setup}`
  return "ERROR: Cannot generate config files and certs for #{droplet[:ipv4]}.\n#{result}" unless $CHILD_STATUS.to_i.zero?

  key_path = "#{ROOT}/certs/digital_ocean#{POSTFIX.tr('-', '_')}"
  ssh_command = "ssh -i #{key_path} -o LogLevel=quiet -o StrictHostKeyChecking=no ubuntu@#{droplet[:ipv4]} 'echo \"SSH Successful!\"'"
  `#{ssh_command}` # Adding new machine to known hosts
  first_deploy = "bundle exec ruby deploy/first_deploy.rb ubuntu #{droplet[:ipv4]} #{TYPE} #{key_path}"
  deploy_result = `#{first_deploy}`
  return "ERROR: Deploy failed on #{droplet[:ipv4]}\n#{deploy_result}" unless $CHILD_STATUS.to_i.zero?

  url = "https://#{droplet[:ipv4]}"
  `curl -k "#{url}"`
  return "ERROR: Deploy successful on #{url} but HTTPS is not working.\n#{deploy_result}" unless $CHILD_STATUS.to_i.zero?

  "SUCCESS: #{url} is up an running!"
end

response = post_json('https://api.digitalocean.com/v2/droplets', body.to_json)
if response.code.to_i < 400
  droplet_response = JSON.parse(response.body)
  ids = droplet_response['droplets'].map { |d| d['id'] }
  sleep(120 * (ids.size.to_f / 10).ceil) # wait for droplets to get network data & run the bootstrap
  droplet_infos = ids.map do |id|
    info = get_json("https://api.digitalocean.com/v2/droplets/#{id}")
    if info.code.to_i < 400
      JSON.parse(info.body)
    else
      info.body
    end
  end
  errors, successes = droplet_infos.partition { |i| i.is_a? String }
  unless errors.empty?
    puts 'ERRORS: The following are unknown droplets'
    puts errors
  end
  unless successes.empty?
    droplets = successes.map do |d|
      {
        id: d['droplet']['id'],
        ipv4: d['droplet']['networks']['v4'].map { |i| i['ip_address'] }.first,
        ipv6: d['droplet']['networks']['v6'].map { |i| i['ip_address'] }.first
      }
    end
    puts droplets.map { |d| setup_droplet(d) }.join("\n\n")
  end
else
  puts 'ERROR: Droplets failed'
  puts response.body
end