#!/usr/bin/python3
import os
import subprocess
from importlib.machinery import SourceFileLoader
backupmanager=SourceFileLoader("BackupManager","/usr/share/n4d/python-plugins/support/BackupManager.py").load_module()
import n4d.responses


class PmbManager:
	
	def __init__(self):
		self.pmb=backupmanager.BackupManager(app='pmb')
		pmbFiles={} 
		pmbDbFiles={'pmb':['/etc/pmb/debian-db.php']} 

		self.pmb.set_app_files(pmbFiles)
		self.pmb.set_app_dbFiles(pmbDbFiles) 

	def restore(self,file_path=None):
		self.pmb.set_backup_name("PmbManager")
		retVal=self.pmb.restore(file_path)
		self.change_pmb_version()
		if retVal[0]:
			return n4d.responses.build_successful_call_response()
		else:
			return n4d.responses.build_failed_call_response('',retVal[1])


	def backup(self,dir='/backup'):
		#self.pmb.set_backup_name("pmb")
		retVal=self.pmb.backup("PmbManager",dir)
		self.last_operations()
		if retVal[0]:
			return n4d.responses.build_successful_call_response(retVal[1])
		else:
			return n4d.responses.build_failed_call_response('',retVal[1])

	def last_operations(self):
		#Regenerate cnames 
		cmd="lliurex-pmb --hosts"
		return (os.system(cmd))
		
	def change_pmb_version(self):

		mysql_command='mysql -uroot -p$(sudo mysql_root_passwd -g) -e '
		#Get bdd_version value frothom parametres table
		sql='"select valeur_param from pmb.parametres where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
		cmd=mysql_command + sql
		p=subprocess.check_output(cmd,shell=True)
		version=p.split("\n")[1]
		
		if version=="v4.47":
			sql='"update pmb.parametres set valeur_param=\'vLlxNemo\' where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
			cmd=mysql_command + sql
			os.system(cmd)
		elif version=="v5.10":
			sql='"update pmb.parametres set valeur_param=\'vLlxPandora\' where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
			cmd=mysql_command + sql
			os.system(cmd)
		elif version=="v5.14":
			sql='"update pmb.parametres set valeur_param=\'vLlxTrusty\' where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
			cmd=mysql_command + sql
			os.system(cmd)
		elif version=="v5.19":
			sql='"update pmb.parametres set valeur_param=\'vLlxXenial\' where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
			cmd=mysql_command + sql
			os.system(cmd)	
		elif version=="v5.28":
			sql='"update pmb.parametres set valeur_param=\'vLlxXenialPlus\' where type_param=\'pmb\' and sstype_param=\'bdd_version\'"'
			cmd=mysql_command + sql
			os.system(cmd)