"""
SofiaSIP testing framework
"""

import servicetest

from twisted.protocols import sip
from twisted.internet import reactor

import dbus
import dbus.glib

class SipProxy(sip.RegisterProxy):
    registry = sip.InMemoryRegistry("127.0.0.1")

    def __init__(self, *args, **kw):
        sip.RegisterProxy.__init__(self, *args, **kw)

    def register(self, message, host, port):
        if hasattr(self, 'registrar_handler'):
            if self.registrar_handler(message, host, port):
                sip.RegisterProxy.register(self, message, host, port)
            else:
                self.unauthorized(message, host, port)

    def handle_request(self, message, addr):
        if message.method == 'REGISTER':
            return sip.RegisterProxy.handle_request(self, message, addr)
        if message.method == 'MESSAGE':
            self.test_handler.handle_event(servicetest.Event('sip-message',
                uri=str(message.uri), headers=message.headers, body=message.body,
                sip_message=message))


def go(register_cb, params=None):
    default_params = {
        'account': 'testacc@127.0.0.1',
        'password': 'testpwd',
        'proxy-host': '127.0.0.1',
        'port': dbus.UInt16(9090),
    }

    if params is not None:
        default_params.update(params)

    handler = servicetest.EventTest()
    servicetest.prepare_test(handler, 'sofiasip', 'sip', default_params)
    handler.data['sip'] = SipProxy()
    handler.data['sip'].test_handler = handler
    reactor.listenUDP(9090, handler.data['sip'])
    handler.data['sip'].registrar_handler = register_cb
    servicetest.run_test(handler)