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

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

"""Vigiboard Controller"""

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 not_, and_, asc

from datetime import datetime

import math

from vigiboard.model import DBSession

from vigiboard.model import Event, EventHistory, CorrEvent, \

                            Host, HostGroup, \

                            StateName, User

from repoze.what.predicates import Any, not_anonymous

from vigiboard.widgets.edit_event import edit_event_status_options

from vigiboard.controllers.vigiboardrequest import VigiboardRequest

from vigiboard.controllers.vigiboard_controller import VigiboardRootController

__all__ = ('RootController', )

class RootController(VigiboardRootController):

    """

    Le controller général de vigiboard

    """

    # XXX Mettre ça dans un fichier de configuration.

    refresh_times = (

        (0, l_('Never')),

        (30, l_('30 seconds')),

        (60, l_('1 minute')),

        (300, l_('5 minutes')),

        (600, l_('10 minutes')),

    )

    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('/')

    @expose('vigiboard.html')

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

    def default(self, page = None, host = None, service = None, output = None,

            trouble_ticket=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 page is None:

            page = 1

        try:

            page = int(page)

        except ValueError:

            abort(404)

        if page < 1:

            page = 1

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

        user = User.by_user_name(username)

        aggregates = VigiboardRequest(user)

        search = {

            'host': '',

            'service': '',

            'output': '',

            'tt': ''

        }

        # Application des filtres si nécessaire

        if host :

            search['host'] = host

            host = host.replace('%', '\\%').replace('_', '\\_') \

                    .replace('*', '%').replace('?', '_')

            aggregates.add_filter(Event.hostname.ilike('%%%s%%' % host))

        if service :

            search['service'] = service

            service = service.replace('%', '\\%').replace('_', '\\_') \

                    .replace('*', '%').replace('?', '_')

            aggregates.add_filter(Event.servicename.ilike('%%%s%%' % service))

        if output :

            search['output'] = output

            output = output.replace('%', '\\%').replace('_', '\\_') \

                    .replace('*', '%').replace('?', '_')

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

        if trouble_ticket :

            search['tt'] = trouble_ticket

            trouble_ticket = trouble_ticket.replace('%', '\\%') \

                    .replace('_', '\\_').replace('*', '%').replace('?', '_')

            aggregates.add_filter(CorrEvent.trouble_ticket.ilike(

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

        # 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'])

        if total_rows <= items_per_page * (page-1):

            page = 1

        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)))

        return dict(

                   events = aggregates.events,

                   rows_info = {

                       'id_first_row': id_first_row + 1,

                       'id_last_row': id_last_row,

                       'total_rows': total_rows,

                   },

                   pages = range(1, nb_pages + 1),

                   page = page,

                   event_edit_status_options = edit_event_status_options,

                   history = [],

                   hist_error = False,

                   plugin_context = aggregates.context_fct,

                   search = search,

                   refresh_times=self.refresh_times,

                )

    @validate(validators={'idcorrevent':validators.String(not_empty=True)},

            error_handler=process_form_errors)

    @expose('json')

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

    def history_dialog(self, idcorrevent):

        """

        JSon renvoyant les éléments pour l'affichage de la fenêtre de dialogue

        contenant des liens internes et externes.

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

        @param id: identifiant de l'événement

        """

        # Obtention de données sur l'événement et sur son historique

        username = request.environ.get('repoze.who.identity'

                    ).get('repoze.who.userid')

        user = User.by_user_name(username)

        event = DBSession.query(

                        CorrEvent.priority,

                        Event,

                 ).join(

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

                    (HostGroup, Event.hostname == HostGroup.hostname),

                 ).filter(HostGroup.idgroup.in_(user.groups)

                 ).filter(CorrEvent.idcorrevent == idcorrevent

                 ).one()

        history = DBSession.query(

                    EventHistory,

                 ).filter(EventHistory.idevent == event[1].idevent

                 ).order_by(asc(EventHistory.timestamp)

                 ).order_by(asc(EventHistory.type_action)).all()

        eventdetails = {}

        for edname, edlink in \

                config['vigiboard_links.eventdetails'].iteritems():

            eventdetails[edname] = edlink[1] % {

                    'idcorrevent': idcorrevent,

                    'host': event[1].hostname,

                    'service': event[1].servicename

                    }

        return dict(

                current_state = StateName.value_to_statename(

                                    event[1].current_state),

                initial_state = StateName.value_to_statename(

                                    event[1].initial_state),

                peak_state = StateName.value_to_statename(

                                    event[1].peak_state),

                idcorrevent = idcorrevent,

                host = event[1].hostname,

                service = event[1].servicename,

                eventdetails = eventdetails,

            )

    @validate(validators={'idcorrevent':validators.String(not_empty=True)},

            error_handler=process_form_errors)

    @expose('vigiboard.html')

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

    def event(self, idcorrevent):

        """

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

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

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

        """

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

        events = VigiboardRequest(User.by_user_name(username))

        events.add_filter(CorrEvent.idcorrevent == idcorrevent)

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

        if events.num_rows() != 1 :

            flash(_('Error in DB'), 'error')

            redirect('/')

        events.format_events(0, 1)

        events.format_history()

        events.generate_tmpl_context() 

        return dict(

                    events = events.events,

                    rows_info = {

                        'id_first_row': 1,

                        'id_last_row': 1,

                        'total_rows': 1,

                    },

                    pages = [1],

                    page = 1,

                    event_edit_status_options = edit_event_status_options,

                    history = events.hist,

                    hist_error = True,

                    plugin_context = events.context_fct,

                    search = {

                        'host': None,

                        'service': None,

                        'output': None,

                        'tt': None

                    },

                   refresh_times=self.refresh_times,

                )

    @validate(validators={'host':validators.NotEmpty(),

        'service':validators.NotEmpty()}, error_handler=process_form_errors)

    @expose('vigiboard.html')

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

    def host_service(self, host, service):

        """

        Affichage de l'historique de l'ensemble des événements correspondant

        au host et 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é

        """

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

        events = VigiboardRequest(User.by_user_name(username))

        events.add_filter(Event.hostname == host,

                Event.servicename == service)

        # XXX On devrait avoir une autre API que ça !!!

        # Supprime le filtre qui empêche d'obtenir des événements fermés

        # (ie: ayant l'état Nagios 'OK' et le statut 'AAClosed').

        if len(events.filter) > 2:

            del events.filter[2]

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

        if events.num_rows() == 0 :

            redirect('/')

        events.format_events(0, events.num_rows())

        events.format_history()

        events.generate_tmpl_context()

        return dict(

                    events = events.events,

                    rows_info = {

                        'id_first_row': 1,

                        'id_last_row': 1,

                        'total_rows': 1,

                    },

                    pages = [1],

                    page = 1,

                    event_edit_status_options = edit_event_status_options,

                    history = events.hist,

                    hist_error = True,

                    plugin_context = events.context_fct,

                    search = {

                        'host': None,

                        'service': None,

                        'output': None,

                        'tt': None

                    },

                    refresh_times=self.refresh_times,

                )

    @validate(validators={

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

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

        "status":validators.OneOf(['NoChange', 'None', 'Acknowledged',

                'AAClosed'])

        }, error_handler=process_form_errors)

    @require(Any(not_anonymous(), msg=_("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['tt']: Nouveau numéro du ticket associé.

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

        """

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

        # il nous faut chacun des identifiants

        if krgv['id'] is None:

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

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

        ids = krgv['id'].split(',')

        if len(ids) > 1 :

            ids = ids[:-1]

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

        events = VigiboardRequest(User.by_user_name(username))

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

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

        if events.num_rows() <= 0 :

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

            redirect('/')

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

        # pour chacun d'eux.

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

        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['status'] != 'NoChange' :

                history = EventHistory(

                        type_action="Acknowlegement change state",

                        idevent=event.idcause,

                        value=krgv['status'],

                        text=_("Changed acknowledgement status from '%s' to '%s'") % (

                            event.status, krgv['status']

                        ),

                        username=username,

                        timestamp=datetime.now(),

                    )

                DBSession.add(history)

                event.status = krgv['status']

        DBSession.flush()

        flash(_('Updated successfully'))

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

    @validate(validators={"plugin_name":validators.OneOf(

        [i for [i, j] in config.get('vigiboard_plugins', [])])},

                error_handler = process_form_errors)

    @expose('json')

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

        """

        Permet aux plugins de pouvoir récupérer des valeurs Json

        """

        plugins = config['vigiboard_plugins']

        if plugins is None:

            return

        plugin = [i for i in plugins if i[0] == plugin_name][0]

        try:

            mypac = __import__(

                'vigiboard.controllers.vigiboard_plugin.' + plugin[0],

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

            plug = getattr(mypac, plugin[1])()

            return plug.controller(*arg, **krgv)

        except:

            raise

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

#                    error_handler = process_form_errors)

    @expose('json')

    def set_fontsize(self, fontsize):

        """

        Save font size

        """

        session['fontsize'] = fontsize

        session.save()

        return dict(ret= 'ok')

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

            error_handler = process_form_errors)

    @expose('json')

    def set_refresh(self, refresh):

        """

        Save refresh time

        """

        session['refresh'] = refresh

        session.save()

        return dict(ret= 'ok')