"""
   SNMP transport class.

   Sends & receives SNMP messages to multiple destinations in bulk.

   Written by Ilya Etingof <ilya@glas.net>, 1999-2002.

"""
import socket

# Import PySNMP components
import role
import v1, v2c

class Error(role.Error):
    """Base class for bulkrole module
    """
    pass

class BadArgument(Error):
    """Bad argument given
    """
    pass

class manager:
    """Send SNMP messages to multiple destinations and receive
       replies.
    """
    def __init__(self, iface=('0.0.0.0', 0)):
        # Initialize defaults
        self.iface = iface
        self.clear()

        # Set defaults to public attributes
        self.retries = 3
        self.timeout = 1

    #
    # Implement list interface
    #

    def __str__(self):
        """
        """
        return str(self._requests)

    def __repr__(self):
        """
        """
        return repr(self._requests)

    def clear(self):
        """Clear the list of sessions and prepare for next round
           of append->dispatch->subscript cycle.
        """
        self._requests = []
        self._responses = []
        self._durty = 0
        
        # [Re-]create SNMP manager transport
        self.transport = role.manager(None, self.iface)

    def append(self, (dst, req)):
        """
           append((dst, req))

           Create transport session destined to "agent" (a tuple of (host, port)
           where host is a string and port is an integer) and queue SNMP
           "request" message (string) to be sent to "agent".

           All queued request messages will be sent upon self.dispatch() method
           invocation.
        """
        if self._durty:
            raise ValueError('List is not valid for update (try clear())')

        if req['request_id'] in map(lambda (dst, req): \
                                    req['request_id'], self._requests):
            raise BadArgument('Duplicate request IDs in queue')

        self._requests.append((dst, req))

    def __setitem__(self, idx, (dst, req)):
        """
        """
        if self._durty:
            raise ValueError('List is not valid for update (try clear())')

        if req['request_id'] in map(lambda (dst, req): \
                                    req['request_id'], self._requests):
            raise BadArgument('Duplicate request IDs in queue')
        
        try:
            self._requests[idx] = (dst, req)

        except IndexError:
            raise IndexError('Request index out of range')

    def __getitem__(self, idx):
        """
        """
        try:
            return self._requests[idx]

        except IndexError:
            raise IndexError('Request index out of range')

    def __len__(self):
        """
        """
        return len(self._requests)
                   
    def count(self, val):
        """XXX
        """
        return self._requests.count(val)

    def index(self, (dst, req)):
        """
        """
        if self._durty:
            raise ValueError('List is not valid for update (try clear())')

        try:
            return self._requests.index((dst, req))

        except ValueError:
            raise ValueError('No such request in queue')

    def insert(self, idx, (dst, req)):
        """
        """
        if self._durty:
            raise ValueError('List is not valid for update (try clear())')

        if req['request_id'] in map(lambda (dst, req): \
                                    req['request_id'], self._requests):
            raise BadArgument('Duplicate request IDs in queue')
        
        try:
            return self._requests.insert(idx, (dst, req))

        except IndexError:
            raise IndexError('Request index out of range')

    def remove(self, (dst, req)):
        """
        """
        try:
            return self._requests.remove((dst, req))

        except ValueError:
            raise ValueError('No such request in queue')

    def pop(self, idx=-1):
        """
        """
        try:
            return self._requests.pop(idx)

        except IndexError:
            raise IndexError('Request index out of range')

    #
    # The main I/O method
    #
    def dispatch(self):
        """
           dispatch()
           
           Send pending SNMP requests and receive replies (or timeout).
        """
        # Indicate that internal queue might change
        self._durty = 1

        # Resolve destination hostnames to IP numbers for later comparation
        try:
            self._requests = map(lambda (dst, req): \
                                 ((socket.gethostbyname(dst[0]), \
                                   dst[1]), req),\
                                 self._requests)

        except socket.error, why:
            raise BadArgument(why)

        # Initialize a list of responses
        self._responses = map(lambda (dst, req): (dst, None), self._requests)
        pending = len(self._responses)
	
        # Initialize retry counter
        retries = self.retries

        # Copy our timeout setting down to transport object
        self.transport.timeout = self.timeout
        
        while retries:
            # Send out requests and prepare for waiting for replies
            for idx in range(len(self._requests)):
                # Skip completed session
                (src, rsp) = self._responses[idx]
                if rsp is not None:
                    continue

                (dst, req) = self._requests[idx]

                try:
                    self.transport.send(req.encode(), dst)
                    
                except role.Error:
                    # Ignore transport errors
                    pass

            while pending:
                # XXX Probably select() based multiplexing would better
                # serve timeouts...
                
                # Wait for response
                (response, src) = self.transport.receive()

                # Stop on timeout
                if response is None:
                    retries = retries - 1
                    break

                # Decode response
                (rsp, rest) = v2c.decode(response)

                # Try to match response message against pending
                # request messages
                for idx in range(len(self._requests)):
                    if (src, rsp) == self._requests[idx]:
                        self._responses[idx] = (src, rsp)
                        pending = pending - 1
                        break

            if not pending:
                break
                
        # Replace list of requests with list of replies
        self._requests = self._responses