# Written by Nick Welch in the years 2005-2008.  Author disclaims copyright.

import os, signal

from Xlib import display
from Xlib.support.connect import get_display

from whimsy.signals import publisher
from whimsy.models.window_manager import window_manager
from whimsy.models.layout import layout
from whimsy.controllers.x_event_controller import x_event_controller
from whimsy.controllers.tick_controller import tick_controller

def wait_signal_handler(*args):
    try:
        os.waitpid(-1, os.WNOHANG | os.WUNTRACED)
    except OSError:
        pass

class main(object):
    def __init__(self):
        os.environ['DISPLAY'] = get_display(None)[0]

        self.dpy    = display.Display()
        self.hub    = publisher()
        self.wm     = window_manager(self.hub, self.dpy)
        self.layout = layout(self.hub, self.wm)
        self.xec    = x_event_controller(self.hub, self.dpy)
        self.ticker = tick_controller(self.hub)

        self.hub.defaults['wm'] = self.wm
        self.hub.defaults['hub'] = self.hub

        self.hub.attach('tick', self.xec.select_and_emit_all)

    def run(self):
        signal.signal(signal.SIGCHLD, wait_signal_handler)
        signal.signal(signal.SIGTERM, lambda signum, frame: self.ticker.stop())
        signal.signal(signal.SIGINT,  lambda signum, frame: self.ticker.stop())
        signal.signal(signal.SIGPIPE, lambda signum, frame: self.ticker.stop())
        signal.signal(signal.SIGUSR1, lambda signum, frame: self.ticker.stop())
        signal.signal(signal.SIGUSR2, lambda signum, frame: self.ticker.stop())

        self.wm.manage()
        self.ticker.tick_forever()