~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2010 Canonical Ltd

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

import urlparse

import webbrowser

from bzrlib import (

    errors,

    hooks,

    msgeditor,

)

from bzrlib.plugins.launchpad import (

    lp_api,

    lp_registration,

)

from lazr.restfulclient import errors as restful_errors

class ProposeMergeHooks(hooks.Hooks):

    """Hooks for proposing a merge on Launchpad."""

    def __init__(self):

        hooks.Hooks.__init__(self)

        self.create_hook(

            hooks.HookPoint(

                'get_prerequisite',

                "Return the prerequisite branch for proposing as merge.",

                (2, 1), None),

        )

        self.create_hook(

            hooks.HookPoint(

                'merge_proposal_body',

                "Return an initial body for the merge proposal message.",

                (2, 1), None),

        )

class Proposer(object):

    hooks = ProposeMergeHooks()

    def __init__(self, tree, source_branch, target_branch, message, reviews,

                 staging=False):

        """Constructor.

        :param tree: The working tree for the source branch.

        :param source_branch: The branch to propose for merging.

        :param target_branch: The branch to merge into.

        :param message: The commit message to use.  (May be None.)

        :param reviews: A list of tuples of reviewer, review type.

        :param staging: If True, propose the merge against staging instead of

            production.

        """

        self.tree = tree

        if staging:

            lp_instance = 'staging'

        else:

            lp_instance = 'production'

        service = lp_registration.LaunchpadService(lp_instance=lp_instance)

        self.launchpad = lp_api.login(service)

        self.source_branch = lp_api.LaunchpadBranch.from_bzr(

            self.launchpad, source_branch)

        if target_branch is None:

            self.target_branch = self.source_branch.get_dev_focus()

        else:

            self.target_branch = lp_api.LaunchpadBranch.from_bzr(

                self.launchpad, target_branch)

        self.commit_message = message

        if reviews == []:

            target_reviewer = self.target_branch.lp.reviewer

            if target_reviewer is None:

                raise errors.BzrCommandError('No reviewer specified')

            self.reviews = [(target_reviewer, '')]

        else:

            self.reviews = [(self.launchpad.people[reviewer], review_type)

                            for reviewer, review_type in

                            reviews]

    def get_comment(self, prerequisite_branch):

        """Determine the initial comment for the merge proposal."""

        info = ["Source: %s\n" % self.source_branch.lp.bzr_identity]

        info.append("Target: %s\n" % self.target_branch.lp.bzr_identity)

        if prerequisite_branch is not None:

            info.append("Prereq: %s\n" % prerequisite_branch.lp.bzr_identity)

        for rdata in self.reviews:

            uniquename = "%s (%s)" % (rdata[0].display_name, rdata[0].name)

            info.append('Reviewer: %s, type "%s"\n' % (uniquename, rdata[1]))

        self.source_branch.bzr.lock_read()

        try:

            self.target_branch.bzr.lock_read()

            try:

                body = self.get_initial_body()

            finally:

                self.target_branch.bzr.unlock()

        finally:

            self.source_branch.bzr.unlock()

        initial_comment = msgeditor.edit_commit_message(''.join(info),

                                                        start_message=body)

        return initial_comment.strip().encode('utf-8')

    def get_initial_body(self):

        """Get a body for the proposal for the user to modify.

        :return: a str or None.

        """

        def list_modified_files():

            lca_tree = self.source_branch.find_lca_tree(

                self.target_branch)

            source_tree = self.source_branch.bzr.basis_tree()

            files = modified_files(lca_tree, source_tree)

            return list(files)

        target_loc = ('bzr+ssh://bazaar.launchpad.net/%s' %

                       self.target_branch.lp.unique_name)

        body = None

        for hook in self.hooks['merge_proposal_body']:

            body = hook({

                'tree': self.tree,

                'target_branch': target_loc,

                'modified_files_callback': list_modified_files,

                'old_body': body,

            })

        return body

    def check_proposal(self):

        """Check that the submission is sensible."""

        if self.source_branch.lp.self_link == self.target_branch.lp.self_link:

            raise errors.BzrCommandError(

                'Source and target branches must be different.')

        for mp in self.source_branch.lp.landing_targets:

            if mp.queue_status in ('Merged', 'Rejected'):

                continue

            if mp.target_branch.self_link == self.target_branch.lp.self_link:

                raise errors.BzrCommandError(

                    'There is already a branch merge proposal: %s' %

                    canonical_url(mp))

    def _get_prerequisite_branch(self):

        hooks = self.hooks['get_prerequisite']

        prerequisite_branch = None

        for hook in hooks:

            prerequisite_branch = hook(

                {'launchpad': self.launchpad,

                 'source_branch': self.source_branch,

                 'target_branch': self.target_branch,

                 'prerequisite_branch': prerequisite_branch})

        return prerequisite_branch

    def create_proposal(self):

        """Perform the submission."""

        prerequisite_branch = self._get_prerequisite_branch()

        if prerequisite_branch is None:

            prereq = None

        else:

            prereq = prerequisite_branch.lp

            prerequisite_branch.update_lp()

        self.source_branch.update_lp()

        reviewers = []

        review_types = []

        for reviewer, review_type in self.reviews:

            review_types.append(review_type)

            reviewers.append(reviewer.self_link)

        initial_comment = self.get_comment(prerequisite_branch)

        try:

            mp = self.source_branch.lp.createMergeProposal(

                target_branch=self.target_branch.lp,

                prerequisite_branch=prereq,

                initial_comment=initial_comment,

                commit_message=self.commit_message, reviewers=reviewers,

                review_types=review_types)

        except restful_errors.HTTPError, e:

            error_lines = []

            for line in e.content.splitlines():

                if line.startswith('Traceback (most recent call last):'):

                    break

                error_lines.append(line)

            raise Exception(''.join(error_lines))

        else:

            webbrowser.open(canonical_url(mp))

def modified_files(old_tree, new_tree):

    """Return a list of paths in the new tree with modified contents."""

    for f, (op, path), c, v, p, n, (ok, k), e in new_tree.iter_changes(

        old_tree):

        if c and k == 'file':

            yield str(path)

def canonical_url(object):

    """Return the canonical URL for a branch."""

    scheme, netloc, path, params, query, fragment = urlparse.urlparse(

        str(object.self_link))

    path = '/'.join(path.split('/')[2:])

    netloc = netloc.replace('api.', 'code.')

    return urlparse.urlunparse((scheme, netloc, path, params, query,

                                fragment))