# -*- coding: utf-8 -*-
import sys
import os
import tarfile
import tempfile
import time
import ConfigParser

class ServerBackupManager:

	def __init__(self):
		
		self.backup_list=["VariablesManager","Hostname","NetworkManager","Dnsmasq","N4dProxy","SlapdManager","PamnssPlugin","SambaManager"]
		#self.backup_list=["VariablesManager","Hostname","NetworkManager","Dnsmasq","N4dProxy","SlapdManager","PamnssPlugin","SambaManager","NetFoldersManager","MysqlManager","CupsManager","ApacheManager"]
		
	#def init

	def backup(self,path=None,service_list=None,netfolders_list=None):
		
		return self.server_basics_backup(path,service_list,netfolders_list)
		
	#def backup
	
	def get_basic_services_list(self):
		
		return self.backup_list
		
	#def get_basic_services_list
		
	
	def server_basics_backup(self,path=None,service_list=None,netfolders_list=None):


		if service_list==None:
			service_list=self.backup_list
		
		if path==None:
			cur_date=time.strftime("%d%m%Y")
			path="/backup/%s"%cur_date
			if not os.path.exists(path):
				os.makedirs(path)
		
		if os.path.exists(path):
			
			file_name=path+"/"+get_backup_name("ServerBackup")
			tar=tarfile.open(file_name,"w:gz")
			
			f=open(path+"/backup_files.conf","w")
			f.write("[ServerBackupManager]\n")
			
			ret={}
			for service in service_list:
				try:

					if service=="NetFoldersManager":
						ret[service]=objects[service].backup(netfolders_list,path)
					else:
						ret[service]=objects[service].backup(path)
						
					if ret[service][0]:
						fname=ret[service][1].split("/")[-1]
						tar.add(ret[service][1],arcname=fname)
						f.write("%s=%s\n"%(service,fname))
						os.remove(ret[service][1])	
			
				except Exception as e:
					ret[service]=[False,str(e)]
					f.write("# [!] %s FAILED: %s\n"%(service,str(e)))
				
		
			f.close()
			
			os.system("lliurex-version -n > %s/lliurex-version"%path)
			tar.add(path+"/backup_files.conf",arcname="backup_files.conf")
			tar.add(path+"/lliurex-version",arcname="lliurex-version")
			os.remove(path+"/backup_files.conf")
			os.remove("%s/lliurex-version"%path)
			tar.close()
			
			os.system("chmod 660 %s"%file_name)
			os.system("chown root:admins %s"%file_name)
			
			return [True,ret,file_name]
			
		else:
			
			return [False,ret,"Backup file not found"]
		
	#def backup
	
	def restore(self,file_name):
		
		return self.server_basics_restore(file_name)
	
	def server_basics_restore(self,file_name):
		
		if os.path.exists(file_name):
		
			tmp_dir=tempfile.mkdtemp()
			tar=tarfile.open(file_name,"r")
			tar.extractall(tmp_dir)
			tar.close()
			if os.path.exists(tmp_dir+"/backup_files.conf"):
				try:
					cfg=ConfigParser.ConfigParser()
					cfg.optionxform=str
					cfg.read(tmp_dir+"/backup_files.conf")
					ret={}
					
					self.restoring_version=None
				
					if os.path.exists(tmp_dir+"/lliurex-version"):
						f=open(tmp_dir+"/lliurex-version")
						self.restoring_version=f.readline().strip("\n")
						self.restoring_version=".".join(self.restoring_version.split(".")[0:2])
					for items in cfg.items("ServerBackupManager"):
						key,value=items
						try:
							if key in objects:
								ret[key]=objects[key].restore(tmp_dir+"/"+value)

							else:
								ret[key]=[False,"Plugin not found"]
						except Exception as e:
							print e
							ret[key]=str(e)
							
					#Fix ipxeboot symlink restore
					ipxeLinkPath=("/var/www/ipxeboot")
					ipxeRealPath=os.path.realpath(ipxeLinkPath)
					try:
						if not(os.path.exists(ipxeRealPath)):
							print ("Fixing ipxeboot symlink...")
							os.remove(ipxeLinkPath)
							os.symlink("/usr/share/llxbootmanager/www-boot",ipxeLinkPath)
					except Exception as e:
						print e
						ret[key]=str(e)
					
					self.final_operations()
					self.restoring_version=None
					return [True,ret]
					
				except Exception as e:
					return [False,str(e)]
					
			else:
				return [False,{},"File not found"]
			
		else:
			return [False,{},"File not found"]
			
	
		return [False,{},"Failed. Did nothing"]
		
		
	#def server_basics_restore
	
	
	def final_operations(self):
		

		try:
			objects["Golem"].startup(None)
			objects["NetFoldersManager"].startup(None)
						
			for item in objects["Golem"].light_get_user_list():
				
				user_info={}
				user_info["profile"]=item[5]
				user_info["uid"]=item[1]
				user_info["uidNumber"]=item[2]
				try:
					objects["Golem"].exist_home_or_create(user_info)
				except:
					pass
						
			objects["Golem"].restore_groups_folders()
			#Enable acls and roadmin just in case
			command='/usr/share/n4d-samba/one-shots/add-roadmin-user.py'
			exitStatus=os.system(command)
			objects["SlapdManager"].load_acls()

			objects["ZeroServerWizardManager"].end_operations()
					
		except Exception as e:
				
			print(e)

		
	#def final_operations

#class ServerBackupManager


if __name__=="__main__":
	
	golem=ServerBackupManager()