import jinja2, glob, os, pathlib, copy

class Templator :

	def __init__(self, config, input_path, output_path, target_path) :
		self.__config = config
		self.__input_path = input_path
		self.__output_path = output_path
		self.__target_path = target_path
		if not self.__target_path.endswith("/") :
			self.__target_path += "/"
		self.__template_env = jinja2.Environment(loader=jinja2.FileSystemLoader(searchpath=self.__input_path))

	def render_global(self) :
		return self.__render("global")

	def render_site(self, server_name=None, first_server=None) :
		if server_name is None :
			server_name = self.__config["SERVER_NAME"]
		if first_server is None :
			first_server = self.__config["SERVER_NAME"].split(" ")[0]
		return self.__render("site", server_name, first_server)

	def __render(self, type, server_name=None, first_server=None) :
		real_config = copy.deepcopy(self.__config)
		if not server_name is None :
			real_config["SERVER_NAME"] = server_name
		if not first_server is None :
			real_config["FIRST_SERVER"] = first_server
		real_config["NGINX_PREFIX"] = self.__target_path
		if real_config["MULTISITE"] == "yes" and type == "site" :
			real_config["NGINX_PREFIX"] += first_server + "/"
		for filename in glob.iglob(self.__input_path + "/" + type + "**/**", recursive=True) :
			if os.path.isfile(filename) :
				relative_filename = filename.replace(self.__input_path, "").replace(type + "/", "")
				template = self.__template_env.get_template(type + "/" + relative_filename)
				template.globals["has_value"] = Templator.has_value
				output = template.render(real_config, all=real_config)
				if real_config["MULTISITE"] == "yes" and type == "site" :
					relative_filename = first_server + "/" + relative_filename
				if "/" in relative_filename :
					directory = relative_filename.replace(relative_filename.split("/")[-1], "")
					pathlib.Path(self.__output_path + "/" + directory).mkdir(parents=True, exist_ok=True)
				with open(self.__output_path + "/" + relative_filename, "w") as f :
					f.write(output)

	@jinja2.contextfunction
	def has_value(context, name, value) :
		for k, v in context.items() :
			if (k == name or k.endswith("_" + name)) and v == value :
				return True
		return False