#!/usr/local/bin/python
import time
import logging
import logging.config
import sys
import signal

from multiprocessing import Process

from airtimefilemonitor.airtimenotifier import AirtimeNotifier
from airtimefilemonitor.airtimeprocessevent import AirtimeProcessEvent
from airtimefilemonitor.mediaconfig import AirtimeMediaConfig

def handleSigTERM(signum, frame):
    logger = logging.getLogger()
    logger.info("Main Process Shutdown, TERM signal caught. %d")
    for p in processes:
        p.terminate()
        logger.info("Killed process. %d", p.pid)

    notifier_daemon_pid = open('/var/run/airtime-notifier.pid', 'r').read()
    os.kill(int(notifier_daemon_pid), 9)
    logger.info("Killed process. %d", int(notifier_daemon_pid))

    sys.exit(0)


# configure logging
try:
    logging.config.fileConfig("logging.cfg")
except Exception, e:
    print 'Error configuring logging: ', e
    sys.exit()

logger = logging.getLogger()
processes = []

try:
    config = AirtimeMediaConfig()
    logger.info("Initializing event processor")
    pe = AirtimeProcessEvent(airtime_config=config)

    notifier = AirtimeNotifier(pe.wm, pe, read_freq=0.1, timeout=0.1, airtime_config=config)
    notifier.coalesce_events()

    #create 5 worker processes
    for i in range(5):
        p = Process(target=notifier.process_file_events, args=(pe.multi_queue,))
        processes.append(p)
        p.start()

    signal.signal(signal.SIGTERM, handleSigTERM)

    logger.info("Setting up monitor")
    response = None
    while response is None:
        response = notifier.api_client.setup_media_monitor()
        time.sleep(5)

    storage_directory = response["stor"].encode('utf-8')
    logger.info("Storage Directory is: %s", storage_directory)
    config.storage_directory = storage_directory

    wdd = pe.watch_directory(storage_directory)
    logger.info("Added watch to %s", storage_directory)
    logger.info("wdd result %s", wdd[storage_directory])

    notifier.loop(daemonize=True, callback=pe.notifier_loop_callback, pid_file='/var/run/airtime-notifier.pid', stdout='/var/log/airtime/media-monitor/media-monitor.log')
    
    for p in processes:
        p.join()

except KeyboardInterrupt:
    notifier.stop()
except Exception, e:
    notifier.stop()
    logger.error('Exception: %s', e)