# -*- 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()