from base import Task
from common import phases
from common.tasks import host


class AddExternalCommands(Task):
	description = 'Determining required external commands for EC2 bootstrapping'
	phase = phases.preparation
	successors = [host.CheckExternalCommands]

	@classmethod
	def run(cls, info):
		if info.manifest.volume['backing'] == 's3':
			info.host_dependencies['euca-bundle-image'] = 'euca2ools'
			info.host_dependencies['euca-upload-bundle'] = 'euca2ools'


class GetInfo(Task):
	description = 'Retrieving instance metadata'
	phase = phases.preparation

	@classmethod
	def run(cls, info):
		import urllib2
		import json
		metadata_url = 'http://169.254.169.254/latest/dynamic/instance-identity/document'
		response = urllib2.urlopen(url=metadata_url, timeout=5)
		info.host = json.load(response)
		return info