from __future__ import absolute_import

from django import forms
from django.utils.translation import ugettext_lazy as _

from sentry.rules.actions.base import EventAction
from sentry.utils import metrics, json
from sentry.models import Integration

from opsgenie import GetTeamRequest
from opsgenie import GetUserRequest

from .utils import ( build_alert_payload, LEVEL_TO_PRIORITY )

class OpsgenieNotifyServiceForm(forms.Form):
    account = forms.ChoiceField(choices=(), widget=forms.Select())
    # not making this a choice field to avoid perf hit
    # instead doing a check on clean()
    team = forms.CharField(required=False, widget=forms.TextInput())
    # not making this a choice field to avoid perf hit
    # instead doing a check on clean()
    username = forms.CharField(required=False, widget=forms.TextInput())
    user_id = forms.HiddenInput() # this will make a check if the user actually exists in clean()
    team_id = forms.HiddenInput() # this will make a check if the team actually exists in clean()
    priority = forms.ChoiceField(choices=(), widget=forms.Select())
    tags = forms.CharField(required=False, widget=forms.TextInput())

    def __init__(self, *args, **kwargs):
        # NOTE: Account maps directly to the integration ID
        account_list = [(i.id, i.name) for i in kwargs.pop('integrations')]
        priorities = kwargs.pop('priorities', None)
        self.team_and_or_user_transformer = kwargs.pop('team_and_or_user_transformer')

        # remove all the extra kwargs before calling super
        super(OpsgenieNotifyServiceForm, self).__init__(*args, **kwargs)

        if account_list:
            self.fields['account'].initial = account_list[0][0]
            self.fields['account'].choices = account_list
            self.fields['account'].widget.choices = self.fields['account'].choices

        if priorities:
            self.fields['priority'].initial = priorities[0][0]
            self.fields['priority'].choices = priorities
            self.fields['priority'].widget.choices = self.fields['priority'].choices


    def clean(self):
        cleaned_data = super(OpsgenieNotifyServiceForm, self).clean()

        account = cleaned_data.get('account', None)
        team = cleaned_data.get('team', '')
        username = cleaned_data.get('username', '')

        if not account:
            raise forms.ValidationError(_("Pleade select the account to send alerts to"),
                                        code='invalid'
                                    )
        if not team and not username:
            raise forms.ValidationError(_("Either username or team should be present to configure the alert"),
                                        code='invalid'
                                    )

        team_id, user_id = self.team_and_or_user_transformer(account, team, username)

        if team and team_id is None and account is not None:
            params = {
                'team': team,
                'account': dict(self.fields['account'].choices).get(int(account)),
            }

            raise forms.ValidationError(
                _('The opsgenie team "%(team)s" does not exist or has not been granted access in the %(account)s Opsgenie account.'),
                code='invalid',
                params=params,
            )

        if username and user_id is None and account is not None:
            params = {
                'user': user,
                'account': dict(self.fields['account'].choices).get(int(account)),
            }

            raise forms.ValidationError(
                _('The opsgenie user "%(user)s" does not exist or has not been granted access in the %(account)s Opsgenie account.'),
                code='invalid',
                params=params,
            )

        if team_id:
            cleaned_data['team_id'] = team_id
        if user_id:
            cleaned_data['user_id'] = user_id

        return cleaned_data

class OpsgenieNotifyServiceAction(EventAction):
    form_cls = OpsgenieNotifyServiceForm
    label = u'Send an alert to the Opsgenie account {account} routed via {team} team and {username} user and show {tags} tag(s) in alert with {priority} priority'

    def __init__(self, *args, **kwargs):
        super(OpsgenieNotifyServiceAction, self).__init__(*args, **kwargs)
        self.form_fields = {
            'account': {
                'type': 'choice',
                'choices': [(i.id, i.name) for i in self.get_integrations()]
            },
            'team': {
                'type': 'string',
                'placeholder': 'i.e Infrastructure'
            },
            'username': {
                'type': 'string',
                'placeholder': 'i.e <EMAIL>'
            },
            'priority': {
                'type': 'choice',
                'choices': self.get_priorities()
            },
            'tags': {
                'type': 'string',
                'placeholder': 'i.e environment,user,app_name'
            }
        }

    def is_enabled(self):
        return self.get_integrations().exists()

    def after(self, event, state):
        if event.group.is_ignored():
            return

        integration_id = self.get_option('account')
        team_id = self.get_option('team_id')
        user_id = self.get_option('user_id')
        priority = self.get_option('priority')
        tags = set(self.get_tags_list())

        try:
            integration = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=integration_id
            )
        except Integration.DoesNotExist:
            # Integration removed, rule still active.
            return

        def send_alert(event, futures):
            rules = [f.rule for f in futures]
            payload = build_alert_payload(event.group, team_id, user_id, priority, event=event, tags=tags, rules=rules)

            client = integration.get_installation(organization_id=self.project.organization.id).get_client()

            try:
                client.alerts.create_alert(payload)
            except Exception as e:
                self.logger.info('rule.fail.opsgenie_post', extra={'error': e.message})

        key = u'opsgenie:{}:{}:{}'.format(integration_id, team_id, user_id)

        metrics.incr('alert.sent', instance='opsgenie.alert', skip_internal=False)
        yield self.future(send_alert, key=key)

    def render_label(self):
        try:
            integration_name = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=self.get_option('account')
            ).name
        except Integration.DoesNotExist:
            integration_name = '[removed]'

        tags = self.get_tags_list()

        return self.label.format(
            account=integration_name,
            team=self.get_option('team') or 'no',
            username=self.get_option('username') or 'no',
            priority=self.get_option('priority') or 'P3',
            tags=u'[{}]'.format(', '.join(tags)) if len(tags)>0 else 'no',
        )

    def get_tags_list(self):
        return [s.strip() for s in self.get_option('tags', '').split(',')]

    def get_priorities(self):
        return [(v, k) for k, v in LEVEL_TO_PRIORITY.iteritems()]

    def get_integrations(self):
        return Integration.objects.filter(
            provider='opsgenie',
            organizations=self.project.organization,
        )

    def get_form_instance(self):
        return self.form_cls(
            self.data,
            integrations=self.get_integrations(),
            priorities=self.get_priorities(),
            team_and_or_user_transformer=self.get_team_and_or_user_id,
        )

    def get_team_and_or_user_id(self, integration_id, team, username):
        try:
            integration = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=integration_id,
            )
        except Integration.DoesNotExist:
            return None

        client = integration.get_installation(organization_id=self.project.organization.id).get_client()

        try:
            resp = client.teams.get_team(GetTeamRequest(identifier=team, identifierType='name'))
            team_id = resp.id
        except Exception as e:
            self.logger.info('rule.opsgenie.team_list_failed', extra={'error': e.message})
            team_id = None

        try:
            resp = client.users.get_user(GetUserRequest(identifier=username))
            user_id = resp.id
        except Exception as e:
            self.logger.info('rule.opsgenie.user_list_failed', extra={'error': e.message})
            user_id = None

        return team_id, user_id

from __future__ import absolute_import

from django import forms
from django.utils.translation import ugettext_lazy as _

from sentry.rules.actions.base import EventAction
from sentry.utils import metrics, json
from sentry.models import Integration

from opsgenie import GetTeamRequest
from opsgenie import GetUserRequest

from .utils import ( build_alert_payload, LEVEL_TO_PRIORITY )

class OpsgenieNotifyServiceForm(forms.Form):
    account = forms.ChoiceField(choices=(), widget=forms.Select())
    # not making this a choice field to avoid perf hit
    # instead doing a check on clean()
    team = forms.CharField(required=False, widget=forms.TextInput())
    # not making this a choice field to avoid perf hit
    # instead doing a check on clean()
    username = forms.CharField(required=False, widget=forms.TextInput())
    user_id = forms.HiddenInput() # this will make a check if the user actually exists in clean()
    team_id = forms.HiddenInput() # this will make a check if the team actually exists in clean()
    priority = forms.ChoiceField(choices=(), widget=forms.Select())
    tags = forms.CharField(required=False, widget=forms.TextInput())

    def __init__(self, *args, **kwargs):
        # NOTE: Account maps directly to the integration ID
        account_list = [(i.id, i.name) for i in kwargs.pop('integrations')]
        priorities = kwargs.pop('priorities', None)
        self.team_and_or_user_transformer = kwargs.pop('team_and_or_user_transformer')

        # remove all the extra kwargs before calling super
        super(OpsgenieNotifyServiceForm, self).__init__(*args, **kwargs)

        if account_list:
            self.fields['account'].initial = account_list[0][0]
            self.fields['account'].choices = account_list
            self.fields['account'].widget.choices = self.fields['account'].choices

        if priorities:
            self.fields['priority'].initial = priorities[0][0]
            self.fields['priority'].choices = priorities
            self.fields['priority'].widget.choices = self.fields['priority'].choices


    def clean(self):
        cleaned_data = super(OpsgenieNotifyServiceForm, self).clean()

        account = cleaned_data.get('account', None)
        team = cleaned_data.get('team', '')
        username = cleaned_data.get('username', '')

        if not account:
            raise forms.ValidationError(_("Pleade select the account to send alerts to"),
                                        code='invalid'
                                    )
        if not team and not username:
            raise forms.ValidationError(_("Either username or team should be present to configure the alert"),
                                        code='invalid'
                                    )

        team_id, user_id = self.team_and_or_user_transformer(account, team, username)

        if team and team_id is None and account is not None:
            params = {
                'team': team,
                'account': dict(self.fields['account'].choices).get(int(account)),
            }

            raise forms.ValidationError(
                _('The opsgenie team "%(team)s" does not exist or has not been granted access in the %(account)s Opsgenie account.'),
                code='invalid',
                params=params,
            )

        if username and user_id is None and account is not None:
            params = {
                'user': user,
                'account': dict(self.fields['account'].choices).get(int(account)),
            }

            raise forms.ValidationError(
                _('The opsgenie user "%(user)s" does not exist or has not been granted access in the %(account)s Opsgenie account.'),
                code='invalid',
                params=params,
            )

        if team_id:
            cleaned_data['team_id'] = team_id
        if user_id:
            cleaned_data['user_id'] = user_id

        return cleaned_data

class OpsgenieNotifyServiceAction(EventAction):
    form_cls = OpsgenieNotifyServiceForm
    label = u'Send an alert to the Opsgenie account {account} routed via {team} team and {username} user and show {tags} tag(s) in alert with {priority} priority'

    def __init__(self, *args, **kwargs):
        super(OpsgenieNotifyServiceAction, self).__init__(*args, **kwargs)
        self.form_fields = {
            'account': {
                'type': 'choice',
                'choices': [(i.id, i.name) for i in self.get_integrations()]
            },
            'team': {
                'type': 'string',
                'placeholder': 'i.e Infrastructure'
            },
            'username': {
                'type': 'string',
                'placeholder': 'i.e <EMAIL>'
            },
            'priority': {
                'type': 'choice',
                'choices': self.get_priorities()
            },
            'tags': {
                'type': 'string',
                'placeholder': 'i.e environment,user,app_name'
            }
        }

    def is_enabled(self):
        return self.get_integrations().exists()

    def after(self, event, state):
        if event.group.is_ignored():
            return

        integration_id = self.get_option('account')
        team_id = self.get_option('team_id')
        user_id = self.get_option('user_id')
        priority = self.get_option('priority')
        tags = set(self.get_tags_list())

        try:
            integration = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=integration_id
            )
        except Integration.DoesNotExist:
            # Integration removed, rule still active.
            return

        def send_alert(event, futures):
            rules = [f.rule for f in futures]
            payload = build_alert_payload(event.group, team_id, user_id, priority, event=event, tags=tags, rules=rules)

            client = integration.get_installation(organization_id=self.project.organization.id).get_client()

            try:
                client.alerts.create_alert(payload)
            except Exception as e:
                self.logger.info('rule.fail.opsgenie_post', extra={'error': e.message})

        key = u'opsgenie:{}:{}:{}'.format(integration_id, team_id, user_id)

        metrics.incr('alert.sent', instance='opsgenie.alert', skip_internal=False)
        yield self.future(send_alert, key=key)

    def render_label(self):
        try:
            integration_name = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=self.get_option('account')
            ).name
        except Integration.DoesNotExist:
            integration_name = '[removed]'

        tags = self.get_tags_list()

        return self.label.format(
            account=integration_name,
            team=self.get_option('team') or 'no',
            username=self.get_option('username') or 'no',
            priority=self.get_option('priority') or 'P3',
            tags=u'[{}]'.format(', '.join(tags)) if len(tags)>0 else 'no',
        )

    def get_tags_list(self):
        return [s.strip() for s in self.get_option('tags', '').split(',')]

    def get_priorities(self):
        return [(v, k) for k, v in LEVEL_TO_PRIORITY.iteritems()]

    def get_integrations(self):
        return Integration.objects.filter(
            provider='opsgenie',
            organizations=self.project.organization,
        )

    def get_form_instance(self):
        return self.form_cls(
            self.data,
            integrations=self.get_integrations(),
            priorities=self.get_priorities(),
            team_and_or_user_transformer=self.get_team_and_or_user_id,
        )

    def get_team_and_or_user_id(self, integration_id, team, username):
        try:
            integration = Integration.objects.get(
                provider='opsgenie',
                organizations=self.project.organization,
                id=integration_id,
            )
        except Integration.DoesNotExist:
            return None

        client = integration.get_installation(organization_id=self.project.organization.id).get_client()

        try:
            resp = client.teams.get_team(GetTeamRequest(identifier=team, identifierType='name'))
            team_id = resp.id
        except Exception as e:
            self.logger.info('rule.opsgenie.team_list_failed', extra={'error': e.message})
            team_id = None

        try:
            resp = client.users.get_user(GetUserRequest(identifier=username))
            user_id = resp.id
        except Exception as e:
            self.logger.info('rule.opsgenie.user_list_failed', extra={'error': e.message})
            user_id = None

        return team_id, user_id

# not making this a choice field to avoid perf hit # instead doing a check on clean() # not making this a choice field to avoid perf hit # instead doing a check on clean() # this will make a check if the user actually exists in clean() # this will make a check if the team actually exists in clean() # NOTE: Account maps directly to the integration ID # remove all the extra kwargs before calling super # Integration removed, rule still active.