# -*- coding: utf-8 -*-

import os
import os.path
import base64
import shutil
import pwd

NET_FOLDER="/net/.lliurex-harvester/"


class TeacherShareManager:

	def __init__(self):
		
		if not os.path.exists(NET_FOLDER):
			os.mkdir(NET_FOLDER)
		
		self.paths={}
		
	#def __init__
	
	def remove_path(self,uid):
		
		if uid in self.paths:
			path,name,ip,port=self.paths[uid]
			self.paths.pop(uid)
			try:
				shutil.rmtree(NET_FOLDER+"."+uid+"_"+name)
			except Exception as e:
				print e
					
		return True
		
	#def remove_path
	
	def add_path(self,user,path,name,ip,port):
		
		dir=NET_FOLDER+"."+user+"_"+name
		
		if user in self.paths:
			try:
				p,n,i,port=self.paths[user]
				print NET_FOLDER+"."+user+"_"+n
				shutil.rmtree(NET_FOLDER+"."+user+"_"+n)
			except Exception as e:
				# folder might not exist
				print e
				
		for item in os.listdir(NET_FOLDER):
			if item.find("."+user)==0:
				shutil.rmtree(NET_FOLDER+item)
				
			
		prevmask = os.umask(0)
		
		try:
			self.paths[user]=(path,name,ip,port)
			os.mkdir(dir)
			teacher_uid=pwd.getpwnam(user)[2]
			teacher_gid=pwd.getpwnam(user)[3]			
			os.chown(dir,teacher_uid,teacher_gid)
			os.chmod(dir,0703)
			os.umask(prevmask)		
			return True
			
		except Exception as e:
			
			print e
			
			if user in self.paths:
				self.paths.pop(uid)
				
			os.umask(prevmask)
			
			return False
		
	#def add_user
	
	def is_configured(self,user,orig_path):
		
		if user in self.paths:
			
			path,ip,name,port=self.paths[user]
			if orig_path[len(orig_path)-1]!="/":
				orig_path+="/"
			if path==orig_path:
				return True
			else:
				return False			
		else:
			return False
		
	#def is_configured
	
	def clear_paths(self):
		
		self.paths={}
		
		return True
		
	#def clear_paths
	
	def get_paths(self):
		
		return self.paths
		
	#def get_paths


#class TeacherShareManager