#!/usr/bin/python3

import tempfile
import shutil
import os
import subprocess
import time
import tarfile
import n4d.responses
from n4d.utils import get_backup_name, n4d_mv


class CupsManager:
	
	
	def __init__(self):
		
		self.backup_files=["/etc/cups/cupsd.conf","/etc/cups/printers.conf"]
		self.backup_dirs=["/etc/cups/ppd/"]
		
		
	#def init
	
	def get_time(self):
		
		return get_backup_name("CupsManager")
		
	#def get_time
	
	
	def backup(self,dir="/backup/"):
		
		try:
		
			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)
			
			tar.close()
			
			#Old n4d:return [True,file_path]
			return n4d.responses.build_successful_call_response(file_path)
			
		except Exception as e:
			#Old n4d: return [False,str(e)]
			return n4d.responses.build_failed_call_response('',str(e))
		
	#def test
	
	def restore(self,file_path=None):


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

		if file_path==None:
			#Old n4d:return [False,"Backup file not found"]
			return n4d.responses.build_successful_call_response(False,'Backup file not found')

		try:

			if os.path.exists(file_path):
				
				tmp_dir=tempfile.mkdtemp()
				tar=tarfile.open(file_path)
				tar.extractall(tmp_dir)
				tar.close()
				
				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):
						cmd="cp -r " + tmp_path +"/* "  + d
						os.system(cmd)
				
				os.system("systemctl restart cups.service")
						
				#Old n4d: return [True,""]
				return n4d.responses.build_successful_call_response(True)
				
		except Exception as e:
				
			#return [False,str(e)]
			return n4d.responses.build_failed_call_response('',str(e))
		
	#def test


	
	
#class CupsManager