# -*- coding: utf-8 -*-

# vim:set expandtab tabstop=4 shiftwidth=4: 

"""Vigiboard Controller"""

from datetime import datetime

from time import mktime

import math

import urllib

from tg.exceptions import HTTPNotFound

from tg import expose, validate, require, flash, \

    tmpl_context, request, config, session, redirect, url

from tw.forms import validators

from pylons.i18n import ugettext as _

from pylons.i18n import lazy_ugettext as l_

from pylons.controllers.util import abort

from sqlalchemy import asc

from sqlalchemy.sql import func

from repoze.what.predicates import Any, not_anonymous

from vigilo.models.configure import DBSession

from vigilo.models import Event, EventHistory, CorrEvent, SupItem, \

                            HostGroup, ServiceGroup, StateName, User

from vigilo.models.functions import sql_escape_like

from vigilo.models.secondary_tables import EVENTSAGGREGATE_TABLE

from vigilo.turbogears.controllers.autocomplete \

    import make_autocomplete_controller

from vigiboard.controllers.vigiboardrequest import VigiboardRequest

from vigiboard.controllers.vigiboard_controller import VigiboardRootController

from vigiboard.widgets.edit_event import edit_event_status_options

from vigiboard.lib.base import BaseController

__all__ = ('RootController', 'get_last_modification_timestamp', 

           'date_to_timestamp')

# pylint: disable-msg=R0201

class RootController(VigiboardRootController):

    """

    Le controller général de vigiboard

    """

    autocomplete = make_autocomplete_controller(BaseController)

    def process_form_errors(self, *argv, **kwargv):

        """

        Gestion des erreurs de validation : On affiche les erreurs

        puis on redirige vers la dernière page accédée.

        """

        for k in tmpl_context.form_errors:

            flash("'%s': %s" % (k, tmpl_context.form_errors[k]), 'error')

        if request.environ.get('HTTP_REFERER') :

            redirect(request.environ.get('HTTP_REFERER'

                ).split(request.environ.get('HTTP_HOST'))[1])

        else :

            redirect('/')

    @validate(validators={

            'page': validators.Int(min=1),

        }, error_handler=process_form_errors)

    @expose('events_table.html')

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def default(self, page=1, hostgroup=None, servicegroup=None,

            host=None, service=None, output=None, trouble_ticket=None,

            from_date=None, to_date=None, *argv, **krgv):

        """

        Page d'accueil de Vigiboard. Elle affiche, suivant la page demandée

        (page 1 par defaut), la liste des événements, rangés par ordre de prise

        en compte, puis de sévérité.

        Pour accéder à cette page, l'utilisateur doit être authentifié.

        @param page: Numéro de la page souhaitée, commence à 1

        @param host: Si l'utilisateur souhaite sélectionner seulement certains

                     événements suivant leur hôte, il peut placer une expression

                     ici en suivant la structure du LIKE en SQL

        @param service: Idem que host mais sur les services

        @param output: Idem que host mais sur le text explicatif

        @param trouble_ticket: Idem que host mais sur les tickets attribués

        """

        if not page:

            page = 1

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        user = User.by_user_name(username)

        aggregates = VigiboardRequest(user)

        aggregates.add_table(

            CorrEvent,

            aggregates.items.c.hostname,

            aggregates.items.c.servicename

        )

        aggregates.add_join((Event, CorrEvent.idcause == Event.idevent))

        aggregates.add_join((aggregates.items, 

            Event.idsupitem == aggregates.items.c.idsupitem))

        aggregates.add_order_by(asc(aggregates.items.c.hostname))

        search = {

            'host': '',

            'service': '',

            'output': '',

            'tt': '',

            'from_date': '',

            'to_date': '',

            'hostgroup': '',

            'servicegroup': '',

        }

        # Application des filtres si nécessaire

        if hostgroup:

            search['hostgroup'] = hostgroup

            hostgroup = sql_escape_like(hostgroup)

            aggregates.add_join((HostGroup, HostGroup.idgroup == \

                aggregates.items.c.idhostgroup))

            aggregates.add_filter(HostGroup.name.ilike('%%%s%%' % hostgroup))

        if servicegroup:

            search['servicegroup'] = servicegroup

            servicegroup = sql_escape_like(servicegroup)

            aggregates.add_join((ServiceGroup, ServiceGroup.idgroup == \

                aggregates.items.c.idservicegroup))

            aggregates.add_filter(

                ServiceGroup.name.ilike('%%%s%%' % servicegroup))

        if host:

            search['host'] = host

            host = sql_escape_like(host)

            aggregates.add_filter(aggregates.items.c.hostname.ilike(

                '%%%s%%' % host))

        if service:

            search['service'] = service

            service = sql_escape_like(service)

            aggregates.add_filter(aggregates.items.c.servicename.ilike(

                '%%%s%%' % service))

        if output:

            search['output'] = output

            output = sql_escape_like(output)

            aggregates.add_filter(Event.message.ilike('%%%s%%' % output))

        if trouble_ticket:

            search['tt'] = trouble_ticket

            trouble_ticket = sql_escape_like(trouble_ticket)

            aggregates.add_filter(CorrEvent.trouble_ticket.ilike(

                '%%%s%%' % trouble_ticket))

        if from_date:

            search['from_date'] = from_date

            try:

                # TRANSLATORS: Format de date et heure.

                from_date = datetime.strptime(

                    from_date, _('%Y-%m-%d %I:%M:%S %p'))

            except ValueError:

                to_date = None

            aggregates.add_filter(CorrEvent.timestamp_active >= from_date)

        if to_date:

            search['to_date'] = to_date

            try:

                # TRANSLATORS: Format de date et heure.

                to_date = datetime.strptime(

                    to_date, _('%Y-%m-%d %I:%M:%S %p'))

            except ValueError:

                to_date = None

            aggregates.add_filter(CorrEvent.timestamp_active <= to_date)

        # Calcul des éléments à afficher et du nombre de pages possibles

        total_rows = aggregates.num_rows()

        items_per_page = int(config['vigiboard_items_per_page'])

        id_first_row = items_per_page * (page-1)

        id_last_row = min(id_first_row + items_per_page, total_rows)

        aggregates.format_events(id_first_row, id_last_row)

        aggregates.generate_tmpl_context()

        nb_pages = int(math.ceil(total_rows / (items_per_page + 0.0)))

        if not total_rows:

            id_first_row = 0

        else:

            id_first_row += 1

        return dict(

            events = aggregates.events,

            plugins = get_plugins_instances(),

            rows_info = {

                'id_first_row': id_first_row,

                'id_last_row': id_last_row,

                'total_rows': total_rows,

            },

            nb_pages = nb_pages,

            page = page,

            event_edit_status_options = edit_event_status_options,

            search = search,

            refresh_times = config['vigiboard_refresh_times'],

        )

    @validate(validators={

            'idcorrevent': validators.Int(not_empty=True),

            'page': validators.Int(min=1),

        }, error_handler=process_form_errors)

    @expose('raw_events_table.html')

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def masked_events(self, idcorrevent, page=1):

        """

        Affichage de la liste des événements bruts masqués dans un

        événement corrélé (agrégés).

        @param idevent: identifiant de l'événement souhaité

        """

        if not page:

            page = 1

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        events = VigiboardRequest(User.by_user_name(username), False)

        events.add_table(

            Event,

            events.items.c.hostname,

            events.items.c.servicename,

        )

        events.add_join((EVENTSAGGREGATE_TABLE, \

            EVENTSAGGREGATE_TABLE.c.idevent == Event.idevent))

        events.add_join((CorrEvent, CorrEvent.idcorrevent == \

            EVENTSAGGREGATE_TABLE.c.idcorrevent))

        events.add_join((events.items, 

            Event.idsupitem == events.items.c.idsupitem))

        events.add_filter(Event.idevent != CorrEvent.idcause)

        events.add_filter(CorrEvent.idcorrevent == idcorrevent)

        # Vérification que l'événement existe

        total_rows = events.num_rows()

        if total_rows < 1:

            flash(_('No masked event or access denied'), 'error')

            redirect('/')

        # Calcul des éléments à afficher et du nombre de pages possibles

        items_per_page = int(config['vigiboard_items_per_page'])

        id_first_row = items_per_page * (page-1)

        id_last_row = min(id_first_row + items_per_page, total_rows)

        events.format_events(id_first_row, id_last_row)

        events.generate_tmpl_context()

        nb_pages = int(math.ceil(total_rows / (items_per_page + 0.0)))

        if not total_rows:

            id_first_row = 0

        else:

            id_first_row += 1

        return dict(

            idcorrevent = idcorrevent,

            events = events.events,

            plugins = get_plugins_instances(),

            rows_info = {

                'id_first_row': id_first_row,

                'id_last_row': id_last_row,

                'total_rows': total_rows,

            },

            nb_pages = nb_pages,

            page = page,

            search = {

                'host': '',

                'service': '',

                'output': '',

                'tt': '',

                'from_date': '',

                'to_date': '',

                'hostgroup': '',

                'servicegroup': '',

            },

           refresh_times=config['vigiboard_refresh_times'],

        )

    @validate(validators={

            'idevent': validators.Int(not_empty=True),

            'page': validators.Int(min=1),

        }, error_handler=process_form_errors)

    @expose('history_table.html')

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def event(self, idevent, page=1):

        """

        Affichage de l'historique d'un événement brut.

        Pour accéder à cette page, l'utilisateur doit être authentifié.

        @param idevent: identifiant de l'événement brut souhaité.

        @type idevent: C{int}

        """

        if not page:

            page = 1

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        events = VigiboardRequest(User.by_user_name(username), False)

        events.add_table(

            Event,

            events.items.c.hostname,

            events.items.c.servicename,

        )

        events.add_join((EVENTSAGGREGATE_TABLE, \

            EVENTSAGGREGATE_TABLE.c.idevent == Event.idevent))

        events.add_join((CorrEvent, CorrEvent.idcorrevent == \

            EVENTSAGGREGATE_TABLE.c.idcorrevent))

        events.add_join((events.items, 

            Event.idsupitem == events.items.c.idsupitem))

        events.add_filter(Event.idevent == idevent)

        if events.num_rows() != 1:

            flash(_('No such event or access denied'), 'error')

            redirect('/')

        events.format_events(0, 1)

        events.generate_tmpl_context()

        history = events.format_history()

        total_rows = history.count()

        items_per_page = int(config['vigiboard_items_per_page'])

        id_first_row = items_per_page * (page-1)

        id_last_row = min(id_first_row + items_per_page, total_rows)

        history_entries = history[id_first_row : id_last_row]

        nb_pages = int(math.ceil(total_rows / (items_per_page + 0.0)))

        if not total_rows:

            id_first_row = 0

        else:

            id_first_row += 1

        return dict(

            idevent = idevent,

            plugins = get_plugins_instances(),

            rows_info = {

                'id_first_row': id_first_row,

                'id_last_row': id_last_row,

                'total_rows': total_rows,

            },

            nb_pages = nb_pages,

            page = page,

            history = history_entries,

            search = {

                'host': '',

                'service': '',

                'output': '',

                'tt': '',

                'from_date': '',

                'to_date': '',

                'hostgroup': '',

                'servicegroup': '',

            },

           refresh_times=config['vigiboard_refresh_times'],

        )

    @validate(

        validators={

            'host': validators.NotEmpty(),

#            'service': validators.NotEmpty(),

            'page': validators.Int(min=1),

        }, 

        error_handler = process_form_errors)

    @expose('events_table.html')

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def item(self, host, service=None, page=1):

        """

        Affichage de l'historique de l'ensemble des événements corrélés

        jamais ouverts sur l'hôte / service demandé.

        Pour accéder à cette page, l'utilisateur doit être authentifié.

        @param host: Nom de l'hôte souhaité.

        @param service: Nom du service souhaité

        """

        if not page:

            page = 1

        idsupitem = SupItem.get_supitem(host, service)

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        aggregates = VigiboardRequest(User.by_user_name(username), False)

        aggregates.add_table(

            CorrEvent,

            aggregates.items.c.hostname,

            aggregates.items.c.servicename,

        )

        aggregates.add_join((Event, CorrEvent.idcause == Event.idevent))

        aggregates.add_join((aggregates.items, 

            Event.idsupitem == aggregates.items.c.idsupitem))

        aggregates.add_filter(aggregates.items.c.idsupitem == idsupitem)

        # Vérification qu'il y a au moins 1 événement qui correspond

        total_rows = aggregates.num_rows()

        if not total_rows:

            flash(_('No access to this host/service or no event yet'), 'error')

            redirect('/')

        items_per_page = int(config['vigiboard_items_per_page'])

        id_first_row = items_per_page * (page-1)

        id_last_row = min(id_first_row + items_per_page, total_rows)

        aggregates.format_events(id_first_row, id_last_row)

        aggregates.generate_tmpl_context()

        nb_pages = int(math.ceil(total_rows / (items_per_page + 0.0)))

        if not total_rows:

            id_first_row = 0

        else:

            id_first_row += 1

        return dict(

            hostname = host,

            servicename = service,

            events = aggregates.events,

            item_label = item_label,

            plugins = get_plugins_instances(),

            rows_info = {

                'id_first_row': id_first_row,

                'id_last_row': id_last_row,

                'total_rows': total_rows,

            },

            nb_pages = nb_pages,

            page = page,

            event_edit_status_options = edit_event_status_options,

            search = {

                'host': '',

                'service': '',

                'output': '',

                'tt': '',

                'from_date': '',

                'to_date': '',

                'hostgroup': '',

                'servicegroup': '',

            },

            refresh_times=config['vigiboard_refresh_times'],

        )

    @validate(validators={

        "id": validators.Regex(r'^[0-9]+(,[0-9]+)*,?$'),

#        "trouble_ticket": validators.Regex(r'^[0-9]*$'),

        "ack": validators.OneOf([

            u'NoChange',

            u'None',

            u'Acknowledged',

            u'AAClosed'

        ])}, error_handler=process_form_errors)

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def update(self, **krgv):

        """

        Mise à jour d'un événement suivant les arguments passés.

        Cela peut être un changement de ticket ou un changement de statut.

        @param krgv['id']: Le ou les identifiants des événements à traiter

        @param krgv['last_modification']: La date de la dernière modification

        dont l'utilisateur est au courant.

        @param krgv['tt']: Nouveau numéro du ticket associé.

        @param krgv['status']: Nouveau status de/des événements.

        """

        # On vérifie que des identifiants ont bien été transmis via

        # le formulaire, et on informe l'utilisateur le cas échéant.

        if krgv['id'] is None:

            flash(_('No event has been selected'), 'warning')

            raise redirect(request.environ.get('HTTP_REFERER', url('/')))

        # Le filtre permet d'éliminer les chaines vides contenues dans le

        # tableau ('a,b,' -> split -> ['a','b',''] -> filter -> ['a','b']).

        ids = map(int, filter(len, krgv['id'].split(',')))

        # Si l'utilisateur édite plusieurs événements à la fois,

        # il nous faut chacun des identifiants

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        events = VigiboardRequest(User.by_user_name(username))

        events.add_table(CorrEvent)

        events.add_join((Event, CorrEvent.idcause == Event.idevent))

        events.add_join((events.items, 

            Event.idsupitem == events.items.c.idsupitem))

        events.add_filter(CorrEvent.idcorrevent.in_(ids))

        events.generate_request()

        idevents = [cause.idcause for cause in events.req]

        # Si des changements sont survenus depuis que la 

        # page est affichée, on en informe l'utilisateur.

        last_modification = get_last_modification_timestamp(idevents, None)

        if last_modification and datetime.fromtimestamp(\

            float(krgv['last_modification'])) < last_modification:

            flash(_('Changes have occurred since the page was last displayed, '

                    'your changes HAVE NOT been saved.'), 'warning')

            raise redirect(request.environ.get('HTTP_REFERER', url('/')))

        # Vérification que au moins un des identifiants existe et est éditable

        if not events.num_rows():

            flash(_('No access to this event'), 'error')

            redirect('/')

        # Modification des événements et création d'un historique

        # pour chacun d'eux.

        for req in events.req:

            if isinstance(req, CorrEvent):

                event = req

            else:

                event = req[0]

            if krgv['trouble_ticket'] != '' :

                history = EventHistory(

                        type_action="Ticket change",

                        idevent=event.idcause,

                        value=krgv['trouble_ticket'],

                        text="Changed trouble ticket from '%s' to '%s'" % (

                            event.trouble_ticket, krgv['trouble_ticket']

                        ),

                        username=username,

                        timestamp=datetime.now(),

                    )

                DBSession.add(history)   

                event.trouble_ticket = krgv['trouble_ticket']

            if krgv['ack'] != 'NoChange' :

                history = EventHistory(

                        type_action="Acknowledgement change state",

                        idevent=event.idcause,

                        value=krgv['ack'],

                        text="Changed acknowledgement status "

                            "from '%s' to '%s'" % (

                            event.status, krgv['ack']

                        ),

                        username=username,

                        timestamp=datetime.now(),

                    )

                DBSession.add(history)

                event.status = krgv['ack']

        DBSession.flush()

        flash(_('Updated successfully'))

        redirect(request.environ.get('HTTP_REFERER', url('/')))

    @validate(validators={

        "plugin_name": validators.OneOf([i[0] for i \

            in config.get('vigiboard_plugins', [])]),

        'idcorrevent': validators.Int(not_empty=True),

        }, error_handler=process_form_errors)

    @expose('json')

    @require(Any(not_anonymous(), msg=l_("You need to be authenticated")))

    def get_plugin_value(self, idcorrevent, plugin_name, *arg, **krgv):

        """

        Permet de récupérer la valeur d'un plugin associée à un CorrEvent

        donné via JSON.

        """

        plugins = config.get('vigiboard_plugins', {})

        # Permet de vérifier si l'utilisateur a bien les permissions

        # pour accéder à cet événement et si l'événement existe.

        username = request.environ['repoze.who.identity']['repoze.who.userid']

        events = VigiboardRequest(User.by_user_name(username))

        events.add_table(CorrEvent.idcorrevent)

        events.add_join((Event, CorrEvent.idcause == Event.idevent))

        events.add_join((events.items, 

            Event.idsupitem == events.items.c.idsupitem))

        events.add_filter(CorrEvent.idcorrevent == idcorrevent)

        # Pas d'événement ou permission refusée. On ne distingue pas

        # les 2 cas afin d'éviter la divulgation d'informations.

        if not events.num_rows():

            raise HTTPNotFound(_('No such incident or insufficient permissions'))

        plugin_class = [p[1] for p in plugins if p[0] == plugin_name]

        if not plugin_class:

            raise HTTPNotFound(_('No such plugin'))

        plugin_class = plugin_class[0]

        try:

            mypac = __import__(

                'vigiboard.controllers.plugins.' + plugin_name,

                globals(), locals(), [plugin_class], -1)

            plugin = getattr(mypac, plugin_class)

            if callable(plugin):

                return plugin().get_value(idcorrevent, *arg, **krgv)

            raise HTTPInternalServerError(_('Not a valid plugin'))

        except ImportError:

            raise HTTPInternalServerError(_('Plugin could not be loaded'))

    @validate(validators={

        "fontsize": validators.Regex(

            r'[0-9]+(pt|px|em|%)',

            regexOps = ('I',)

        )}, error_handler = process_form_errors)

    @expose('json')

    def set_fontsize(self, fontsize):

        """Enregistre la taille de la police dans les préférences."""

        session['fontsize'] = fontsize

        session.save()

        return dict()

    @validate(validators={"refresh": validators.Int()},

            error_handler=process_form_errors)

    @expose('json')

    def set_refresh(self, refresh):

        """Enregistre le temps de rafraichissement dans les préférences."""

        session['refresh'] = refresh

        session.save()

        return dict()

    @expose('json')

    def set_theme(self, theme):

        """Enregistre le thème à utiliser dans les préférences."""

        # On sauvegarde l'ID du thème sans vérifications

        # car les thèmes (styles CSS) sont définies dans

        # les packages de thèmes (ex: vigilo-themes-default).

        # La vérification de la valeur est faite dans les templates.

        session['theme'] = theme

        session.save()

        return dict()

def get_last_modification_timestamp(event_id_list, 

                                    value_if_none=datetime.now()):

    """

    Récupère le timestamp de la dernière modification 

    opérée sur l'un des événements dont l'identifiant

    fait partie de la liste passée en paramètre.

    """

    last_modification_timestamp = DBSession.query(

                                func.max(EventHistory.timestamp),

                         ).filter(EventHistory.idevent.in_(event_id_list)

                         ).scalar()

    if not last_modification_timestamp:

        if not value_if_none:

            return None

        else:

            last_modification_timestamp = value_if_none

    return datetime.fromtimestamp(mktime(

        last_modification_timestamp.timetuple()))

def get_plugins_instances():

    """

    Renvoie une liste d'instances de plugins pour VigiBoard.

    @return: Liste de tuples contenant le nom du plugin et l'instance associé.

    @rtype: C{list} of C{tuple}

    """

    plugins = config.get('vigiboard_plugins', [])

    plugins_instances = []

    for (plugin_name, plugin_class) in plugins:

        try:

            mypac = __import__(

                'vigiboard.controllers.plugins.' + plugin_name,

                globals(), locals(), [plugin_class], -1)

            plugin = getattr(mypac, plugin_class)

            if callable(plugin):

                plugins_instances.append((plugin_name, plugin()))

        except ImportError:

            pass

    return plugins_instances