# npackage example  https://svn.lliurex.net/pandora/n4d-ldap/trunk
# jinja2 http://jinja.pocoo.org/docs/templates

from jinja2 import Environment
from jinja2.loaders import FileSystemLoader
from jinja2 import Template
import tempfile
import shutil
import os
import subprocess
import tarfile
import time
import glob


class ApacheManager:

	def __init__(self):
		#Load template file
		self.tpl_env = Environment(loader=FileSystemLoader('/usr/share/n4d/templates/apache'))
		self.backup_files=[]
		self.backup_dirs=["/etc/apache2/","/etc/easysites/","/var/www/","/net/server-sync/easy-sites"]
		self.easysites_dirs=["/var/lib/lliurex-www/links/","/usr/share/lliurex-www/srv/icons/"]

	#def init
	
	def startup(self,options):
		# executed when launching n4d
		pass
		
	#def startup

	def apt(self):
		# executed after apt operations
		pass
		
	#def apt
	
	# service test and backup functions #
	def test(self):

		pass

	#def makedir
	def makedir(self,dir=None):

		if not os.path.isdir(dir):
			os.makedirs(dir)
		return [True]

	#def timestamp backup
	def get_time(self):
		return get_backup_name("ApacheManager")

	#def backup
	def backup(self,dir="/backup"):
		
		try:
		
			self.makedir(dir)
			file_path=dir+"/"+self.get_time()
			tar=tarfile.open(file_path,"w:gz")
				
			for f in self.backup_files:
				if os.path.exists(f):
					tar.add(f)

			for d in self.backup_dirs:
				if os.path.exists(d):
					tar.add(d)
			
			'''
			self.list_easy_sites = glob.glob('/var/www/srv/'+'easy-'+'*') + glob.glob("/var/www/srv/links/easy-*")
			
			for j in self.list_easy_sites:
				if os.path.exists(j):
					tar.add(j)

			
			self.list_easy_site_hide=glob.glob("/var/www/srv/links/hide_links/easy-*")
			for j in self.list_easy_site_hide:
				if os.path.exists(j):
					tar.add(j)	
					
			self.list_easy_site_icons=glob.glob("/var/www/srv/icons/easy-*")
			for j in self.list_easy_site_icons:
				if os.path.exists(j):
					tar.add(j)				
			'''
			if os.path.exists(self.easysites_dirs[0]):
				easy_links=glob.glob(self.easysites_dirs[0]+'easy-'+'*')
				for j in easy_links:
					if os.path.exists(j):
						tar.add(j)
			
			if os.path.exists(self.easysites_dirs[1]):
				easy_icons=glob.glob(self.easysites_dirs[1]+'easy-'+'*')
				for j in easy_icons:
					if os.path.exists(j):
						tar.add(j)			
			
			dir="/net/server-sync/easy-sites"
			if os.path.exists(dir):
				cmd="getfacl -pR %s > /tmp/easy-sites.acl"%dir
				os.system(cmd)
				tar.add("/tmp/easy-sites.acl",arcname="easy-sites.acl")
				os.remove("/tmp/easy-sites.acl")
				
				
			os.system("a2query -m > /tmp/mods_available")
			
			tar.add("/tmp/mods_available",arcname="llx-backup_mods_available")
			tar.close()

			return [True,file_path]

		except Exception as e:
			return [False,str(e)]

	#def restore backup
	def restore(self,file_path=None):


		if file_path==None:
			dir="/backup"
			for f in sorted(os.listdir(dir),reverse=True):
				if "ApacheManager" in f:
					file_path=dir+"/"+f
					break


		try:
			if os.path.exists(file_path):
				tmp_dir=tempfile.mkdtemp()
				tar=tarfile.open(file_path)
				tar.extractall(tmp_dir)
				tar.close()

				os.system("rm -rf %s"%tmp_dir+"/etc/apache2/mods-available")
				os.system("rm -rf %s"%tmp_dir+"/etc/apache2/mods-enabled")
				

				try:
					if objects["ServerBackupManager"].restoring_version=="14.06":
						
						print "[ApacheManager] Removing old apache conf files..."
						
						files_to_remove=["/etc/apache2/apache2.conf","/etc/apache2/envvars","/etc/apache2/magic","/etc/apache2/ports.conf","/etc/apache2/httpd.conf","/var/www/index.html"]
						
						for f in files_to_remove:
							print tmp_dir+f,os.path.exists(tmp_dir+f)
							if os.path.exists(tmp_dir+f):
								os.remove(tmp_dir+f)
						os.system("rm -rf %s"%tmp_dir+"/etc/apache2/conf.d")
						os.system("ls -la %s"%tmp_dir+"/etc/apache2/")
						
						
				except Exception as llx1406e:
					print llx1406e
			


				for f in self.backup_files:
					tmp_path=tmp_dir+f
					if os.path.exists(tmp_path):
						shutil.copy(tmp_path,f)

				for d in self.backup_dirs:
					tmp_path=tmp_dir+d
					if os.path.exists(tmp_path):
						self.makedir(d)
						cmd="cp -r " + tmp_path +"/* "  + d
						os.system(cmd)

				try:
					for d in self.easysites_dirs:
						tmp_path=tmp_dir+d
						tmp_files=glob.glob(tmp_path+'easy-'+'*')
						for file in tmp_files:
							if os.path.exists(file):
								cmd="cp "+file+" "+d
								os.system(cmd)
								if os.path.exists(os.path.join(d,os.path.basename(file))):
									if 'lib/lliurex-www/links' in os.path.join(d,os.path.basename(file)):
										cmd="chown www-data:www-data "+ os.path.join(d,os.path.basename(file))
										os.system(cmd)
				
				except:
					pass						
				#Add pmb redirection to port 800 (apache2-lliurex)
				sw_loadProxyMods=False
				for confFile in ["/etc/apache2/sites-enabled/pmb.conf","/etc/apache2/sites-enabled/opac.conf"]:
					if os.path.exists(confFile):
						sw_loadProxyMods=True
						site=confFile.split('/')[-1]
						site=site.replace('.conf','')
						confF=open(confFile,'r')
						contentF=confF.readlines()
						confF.close()
						pos=0
						aliasPos=0
						for line in contentF:
							pos=pos+1
							if "Alias " in line:
								aliasPos=pos
							if "ProxyPass" in line:
								aliasPos=-1
								break

						if aliasPos>0:
							contentF.insert(aliasPos,"\n\tProxyPass / http://"+site+":800/\n\tProxyPassReverse / http://"+site+":800/\n")
							confF=open(confFile,'w')
							confF.writelines(contentF)
							confF.close()

				if sw_loadProxyMods:	
					#Ensure that proxy mods are enabled
					cmd="/usr/sbin/a2enmod proxy proxy_http"
					os.system(cmd)

				if os.path.exists("/net/server-sync/easy-sites") and os.path.exists(tmp_dir+"/easy-sites.acl"):
					os.system("setfacl -R --restore=%s/easy-sites.acl"%tmp_dir)
					
				
				
				# # 14.06 -> 15.05 CASE #############################
				
				try:
					if objects["ServerBackupManager"].restoring_version=="14.06":
						
						print("[ApacheManager] Fixing files ...")
						
						dir="/etc/apache2/sites-available/"
						
						for item in os.walk(dir):
							path,data,files=item
						
						for f in files:
							path=dir+f
							if not os.path.exists(path):
								continue
							if f=="default":
								os.remove(path)
								continue
								if os.path.exists(dir+"000-default.conf"):
									os.remove(dir+"000-default.conf")
								os.symlink("/etc/apache2/sites-available/llx-default.conf",dir+"000-default.conf")
							elif f=="default.lliurex":
								os.remove(path)
								continue
							elif f=="000-default.orig":
								pass
							elif not f.endswith(".conf"):
								shutil.move(path,path+".conf")

						print dir,"DONE"

						dir="/etc/apache2/sites-enabled/"
						files=[]
						
						for item in os.walk(dir):
							path,data,files=item

						
						for f in files:
							path=dir+f
							link=os.readlink(path)
							if link.startswith("../"):
								link=link.replace("../","/etc/apache2/")
							
							if f=="000-default":
								os.remove(path)
								if os.path.exists("/etc/apache2/sites-enabled/000-default.conf"):
									os.remove("/etc/apache2/sites-enabled/000-default.conf")
								os.symlink("/etc/apache2/sites-available/000-default.conf","/etc/apache2/sites-enabled/000-default.conf")
								continue
							

							
							if not path.endswith(".conf"):
								os.remove(path)
								if not os.path.exists(link):
									link+=".conf"
								
								if not os.path.exists(link):
									continue
								if os.path.exists(path+".conf"):
									os.remove(path+".conf")
								os.symlink(link,path+".conf")
								
							else:
								if not os.path.exists(link):
									os.remove(path)
								
						
				except Exception as llx14_ex:
					print llx14_ex
					
				# #END OF 14.06->15.05 CASE # ############
				
				os.system("service apache2 restart")

                                return [True,""]

			else:
				return [False,"Backup file not found"]

		except Exception as e:

			return [False,str(e)]


	#def restore

	def load_exports(self):
		#Get template
		template_server = self.tpl_env.get_template("server")
		list_variables = {}
		
		#Inicialize INTERNAL_DOMAIN
		list_variables['INTERNAL_DOMAIN'] = objects['VariablesManager'].get_variable('INTERNAL_DOMAIN')
		#If INT_DOMAIN is not defined calculate it with args values
		if  list_variables['INTERNAL_DOMAIN'] == None:
			return {'status':False,'msg':'Variable INTERNAL_DOMAIN not defined'}
		
		#Inicialize INTERNAL_DOMAIN
		list_variables['HOSTNAME'] = objects['VariablesManager'].get_variable('HOSTNAME')
		#If INT_DOMAIN is not defined calculate it with args values
		if  list_variables['HOSTNAME'] == None:
			return {'status':False,'msg':'Variable HOSTNAME not defined'}
		
		###########################
		#Setting VARS
		###########################
		
		#Set HTTP_PATH 
		list_variables['HTTP_PATH'] = objects['VariablesManager'].get_variable('HTTP_PATH')
		#If variable PROXY_ENABLED is not defined calculate it with args values
		if  list_variables['HTTP_PATH'] == None:
			status,list_variables['HTTP_PATH'] = objects['VariablesManager'].init_variable('HTTP_PATH',{'PATH':'/var/www/'})
		
		#Encode vars to UTF-8
		string_template = template_server.render(list_variables).encode('UTF-8')
		#Open template file
		fd, tmpfilepath = tempfile.mkstemp()
		new_export_file = open(tmpfilepath,'w')
		new_export_file.write(string_template)
		new_export_file.close()
		os.close(fd)
		#Write template values
		n4d_mv(tmpfilepath,'/etc/apache2/sites-available/server',True,'root','root','0644',False )
		#Restart service
		subprocess.Popen(['a2ensite','server.conf'],stdout=subprocess.PIPE).communicate()
		subprocess.Popen(['/etc/init.d/apache2','reload'],stdout=subprocess.PIPE).communicate()
		return {'status':True,'msg':'Exports written'}
	#def load_exports

	def reboot_apache(self):
		#Restart apache service
		subprocess.Popen(['/etc/init.d/apache2','restart'],stdout=subprocess.PIPE).communicate()
		return {'status':True,'msg':'apache2 rebooted'}
	#def reboot_squid

	# ######################### #
	
#class ApacheManager