import dbus
import dbus.service
import dbus.mainloop.glib
import gobject
import logging
import threading

""" MockFlimflam provides a select few methods from the flimflam
    DBus API so that we can track "dbus-send" invocations sent
    by the shill init scripts.  It could be used as a kernel for
    a test of other facilities that use the shill/flimflam DBus
    API and at that point it should be moved out of this specific
    test. """

class MethodCall(object):
    """ A logged method call to the DBus API. """
    def __init__(self, method, argument):
        self.method = method
        self.argument = argument

class FlimflamManager(dbus.service.Object):
    """ The flimflam DBus Manager object instance.  Methods in this
        object are called whenever a DBus RPC method is invoked. """
    def __init__(self, bus, object_path):
        dbus.service.Object.__init__(self, bus, object_path)
        self.method_calls = []

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='s', out_signature='o')
    def CreateProfile(self, profile):
        self.add_method_call('CreateProfile', profile)
        return '/'

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='s', out_signature='')
    def RemoveProfile(self, profile):
        self.add_method_call('RemoveProfile', profile)

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='s', out_signature='o')
    def PushProfile(self, profile):
        self.add_method_call('PushProfile', profile)
        return '/'

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='ss', out_signature='o')
    def InsertUserProfile(self, profile, user_hash):
        self.add_method_call('InsertUserProfile', (profile, user_hash))
        return '/'

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='s', out_signature='')
    def PopProfile(self, profile):
        self.add_method_call('PopProfile', profile)

    @dbus.service.method('org.chromium.flimflam.Manager',
                         in_signature='', out_signature='')
    def PopAllUserProfiles(self):
        self.add_method_call('PopAllUserProfiles', '')

    def add_method_call(self, method, arg):
        print "Called method %s" % method
        logging.info("Mock Flimflam method %s called with argument %s" %
                     (method, arg))
        self.method_calls.append(MethodCall(method, arg))

    def get_method_calls(self):
        method_calls = self.method_calls
        self.method_calls = []
        return method_calls

class MockFlimflam(threading.Thread):
    """ This thread object instantiates a mock flimflam manager and
        runs a mainloop that receives DBus API messages. """
    FLIMFLAM = "org.chromium.flimflam"
    def __init__(self):
        threading.Thread.__init__(self)
        gobject.threads_init()

    def run(self):
        dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
        self.bus = dbus.SystemBus()
        name = dbus.service.BusName(self.FLIMFLAM, self.bus)
        self.manager = FlimflamManager(self.bus, '/')
        self.mainloop = gobject.MainLoop()
        self.mainloop.run()

    def quit(self):
        self.mainloop.quit()

    def get_method_calls(self):
        return self.manager.get_method_calls()


if __name__ == '__main__':
    MockFlimflam().run()