backend/src/appenlight/views/integrations/jira.py

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

# Copyright 2010 - 2017 RhodeCode GmbH and the AppEnlight project authors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import copy
import logging
from appenlight.models.integrations.jira import JiraIntegration, \
    JiraClient, IntegrationException
from appenlight.models.report_comment import ReportComment
from appenlight.models.services.report_group import ReportGroupService
from pyramid.view import view_config
from appenlight import forms
from datetime import datetime
from pyramid.httpexceptions import HTTPFound, HTTPUnprocessableEntity
from webob.multidict import MultiDict

log = logging.getLogger(__name__)

from . import IntegrationView


class JiraView(IntegrationView):
    def create_client(self, form=None):
        """
        Creates a client that can make authenticated requests to jira
        """
        if self.integration and not form:
            user_name = self.integration.config['user_name']
            password = self.integration.config['password']
            host_name = self.integration.config['host_name']
            project = self.integration.config['project']
        else:
            user_name, password = form.user_name.data, form.password.data
            host_name = form.host_name.data
            project = form.host_name.data
        client = JiraClient(user_name, password, host_name, project,
                            request=self.request)
        return client

    @view_config(route_name='integrations_id',
                 match_param=['action=info', 'integration=jira'],
                 renderer='json')
    def get_jira_info(self):
        """
        Get basic metadata - assignees and priority levels from tracker
        """
        try:
            client = self.create_client()
        except IntegrationException as e:
            self.request.response.status_code = 503
            return {'error_messages': [str(e)]}
        assignees = []
        priorities = []
        issue_types = []
        possible_issue_types = client.get_issue_types(self.request)
        for issue_type in possible_issue_types:
            for field in issue_type['fields']:
                if field['id'] == 'assignee':
                    assignees = field['values']
                if field['id'] == 'priority':
                    priorities = field['values']
            issue_types.append({'name':issue_type['name'],
                                'id':issue_type['id']})
        return {'assignees': assignees,
                'priorities': priorities,
                'issue_types': issue_types}

    @view_config(route_name='integrations_id',
                 match_param=['action=create-issue',
                              'integration=jira'],
                 renderer='json')
    def create_issue(self):
        """
        Creates a new issue in jira from report group
        """
        report = ReportGroupService.by_id(
            self.request.unsafe_json_body['group_id'])
        form_data = {
            'title': self.request.unsafe_json_body.get('title',
                                                       'Unknown Title'),
            'content': self.request.unsafe_json_body.get('content', ''),
            'issue_type': self.request.unsafe_json_body['issue_type']['id'],
            'priority': self.request.unsafe_json_body['priority']['id'],
            'responsible': self.request.unsafe_json_body['responsible']['id'],
            'project': self.integration.config['project']
        }
        try:
            client = self.create_client()
            issue = client.create_issue(form_data, request=self.request)
        except IntegrationException as e:
            self.request.response.status_code = 503
            return {'error_messages': [str(e)]}

        comment_body = 'Jira issue created: %s ' % issue['web_url']
        comment = ReportComment(owner_id=self.request.user.id,
                                report_time=report.first_timestamp,
                                body=comment_body)
        report.comments.append(comment)
        return True

    @view_config(route_name='integrations_id',
                 match_param=['action=setup', 'integration=jira'],
                 renderer='json',
                 permission='edit')
    def setup(self):
        """
        Validates and creates integration between application and jira
        """
        resource = self.request.context.resource
        form = forms.IntegrationJiraForm(
            MultiDict(self.request.safe_json_body or {}),
            csrf_context=self.request, **self.integration_config)
        if self.request.method == 'POST' and form.validate():
            integration_config = {
                'user_name': form.user_name.data,
                'password': form.password.data,
                'host_name': form.host_name.data,
                'project': form.project.data
            }
            if not self.integration:
                # add new integration
                self.integration = JiraIntegration(
                    modified_date=datetime.utcnow(),
                )
                self.request.session.flash('Integration added')
                resource.integrations.append(self.integration)
            else:
                self.request.session.flash('Integration updated')
            self.integration.config = integration_config
            return integration_config
        elif self.request.method == 'POST':
            return HTTPUnprocessableEntity(body=form.errors_json)

        to_return = copy.deepcopy(self.integration_config)
        to_return.pop('password', None)
        return to_return