from __future__ import unicode_literals

import sys
from datetime import datetime, timedelta

import utils

CHECK_INTERVAL = 10 # minutes; The actual interval is 5 minutes but as other
                    # tests might restart Icinga we need to take any
                    # rescheduling into account

TABLE_PREFIX = 'icinga_'
TABLES = [
    # Central tables
    'instances',
    'objects',
    # Debugging tables
    'conninfo',
    # Historical tables
    'acknowledgements',
    'commenthistory',
    'contactnotifications',
    'dbversion',
    'downtimehistory',
    'eventhandlers',
    'externalcommands',
    'flappinghistory',
    'hostchecks',
    'logentries',
    'notifications',
    'processevents',
    'servicechecks',
    'statehistory',
    'systemcommands',
    # Current status tables
    'comments',
    'customvariablestatus',
    'hoststatus',
    'programstatus',
    'runtimevariables',
    'scheduleddowntime',
    'servicestatus',
    'contactstatus',
    # Configuration tables
    'commands',
    'configfiles',
    'configfilevariables',
    'contact_addresses',
    'contact_notificationcommands',
    'contactgroup_members',
    'contactgroups',
    'contactnotificationmethods',
    'contacts',
    'customvariables',
    'host_contactgroups',
    'host_contacts',
    'host_parenthosts',
    'hostdependencies',
    'hostescalation_contactgroups',
    'hostescalation_contacts',
    'hostescalations',
    'hostgroup_members',
    'hostgroups',
    'hosts',
    'service_contactgroups',
    'service_contacts',
    'servicedependencies',
    'serviceescalation_contactgroups',
    'serviceescalation_contacts',
    'serviceescalations',
    'servicegroup_members',
    'servicegroups',
    'services',
    'timeperiod_timeranges',
    'timeperiods'
    ]
EXAMPLE_CONFIG = {
    'localhost': ['disk', 'http', 'icinga', 'load', 'ping4',
                  'ping6', 'procs', 'ssh', 'users'],
    'nsca-ng': ['PassiveService1', 'PassiveService2']
}


def validate_tables(tables):
    """
    Return whether all tables of the IDO database scheme exist in
    the given table listing

    """
    utils.Logger.info('Checking database scheme... (tables)\n')
    failures = False
    for table in (TABLE_PREFIX + n for n in TABLES):
        if table in tables:
            utils.Logger.ok('Found table "{0}" in database\n'.format(table))
        else:
            utils.Logger.fail('Could not find table "{0}" in database\n'
                              ''.format(table))
            failures = True

    return not failures


def verify_host_config(config_data):
    """
    Return whether the example hosts exist in the given "hosts" table

    """
    utils.Logger.info('Checking example host configuration...\n')
    failures = False
    for hostname in EXAMPLE_CONFIG:
        if not any(1 for e in config_data if e['alias'] == hostname):
            utils.Logger.fail('Could not find host "{0}"\n'.format(hostname))
            failures = True
        else:
            utils.Logger.ok('Found host "{0}"\n'.format(hostname))

    return not failures


def verify_service_config(config_data):
    """
    Return whether the example services exist in the given "services" table

    """
    utils.Logger.info('Checking example service configuration...\n')
    failures = False
    for hostname, servicename in ((h, s) for h, ss in EXAMPLE_CONFIG.iteritems()
                                         for s in ss):
        if not any(1 for c in config_data
                     if c['alias'] == hostname and
                        c['display_name'] == servicename):
            utils.Logger.fail('Could not find service "{0}" on host "{1}"\n'
                              ''.format(servicename, hostname))
            failures = True
        else:
            utils.Logger.ok('Found service "{0}" on host "{1}"\n'
                            ''.format(servicename, hostname))

    return not failures


def check_last_host_status_update(check_info):
    """
    Return whether the example hosts are checked as scheduled

    """
    utils.Logger.info('Checking last host status updates...\n')
    failures = False
    for host_info in check_info:
        if host_info['alias'] == 'localhost':
            last_check = datetime.fromtimestamp(float(host_info['last_check']))
            if datetime.now() - last_check > timedelta(minutes=CHECK_INTERVAL,
                                                       seconds=10):
                utils.Logger.fail('The last status update of host "{0}" was'
                                  ' more than {1} minutes ago\n'
                                  ''.format(host_info['alias'], CHECK_INTERVAL))
                failures = True
            else:
                utils.Logger.ok('Host "{0}" is being updated\n'
                                ''.format(host_info['alias']))
        elif host_info['alias'] == 'nsca-ng':
            if float(host_info['last_check']) > 0:
                utils.Logger.fail('The host "{0}" was checked even'
                                  ' though it has no check service'
                                  ''.format(host_info['alias']))
                failures = True
            else:
                utils.Logger.ok('Host "{0}" is not being checked because '
                                'there is no check service\n'
                                ''.format(host_info['alias']))
        else:
            utils.Logger.info('Skipping host "{0}"\n'
                              ''.format(host_info['alias']))

    return not failures


def check_last_service_status_update(check_info):
    """
    Return whether the example services are checked as scheduled

    """
    utils.Logger.info('Checking last service status updates...\n')
    failures = False
    for svc_info in check_info:
        if svc_info['display_name'] in EXAMPLE_CONFIG.get(svc_info['alias'], []):
            last_check = datetime.fromtimestamp(float(svc_info['last_check']))
            if datetime.now() - last_check > timedelta(minutes=CHECK_INTERVAL,
                                                       seconds=10):
                utils.Logger.fail('The last status update of service "{0}" on '
                                  'host "{1}" was more than {2} minutes ago\n'
                                  ''.format(svc_info['display_name'],
                                            svc_info['alias'],
                                            CHECK_INTERVAL))
                failures = True
            else:
                utils.Logger.ok('Service "{0}" on host "{1}" is being updated\n'
                                ''.format(svc_info['display_name'],
                                          svc_info['alias']))
        else:
            utils.Logger.info('Skipping service "{0}" on host "{1}"\n'
                              ''.format(svc_info['display_name'],
                                        svc_info['alias']))

    return not failures


def check_logentries(logentry_info):
    """
    Return whether the given logentry originates from host "localhost"
    and refers to its very last hard status change

    """
    utils.Logger.info('Checking status log for host "localhost"...\n')
    if logentry_info and logentry_info[0]['alias'] == 'localhost':
        entry_time = datetime.fromtimestamp(float(logentry_info[0]['entry_time']))
        state_time = datetime.fromtimestamp(float(logentry_info[0]['state_time']))
        if entry_time - state_time > timedelta(seconds=10):
            utils.Logger.fail('The last hard state of host "localhost"'
                              ' seems not to have been logged\n')
            return False
    else:
        utils.Logger.fail('No logs found in the IDO for host "localhost"\n')
        return False

    utils.Logger.ok('The last hard state of host "localhost"'
                    ' was properly logged\n')
    return True