import json
import os
import subprocess

class LlxBootManager:
	def __init__(self):
		self.cfgpath="/etc/llxbootmanager/bootcfg.json"
		# Cal canviar la ruta!!!
		self.php_path="/usr/share/llxbootmanager/www-boot"
		# Clients Boot Config
		self.clients_conf_path="/etc/llxbootmanager/clients.json"
		pass
	#def __init__


	def getBootList(self):
		'''
		Calculates and return all the options for the iPXE Boot Menu
		'''

		try:
			command=["php", self.php_path+"/getmenujson.php"]
			proc = subprocess.Popen(command,  stdout=subprocess.PIPE, cwd=self.php_path)
			out, err = proc.communicate()
			return json.loads(out)
		except Exception as e:
			print "Exception: "+str(e)
			return -1



	def getBootOrder(self):
		'''
		Returns the boot order for iPE Boot Menu
		'''
		try:
			json_data=open(self.cfgpath);
			data=json.load(json_data);
			return data["bootorder"];
		except Exception as e:
			print "Exception: "+str(e)
			return -1

	def pushToBootList(self, label):
		'''
		Adds new label to Boot order for iPE Boot Menu
		'''
		
		try:
			print "label is "+label
			json_data=open(self.cfgpath);
			data=json.load(json_data);

			# Removing label before push it
			data['bootorder']=list(filter(lambda a: a !=label.encode("utf-8"), data['bootorder']))
			# Cleaning
			data['bootorder']=list(filter(None, data['bootorder']))

			# Push label
			data["bootorder"].append(label.encode("utf-8"));
			return (self.setBootOrder(*data["bootorder"]));

		except Exception as e:
			print "Exception: "+str(e)
			return -1
		
	def removeFromBootList(self, label):
		'''
		Adds new label to Boot order for iPE Boot Menu
		'''
		
		try:
			print "label is "+label
			json_data=open(self.cfgpath);
			data=json.load(json_data);

			# Removing label occurences
			data['bootorder']=list(filter(lambda a: a !=label.encode("utf-8"), data['bootorder']))
			# Cleaning
			data['bootorder']=list(filter(None, data['bootorder']))

			return (self.setBootOrder(*data["bootorder"]));

		except Exception as e:
			print "Exception: "+str(e)
			return -1

	def prependBootList(self, label):
		'''
		Appending Boot List with label and returns the boot order for iPE Boot Menu
		'''
		try:
			print "[LlxBootManager] Prepending label "+label+" to Boot List."
			json_data=open(self.cfgpath);
			data=json.load(json_data);

			# Removing label before push it
			data['bootorder']=list(filter(lambda a: a !=label.encode("utf-8"), data['bootorder']))

			data["bootorder"].insert(0,label.encode("utf-8"));
			return (self.setBootOrder(*data["bootorder"]));

		except Exception as e:
			print "Exception: "+str(e)
			return -1



	def getBootTimer(self):
		'''
		Returns menu time for PXE Menu
		'''
		try:
			json_data=open(self.cfgpath);
			data=json.load(json_data);
			return data["timeout"]
		except Exception as e:
			print "Exception: "+str(e)
			return -1

	def setBootOrder(self, *order):
		'''
		Set Boot order order for iPE Boot Menu
		'''
		try:
			time=self.getBootTimer();

			order=list(filter(None, order))

			bootcfg= { "bootorder": order, "timeout": time }

			bootcfg_string = unicode(json.dumps(bootcfg,indent=4,encoding="utf-8",ensure_ascii=False)).encode("utf-8")

			f = open(self.cfgpath,'w')
			f.writelines(bootcfg_string)
			f.close()

		except Exception as e:
			print "Exception: "+str(e)
			return -1



		pass

	def setBootTimer(self, time):
		'''
		Set timeout for pxe menu
		'''
		bootorder=self.getBootOrder();

		bootcfg= { "bootorder": bootorder, "timeout": time }

		bootcfg_string = unicode(json.dumps(bootcfg,indent=4,encoding="utf-8",ensure_ascii=False)).encode("utf-8")

		f = open(self.cfgpath,'w')
		f.writelines(bootcfg_string)
		f.close()
		pass


	# Methods to configure boot clients

	def getClientsConfig(self):
		'''
		Returns content for /etc/llxbootmanager/clients.json
		'''
		try:
			f = open(self.clients_conf_path,'r');
			data = (json.load(f));
			f.close();
			return (json.dumps(data));
		except Exception as e:
                        return False

		pass


        def getClientConfig(self, mac):
                '''
                Return boot for an specific mac
                '''

                try:
                        f=open(self.clients_conf_path, 'r')
                        clients = (json.load(f))
                        f.close()

                        for cl in clients["clients"]:
                                if (cl["mac"]==mac):
                                        return cl["boot"]

                        # if not fount...
                        return False
                                
                except Exception as e:
                        return False



	def setClientConfig(self, *args):
		'''
		configures boot for certain mac. args[0] specifies mac, and args[1] specifies boot
		if boot (args[1]) is unspecified, removes it from config
		'''

		if(len(args)!=1 and len(args)!=2):
			return False;

		# Reading config file
		try:
	                f = open(self.clients_conf_path,'r');
        	        clients = (json.load(f));
			f.close();
		except Exception as e:
			# File does not exists. Create an empty json...
			clients={"clients":[]}

		if(len(args)>=1):
			# Remove item if exists. It runs if we want to delete an item or modify any existent
			mac=args[0]

                        for client in clients["clients"]:
                                #print "compare: "+client["mac"]+" amb "+mac
                                if (client["mac"]==mac):
                                        #print "FOUND"
                                        clients["clients"].remove(client)

		if(len(args)==2): # If we are specifying any boot option, let's add it
			boot=args[1]

			newclient={"mac":mac, "boot":boot}
			clients["clients"].append(newclient);

		# Finally Save Results
		with open(self.clients_conf_path, 'w') as f:
			json.dump(clients, f)

		pass