/brz/remove-bazaar

# Copyright (C) 2010, 2011 Canonical Ltd

# Copyright (C) 2018 Breezy Developers

#

# 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

"""Support for Launchpad."""

import re

import shutil

import tempfile

from ...propose import (

    Hoster,

    LabelsUnsupported,

    MergeProposal,

    MergeProposalBuilder,

    MergeProposalExists,

    UnsupportedHoster,

    )

from ... import (

    branch as _mod_branch,

    controldir,

    errors,

    hooks,

    urlutils,

    )

from ...git.urls import git_url_to_bzr_url

from ...lazy_import import lazy_import

lazy_import(globals(), """

from breezy.plugins.launchpad import (

    lp_api,

    uris as lp_uris,

    )

from launchpadlib import uris

""")

from ...transport import get_transport

# TODO(jelmer): Make selection of launchpad staging a configuration option.

def status_to_lp_mp_statuses(status):

    statuses = []

    if status in ('open', 'all'):

        statuses.extend([

            'Work in progress',

            'Needs review',

            'Approved',

            'Code failed to merge',

            'Queued'])

    if status in ('closed', 'all'):

        statuses.extend(['Rejected', 'Superseded'])

    if status in ('merged', 'all'):

        statuses.append('Merged')

    return statuses

def plausible_launchpad_url(url):

    if url is None:

        return False

    if url.startswith('lp:'):

        return True

    regex = re.compile(r'([a-z]*\+)*(bzr\+ssh|http|ssh|git|https)'

                       r'://(bazaar|git).*\.launchpad\.net')

    return bool(regex.match(url))

class WebserviceFailure(Exception):

    def __init__(self, message):

        self.message = message

def _call_webservice(call, *args, **kwargs):

    """Make a call to the webservice, wrapping failures.

    :param call: The call to make.

    :param *args: *args for the call.

    :param **kwargs: **kwargs for the call.

    :return: The result of calling call(*args, *kwargs).

    """

    from lazr.restfulclient import errors as restful_errors

    try:

        return call(*args, **kwargs)

    except restful_errors.HTTPError as e:

        error_lines = []

        for line in e.content.splitlines():

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

                break

            error_lines.append(line)

        raise WebserviceFailure(b''.join(error_lines))

class LaunchpadMergeProposal(MergeProposal):

    def __init__(self, mp):

        self._mp = mp

    def get_source_branch_url(self):

        if self._mp.source_branch:

            return self._mp.source_branch.bzr_identity

        else:

            return git_url_to_bzr_url(

                self._mp.source_git_repository.git_identity,

                ref=self._mp.source_git_path.encode('utf-8'))

    def get_target_branch_url(self):

        if self._mp.target_branch:

            return self._mp.target_branch.bzr_identity

        else:

            return git_url_to_bzr_url(

                self._mp.target_git_repository.git_identity,

                ref=self._mp.target_git_path.encode('utf-8'))

    @property

    def url(self):

        return lp_api.canonical_url(self._mp)

    def is_merged(self):

        return (self._mp.queue_status == 'Merged')

    def is_closed(self):

        return (self._mp.queue_status in ('Rejected', 'Superseded'))

    def reopen(self):

        self._mp.setStatus(status='Needs review')

    def get_description(self):

        return self._mp.description

    def set_description(self, description):

        self._mp.description = description

        self._mp.lp_save()

    def get_commit_message(self):

        return self._mp.commit_message

    def set_commit_message(self, commit_message):

        self._mp.commit_message = commit_message

        self._mp.lp_save()

    def close(self):

        self._mp.setStatus(status='Rejected')

    def can_be_merged(self):

        if not self._mp.preview_diff:

            # Maybe?

            return True

        return not bool(self._mp.preview_diff.conflicts)

    def get_merged_by(self):

        merge_reporter = self._mp.merge_reporter

        if merge_reporter is None:

            return None

        return merge_reporter.name

    def get_merged_at(self):

        return self._mp.date_merged

    def merge(self, commit_message=None):

        target_branch = _mod_branch.Branch.open(

            self.get_target_branch_url())

        source_branch = _mod_branch.Branch.open(

            self.get_source_branch_url())

        # TODO(jelmer): Ideally this would use a memorytree, but merge doesn't

        # support that yet.

        # tree = target_branch.create_memorytree()

        tmpdir = tempfile.mkdtemp()

        try:

            tree = target_branch.create_checkout(

                to_location=tmpdir, lightweight=True)

            tree.merge_from_branch(source_branch)

            tree.commit(commit_message or self._mp.commit_message)

        finally:

            shutil.rmtree(tmpdir)

class Launchpad(Hoster):

    """The Launchpad hosting service."""

    name = 'launchpad'

    # https://bugs.launchpad.net/launchpad/+bug/397676

    supports_merge_proposal_labels = False

    supports_merge_proposal_commit_message = True

    merge_proposal_description_format = 'plain'

    def __init__(self, staging=False):

        self._staging = staging

        if staging:

            lp_base_url = uris.STAGING_SERVICE_ROOT

        else:

            lp_base_url = uris.LPNET_SERVICE_ROOT

        self.launchpad = lp_api.connect_launchpad(lp_base_url, version='devel')

    @property

    def base_url(self):

        return lp_api.uris.web_root_for_service_root(

            str(self.launchpad._root_uri))

    def __repr__(self):

        return "Launchpad(staging=%s)" % self._staging

    def hosts(self, branch):

        # TODO(jelmer): staging vs non-staging?

        return plausible_launchpad_url(branch.user_url)

    @classmethod

    def probe_from_url(cls, url, possible_transports=None):

        if plausible_launchpad_url(url):

            return Launchpad()

        raise UnsupportedHoster(url)

    def _get_lp_git_ref_from_branch(self, branch):

        url, params = urlutils.split_segment_parameters(branch.user_url)

        (scheme, user, password, host, port, path) = urlutils.parse_url(

            url)

        repo_lp = self.launchpad.git_repositories.getByPath(

            path=path.strip('/'))

        try:

            ref_path = params['ref']

        except KeyError:

            branch_name = params.get('branch', branch.name)

            if branch_name:

                ref_path = 'refs/heads/%s' % branch_name

            else:

                ref_path = repo_lp.default_branch

        ref_lp = repo_lp.getRefByPath(path=ref_path)

        return (repo_lp, ref_lp)

    def _get_lp_bzr_branch_from_branch(self, branch):

        return self.launchpad.branches.getByUrl(

            url=urlutils.unescape(branch.user_url))

    def _get_derived_git_path(self, base_path, owner, project):

        base_repo = self.launchpad.git_repositories.getByPath(path=base_path)

        if project is None:

            project = urlutils.parse_url(base_repo.git_ssh_url)[-1].strip('/')

        if project.startswith('~'):

            project = '/'.join(base_path.split('/')[1:])

        # TODO(jelmer): Surely there is a better way of creating one of these

        # URLs?

        return "~%s/%s" % (owner, project)

    def _publish_git(self, local_branch, base_path, name, owner, project=None,

                     revision_id=None, overwrite=False, allow_lossy=True):

        to_path = self._get_derived_git_path(base_path, owner, project)

        to_transport = get_transport("git+ssh://git.launchpad.net/" + to_path)

        try:

            dir_to = controldir.ControlDir.open_from_transport(to_transport)

        except errors.NotBranchError:

            # Didn't find anything

            dir_to = None

        if dir_to is None:

            try:

                br_to = local_branch.create_clone_on_transport(

                    to_transport, revision_id=revision_id, name=name)

            except errors.NoRoundtrippingSupport:

                br_to = local_branch.create_clone_on_transport(

                    to_transport, revision_id=revision_id, name=name,

                    lossy=True)

        else:

            try:

                dir_to = dir_to.push_branch(

                    local_branch, revision_id, overwrite=overwrite, name=name)

            except errors.NoRoundtrippingSupport:

                if not allow_lossy:

                    raise

                dir_to = dir_to.push_branch(

                    local_branch, revision_id, overwrite=overwrite, name=name,

                    lossy=True)

            br_to = dir_to.target_branch

        return br_to, (

            "https://git.launchpad.net/%s/+ref/%s" % (to_path, name))

    def _get_derived_bzr_path(self, base_branch, name, owner, project):

        if project is None:

            base_branch_lp = self._get_lp_bzr_branch_from_branch(base_branch)

            project = '/'.join(base_branch_lp.unique_name.split('/')[1:-1])

        # TODO(jelmer): Surely there is a better way of creating one of these

        # URLs?

        return "~%s/%s/%s" % (owner, project, name)

    def get_push_url(self, branch):

        (vcs, user, password, path, params) = self._split_url(branch.user_url)

        if vcs == 'bzr':

            branch_lp = self._get_lp_bzr_branch_from_branch(branch)

            return branch_lp.bzr_identity

        elif vcs == 'git':

            return urlutils.join_segment_parameters(

                "git+ssh://git.launchpad.net/" + path, params)

        else:

            raise AssertionError

    def _publish_bzr(self, local_branch, base_branch, name, owner,

                     project=None, revision_id=None, overwrite=False,

                     allow_lossy=True):

        to_path = self._get_derived_bzr_path(base_branch, name, owner, project)

        to_transport = get_transport("lp:" + to_path)

        try:

            dir_to = controldir.ControlDir.open_from_transport(to_transport)

        except errors.NotBranchError:

            # Didn't find anything

            dir_to = None

        if dir_to is None:

            br_to = local_branch.create_clone_on_transport(

                to_transport, revision_id=revision_id)

        else:

            br_to = dir_to.push_branch(

                local_branch, revision_id, overwrite=overwrite).target_branch

        return br_to, ("https://code.launchpad.net/" + to_path)

    def _split_url(self, url):

        url, params = urlutils.split_segment_parameters(url)

        (scheme, user, password, host, port, path) = urlutils.parse_url(url)

        path = path.strip('/')

        if host.startswith('bazaar.'):

            vcs = 'bzr'

        elif host.startswith('git.'):

            vcs = 'git'

        else:

            raise ValueError("unknown host %s" % host)

        return (vcs, user, password, path, params)

    def publish_derived(self, local_branch, base_branch, name, project=None,

                        owner=None, revision_id=None, overwrite=False,

                        allow_lossy=True):

        """Publish a branch to the site, derived from base_branch.

        :param base_branch: branch to derive the new branch from

        :param new_branch: branch to publish

        :param name: Name of the new branch on the remote host

        :param project: Optional project name

        :param owner: Optional owner

        :return: resulting branch

        """

        if owner is None:

            owner = self.launchpad.me.name

        (base_vcs, base_user, base_password, base_path,

            base_params) = self._split_url(base_branch.user_url)

        # TODO(jelmer): Prevent publishing to development focus

        if base_vcs == 'bzr':

            return self._publish_bzr(

                local_branch, base_branch, name, project=project, owner=owner,

                revision_id=revision_id, overwrite=overwrite,

                allow_lossy=allow_lossy)

        elif base_vcs == 'git':

            return self._publish_git(

                local_branch, base_path, name, project=project, owner=owner,

                revision_id=revision_id, overwrite=overwrite,

                allow_lossy=allow_lossy)

        else:

            raise AssertionError('not a valid Launchpad URL')

    def get_derived_branch(self, base_branch, name, project=None, owner=None):

        if owner is None:

            owner = self.launchpad.me.name

        (base_vcs, base_user, base_password, base_path,

            base_params) = self._split_url(base_branch.user_url)

        if base_vcs == 'bzr':

            to_path = self._get_derived_bzr_path(

                base_branch, name, owner, project)

            return _mod_branch.Branch.open("lp:" + to_path)

        elif base_vcs == 'git':

            to_path = self._get_derived_git_path(

                base_path.strip('/'), owner, project)

            to_url = urlutils.join_segment_parameters(

                "git+ssh://git.launchpad.net/" + to_path,

                {'branch': name})

            return _mod_branch.Branch.open(to_url)

        else:

            raise AssertionError('not a valid Launchpad URL')

    def iter_proposals(self, source_branch, target_branch, status='open'):

        (base_vcs, base_user, base_password, base_path,

            base_params) = self._split_url(target_branch.user_url)

        statuses = status_to_lp_mp_statuses(status)

        if base_vcs == 'bzr':

            target_branch_lp = self.launchpad.branches.getByUrl(

                url=target_branch.user_url)

            source_branch_lp = self.launchpad.branches.getByUrl(

                url=source_branch.user_url)

            for mp in target_branch_lp.getMergeProposals(status=statuses):

                if mp.source_branch_link != source_branch_lp.self_link:

                    continue

                yield LaunchpadMergeProposal(mp)

        elif base_vcs == 'git':

            (source_repo_lp, source_branch_lp) = (

                self._get_lp_git_ref_from_branch(source_branch))

            (target_repo_lp, target_branch_lp) = (

                self._get_lp_git_ref_from_branch(target_branch))

            for mp in target_branch_lp.getMergeProposals(status=statuses):

                if (target_branch_lp.path != mp.target_git_path or

                        target_repo_lp != mp.target_git_repository or

                        source_branch_lp.path != mp.source_git_path or

                        source_repo_lp != mp.source_git_repository):

                    continue

                yield LaunchpadMergeProposal(mp)

        else:

            raise AssertionError('not a valid Launchpad URL')

    def get_proposer(self, source_branch, target_branch):

        (base_vcs, base_user, base_password, base_path,

            base_params) = self._split_url(target_branch.user_url)

        if base_vcs == 'bzr':

            return LaunchpadBazaarMergeProposalBuilder(

                self, source_branch, target_branch)

        elif base_vcs == 'git':

            return LaunchpadGitMergeProposalBuilder(

                self, source_branch, target_branch)

        else:

            raise AssertionError('not a valid Launchpad URL')

    @classmethod

    def iter_instances(cls):

        yield cls()

    def iter_my_proposals(self, status='open'):

        statuses = status_to_lp_mp_statuses(status)

        for mp in self.launchpad.me.getMergeProposals(status=statuses):

            yield LaunchpadMergeProposal(mp)

    def iter_my_forks(self):

        # Launchpad doesn't really have the concept of "forks"

        return iter([])

    def get_proposal_by_url(self, url):

        # Launchpad doesn't have a way to find a merge proposal by URL.

        (scheme, user, password, host, port, path) = urlutils.parse_url(

            url)

        LAUNCHPAD_CODE_DOMAINS = [

            ('code.%s' % domain) for domain in lp_uris.LAUNCHPAD_DOMAINS.values()]

        if host not in LAUNCHPAD_CODE_DOMAINS:

            raise UnsupportedHoster(url)

        # TODO(jelmer): Check if this is a launchpad URL. Otherwise, raise

        # UnsupportedHoster

        # See https://api.launchpad.net/devel/#branch_merge_proposal

        # the syntax is:

        # https://api.launchpad.net/devel/~<author.name>/<project.name>/<branch.name>/+merge/<id>

        api_url = str(self.launchpad._root_uri) + path

        mp = self.launchpad.load(api_url)

        return LaunchpadMergeProposal(mp)

class LaunchpadBazaarMergeProposalBuilder(MergeProposalBuilder):

    def __init__(self, lp_host, source_branch, target_branch,

                 staging=None, approve=None, fixes=None):

        """Constructor.

        :param source_branch: The branch to propose for merging.

        :param target_branch: The branch to merge into.

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

            production.

        :param approve: If True, mark the new proposal as approved immediately.

            This is useful when a project permits some things to be approved

            by the submitter (e.g. merges between release and deployment

            branches).

        """

        self.lp_host = lp_host

        self.launchpad = lp_host.launchpad

        self.source_branch = source_branch

        self.source_branch_lp = self.launchpad.branches.getByUrl(

            url=source_branch.user_url)

        if target_branch is None:

            self.target_branch_lp = self.source_branch_lp.get_target()

            self.target_branch = _mod_branch.Branch.open(

                self.target_branch_lp.bzr_identity)

        else:

            self.target_branch = target_branch

            self.target_branch_lp = self.launchpad.branches.getByUrl(

                url=target_branch.user_url)

        self.approve = approve

        self.fixes = fixes

    def get_infotext(self):

        """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)

        return ''.join(info)

    def get_initial_body(self):

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

        :return: a str or None.

        """

        if not self.hooks['merge_proposal_body']:

            return None

        def list_modified_files():

            lca_tree = self.source_branch_lp.find_lca_tree(

                self.target_branch_lp)

            source_tree = self.source_branch.basis_tree()

            files = modified_files(lca_tree, source_tree)

            return list(files)

        with self.target_branch.lock_read(), \

                self.source_branch.lock_read():

            body = None

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

                body = hook({

                    'target_branch': self.target_branch_lp.bzr_identity,

                    '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 MergeProposalExists(lp_api.canonical_url(mp))

    def approve_proposal(self, mp):

        with self.source_branch.lock_read():

            _call_webservice(

                mp.createComment,

                vote=u'Approve',

                subject='',  # Use the default subject.

                content=u"Rubberstamp! Proposer approves of own proposal.")

            _call_webservice(mp.setStatus, status=u'Approved',

                             revid=self.source_branch.last_revision())

    def create_proposal(self, description, reviewers=None, labels=None,

                        prerequisite_branch=None, commit_message=None,

                        work_in_progress=False):

        """Perform the submission."""

        if labels:

            raise LabelsUnsupported(self)

        if prerequisite_branch is not None:

            prereq = self.launchpad.branches.getByUrl(

                url=prerequisite_branch.user_url)

        else:

            prereq = None

        if reviewers is None:

            reviewer_objs = []

        else:

            reviewer_objs = []

            for reviewer in reviewers:

                if '@' in reviewer:

                    reviewer_obj = self.launchpad.people.getByEmail(email=reviewer)

                else:

                    reviewer_obj = self.launchpad.people[reviewer]

                reviewer_objs.append(reviewer_obj)

        try:

            mp = _call_webservice(

                self.source_branch_lp.createMergeProposal,

                target_branch=self.target_branch_lp,

                prerequisite_branch=prereq,

                initial_comment=description.strip(),

                commit_message=commit_message,

                needs_review=(not work_in_progress),

                reviewers=[reviewer.self_link for reviewer in reviewer_objs],

                review_types=['' for reviewer in reviewer_objs])

        except WebserviceFailure as e:

            # Urgh.

            if (b'There is already a branch merge proposal '

                    b'registered for branch ') in e.message:

                raise MergeProposalExists(self.source_branch.user_url)

            raise

        if self.approve:

            self.approve_proposal(mp)

        if self.fixes:

            if self.fixes.startswith('lp:'):

                self.fixes = self.fixes[3:]

            _call_webservice(

                mp.linkBug,

                bug=self.launchpad.bugs[int(self.fixes)])

        return LaunchpadMergeProposal(mp)

class LaunchpadGitMergeProposalBuilder(MergeProposalBuilder):

    def __init__(self, lp_host, source_branch, target_branch,

                 staging=None, approve=None, fixes=None):

        """Constructor.

        :param source_branch: The branch to propose for merging.

        :param target_branch: The branch to merge into.

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

            production.

        :param approve: If True, mark the new proposal as approved immediately.

            This is useful when a project permits some things to be approved

            by the submitter (e.g. merges between release and deployment

            branches).

        """

        self.lp_host = lp_host

        self.launchpad = lp_host.launchpad

        self.source_branch = source_branch

        (self.source_repo_lp,

            self.source_branch_lp) = self.lp_host._get_lp_git_ref_from_branch(

                source_branch)