from twisted.application import service as tw_service, strports
from slavealloc.data import setup
from slavealloc.daemon import service
from slavealloc.daemon.http import site


class Allocator(tw_service.MultiService):
    def __init__(self, http_port, db_url, db_kwargs={},
                 base_url='/',
                 run_allocator=False, run_ui=False):
        tw_service.MultiService.__init__(self)

        setup.setup(db_url, db_kwargs)

        if run_allocator:
            self.allocator = service.AllocatorService()
            self.allocator.setServiceParent(self)
        else:
            self.allocator = None

        if run_allocator or run_ui:
            self.site = site.Site(self.allocator,
                                  run_allocator=run_allocator, run_ui=run_ui,
                                  base_url=base_url)
            self.httpservice = strports.service(http_port, self.site)
            self.httpservice.setServiceParent(self)