##############################################################################
# -*- coding: utf-8 -*-
# Project:     ControlIES
# Module:    LdapConnection.py
# Purpose:     Connection with ldap server
# Language:    Python 2.5
# Date:        7-Feb-2011.
# Ver:        7-Feb-2011.
# Author:    Manuel Mora Gordillo
# Copyright:    2011 - Manuel Mora Gordillo <manuito @no-spam@ gmail.com>
#
# ControlIES is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# ControlIES is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with ControlAula. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

import ldap
import logging
from zope.interface import Interface, Attribute, implements

class ILdapConnection(Interface):
    connection=Attribute("a ldap connection")
    
class LdapConnection(object):
    implements(ILdapConnection)
    
    def __init__(self,session):
		self.host = ""
		self.user = ""
		self.passwd = ""
		pass

    def setCredentials(self,host,user,passwd):
		self.host = host
		self.user = user
		self.passwd = passwd

    def validation(self):
		if self.host == "":
			return "host"

		if self.user == "":
			return "user"

		if self.passwd == "":
			return "password"

		return "OK"

    def process(self):
		val = self.validation()
        	
		if val != "OK":
			return val

		auth = self.connect()
		return auth

    def connect(self):
		self.connection=ldap.open(self.host)
		try:
			self.connection.simple_bind_s("cn="+self.user+",ou=People,dc=instituto,dc=extremadura,dc=es",self.passwd)
		except ldap.INVALID_CREDENTIALS:
			logging.getLogger().debug('LDAP user or password incorrect')
			return False
		except ldap.LDAPError:
			logging.getLogger().debug('LDAP error connect')
			return False

		return True

    def getConnect(self):
		return self.connection
		
    def search(self,baseDN,filter,retrieveAttributes):
		
		try:
			ldap_result_id = self.connection.search(baseDN+",dc=instituto,dc=extremadura,dc=es", ldap.SCOPE_SUBTREE, filter, retrieveAttributes)
			result_set = []
			while 1:
				result_type, result_data = self.connection.result(ldap_result_id, 0)
				if (result_data == []):
					break
				else:
					if result_type == ldap.RES_SEARCH_ENTRY:
						result_set.append(result_data)
			return result_set
		except ldap.LDAPError, e:
			logging.getLogger().debug('LDAP error search')
		
		"""result = con.search_s( base_dn, ldap.SCOPE_SUBTREE, filter, attrs )
		return result"""

    def add(self,baseDN,attr):
		try:
			self.connection.add_s(baseDN+",dc=instituto,dc=extremadura,dc=es", attr)
			
		except ldap.ALREADY_EXISTS:
			logging.getLogger().debug("LDAP already exists %s" % (baseDN))
		except ldap.OPERATIONS_ERROR:
			logging.getLogger().debug("LDAP operation error %s" % (baseDN))
		except ldap.NO_SUCH_OBJECT:
			logging.getLogger().debug("LDAP no such object %s" % (baseDN))
			
		return True

    def modify(self,baseDN,attr):
		try:
			self.connection.modify_s(baseDN+",dc=instituto,dc=extremadura,dc=es", attr)
			
		except ldap.OPERATIONS_ERROR:
			print "error"
		except ldap.NO_SUCH_OBJECT:
			print "no_such_object"
			
		return True
		
    def delete(self,baseDN):
		try:
			self.connection.delete_s(baseDN+",dc=instituto,dc=extremadura,dc=es")
			
		except ldap.OPERATIONS_ERROR:
			print "error"
		except ldap.NO_SUCH_OBJECT:
			print "no_such_object"
			
		return True
		
    """def searchClassroomComputers(self,classroom):

		''' How many groups? '''
		base_dn = 'cn=THINCLIENTS,cn=DHCP Config,dc=instituto,dc=extremadura,dc=es'
		filter = '(cn=group*)'
		attrs = ['cn']
		groups = self.search(self,base_dn,filter,attrs)

		numberDesktops=0;
		for i in range(0,len(groups)):
			''' search computers of different groups '''
			base_dn = 'cn='+groups[i][0]['cn'][0]+',cn=THINCLIENTS,cn=DHCP Config,dc=instituto,dc=extremadura,dc=es'
			filter = '(cn='+classroom+'-o*)'
			attrs = ['cn','dhcpHWAddress']
			computers = self.search(self,base_dn,filter,attrs)

			for j in range(0,len(computers)):
				self.Desktops[numberDesktops] = {'desktop':computers[j][1]['cn'][0] , 'mac':computers[j][1]['dhcpHWAddress'][0]}
				numberDesktops = numberDesktops + 1"""