import tg

from tg import expose, flash, require, url, request, redirect, validate, tmpl_context

from tw.forms import validators 

from sets import Set

from sqlalchemy import not_ , and_ , asc , desc, or_

from tw.jquery import JQueryUIDialog

from vigiboard.model import DBSession, metadata

from vigiboard.model.vigiboard_bdd import *

from vigiboard.model.auth import Permission

from repoze.what.predicates import Any,not_anonymous

from vigiboard.widgets.edit_event import Edit_Event_Form , Search_Form , edit_event_status_options

def GetUserGroups():

	gr = DBSession.query(Groups.name).join(

	    ( GroupPermissions , Groups.name == GroupPermissions.groupname ),

	    ( Permission , Permission.permission_id == GroupPermissions.idpermission )

	    ).filter(Permission.permission_name.in_(request.environ.get('repoze.who.identity').get('permissions')))

	lst_grp = Set([i.name for i in gr])

	lst_tmp = lst_grp

	while len(lst_tmp) > 0:

		gr = DBSession.query(Groups.name).filter(Groups.parent.in_(lst_tmp))

		tmp = Set([])

		for i in gr :

			tmp.add(i.name)

			lst_grp.add(i.name)

		lst_tmp = tmp

	return lst_grp

def FormatEventsImgStatu (event):

	if event.active and event.status == 'AAClosed':

		return { 'src': url('/images/vigiboard/crossed.png') }

	elif event.status == 'Acknowledged' :

		return { 'src': url('/images/vigiboard/checked.png') }

	else:

		return None

def FormatEvents(events,first_row,last_row):

	ev = []

	i=0

	severity = { 0 : 'Minor' , 1 : 'Minor', 2 : 'Minor', 3 : 'Minor', 4 : 'Minor', 5 : 'Minor' , 6 : 'Major' , 7 : 'Critical' }

	class_tr = ['odd','even']

	class_ack = { 'Acknowledged' : 'Ack' , 'None' : '' , 'AAClosed' : 'Ack' }

	ids = []

	for event in events[first_row : last_row]:

		ids.append(event.idevent)

		if event.active :

			ev.append([

				event,

				{'class': class_tr[i%2]},

				{'class' : severity[event.severity] + class_ack[event.status]},

				{'class' : severity[event.severity] + class_ack[event.status] },

				{'src' : url('/images/vigiboard/%s2.png' % severity[event.severity].upper())},

				FormatEventsImgStatu(event)

				])

		else :

			ev.append([

				event,

				{'class': class_tr[i%2]},

				{'class' : severity[event.severity] + class_ack[event.status] },

				{'class' : 'Cleared' + class_ack[event.status] },

				{'src' : url('/images/vigiboard/%s2.png' % severity[event.severity].upper())},

				FormatEventsImgStatu(event)

				])

		i=i+1

	return ev,ids

def FormatHistory (idevents,hostname,servicename):

	history = DBSession.query(EventHistory).filter(EventHistory.idevent.in_(idevents)).order_by(desc(EventHistory.timestamp))

	severity = { 0 : 'None' , 1 : 'OK', 2 : 'Suppressed', 3 : 'Initial', 4 : 'Maintenance', 5 : 'Minor' , 6 : 'Major' , 7 : 'Critical' }

	hist = []

	i = 0

	class_tr = ['odd','even']

	for h in history :

		if h.value :

			hist.append([

				h.idhistory,

				hostname,

				servicename,

				h.timestamp,

				h.username,

				h.type_action,

				severity[min(int(h.value),7)],

				h.text,

				{'class' : class_tr[i%2]},

				{'class':severity[min(int(h.value),7)]}

			])

		else:

			hist.append([

				h.idhistory,

				hostname,

				servicename,

				h.timestamp,

				h.username,

				h.type_action,

				severity[0],

				h.text,

				{'class' : class_tr[i%2]},

				{'class':severity[0]}

			])	

		i = i+1

	return hist

def GenerateTmplContext():

	tmpl_context.edit_eventdialog = JQueryUIDialog(id='Edit_EventsDialog',autoOpen=False,title='Edit Event')

        tmpl_context.searchdialog = JQueryUIDialog(id='SearchDialog',autoOpen=False,title='Search Event')

        tmpl_context.historydialog = JQueryUIDialog(id='HistoryDialog',autoOpen=False,title='History')

        tmpl_context.edit_event_form = Edit_Event_Form('edit_event_form',action=url('/update_event'))

        tmpl_context.search_form = Search_Form('search_form',action=url('/vigiboard'))

    @expose('vigiboard.templates.vigiboard')

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

    def vigiboard(self,page='1',host=None,service=None,output=None,trouble_ticket=None):

	   try :

	   	page = int(page)

	   except :

		page = 1

	   if page < 1 :

		  page = 1

	   events = DBSession.query(Events).join(

		( Host, Events.hostname == Host.name ),

		( Service, Events.servicename == Service.name ),

		( HostGroups , Host.name == HostGroups.hostname )

		).filter(HostGroups.groupname.in_(GetUserGroups())

		).filter(not_(and_(Events.active == False,Events.status == 'AAClosed'))

		).filter(Events.timestamp_active != None

		).filter(not_(Events.timestamp_active.like('0000-00-00 00:00:00'))

		)

	   if host :

		   events = events.filter(Events.hostname.like('%%%s%%' % host))

	   if service :

		   events = events.filter(Events.servicename.like('%%%s%%' % service))

	   if output :	

		   events = events.filter(Events.output.like('%%%s%%' % output))

	   if trouble_ticket :

		   events = events.filter(Events.trouble_ticket.like('%%%s%%' % trouble_ticket))

	   events= events.order_by(asc(Events.status)

		).order_by(desc(Events.active)

		).order_by(desc(Events.severity)

		).order_by(asc(Events.hostname)

		).order_by(desc(Events.timestamp))

	   total_row = events.count()

	   item_per_page = int(tg.config['vigiboard_item_per_page'])

	   if total_row <= item_per_page * (page-1) :

		page = 1

	   id_first_row = item_per_page * (page-1)

	   id_last_row = min(id_first_row + item_per_page,total_row)

	   GenerateTmplContext()

	   events , ids = FormatEvents(events,id_first_row,id_last_row)

	   return dict(

			   events=events,

			   id_first_row=id_first_row,

			   id_last_row=id_last_row,

			   total_row=total_row,

			   pages=range(1,(total_row/item_per_page) + 2),

			   page=page,

			   event_edit_status_options=edit_event_status_options,

			   history=[],

			   hist_error = False

		)

    @expose('json')

    def vigiboard_HistoryDialog ( self , id ) :

	    events = DBSession.query(Events.severity,Events.idevent,Events.hostname,Events.servicename).filter(Events.idevent == id)[0]

	    initial_state = DBSession.query(EventHistory).filter(EventHistory.idevent == id).order_by(asc(EventHistory.timestamp)).order_by(asc(EventHistory.type_action))

	    if initial_state.count() > 0 :

		    initial_state = initial_state[0].value

	    else :

		    initial_state = 0

	    severity = { 0 : 'None' , 1 : 'OK', 2 : 'Suppressed', 3 : 'Initial', 4 : 'Maintenance', 5 : 'Minor' , 6 : 'Major' , 7 : 'Critical' }

	    return dict(

		initial_state=severity[int(initial_state)],

		current_state=severity[events.severity],

		idevent = events.idevent,

		host = events.hostname,

		service = events.servicename,

		nagios_link = tg.config['vigiboard_links.nagios'] % {'idevent':events.idevent},

		metrology_link = tg.config['vigiboard_links.metrology'] % {'idevent':events.idevent},

		security_link = tg.config['vigiboard_links.security'] % {'idevent':events.idevent},

		servicetype_link = tg.config['vigiboard_links.servicetype'] % {'idevent':events.idevent}

		)

    @expose('vigiboard.templates.vigiboard')

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

    def vigiboard_event(self,idevent):

	   events = DBSession.query(Events).join(

		( Host, Events.hostname == Host.name ),

		( Service, Events.servicename == Service.name ),

		( HostGroups , Host.name == HostGroups.hostname )

		).filter(HostGroups.groupname.in_(GetUserGroups())

	   	).filter(not_(and_(Events.active == False,Events.status == 'AAClosed'))

		).filter(Events.timestamp_active != None

		).filter(not_(Events.timestamp_active.like('0000-00-00 00:00:00'))

		).filter(Events.idevent == idevent)

	   if events.count() != 1 :

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

		redirect('/vigiboard')

   	   GenerateTmplContext()

	   events , ids = FormatEvents(events,0,1)

	   return dict(

			   events=events,

			   id_first_row=1,

			   id_last_row=1,

			   total_row=1,

			   pages=[1],

			   page=1,

			   event_edit_status_options=edit_event_status_options,

			   history=FormatHistory(ids,events[0][0].hostname,events[0][0].servicename),

			   hist_error = True

		)

    @expose('vigiboard.templates.vigiboard')

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

    def vigiboard_host_service(self,host,service):

	   events = DBSession.query(Events).join(

		( Host, Events.hostname == Host.name ),

		( Service, Events.servicename == Service.name ),

		( HostGroups , Host.name == HostGroups.hostname )

		).filter(HostGroups.groupname.in_(GetUserGroups())

	   	).filter(not_(and_(Events.active == False,Events.status == 'AAClosed'))

		).filter(Events.timestamp_active != None

		).filter(not_(Events.timestamp_active.like('0000-00-00 00:00:00'))

		).filter(Events.hostname == host

		).filter(Events.servicename == service)

	   if events.count() == 0 :

		redirect('/vigiboard')

	   GenerateTmplContext()

	   events , ids = FormatEvents(events,0,events.count())

	   return dict(

			   events=events,

			   id_first_row=1,

			   id_last_row=1,

			   total_row=1,

			   pages=[1],

			   page=1,

			   event_edit_status_options=edit_event_status_options,

			   history=FormatHistory(ids,host,service),

			   hist_error = True

		)

    def edit_event_form_errors (self,**argv):

	    flash(_(tmpl_context.form_errors),'error')

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

    @expose('vigiboard.templates.vigiboard_update')

    @validate(validators={

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

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

	    "status":validators.OneOf(['None', 'Acknowledged', 'AAClosed'])

	    },error_handler=edit_event_form_errors)

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

    def update_event(self,*argv,**krgv):

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

	   if len(ids) > 1 :

		   ids = ids[:-1]

	   events = DBSession.query(Events).join(

		( Host, Events.hostname == Host.name ),

		( Service, Events.servicename == Service.name ),

		( HostGroups , Host.name == HostGroups.hostname )

		).filter(HostGroups.groupname.in_(GetUserGroups())

		).filter(Events.idevent.in_(ids))

	   if events.count() <= 0 :

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

		   redirect('/vigiboard')

	   for event in events :

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

		   	event.trouble_ticket = krgv['tt']

		   	history = EventHistory(type_action="Ticket change",idevent=event.idevent,value='',text='',username=request.environ.get('repoze.who.identity').get('repoze.who.userid'))

		   	DBSession.add(history)	

		   if events[0].status != krgv['status'] :

		   	event.status = krgv['status']

                   	history = EventHistory(type_action="Acknowlegement change state",idevent=event.idevent,value='',text='',username=request.environ.get('repoze.who.identity').get('repoze.who.userid'))

                   	DBSession.add(history)

   	   flash(_('Updated successfully'))

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