# branchmap.py - logic to computes, maintain and stores branchmap for local repo

#

# Copyright 2005-2007 Olivia Mackall <olivia@selenic.com>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

import struct

from .node import (

    bin,

    hex,

    nullrev,

)

from typing import (

    Any,

    Callable,

    Dict,

    Iterable,

    List,

    Optional,

    Set,

    TYPE_CHECKING,

    Tuple,

    Union,

)

from . import (

    encoding,

    error,

    obsolete,

    scmutil,

    util,

)

from .utils import (

    repoviewutil,

    stringutil,

)

if TYPE_CHECKING:

    from . import localrepo

    assert [localrepo]

subsettable = repoviewutil.subsettable

calcsize = struct.calcsize

pack_into = struct.pack_into

unpack_from = struct.unpack_from

class BranchMapCache:

    """mapping of filtered views of repo with their branchcache"""

    def __init__(self):

        self._per_filter = {}

    def __getitem__(self, repo):

        self.updatecache(repo)

        bcache = self._per_filter[repo.filtername]

        assert bcache._filtername == repo.filtername, (

            bcache._filtername,

            repo.filtername,

        )

        return bcache

    def update_disk(self, repo):

        """ensure and up-to-date cache is (or will be) written on disk

        The cache for this repository view is updated  if needed and written on

        disk.

        If a transaction is in progress, the writing is schedule to transaction

        close. See the `BranchMapCache.write_dirty` method.

        This method exist independently of __getitem__ as it is sometime useful

        to signal that we have no intend to use the data in memory yet.

        """

        self.updatecache(repo)

        bcache = self._per_filter[repo.filtername]

        assert bcache._filtername == repo.filtername, (

            bcache._filtername,

            repo.filtername,

        )

    def updatecache(self, repo):

        """Update the cache for the given filtered view on a repository"""

        # This can trigger updates for the caches for subsets of the filtered

        # view, e.g. when there is no cache for this filtered view or the cache

        # is stale.

        cl = repo.changelog

        filtername = repo.filtername

        bcache = self._per_filter.get(filtername)

        if bcache is None or not bcache.validfor(repo):

            # cache object missing or cache object stale? Read from disk

            bcache = branchcache.fromfile(repo)

        revs = []

        if bcache is None:

            # no (fresh) cache available anymore, perhaps we can re-use

            # the cache for a subset, then extend that to add info on missing

            # revisions.

            subsetname = subsettable.get(filtername)

            if subsetname is not None:

                subset = repo.filtered(subsetname)

                bcache = self[subset].copy(repo)

                extrarevs = subset.changelog.filteredrevs - cl.filteredrevs

                revs.extend(r for r in extrarevs if r <= bcache.tiprev)

            else:

                # nothing to fall back on, start empty.

                bcache = branchcache(repo)

        revs.extend(cl.revs(start=bcache.tiprev + 1))

        if revs:

            bcache.update(repo, revs)

        assert bcache.validfor(repo), filtername

        self._per_filter[repo.filtername] = bcache

    def replace(self, repo, remotebranchmap):

        """Replace the branchmap cache for a repo with a branch mapping.

        This is likely only called during clone with a branch map from a

        remote.

        """

        cl = repo.changelog

        clrev = cl.rev

        clbranchinfo = cl.branchinfo

        rbheads = []

        closed = set()

        for bheads in remotebranchmap.values():

            rbheads += bheads

            for h in bheads:

                r = clrev(h)

                b, c = clbranchinfo(r)

                if c:

                    closed.add(h)

        if rbheads:

            rtiprev = max((int(clrev(node)) for node in rbheads))

            cache = branchcache(

                repo,

                remotebranchmap,

                repo[rtiprev].node(),

                rtiprev,

                closednodes=closed,

            )

            # Try to stick it as low as possible

            # filter above served are unlikely to be fetch from a clone

            for candidate in (b'base', b'immutable', b'served'):

                rview = repo.filtered(candidate)

                if cache.validfor(rview):

                    cache._filtername = candidate

                    self._per_filter[candidate] = cache

                    cache._dirty = True

                    cache.write(rview)

                    return

    def clear(self):

        self._per_filter.clear()

    def write_dirty(self, repo):

        unfi = repo.unfiltered()

        for filtername in repoviewutil.get_ordered_subset():

            cache = self._per_filter.get(filtername)

            if cache is None:

                continue

            if cache._dirty:

                if filtername is None:

                    repo = unfi

                else:

                    repo = unfi.filtered(filtername)

                cache.write(repo)

def _unknownnode(node):

    """raises ValueError when branchcache found a node which does not exists"""

    raise ValueError('node %s does not exist' % node.hex())

def _branchcachedesc(repo):

    if repo.filtername is not None:

        return b'branch cache (%s)' % repo.filtername

    else:

        return b'branch cache'

class _BaseBranchCache:

    """A dict like object that hold branches heads cache.

    This cache is used to avoid costly computations to determine all the

    branch heads of a repo.

    The cache is serialized on disk in the following format:

    <tip hex node> <tip rev number> [optional filtered repo hex hash]

    <branch head hex node> <open/closed state> <branch name>

    <branch head hex node> <open/closed state> <branch name>

    ...

    The first line is used to check if the cache is still valid. If the

    branch cache is for a filtered repo view, an optional third hash is

    included that hashes the hashes of all filtered and obsolete revisions.

    The open/closed state is represented by a single letter 'o' or 'c'.

    This field can be used to avoid changelog reads when determining if a

    branch head closes a branch or not.

    """

    def __init__(

        self,

        repo: "localrepo.localrepository",

        entries: Union[

            Dict[bytes, List[bytes]], Iterable[Tuple[bytes, List[bytes]]]

        ] = (),

        closed_nodes: Optional[Set[bytes]] = None,

    ) -> None:

        """hasnode is a function which can be used to verify whether changelog

        has a given node or not. If it's not provided, we assume that every node

        we have exists in changelog"""

        # closednodes is a set of nodes that close their branch. If the branch

        # cache has been updated, it may contain nodes that are no longer

        # heads.

        if closed_nodes is None:

            closed_nodes = set()

        self._closednodes = set(closed_nodes)

        self._entries = dict(entries)

    def __iter__(self):

        return iter(self._entries)

    def __setitem__(self, key, value):

        self._entries[key] = value

    def __getitem__(self, key):

        return self._entries[key]

    def __contains__(self, key):

        return key in self._entries

    def iteritems(self):

        return self._entries.items()

    items = iteritems

    def hasbranch(self, label):

        """checks whether a branch of this name exists or not"""

        return label in self._entries

    def _branchtip(self, heads):

        """Return tuple with last open head in heads and false,

        otherwise return last closed head and true."""

        tip = heads[-1]

        closed = True

        for h in reversed(heads):

            if h not in self._closednodes:

                tip = h

                closed = False

                break

        return tip, closed

    def branchtip(self, branch):

        """Return the tipmost open head on branch head, otherwise return the

        tipmost closed head on branch.

        Raise KeyError for unknown branch."""

        return self._branchtip(self[branch])[0]

    def iteropen(self, nodes):

        return (n for n in nodes if n not in self._closednodes)

    def branchheads(self, branch, closed=False):

        heads = self._entries[branch]

        if not closed:

            heads = list(self.iteropen(heads))

        return heads

    def iterbranches(self):

        for bn, heads in self.items():

            yield (bn, heads) + self._branchtip(heads)

    def iterheads(self):

        """returns all the heads"""

        return self._entries.values()

    def update(self, repo, revgen):

        """Given a branchhead cache, self, that may have extra nodes or be

        missing heads, and a generator of nodes that are strictly a superset of

        heads missing, this function updates self to be correct.

        """

        starttime = util.timer()

        cl = repo.changelog

        # collect new branch entries

        newbranches = {}

        getbranchinfo = repo.revbranchcache().branchinfo

        max_rev = -1

        for r in revgen:

            branch, closesbranch = getbranchinfo(r)

            newbranches.setdefault(branch, []).append(r)

            if closesbranch:

                self._closednodes.add(cl.node(r))

            max_rev = max(max_rev, r)

        if max_rev < 0:

            msg = "running branchcache.update without revision to update"

            raise error.ProgrammingError(msg)

        # Delay fetching the topological heads until they are needed.

        # A repository without non-continous branches can skip this part.

        topoheads = None

        # If a changeset is visible, its parents must be visible too, so

        # use the faster unfiltered parent accessor.

        parentrevs = repo.unfiltered().changelog.parentrevs

        # Faster than using ctx.obsolete()

        obsrevs = obsolete.getrevs(repo, b'obsolete')

        for branch, newheadrevs in newbranches.items():

            # For every branch, compute the new branchheads.

            # A branchhead is a revision such that no descendant is on

            # the same branch.

            #

            # The branchheads are computed iteratively in revision order.

            # This ensures topological order, i.e. parents are processed

            # before their children. Ancestors are inclusive here, i.e.

            # any revision is an ancestor of itself.

            #

            # Core observations:

            # - The current revision is always a branchhead for the

            #   repository up to that point.

            # - It is the first revision of the branch if and only if

            #   there was no branchhead before. In that case, it is the

            #   only branchhead as there are no possible ancestors on

            #   the same branch.

            # - If a parent is on the same branch, a branchhead can

            #   only be an ancestor of that parent, if it is parent

            #   itself. Otherwise it would have been removed as ancestor

            #   of that parent before.

            # - Therefore, if all parents are on the same branch, they

            #   can just be removed from the branchhead set.

            # - If one parent is on the same branch and the other is not

            #   and there was exactly one branchhead known, the existing

            #   branchhead can only be an ancestor if it is the parent.

            #   Otherwise it would have been removed as ancestor of

            #   the parent before. The other parent therefore can't have

            #   a branchhead as ancestor.

            # - In all other cases, the parents on different branches

            #   could have a branchhead as ancestor. Those parents are

            #   kept in the "uncertain" set. If all branchheads are also

            #   topological heads, they can't have descendants and further

            #   checks can be skipped. Otherwise, the ancestors of the

            #   "uncertain" set are removed from branchheads.

            #   This computation is heavy and avoided if at all possible.

            bheads = self._entries.get(branch, [])

            bheadset = {cl.rev(node) for node in bheads}

            uncertain = set()

            for newrev in sorted(newheadrevs):

                if newrev in obsrevs:

                    # We ignore obsolete changesets as they shouldn't be

                    # considered heads.

                    continue

                if not bheadset:

                    bheadset.add(newrev)

                    continue

                parents = [p for p in parentrevs(newrev) if p != nullrev]

                samebranch = set()

                otherbranch = set()

                obsparents = set()

                for p in parents:

                    if p in obsrevs:

                        # We ignored this obsolete changeset earlier, but now

                        # that it has non-ignored children, we need to make

                        # sure their ancestors are not considered heads. To

                        # achieve that, we will simply treat this obsolete

                        # changeset as a parent from other branch.

                        obsparents.add(p)

                    elif p in bheadset or getbranchinfo(p)[0] == branch:

                        samebranch.add(p)

                    else:

                        otherbranch.add(p)

                if not (len(bheadset) == len(samebranch) == 1):

                    uncertain.update(otherbranch)

                    uncertain.update(obsparents)

                bheadset.difference_update(samebranch)

                bheadset.add(newrev)

            if uncertain:

                if topoheads is None:

                    topoheads = set(cl.headrevs())

                if bheadset - topoheads:

                    floorrev = min(bheadset)

                    if floorrev <= max(uncertain):

                        ancestors = set(cl.ancestors(uncertain, floorrev))

                        bheadset -= ancestors

            if bheadset:

                self[branch] = [cl.node(rev) for rev in sorted(bheadset)]

        duration = util.timer() - starttime

        repo.ui.log(

            b'branchcache',

            b'updated %s in %.4f seconds\n',

            _branchcachedesc(repo),

            duration,

        )

        return max_rev

class branchcache(_BaseBranchCache):

    """Branchmap info for a local repo or repoview"""

    _base_filename = b"branch2"

    def __init__(

        self,

        repo: "localrepo.localrepository",

        entries: Union[

            Dict[bytes, List[bytes]], Iterable[Tuple[bytes, List[bytes]]]

        ] = (),

        tipnode: Optional[bytes] = None,

        tiprev: Optional[int] = nullrev,

        filteredhash: Optional[bytes] = None,

        closednodes: Optional[Set[bytes]] = None,

        hasnode: Optional[Callable[[bytes], bool]] = None,

        verify_node: bool = False,

    ) -> None:

        """hasnode is a function which can be used to verify whether changelog

        has a given node or not. If it's not provided, we assume that every node

        we have exists in changelog"""

        self._filtername = repo.filtername

        if tipnode is None:

            self.tipnode = repo.nullid

        else:

            self.tipnode = tipnode

        self.tiprev = tiprev

        self.filteredhash = filteredhash

        self._dirty = False

        super().__init__(repo=repo, entries=entries, closed_nodes=closednodes)

        # closednodes is a set of nodes that close their branch. If the branch

        # cache has been updated, it may contain nodes that are no longer

        # heads.

        # Do we need to verify branch at all ?

        self._verify_node = verify_node

        # branches for which nodes are verified

        self._verifiedbranches = set()

        self._hasnode = None

        if self._verify_node:

            self._hasnode = repo.changelog.hasnode

    def validfor(self, repo):

        """check that cache contents are valid for (a subset of) this repo

        - False when the order of changesets changed or if we detect a strip.

        - True when cache is up-to-date for the current repo or its subset."""

        try:

            node = repo.changelog.node(self.tiprev)

        except IndexError:

            # changesets were stripped and now we don't even have enough to

            # find tiprev

            return False

        if self.tipnode != node:

            # tiprev doesn't correspond to tipnode: repo was stripped, or this

            # repo has a different order of changesets

            return False

        tiphash = scmutil.filteredhash(repo, self.tiprev, needobsolete=True)

        # hashes don't match if this repo view has a different set of filtered

        # revisions (e.g. due to phase changes) or obsolete revisions (e.g.

        # history was rewritten)

        return self.filteredhash == tiphash

    @classmethod

    def fromfile(cls, repo):

        f = None

        try: