deploy_to_amazon_ec2

#!/usr/bin/python

# A simple script for building and deploying artifacts to Amazon EC2.
# EC2 instancc must be 'Ubuntu 16.04.1 x64'.

import os
import sys
import os
import subprocess

if len(sys.argv) < 4:
	print "Not enough parameters."
	print "Example: ./deploy_into_amazon_ec2 <address> <port> <private key>"
	quit()

credentials = sys.argv[1]
port = sys.argv[2]
pem_file = sys.argv[3]

swiftc_path = subprocess.check_output("which swiftc", shell=True)

head, tail = os.path.split(swiftc_path.rstrip())
root, tail = os.path.split(head)
shared = os.path.join(root, "lib", "swift", "linux")

if not os.path.isdir(shared):
	print "Shared objects directory not found at: " + shared
	quit()

print "Found shared objects directory: " + shared
print "Building project..."

try:
	subprocess.check_output("swift build -v -c release -Xlinker -rpath -Xlinker .", shell=True)
except:
	quit()

print "Build ok."
print 'Checking SSH connection...'

try:
	subprocess.check_output('ssh -i "' + pem_file + '" ' + credentials + " 'pwd'", shell=True)
except:
	print "Unexpected error:", sys.exc_info()
	quit()

print 'Installing libcurl4-openssl-dev (it can take some time)...'

try:
	subprocess.check_output('ssh -i "' + pem_file + '" ' + credentials + " 'sudo apt-get -y install libcurl4-openssl-dev'", shell=True)
except:
	print "Unexpected error:", sys.exc_info()
	quit()

print 'Checking if service is on...'

try:
	pid = subprocess.check_output('ssh -i "' + pem_file + '" ' + credentials + " 'lsof -t -i:" + port + "'" , stderr=subprocess.STDOUT, shell=True).rstrip()
	print "Service found PID = " + pid
	subprocess.check_output('ssh -i "' + pem_file + '" ' + credentials + " 'kill -9 $(lsof -t -i:" + port + ")'", stderr=subprocess.STDOUT, shell=True).rstrip()
	print "Service killed."
except subprocess.CalledProcessError, e:
	if e.returncode == 1:
		print "Service not found on port: " + str(port) + ". No restart needed."
	else:
		print "Unexpected error:", sys.exc_info()
		print e.output
		quit()

print "Syncing Swift core shared objects..."

try:
	subprocess.check_output('rsync -PazSHAX --rsh "ssh -i ' + pem_file + '" --rsync-path "sudo rsync" ' + os.path.join(shared, "*") + " " + credentials +":/home/ubuntu/swiftx", stderr=subprocess.STDOUT, shell=True)
except:
	print "Unexpected error:", sys.exc_info()
	quit()

print "Sending release artifact..."

try:
	subprocess.check_output('rsync --rsh "ssh -i ' + pem_file + '" --rsync-path "sudo rsync" ./.build/release/swiftX ' + credentials + ":/home/ubuntu/swiftx/release", shell=True)
except:
	print "Failed to rsync executable: ", sys.exc_info()
	quit()

print "Starting server..."

try:
	subprocess.check_output('ssh -i "' + pem_file + '" ' + credentials + " '/home/ubuntu/swiftx/release < /dev/null > /tmp/swiftx_log 2>&1 &'", shell=True)
except:
	print "Failed to run executable: ", sys.exc_info()
	quit()

print "Server is running. Make sure you enabled " + port + " at Security Grups settings."