from servicebase import ServiceBase
import requests
import json

class ServiceDelegate(ServiceBase) :

	current_auth_header = None
	return_payload = {}
	host = None

	def get_arguments(cls) :
		return ['-n', '--net', 'store_true', "Return network information about the subject"]

	def startup(self) :
		for requirement in ['host','username','key'] :
			if requirement not in self.config or (requirement in self.config and (self.config[requirement] is '' or type(self.config[requirement]) is not str)):
				self.return_payload.update({'error':'Missing required config option ' + requirement})
				return
		self.host = self.config['host']
		self.current_auth_header = self.get_bc_auth_header(self.config['host'],self.config['username'],self.config['key'])

	def perform_lookup(self,host_tuple) :
		self.return_payload = {}
		if self.current_auth_header :
			object = self.search_bc_object(host_tuple)
			self.return_payload.update({'object':object})
			parent = self.search_bc_parent_object(host_tuple)
			self.return_payload.update({'parent':parent})
			return self.return_payload

	def shutdown(self) :
		if self.host and self.current_auth_header :
			logout = requests.get("https://" + self.host + "/Services/REST/v1/logout",headers=self.current_auth_header)
			if not logout.text.endswith("successfully logged out.\"") :
				self.return_payload.update({'error':'Unable to log out of BlueCat API session'})


	def get_bc_auth_header(self,host,username,password) :
		auth_token = requests.get("https://" + host + "/Services/REST/v1/login?username=" + username + "&password=" + password).text[17:71]
		if auth_token.startswith('BAMAuthToken') :
			return { "Authorization" : auth_token }
		else :
			return None

	def search_bc_object(self,host_tuple) :
		"""Searches BC for the subject."""
		response = requests.get("https://" + self.host + "/Services/REST/v1/searchByObjectTypes?keyword=" + host_tuple[2][0] + "&types=IP4Address,IP6Address&start=0&count=10",headers=self.current_auth_header)
		return self.clean_response(response)

	def search_bc_parent_object(self,host_tuple) :
		"""Search for and return information about the objects parent, if any"""
		if 'object' in self.return_payload and 'id' in self.return_payload['object'] :
			response =  requests.get("https://" + self.host + "/Services/REST/v1/getParent?entityId=" + str(self.return_payload['object']['id']), headers=self.current_auth_header)
			return self.clean_response(response)

	def clean_response(self,response) :
		parsed = {}
		try :
			parsed = json.loads(response.text)
		except Exception :
			return parsed
		if type(parsed) is list and len(parsed) > 0 :
			parsed = parsed[0]
		if type(parsed) is dict :
			if 'properties' in parsed :
				parsed["properties"] = parsed["properties"][:-1]
				bad_delim_props = [x for x in parsed["properties"].split("|") if x.count('=') > 1]
				if len(bad_delim_props) > 0 :
					parsed.update({"unknown":bad_delim_props})
				attributes = dict(x.split("=") for x in parsed["properties"].split("|") if x.count('=') == 1)
				parsed.pop("properties", None)
				parsed.update(attributes)
			return parsed