serverdensity-heroku 0.0.3

data/bin/sd-deploy.py

@@ -0,0 +1,317 @@
+'''
+	Server Density
+	www.serverdensity.com
+	----
+	Server monitoring agent for Linux, FreeBSD and Mac OS X
+
+	Licensed under Simplified BSD License (see LICENSE)
+'''
+
+#
+# Why are you using this?
+#
+import time
+print 'Note: This script is for automating deployments and is not the normal way to install the SD agent. See http://www.serverdensity.com/docs/agent/installation/'
+print 'Continuing in 4 seconds...'
+time.sleep(4)
+
+#
+# Argument checks
+#
+import sys
+
+if len(sys.argv) < 5:
+	print 'Usage: python sd-deploy.py [API URL] [SD URL] [username] [password] [[init]]'
+	sys.exit(2)	
+
+#
+# Get server details
+#
+
+import socket	
+
+# IP
+try:
+	serverIp = socket.gethostbyname(socket.gethostname())
+	
+except socket.error, e:
+	print 'Unable to get server IP: ' + str(e)
+	sys.exit(2)
+	
+# Hostname
+try:
+	serverHostname = hostname = socket.getfqdn()
+	
+except socket.error, e:
+	print 'Unable to get server hostname: ' + str(e)
+	sys.exit(2)
+
+#
+# Get latest agent version
+#
+
+print '1/4: Downloading latest agent version';
+		
+import httplib
+import urllib2
+
+# Request details
+try: 
+	requestAgent = urllib2.urlopen('http://www.serverdensity.com/agentupdate/')
+	responseAgent = requestAgent.read()
+	
+except urllib2.HTTPError, e:
+	print 'Unable to get latest version info - HTTPError = ' + str(e)
+	sys.exit(2)
+	
+except urllib2.URLError, e:
+	print 'Unable to get latest version info - URLError = ' + str(e)
+	sys.exit(2)
+	
+except httplib.HTTPException, e:
+	print 'Unable to get latest version info - HTTPException'
+	sys.exit(2)
+	
+except Exception, e:
+	import traceback
+	print 'Unable to get latest version info - Exception = ' + traceback.format_exc()
+	sys.exit(2)
+
+#
+# Define downloader function
+#
+
+import md5 # I know this is depreciated, but we still support Python 2.4 and hashlib is only in 2.5. Case 26918
+import urllib
+
+def downloadFile(agentFile, recursed = False):
+	print 'Downloading ' + agentFile['name']
+	
+	downloadedFile = urllib.urlretrieve('http://www.serverdensity.com/downloads/sd-agent/' + agentFile['name'])
+	
+	# Do md5 check to make sure the file downloaded properly
+	checksum = md5.new()
+	f = file(downloadedFile[0], 'rb')
+	
+	# Although the files are small, we can't guarantee the available memory nor that there
+	# won't be large files in the future, so read the file in small parts (1kb at time)
+	while True:
+		part = f.read(1024)
+		
+		if not part: 
+			break # end of file
+	
+		checksum.update(part)
+		
+	f.close()
+	
+	# Do we have a match?
+	if checksum.hexdigest() == agentFile['md5']:
+		return downloadedFile[0]
+		
+	else:
+		# Try once more
+		if recursed == False:
+			downloadFile(agentFile, True)
+		
+		else:
+			print agentFile['name'] + ' did not match its checksum - it is corrupted. This may be caused by network issues so please try again in a moment.'
+			sys.exit(2)
+
+#
+# Install the agent files
+#
+
+# We need to return the data using JSON. As of Python 2.6+, there is a core JSON
+# module. We have a 2.4/2.5 compatible lib included with the agent but if we're
+# on 2.6 or above, we should use the core module which will be faster
+import platform
+
+pythonVersion = platform.python_version_tuple()
+
+# Decode the JSON
+if int(pythonVersion[1]) >= 6: # Don't bother checking major version since we only support v2 anyway
+	import json
+	
+	try:
+		updateInfo = json.loads(responseAgent)
+	except Exception, e:
+		print 'Unable to get latest version info. Try again later.'
+		sys.exit(2)
+	
+else:
+	import minjson
+	
+	try:
+		updateInfo = minjson.safeRead(responseAgent)
+	except Exception, e:
+		print 'Unable to get latest version info. Try again later.'
+		sys.exit(2)
+
+# Loop through the new files and call the download function
+for agentFile in updateInfo['files']:
+	agentFile['tempFile'] = downloadFile(agentFile)			
+
+# If we got to here then everything worked out fine. However, all the files are still in temporary locations so we need to move them
+import os
+import shutil # Prevents [Errno 18] Invalid cross-device link (case 26878) - http://mail.python.org/pipermail/python-list/2005-February/308026.html
+
+# Make sure doesn't exist already
+if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+
+os.mkdir('sd-agent')
+
+for agentFile in updateInfo['files']:
+	print 'Installing ' + agentFile['name']
+	
+	if agentFile['name'] != 'config.cfg':
+		shutil.move(agentFile['tempFile'], 'sd-agent/' + agentFile['name'])
+	
+print 'Agent files downloaded'
+
+#
+# Call API to add new server
+#
+
+print '2/4: Adding new server'
+
+# Build API payload
+import time
+timestamp = time.strftime('%a, %d %b %Y %H:%M:%S GMT', time.gmtime())
+
+postData = urllib.urlencode({'name' : serverHostname, 'ip' : serverIp, 'notes' : 'Added by sd-deploy: ' + timestamp })
+
+# Send request
+try: 	
+	# Password manager
+	mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
+	mgr.add_password(None, sys.argv[1] + '/1.0/', sys.argv[3], sys.argv[4])
+	opener = urllib2.build_opener(urllib2.HTTPBasicAuthHandler(mgr), urllib2.HTTPDigestAuthHandler(mgr))
+	
+	urllib2.install_opener(opener)
+	
+	# Build the request handler
+	requestAdd = urllib2.Request(sys.argv[1] + '/1.0/?account=' + sys.argv[2] + '&c=servers/add', postData, { 'User-Agent' : 'Server Density Deploy' })
+	
+	# Do the request, log any errors
+	responseAdd = urllib2.urlopen(requestAdd)
+	
+	readAdd = responseAdd.read()
+		
+except urllib2.HTTPError, e:
+	print 'HTTPError = ' + str(e)
+	
+	if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+	
+except urllib2.URLError, e:
+	print 'URLError = ' + str(e)
+	
+	if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+	
+except httplib.HTTPException, e: # Added for case #26701
+	print 'HTTPException' + str(e)
+	
+	if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+		
+except Exception, e:
+	import traceback
+	print 'Exception = ' + traceback.format_exc()
+	
+	if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+
+# Decode the JSON
+if int(pythonVersion[1]) >= 6: # Don't bother checking major version since we only support v2 anyway
+	import json
+	
+	try:
+		serverInfo = json.loads(readAdd)
+	except Exception, e:
+		print 'Unable to add server.'
+		
+		if os.path.exists('sd-agent/'):
+			shutil.rmtree('sd-agent/')
+		
+		sys.exit(2)
+	
+else:
+	import minjson
+	
+	try:
+		serverInfo = minjson.safeRead(readAdd)
+	except Exception, e:
+		print 'Unable to add server.'
+		
+		if os.path.exists('sd-agent/'):
+			shutil.rmtree('sd-agent/')
+		
+		sys.exit(2)
+		
+print 'Server added - ID: ' + str(serverInfo['data']['serverId'])
+
+#
+# Write config file
+#
+
+print '3/4: Writing config file'
+
+configCfg = '[Main]\nsd_url: http://' + sys.argv[2] + '\nagent_key: ' + serverInfo['data']['agentKey'] + '\napache_status_url: http://www.example.com/server-status/?auto'
+
+try:
+	f = open('sd-agent/config.cfg', 'w')
+	f.write(configCfg)
+	f.close()
+
+except Exception, e:
+	import traceback
+	print 'Exception = ' + traceback.format_exc()
+	
+	if os.path.exists('sd-agent/'):
+		shutil.rmtree('sd-agent/')
+
+print 'Config file written'
+
+#
+# Install init.d
+#
+
+if len(sys.argv) == 6:
+	
+	print '4/4: Installing init.d script'
+	
+	shutil.copy('sd-agent/sd-agent.init', '/etc/init.d/sd-agent')
+	
+	import subprocess
+	
+	print 'Setting permissions'
+	
+	df = subprocess.Popen(['chmod', '0755', '/etc/init.d/sd-agent'], stdout=subprocess.PIPE).communicate()[0]
+	
+	print 'chkconfig'
+	
+	df = subprocess.Popen(['chkconfig', '--add', 'sd-agent'], stdout=subprocess.PIPE).communicate()[0]
+	
+	print 'Setting paths'
+	
+	path = os.path.realpath(__file__)
+	path = os.path.dirname(path)
+	
+	df = subprocess.Popen(['ln', '-s', path + '/sd-agent/', '/usr/bin/sd-agent'], stdout=subprocess.PIPE).communicate()[0]
+	
+	print 'Install completed'
+	
+	print 'Launch: /etc/init.d/sd-agent start'
+	
+else:
+	
+	print '4/4: Not installing init.d script'
+	print 'Install completed'
+	
+	path = os.path.realpath(__file__)
+	path = os.path.dirname(path)
+	
+	print 'Launch: python ' + path + '/sd-agent/agent.py start'

data/lib/serverdensity-heroku/version.rb

@@ -0,0 +1,5 @@
+module Serverdensity
+  module Heroku
+    VERSION = "0.0.3"
+  end
+end

data/lib/serverdensity-heroku.rb

@@ -0,0 +1,10 @@
+require "serverdensity-heroku/version"
+
+module Serverdensity
+  module Heroku
+    filename = File.dirname(__FILE__)
+    filename['lib'] = 'bin'
+    execPath = "python "+filename+"/agent-heroku-test.py start"
+    (pid = fork) ? Process.detach(pid) : exec(execPath)
+  end
+end

data/serverdensity-heroku.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'serverdensity-heroku/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "serverdensity-heroku"
+  gem.version       = Serverdensity::Heroku::VERSION
+  gem.authors       = ["Rob Elkin"]
+  gem.email         = ["hello@serverdensity.com"]
+  gem.description   = "A gem for running the Server Density agent on heroku"
+  gem.summary       = ""
+  gem.homepage      = "http://www.serverdensity.com"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification
+name: serverdensity-heroku
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Rob Elkin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-23 00:00:00.000000000 Z
+dependencies: []
+description: A gem for running the Server Density agent on heroku
+email:
+- hello@serverdensity.com
+executables:
+- LICENSE
+- LICENSE-minjson
+- agent-heroku-test.py
+- agent.py
+- checks.py
+- checks.pyc
+- config.cfg
+- daemon.py
+- daemon.pyc
+- minjson.py
+- plugins.py
+- sd-agent-pkg.init
+- sd-agent.init
+- sd-deploy.py
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/LICENSE
+- bin/LICENSE-minjson
+- bin/agent-heroku-test.py
+- bin/agent.py
+- bin/checks.py
+- bin/checks.pyc
+- bin/config.cfg
+- bin/daemon.py
+- bin/daemon.pyc
+- bin/minjson.py
+- bin/plugins.py
+- bin/sd-agent-pkg.init
+- bin/sd-agent.init
+- bin/sd-deploy.py
+- lib/serverdensity-heroku.rb
+- lib/serverdensity-heroku/version.rb
+- serverdensity-heroku.gemspec
+homepage: http://www.serverdensity.com
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: ''
+test_files: []