async def query_windows(display: X.Display, window_factory, windows, view):
"""Signal handler for SIGUSR1.
Searches for added and removed tmux clients.
Added clients: additional windows will be mapped
Removed clients: existing windows will be destroyed
"""
parent_window_infos = xutil.get_parent_window_infos(display)
view.offset = tmux_util.get_offset()
map_parent_window_id_info = {
info.window_id: info
for info in parent_window_infos
}
parent_window_ids = map_parent_window_id_info.keys()
map_current_windows = {window.parent_id: window for window in windows}
current_window_ids = map_current_windows.keys()
diff_window_ids = parent_window_ids ^ current_window_ids
added_window_ids = diff_window_ids & parent_window_ids
removed_window_ids = diff_window_ids & current_window_ids
draw = added_window_ids or removed_window_ids
if added_window_ids:
windows += window_factory.create(
*[map_parent_window_id_info.get(wid) for wid in added_window_ids])
if removed_window_ids:
windows -= [map_current_windows.get(wid) for wid in removed_window_ids]
if draw:
windows.draw()
def main(options):
if options['--silent']:
try:
outfile = os.open(os.devnull, os.O_WRONLY)
os.close(sys.stderr.fileno())
os.dup2(outfile, sys.stderr.fileno())
finally:
os.close(outfile)
display = xutil.get_display()
screen = display.screen()
window_infos = xutil.get_parent_window_infos()
loop = asyncio.get_event_loop()
executor = thread.DaemonThreadPoolExecutor(max_workers=2)
parser_object = (parser.ParserOption(options['--parser'])
.parser_class())
image_loader = (loading.ImageLoaderOption(options['--loader'])
.loader_class())
error_handler = error_processor_factory(parser_object)
view = View()
tools = Tools(image_loader, parser_object, error_handler)
window_factory = ui.OverlayWindow.Factory(display, view)
windows = batch.BatchList(window_factory.create(*window_infos))
image_loader.register_error_handler(error_handler)
view.screen_width = screen.width_in_pixels
view.screen_height = screen.height_in_pixels
if tmux_util.is_used():
atexit.register(setup_tmux_hooks())
view.offset = tmux_util.get_offset()
with windows, image_loader:
loop.set_default_executor(executor)
for sig in (signal.SIGINT, signal.SIGTERM):
loop.add_signal_handler(
sig, shutdown_factory(loop))
loop.add_signal_handler(
signal.SIGUSR1,
lambda: asyncio.ensure_future(query_windows(
window_factory, windows, view)))
loop.add_signal_handler(
signal.SIGWINCH,
lambda: asyncio.ensure_future(
reset_terminal_info(windows)))
asyncio.ensure_future(process_xevents(loop, display, windows))
asyncio.ensure_future(process_commands(
loop, shutdown_factory(loop),
windows, view, tools))
try:
loop.run_forever()
finally:
loop.close()
executor.shutdown(wait=False)
def main_layer(options):
display = xutil.get_display()
window_infos = xutil.get_parent_window_infos()
loop = asyncio.get_event_loop()
executor = thread.DaemonThreadPoolExecutor(max_workers=2)
parser_class = parser.ParserOption(options['--parser']).parser_class
view = ui.View()
window_factory = ui.OverlayWindow.Factory(display, view)
windows = batch.BatchList(window_factory.create(*window_infos))
if tmux_util.is_used():
atexit.register(setup_tmux_hooks())
view.offset = tmux_util.get_offset()
if options['--silent']:
sys.stderr = open('/dev/null', 'w')
with windows:
loop.set_default_executor(executor)
for sig in (signal.SIGINT, signal.SIGTERM):
loop.add_signal_handler(
sig, shutdown_factory(loop))
loop.add_signal_handler(
signal.SIGUSR1,
lambda: asyncio.ensure_future(query_windows(
window_factory, windows, view)))
asyncio.ensure_future(main_xevents(loop, display, windows))
asyncio.ensure_future(main_commands(
loop, shutdown_factory(loop), parser_class(),
windows, view))
try:
loop.run_forever()
finally:
loop.close()
executor.shutdown(wait=False)