backend/src/appenlight/views/integrations/bitbucket.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.

from appenlight.models.integrations.bitbucket import (
    BitbucketIntegration,
    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
import logging
from datetime import datetime
from pyramid.httpexceptions import HTTPUnprocessableEntity
from webob.multidict import MultiDict

log = logging.getLogger(__name__)

from . import IntegrationView


class BitbucketView(IntegrationView):
    @view_config(
        route_name="integrations_id",
        match_param=["action=info", "integration=bitbucket"],
        renderer="json",
    )
    def get_bitbucket_info(self):
        """
        Grab information about possible priority levels and assignable users
        """
        try:
            client = BitbucketIntegration.create_client(
                self.request,
                self.integration.config["user_name"],
                self.integration.config["repo_name"],
            )
        except IntegrationException as e:
            self.request.response.status_code = 503
            return {"error_messages": [str(e)]}
        assignees = client.get_assignees()
        priorities = client.get_priorities()
        return {"assignees": assignees, "priorities": priorities}

    @view_config(
        route_name="integrations_id",
        match_param=["action=create-issue", "integration=bitbucket"],
        renderer="json",
    )
    def create_issue(self):
        """
        Creates a new issue in bitbucket issue tracker 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", ""),
            "kind": "bug",
            "priority": self.request.unsafe_json_body["priority"],
            "responsible": self.request.unsafe_json_body["responsible"]["user"],
        }

        try:
            client = BitbucketIntegration.create_client(
                self.request,
                self.integration.config["user_name"],
                self.integration.config["repo_name"],
            )
            issue = client.create_issue(form_data)
        except IntegrationException as e:
            self.request.response.status_code = 503
            return {"error_messages": [str(e)]}

        comment_body = "Bitbucket 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=bitbucket"],
        renderer="json",
        permission="edit",
    )
    def setup(self):
        """
        Validates and creates integration between application and bitbucket
        """
        resource = self.request.context.resource
        form = forms.IntegrationBitbucketForm(
            MultiDict(self.request.safe_json_body or {}),
            csrf_context=self.request,
            **self.integration_config
        )
        if self.request.method == "POST" and form.validate():
            integration_config = {
                "repo_name": form.repo_name.data,
                "user_name": form.user_name.data,
                "host_name": "https://bitbucket.org",
            }
            if not self.integration:
                # add new integration
                self.integration = BitbucketIntegration(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)
        return self.integration_config