#!/usr/bin/env python3
# -*- coding: utf-8 -*-  

import os
import time

from Logger import Logger
from Setup import ConnectionSetup
from config import settings, setup_logger

log = setup_logger()
lockpath = os.path.join(settings.locks, settings.lockname)


def check_dirs_and_files():
    # log
    if not os.path.exists(settings.logs):
        os.mkdir(settings.logs, 0o000755)
    if not os.path.exists(os.path.join(settings.logs, settings.exceptionlog)):
        file = open(os.path.join(settings.logs, settings.exceptionlog), 'w')
        file.write("<exceptions></exceptions>")
        file.close()
    # lock
    if not os.path.exists(settings.locks):
        os.mkdir(settings.locks, 0o000755)
    # records
    if not os.path.exists(settings.records):
        os.mkdir(settings.records, 0o000755)


def obtain_lock():
    if not os.path.exists(lockpath):
        lock = open(lockpath, 'w')
        lock.write(str(time.time()))
        lock.close()
        return True
    return False


def free_lock():
    if os.path.exists(lockpath):
        os.remove(lockpath)


def format_host(host):
    return "%s:%d" % (host['name'], host['port'])


if __name__ == "__main__":
    try:
        input = raw_input
    except NameError:
        pass
    check_dirs_and_files()
    conSetup = ConnectionSetup(log)
    connectedSensors = []
    connections = []
    try:
        log.info("setting up all sensors")
        while True:
            if obtain_lock():
                logger = Logger(log)
                for con in settings.hosts:
                    try:
                        log.info("connecting to host '" + str(con) + "'")
                        con = settings.hosts[con]
                        connection, sensors = conSetup.setup_connection_and_sensors(con['host'], con['sensors'], logger)
                        connections.append(connection)
                        connectedSensors.append(sensors)
                        log.info("started logging at " + format_host(con['host']))
                    except Exception as inst:
                        # connection failed, log and exit
                        # TODO: logger.print_exception(inst)
                        log.error("connection failed: " + str(inst))
                # noinspection PyUnboundLocalVariable
                input("Press key to restart\n")
                log.info("stop logging... @" + time.ctime() + "\n")
                conSetup.disconnect_any(connections)
                free_lock()
            else:
                log.critical("lock collision: lock 'all' active")
            log.info("wait for retry (" + str(settings.waitDelay) + ")")
            time.sleep(settings.waitDelay)
    except KeyboardInterrupt:
        log.info("keyboard-interrupt happened, cleaning up")
        conSetup.disconnect_any(connections)
        free_lock()