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

from twisted.web import soap, xmlrpc

from ..controller.viewinterface import Interface

from ..error import InterfaceException, ValidateException

class TwistedInterface(Interface):

    """Class that addes needed function for XML-RPC/SOAP"""

    def __init__(self):

        Interface.__init__(self)

    def listMethods(self):

        """Since we override lookupProcedure, its suggested to override

        listProcedures too.

        """

        return self.listProcedures()

    def listProcedures(self):

        """returns a list of all functions that are allowed to call via XML-RPC."""

        return ['listMethods','status','sms','fax','mail','routes','defaultRoute','bill','telnumber','email']

class XMLRPCInterface(TwistedInterface,xmlrpc.XMLRPC): 

    """XML-RPC interface"""

    def __init__(self):

        xmlrpc.XMLRPC.__init__(self) 

        TwistedInterface.__init__(self)

    def lookupProcedure(self, procedurePath):

        if procedurePath not in self.listProcedures():

            raise xmlrpc.NoSuchFunction(self.NOT_FOUND,

                        "procedure %s not found" % procedurePath)

        try:

            return getattr(self,procedurePath)

        except KeyError:

            raise xmlrpc.NoSuchFunction(self.NOT_FOUND,

                        "procedure %s not found" % procedurePath)

    def _ebRender(self, failure):

        if isinstance(failure.value, InterfaceException):

            return xmlrpc.Fault(failure.value.code, failure.value.msg)

        if isinstance(failure.value, ValidateException):

            return xmlrpc.Fault(failure.value.code, failure.value.msg)

        return xmlrpc.XMLRPC._ebRender(self, failure)

class SOAPInterface(TwistedInterface,soap.SOAPPublisher): 

    """SOAP interface"""

    def __init__(self):

        soap.SOAPPublisher.__init__(self) 

        TwistedInterface.__init__(self)

    def lookupFunction(self, functionName):

        if functionName in self.listProcedures():

            function = getattr(self, functionName, None)

            if function:

                return function, getattr(function, "useKeywords", False)

            return None

        else:

            return None

def appendResource(root):

    """adding XML-RPC and SOAP to root."""

    root.putChild('RPC2', XMLRPCInterface())

    root.putChild('SOAP', SOAPInterface())

    return root

if __name__ == '__main__':

    from twisted.web import resource, server

    from twisted.internet import reactor

    root = resource.Resource()

    root = appendResource(root)

    reactor.listenTCP(7080, server.Site(root))

    reactor.run()