from Config import Config

class AutoConf :

	def __init__(self, swarm) :
		self.__swarm = swarm
		self.__instances = {}
		self.__sites = {}
		self.__config = Config(self.__swarm)

	def pre_process(self, objs) :
		for instance in objs :
			(id, name, labels) = self.__get_infos(instance)
			if "bunkerized-nginx.AUTOCONF" in labels :
				if self.__swarm :
					self.__process_instance(instance, "create", id, name, labels)
				else :
					if instance.status in ("restarting", "running", "created", "exited") :
						self.__process_instance(instance, "create", id, name, labels)
					if instance.status == "running" :
						self.__process_instance(instance, "start", id, name, labels)

		for server in objs :
			(id, name, labels) = self.__get_infos(server)
			if "bunkerized-nginx.SERVER_NAME" in labels :
				if self.__swarm :
					self.__process_server(server, "create", id, name, labels)
				else :
					if server.status in ("restarting", "running", "created", "exited") :
						self.__process_server(server, "create", id, name, labels)
					if server.status == "running" :
						self.__process_server(server, "start", id, name, labels)

	def process(self, obj, event) :
		(id, name, labels) = self.__get_infos(obj)
		if "bunkerized-nginx.AUTOCONF" in labels :
			self.__process_instance(obj, event, id, name, labels)
		elif "bunkerized-nginx.SERVER_NAME" in labels :
			self.__process_server(obj, event, id, name, labels)

	def __get_infos(self, obj) :
		if self.__swarm :
        	        id = obj["Actor"]["ID"]
			name = obj["Actor"]["Attributes"]["name"]
	                labels = obj.attrs["Spec"]["Labels"]
		else :
			id = obj.id
			name = obj.name
			labels = obj.labels
		return (id, name, labels)

	def __process_instance(self, instance, event, id, name, labels) :
		if event == "create" :
			self.__instances[id] = obj
			utils.log("[*] bunkerized-nginx instance created : " + name + " / " + id)
		elif event == "start" :
			self.__instances[id].reload()
			utils.log("[*] bunkerized-nginx instance started : " + name + " / " + id)
		elif event == "die" :
			self.__instances[id].reload()
			utils.log("[*] bunkerized-nginx instance stopped : " + name + " / " + id)
		elif event == "destroy" or event == "remove" :
			del self.__instances[id]
			utils.log("[*] bunkerized-nginx instance removed : " + name + " / " + id)

	def __process_server(self, instance, event, id, name, labels) :
		vars = { k.replace("bunkerized-nginx.", "", 1) : v for k, v in labels.items() if k.startswith("bunkerized-nginx.")}
		if event == "create" :
			if self.__config.generate(instances, vars) :
				utils.log("[*] Generated config for " + vars["SERVER_NAME"])
				self.__servers[id] = obj
				if self.__swarm :
					if self.__config.activate(instances, vars) :
						utils.log("[*] Activated config for " + vars["SERVER_NAME"])
					else :
						utils.log("[!] Can't activate config for " + vars["SERVER_NAME"])
			else :
				utils.log("[!] Can't generate config for " + vars["SERVER_NAME"])
		elif event == "start" :
			if id in self.__servers :
				self.__servers[id].reload()
				if self.__config.activate(instances, vars) :
					utils.log("[*] Activated config for " + vars["SERVER_NAME"])
				else :
					utils.log("[!] Can't activate config for " + vars["SERVER_NAME"])
		elif event == "die" :
			if id in self.__servers :
				self.__servers[id].reload()
				if self.__config.deactivate(instances, vars) :
					utils.log("[*] Deactivated config for " + vars["SERVER_NAME"])
				else :
					utils.log("[!] Can't deactivate config for " + vars["SERVER_NAME"])
		elif event == "destroy" or event == "remove" :
			if id in self.__servers :
				if self.__swarm :
					if self.__config.deactivate(instances, vars) :
						utils.log("[*] Deactivated config for " + vars["SERVER_NAME"])
					else :
						utils.log("[!] Can't deactivate config for " + vars["SERVER_NAME"])+
				del self.__servers[id]
				if self.__config.remove(vars) :
					utils.log("[*] Removed config for " + vars["SERVER_NAME"])
				else :
					utils.log("[!] Can't remove config for " + vars["SERVER_NAME"])