import logging
from pathlib import Path
from functools import partial

from execlog import util
from execlog import ChainRouter, Event
from execlog.routers import PathRouter
from execlog.listeners import PathListener


logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
logger.addHandler(util.generic.TqdmLoggingHandler())

# router setup
router1 = PathRouter()
router2 = PathRouter()
router3 = PathRouter()

chain_router = ChainRouter([router1, router2, router3])

# router-1
router1.register('endpoint_proxy', partial(print, 'R1-1 ::'))

# router-2
router2.register('endpoint_proxy', partial(print, 'R2-1 ::'))
router2.register('endpoint_proxy', partial(print, 'R2-2 ::'))

# router-3
router3.register('endpoint_proxy', partial(print, 'R3-1 ::'))
router3.register('endpoint_proxy', partial(print, 'R3-2 ::'))
router3.register('endpoint_proxy', partial(print, 'R3-3 ::'))

events = [
    Event(endpoint='endpoint_proxy', name='file1'),
    Event(endpoint='endpoint_proxy', name='file2'),
    Event(endpoint='endpoint_proxy', name='file3'),
]

if __name__ == '__main__':
    futures = chain_router.submit(events)
    chain_router.wait_on_futures(futures)
    
    #listener = chain_router.get_listener()
    #listener.start()

    #file_a = Path('endpoint_proxy/fileA')
    #file_a.write_text('test text')
    #file_a.unlink()