~launchpad-pqm/launchpad/devel

# Copyright 2009-2011 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""In-memory doubles of core codehosting objects."""

__metaclass__ = type

__all__ = [

    'InMemoryFrontend',

    'XMLRPCWrapper',

    ]

import operator

from xmlrpclib import Fault

from bzrlib.urlutils import (

    escape,

    unescape,

    )

from twisted.internet import defer

from zope.component import (

    adapter,

    getSiteManager,

    )

from zope.interface import implementer

from canonical.database.constants import UTC_NOW

from lp.xmlrpc import faults

from lp.app.validators import LaunchpadValidationError

from lp.app.validators.name import valid_name

from lp.code.bzr import (

    BranchFormat,

    ControlFormat,

    RepositoryFormat,

    )

from lp.code.enums import BranchType

from lp.code.errors import UnknownBranchTypeError

from lp.code.interfaces.branch import IBranch

from lp.code.interfaces.branchtarget import IBranchTarget

from lp.code.interfaces.codehosting import (

    BRANCH_ALIAS_PREFIX,

    branch_id_alias,

    BRANCH_ID_ALIAS_PREFIX,

    BRANCH_TRANSPORT,

    CONTROL_TRANSPORT,

    LAUNCHPAD_ANONYMOUS,

    LAUNCHPAD_SERVICES,

    )

from lp.code.interfaces.linkedbranch import ICanHasLinkedBranch

from lp.code.model.branchnamespace import BranchNamespaceSet

from lp.code.model.branchtarget import (

    PackageBranchTarget,

    ProductBranchTarget,

    )

from lp.code.xmlrpc.codehosting import datetime_from_tuple

from lp.registry.errors import InvalidName

from lp.registry.interfaces.pocket import PackagePublishingPocket

from lp.services.utils import iter_split

from lp.services.xmlrpc import LaunchpadFault

from lp.testing.factory import ObjectFactory

class FakeStore:

    """Fake store that implements find well enough to pass tests.

    This is needed because some of the `test_codehosting` tests use

    assertSqlAttributeEqualsDate, which relies on ORM behaviour. Here, we fake

    enough of the ORM to pass the tests.

    """

    def __init__(self, object_set):

        self._object_set = object_set

    def find(self, cls, **kwargs):

        """Implement Store.find that takes two attributes: id and one other.

        This is called by `assertSqlAttributeEqualsDate`, which relies on

        `find` returning either a single match or None. Returning a match

        implies that the given attribute has the expected value. Returning

        None implies the opposite.

        """

        branch_id = kwargs.pop('id')

        assert len(kwargs) == 1, (

            'Expected only id and one other. Got %r' % kwargs)

        attribute = kwargs.keys()[0]

        expected_value = kwargs[attribute]

        branch = self._object_set.get(branch_id)

        if branch is None:

            return FakeResult(None)

        if expected_value is getattr(branch, attribute):

            return FakeResult(branch)

        return None

class FakeResult:

    """As with FakeStore, just enough of a result to pass tests."""

    def __init__(self, branch):

        self._branch = branch

    def one(self):

        return self._branch

class FakeDatabaseObject:

    """Base class for fake database objects."""

    def _set_object_set(self, object_set):

        self.__storm_object_info__ = {'store': FakeStore(object_set)}

class ObjectSet:

    """Generic set of database objects."""

    def __init__(self):

        self._objects = {}

        self._next_id = 1

    def _add(self, db_object):

        self._objects[self._next_id] = db_object

        db_object.id = self._next_id

        self._next_id += 1

        db_object._set_object_set(self)

        return db_object

    def _delete(self, db_object):

        del self._objects[db_object.id]

    def __iter__(self):

        return self._objects.itervalues()

    def _find(self, **kwargs):

        [(key, value)] = kwargs.items()

        for obj in self:

            if getattr(obj, key) == value:

                return obj

    def get(self, id):

        return self._objects.get(id, None)

    def getByName(self, name):

        return self._find(name=name)

class BranchSet(ObjectSet):

    """Extra find methods for branches."""

    def getByUniqueName(self, unique_name):

        return self._find(unique_name=unique_name)

class FakeSourcePackage:

    """Fake ISourcePackage."""

    def __init__(self, sourcepackagename, distroseries):

        self.sourcepackagename = sourcepackagename

        self.distroseries = distroseries

    def __hash__(self):

        return hash((self.sourcepackagename.id, self.distroseries.id))

    def __eq__(self, other):

        return (self.sourcepackagename.id == other.sourcepackagename.id

                and self.distroseries.id == other.distroseries.id)

    def __ne__(self, other):

        return not (self == other)

    @property

    def distribution(self):

        if self.distroseries is not None:

            return self.distroseries.distribution

        else:

            return None

    @property

    def development_version(self):

        name = '%s-devel' % self.distribution.name

        dev_series = self._distroseries_set.getByName(name)

        if dev_series is None:

            dev_series = FakeDistroSeries(name, self.distribution)

            self._distroseries_set._add(dev_series)

        return self.__class__(self.sourcepackagename, dev_series)

    @property

    def path(self):

        return '%s/%s/%s' % (

            self.distribution.name,

            self.distroseries.name,

            self.sourcepackagename.name)

    def getBranch(self, pocket):

        return self.distroseries._linked_branches.get(

            (self, pocket), None)

    def setBranch(self, pocket, branch, registrant):

        self.distroseries._linked_branches[self, pocket] = branch

class SourcePackageNameSet(ObjectSet):

    def new(self, name_string):

        if not valid_name(name_string):

            raise InvalidName(name_string)

        return self._add(FakeSourcePackageName(name_string))

@adapter(FakeSourcePackage)

@implementer(IBranchTarget)

def fake_source_package_to_branch_target(fake_package):

    return PackageBranchTarget(fake_package)

class FakeBranch(FakeDatabaseObject):

    """Fake branch object."""

    def __init__(self, branch_type, name, owner, url=None, product=None,

                 stacked_on=None, private=False, registrant=None,

                 distroseries=None, sourcepackagename=None):

        self.branch_type = branch_type

        self.last_mirror_attempt = None

        self.last_mirrored = None

        self.last_mirrored_id = None

        self.next_mirror_time = None

        self.url = url

        self.mirror_failures = 0

        self.name = name

        self.owner = owner

        self.stacked_on = None

        self.mirror_status_message = None

        self.stacked_on = stacked_on

        self.private = private

        self.product = product

        self.registrant = registrant

        self._mirrored = False

        self.distroseries = distroseries

        self.sourcepackagename = sourcepackagename

    @property

    def unique_name(self):

        if self.product is None:

            if self.distroseries is None:

                product = '+junk'

            else:

                product = '%s/%s/%s' % (

                    self.distroseries.distribution.name,

                    self.distroseries.name,

                    self.sourcepackagename.name)

        else:

            product = self.product.name

        return '~%s/%s/%s' % (self.owner.name, product, self.name)

    def getPullURL(self):

        return 'lp-fake:///' + self.unique_name

    @property

    def target(self):

        if self.product is None:

            if self.distroseries is None:

                target = self.owner

            else:

                target = self.sourcepackage

        else:

            target = self.product

        return IBranchTarget(target)

    def requestMirror(self):

        self.next_mirror_time = UTC_NOW

class FakePerson(FakeDatabaseObject):

    """Fake person object."""

    is_team = False

    def __init__(self, name):

        self.name = self.displayname = name

    def inTeam(self, person_or_team):

        if self is person_or_team:

            return True

        if not person_or_team.is_team:

            return False

        return self in person_or_team._members

class FakeTeam(FakePerson):

    """Fake team."""

    is_team = True

    def __init__(self, name, members=None):

        super(FakeTeam, self).__init__(name)

        if members is None:

            self._members = []

        else:

            self._members = list(members)

class FakeProduct(FakeDatabaseObject):

    """Fake product."""

    def __init__(self, name, owner):

        self.name = name

        self.owner = owner

        self.bzr_path = name

        self.development_focus = FakeProductSeries(self, 'trunk')

        self.series = {

            'trunk': self.development_focus,

            }

    def getSeries(self, name):

        return self.series.get(name, None)

@adapter(FakeProduct)

@implementer(IBranchTarget)

def fake_product_to_branch_target(fake_product):

    """Adapt a `FakeProduct` to `IBranchTarget`."""

    return ProductBranchTarget(fake_product)

@adapter(FakeProduct)

@implementer(ICanHasLinkedBranch)

def fake_product_to_can_has_linked_branch(fake_product):

    """Adapt a `FakeProduct` to `ICanHasLinkedBranch`."""

    return fake_product.development_focus

class FakeProductSeries(FakeDatabaseObject):

    """Fake product series."""

    branch = None

    def __init__(self, product, name):

        self.product = product

        self.name = name

    @property

    def bzr_path(self):

        if self.product.development_focus is self:

            return self.product.name

        else:

            return "%s/%s" % (self.product.name, self.name)

@adapter(FakeProductSeries)

@implementer(ICanHasLinkedBranch)

def fake_productseries_to_can_has_linked_branch(fake_productseries):

    """Adapt a `FakeProductSeries` to `ICanHasLinkedBranch`."""

    return fake_productseries

class FakeScriptActivity(FakeDatabaseObject):

    """Fake script activity."""

    def __init__(self, name, hostname, date_started, date_completed):

        self.id = self.name = name

        self.hostname = hostname

        self.date_started = datetime_from_tuple(date_started)

        self.date_completed = datetime_from_tuple(date_completed)

class FakeDistribution(FakeDatabaseObject):

    def __init__(self, name):

        self.name = name

class FakeDistroSeries(FakeDatabaseObject):

    """Fake distroseries."""

    def __init__(self, name, distribution):

        self.name = name

        self.distribution = distribution

        self._linked_branches = {}

class FakeSourcePackageName(FakeDatabaseObject):

    """Fake SourcePackageName."""

    def __init__(self, name):

        self.name = name

DEFAULT_PRODUCT = object()

class FakeObjectFactory(ObjectFactory):

    def __init__(self, branch_set, person_set, product_set, distribution_set,

                 distroseries_set, sourcepackagename_set):

        super(FakeObjectFactory, self).__init__()

        self._branch_set = branch_set

        self._person_set = person_set

        self._product_set = product_set

        self._distribution_set = distribution_set

        self._distroseries_set = distroseries_set

        self._sourcepackagename_set = sourcepackagename_set

    def makeBranch(self, branch_type=None, stacked_on=None, private=False,

                   product=DEFAULT_PRODUCT, owner=None, name=None,

                   registrant=None, sourcepackage=None):

        if branch_type is None:

            branch_type = BranchType.HOSTED

        if branch_type == BranchType.MIRRORED:

            url = self.getUniqueURL()

        else:

            url = None

        if name is None:

            name = self.getUniqueString()

        if owner is None:

            owner = self.makePerson()

        if product is DEFAULT_PRODUCT:

            product = self.makeProduct()

        if registrant is None:

            registrant = self.makePerson()

        if sourcepackage is None:

            sourcepackagename = None

            distroseries = None

        else:

            sourcepackagename = sourcepackage.sourcepackagename

            distroseries = sourcepackage.distroseries

        IBranch['name'].validate(unicode(name))

        branch = FakeBranch(

            branch_type, name=name, owner=owner, url=url,

            stacked_on=stacked_on, product=product, private=private,

            registrant=registrant, distroseries=distroseries,

            sourcepackagename=sourcepackagename)

        self._branch_set._add(branch)

        return branch

    def makeAnyBranch(self, **kwargs):

        return self.makeProductBranch(**kwargs)

    def makePackageBranch(self, sourcepackage=None, **kwargs):

        if sourcepackage is None:

            sourcepackage = self.makeSourcePackage()

        return self.makeBranch(

            product=None, sourcepackage=sourcepackage, **kwargs)

    def makePersonalBranch(self, owner=None, **kwargs):

        if owner is None:

            owner = self.makePerson()

        return self.makeBranch(

            owner=owner, product=None, sourcepackage=None, **kwargs)

    def makeProductBranch(self, product=None, **kwargs):

        if product is None:

            product = self.makeProduct()

        return self.makeBranch(product=product, sourcepackage=None, **kwargs)

    def makeDistribution(self):

        distro = FakeDistribution(self.getUniqueString())

        self._distribution_set._add(distro)

        return distro

    def makeDistroSeries(self):

        distro = self.makeDistribution()

        distroseries_name = self.getUniqueString()

        distroseries = FakeDistroSeries(distroseries_name, distro)

        self._distroseries_set._add(distroseries)

        return distroseries

    def makeSourcePackageName(self):

        return self._sourcepackagename_set.new(self.getUniqueString())

    def makeSourcePackage(self, distroseries=None, sourcepackagename=None):

        if distroseries is None:

            distroseries = self.makeDistroSeries()

        if sourcepackagename is None:

            sourcepackagename = self.makeSourcePackageName()

        package = FakeSourcePackage(sourcepackagename, distroseries)

        package._distroseries_set = self._distroseries_set

        return package

    def makeTeam(self, owner):

        team = FakeTeam(name=self.getUniqueString(), members=[owner])

        self._person_set._add(team)

        return team

    def makePerson(self, name=None):

        if name is None:

            name = self.getUniqueString()

        person = FakePerson(name=name)

        self._person_set._add(person)

        return person

    def makeProduct(self, name=None, owner=None):

        if name is None:

            name = self.getUniqueString()

        if owner is None:

            owner = self.makePerson()

        product = FakeProduct(name, owner)

        self._product_set._add(product)

        return product

    def makeProductSeries(self, product, name=None):

        if name is None:

            name = self.getUniqueString()

        series = FakeProductSeries(product, name)

        product.series[name] = series

        return series

    def enableDefaultStackingForProduct(self, product, branch=None):

        """Give 'product' a default stacked-on branch.

        :param product: The product to give a default stacked-on branch to.

        :param branch: The branch that should be the default stacked-on

            branch.  If not supplied, a fresh branch will be created.

        """

        if branch is None:

            branch = self.makeBranch(product=product)

        product.development_focus.branch = branch

        branch.last_mirrored_id = 'rev1'

        return branch

    def enableDefaultStackingForPackage(self, package, branch):

        """Give 'package' a default stacked-on branch.

        :param package: The package to give a default stacked-on branch to.

        :param branch: The branch that should be the default stacked-on

            branch.

        """

        package.development_version.setBranch(

            PackagePublishingPocket.RELEASE, branch, branch.owner)

        branch.last_mirrored_id = 'rev1'

        return branch

class FakeCodehosting:

    def __init__(self, branch_set, person_set, product_set, distribution_set,

                 distroseries_set, sourcepackagename_set, factory,

                 script_activity_set):

        self._branch_set = branch_set

        self._person_set = person_set

        self._product_set = product_set

        self._distribution_set = distribution_set

        self._distroseries_set = distroseries_set

        self._sourcepackagename_set = sourcepackagename_set

        self._factory = factory

        self._script_activity_set = script_activity_set

    def acquireBranchToPull(self, branch_type_names):

        if not branch_type_names:

            branch_type_names = 'HOSTED', 'MIRRORED', 'IMPORTED'

        branch_types = []

        for branch_type_name in branch_type_names:

            try:

                branch_types.append(BranchType.items[branch_type_name])

            except KeyError:

                raise UnknownBranchTypeError(

                    'Unknown branch type: %r' % (branch_type_name,))

        branches = sorted(

            [branch for branch in self._branch_set

             if branch.next_mirror_time is not None

             and branch.branch_type in branch_types],

            key=operator.attrgetter('next_mirror_time'))

        if branches:

            branch = branches[-1]

            # Mark it as started mirroring.

            branch.last_mirror_attempt = UTC_NOW

            branch.next_mirror_time = None

            default_branch = branch.target.default_stacked_on_branch

            if default_branch is None:

                default_branch_name = ''

            elif (branch.branch_type == BranchType.MIRRORED

                  and default_branch.private):

                default_branch_name = ''

            else:

                default_branch_name = '/' + default_branch.unique_name

            return (branch.id, branch.getPullURL(), branch.unique_name,

                    default_branch_name, branch.branch_type.name)

        else:

            return ()

    def mirrorFailed(self, branch_id, reason):

        branch = self._branch_set.get(branch_id)

        if branch is None:

            return faults.NoBranchWithID(branch_id)

        branch.mirror_failures += 1

        branch.mirror_status_message = reason

        return True

    def recordSuccess(self, name, hostname, date_started, date_completed):

        self._script_activity_set._add(

            FakeScriptActivity(name, hostname, date_started, date_completed))

        return True

    def _parseUniqueName(self, branch_path):

        """Return a dict of the parsed information and the branch name."""

        try:

            namespace_path, branch_name = branch_path.rsplit('/', 1)

        except ValueError:

            raise faults.PermissionDenied(

                "Cannot create branch at '/%s'" % branch_path)

        data = BranchNamespaceSet().parse(namespace_path)

        return data, branch_name

    def _createBranch(self, registrant, branch_path):

        """The guts of the create branch method.

        Raises exceptions on error conditions.

        """

        to_link = None

        if branch_path.startswith(BRANCH_ALIAS_PREFIX + '/'):

            branch_path = branch_path[len(BRANCH_ALIAS_PREFIX) + 1:]

            if branch_path.startswith('~'):

                data, branch_name = self._parseUniqueName(branch_path)

            else:

                tokens = branch_path.split('/')

                data = {

                    'person': registrant.name,

                    'product': tokens[0],

                    }

                branch_name = 'trunk'

                # check the series

                product = self._product_set.getByName(data['product'])

                if product is not None:

                    if len(tokens) > 1:

                        series = product.getSeries(tokens[1])

                        if series is None:

                            raise faults.NotFound(

                                "No such product series: '%s'." % tokens[1])

                        else:

                            to_link = ICanHasLinkedBranch(series)

                    else:

                        to_link = ICanHasLinkedBranch(product)

                # don't forget the link.

        else:

            data, branch_name = self._parseUniqueName(branch_path)

        owner = self._person_set.getByName(data['person'])

        if owner is None:

            raise faults.NotFound(

                "User/team '%s' does not exist." % (data['person'],))

        # The real code consults the branch creation policy of the product. We

        # don't need to do so here, since the tests above this layer never

        # encounter that behaviour. If they *do* change to rely on the branch

        # creation policy, the observed behaviour will be failure to raise

        # exceptions.

        if not registrant.inTeam(owner):

            raise faults.PermissionDenied(

                ('%s cannot create branches owned by %s'

                 % (registrant.displayname, owner.displayname)))

        product = sourcepackage = None

        if data['product'] == '+junk':

            product = None

        elif data['product'] is not None:

            product = self._product_set.getByName(data['product'])