# repository.py - Interfaces and base classes for repositories and peers.

# coding: utf-8

#

# Copyright 2017 Gregory Szorc <gregory.szorc@gmail.com>

#

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

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

from __future__ import annotations

import typing

from typing import (

    Any,

    Protocol,

)

from ..i18n import _

from .. import error

from . import util as interfaceutil

if typing.TYPE_CHECKING:

    # Almost all mercurial modules are only imported in the type checking phase

    # to avoid circular imports

    from ..utils import (

        urlutil,

    )

    # TODO: create a Protocol class, since importing uimod here causes a cycle

    #  that confuses pytype.

    Ui = Any

# Local repository feature string.

# Revlogs are being used for file storage.

REPO_FEATURE_REVLOG_FILE_STORAGE = b'revlogfilestorage'

# The storage part of the repository is shared from an external source.

REPO_FEATURE_SHARED_STORAGE = b'sharedstore'

# LFS supported for backing file storage.

REPO_FEATURE_LFS = b'lfs'

# Repository supports being stream cloned.

REPO_FEATURE_STREAM_CLONE = b'streamclone'

# Repository supports (at least) some sidedata to be stored

REPO_FEATURE_SIDE_DATA = b'side-data'

# Files storage may lack data for all ancestors.

REPO_FEATURE_SHALLOW_FILE_STORAGE = b'shallowfilestorage'

REVISION_FLAG_CENSORED = 1 << 15

REVISION_FLAG_ELLIPSIS = 1 << 14

REVISION_FLAG_EXTSTORED = 1 << 13

REVISION_FLAG_HASCOPIESINFO = 1 << 12

REVISION_FLAGS_KNOWN = (

    REVISION_FLAG_CENSORED

    | REVISION_FLAG_ELLIPSIS

    | REVISION_FLAG_EXTSTORED

    | REVISION_FLAG_HASCOPIESINFO

)

CG_DELTAMODE_STD = b'default'

CG_DELTAMODE_PREV = b'previous'

CG_DELTAMODE_FULL = b'fulltext'

CG_DELTAMODE_P1 = b'p1'

## Cache related constants:

#

# Used to control which cache should be warmed in a repo.updatecaches(…) call.

# Warm branchmaps of all known repoview's filter-level

CACHE_BRANCHMAP_ALL = b"branchmap-all"

# Warm branchmaps of repoview's filter-level used by server

CACHE_BRANCHMAP_SERVED = b"branchmap-served"

# Warm internal changelog cache (eg: persistent nodemap)

CACHE_CHANGELOG_CACHE = b"changelog-cache"

# check of a branchmap can use the "pure topo" mode

CACHE_BRANCHMAP_DETECT_PURE_TOPO = b"branchmap-detect-pure-topo"

# Warm full manifest cache

CACHE_FULL_MANIFEST = b"full-manifest"

# Warm file-node-tags cache

CACHE_FILE_NODE_TAGS = b"file-node-tags"

# Warm internal manifestlog cache (eg: persistent nodemap)

CACHE_MANIFESTLOG_CACHE = b"manifestlog-cache"

# Warn rev branch cache

CACHE_REV_BRANCH = b"rev-branch-cache"

# Warm tags' cache for default repoview'

CACHE_TAGS_DEFAULT = b"tags-default"

# Warm tags' cache for  repoview's filter-level used by server

CACHE_TAGS_SERVED = b"tags-served"

# the cache to warm by default after a simple transaction

# (this is a mutable set to let extension update it)

CACHES_DEFAULT = {

    CACHE_BRANCHMAP_SERVED,

}

# the caches to warm when warming all of them

# (this is a mutable set to let extension update it)

CACHES_ALL = {

    CACHE_BRANCHMAP_SERVED,

    CACHE_BRANCHMAP_ALL,

    CACHE_BRANCHMAP_DETECT_PURE_TOPO,

    CACHE_REV_BRANCH,

    CACHE_CHANGELOG_CACHE,

    CACHE_FILE_NODE_TAGS,

    CACHE_FULL_MANIFEST,

    CACHE_MANIFESTLOG_CACHE,

    CACHE_TAGS_DEFAULT,

    CACHE_TAGS_SERVED,

}

# the cache to warm by default on simple call

# (this is a mutable set to let extension update it)

CACHES_POST_CLONE = CACHES_ALL.copy()

CACHES_POST_CLONE.discard(CACHE_FILE_NODE_TAGS)

CACHES_POST_CLONE.discard(CACHE_REV_BRANCH)

class _ipeerconnection(Protocol):

    """Represents a "connection" to a repository.

    This is the base interface for representing a connection to a repository.

    It holds basic properties and methods applicable to all peer types.

    This is not a complete interface definition and should not be used

    outside of this module.

    """

    ui: Ui

    """ui.ui instance"""

    path: urlutil.path | None

    """a urlutil.path instance or None"""

    def url(self):

        """Returns a URL string representing this peer.

        Currently, implementations expose the raw URL used to construct the

        instance. It may contain credentials as part of the URL. The

        expectations of the value aren't well-defined and this could lead to

        data leakage.

        TODO audit/clean consumers and more clearly define the contents of this

        value.

        """

    def local(self):

        """Returns a local repository instance.

        If the peer represents a local repository, returns an object that

        can be used to interface with it. Otherwise returns ``None``.

        """

    def canpush(self):

        """Returns a boolean indicating if this peer can be pushed to."""

    def close(self):

        """Close the connection to this peer.

        This is called when the peer will no longer be used. Resources

        associated with the peer should be cleaned up.

        """

class ipeercapabilities(Protocol):

    """Peer sub-interface related to capabilities."""

    def capable(self, name):

        """Determine support for a named capability.

        Returns ``False`` if capability not supported.

        Returns ``True`` if boolean capability is supported. Returns a string

        if capability support is non-boolean.

        Capability strings may or may not map to wire protocol capabilities.

        """

    def requirecap(self, name, purpose):

        """Require a capability to be present.

        Raises a ``CapabilityError`` if the capability isn't present.

        """

class ipeercommands(Protocol):

    """Client-side interface for communicating over the wire protocol.

    This interface is used as a gateway to the Mercurial wire protocol.

    methods commonly call wire protocol commands of the same name.

    """

    def branchmap(self):

        """Obtain heads in named branches.

        Returns a dict mapping branch name to an iterable of nodes that are

        heads on that branch.

        """

    def capabilities(self):

        """Obtain capabilities of the peer.

        Returns a set of string capabilities.

        """

    def get_cached_bundle_inline(self, path):

        """Retrieve a clonebundle across the wire.

        Returns a chunkbuffer

        """

    def clonebundles(self):

        """Obtains the clone bundles manifest for the repo.

        Returns the manifest as unparsed bytes.

        """

    def debugwireargs(self, one, two, three=None, four=None, five=None):

        """Used to facilitate debugging of arguments passed over the wire."""

    def getbundle(self, source, **kwargs):

        """Obtain remote repository data as a bundle.

        This command is how the bulk of repository data is transferred from

        the peer to the local repository

        Returns a generator of bundle data.

        """

    def heads(self):

        """Determine all known head revisions in the peer.

        Returns an iterable of binary nodes.

        """

    def known(self, nodes):

        """Determine whether multiple nodes are known.

        Accepts an iterable of nodes whose presence to check for.

        Returns an iterable of booleans indicating of the corresponding node

        at that index is known to the peer.

        """

    def listkeys(self, namespace):

        """Obtain all keys in a pushkey namespace.

        Returns an iterable of key names.

        """

    def lookup(self, key):

        """Resolve a value to a known revision.

        Returns a binary node of the resolved revision on success.

        """

    def pushkey(self, namespace, key, old, new):

        """Set a value using the ``pushkey`` protocol.

        Arguments correspond to the pushkey namespace and key to operate on and

        the old and new values for that key.

        Returns a string with the peer result. The value inside varies by the

        namespace.

        """

    def stream_out(self):

        """Obtain streaming clone data.

        Successful result should be a generator of data chunks.

        """

    def unbundle(self, bundle, heads, url):

        """Transfer repository data to the peer.

        This is how the bulk of data during a push is transferred.

        Returns the integer number of heads added to the peer.

        """

class ipeerlegacycommands(Protocol):

    """Interface for implementing support for legacy wire protocol commands.

    Wire protocol commands transition to legacy status when they are no longer

    used by modern clients. To facilitate identifying which commands are

    legacy, the interfaces are split.

    """

    def between(self, pairs):

        """Obtain nodes between pairs of nodes.

        ``pairs`` is an iterable of node pairs.

        Returns an iterable of iterables of nodes corresponding to each

        requested pair.

        """

    def branches(self, nodes):

        """Obtain ancestor changesets of specific nodes back to a branch point.

        For each requested node, the peer finds the first ancestor node that is

        a DAG root or is a merge.

        Returns an iterable of iterables with the resolved values for each node.

        """

    def changegroup(self, nodes, source):

        """Obtain a changegroup with data for descendants of specified nodes."""

    def changegroupsubset(self, bases, heads, source):

        pass

class ipeercommandexecutor(Protocol):

    """Represents a mechanism to execute remote commands.

    This is the primary interface for requesting that wire protocol commands

    be executed. Instances of this interface are active in a context manager

    and have a well-defined lifetime. When the context manager exits, all

    outstanding requests are waited on.

    """

    def callcommand(self, name, args):

        """Request that a named command be executed.

        Receives the command name and a dictionary of command arguments.

        Returns a ``concurrent.futures.Future`` that will resolve to the

        result of that command request. That exact value is left up to

        the implementation and possibly varies by command.

        Not all commands can coexist with other commands in an executor

        instance: it depends on the underlying wire protocol transport being

        used and the command itself.

        Implementations MAY call ``sendcommands()`` automatically if the

        requested command can not coexist with other commands in this executor.

        Implementations MAY call ``sendcommands()`` automatically when the

        future's ``result()`` is called. So, consumers using multiple

        commands with an executor MUST ensure that ``result()`` is not called

        until all command requests have been issued.

        """

    def sendcommands(self):

        """Trigger submission of queued command requests.

        Not all transports submit commands as soon as they are requested to

        run. When called, this method forces queued command requests to be

        issued. It will no-op if all commands have already been sent.

        When called, no more new commands may be issued with this executor.

        """

    def close(self):

        """Signal that this command request is finished.

        When called, no more new commands may be issued. All outstanding

        commands that have previously been issued are waited on before

        returning. This not only includes waiting for the futures to resolve,

        but also waiting for all response data to arrive. In other words,

        calling this waits for all on-wire state for issued command requests

        to finish.

        When used as a context manager, this method is called when exiting the

        context manager.

        This method may call ``sendcommands()`` if there are buffered commands.

        """

class ipeerrequests(Protocol):

    """Interface for executing commands on a peer."""

    def commandexecutor(self):

        """A context manager that resolves to an ipeercommandexecutor.

        The object this resolves to can be used to issue command requests

        to the peer.

        Callers should call its ``callcommand`` method to issue command

        requests.

        A new executor should be obtained for each distinct set of commands

        (possibly just a single command) that the consumer wants to execute

        as part of a single operation or round trip. This is because some

        peers are half-duplex and/or don't support persistent connections.

        e.g. in the case of HTTP peers, commands sent to an executor represent

        a single HTTP request. While some peers may support multiple command

        sends over the wire per executor, consumers need to code to the least

        capable peer. So it should be assumed that command executors buffer

        called commands until they are told to send them and that each

        command executor could result in a new connection or wire-level request

        being issued.

        """

class ipeerbase(_ipeerconnection, ipeercapabilities, ipeerrequests):

    """Unified interface for peer repositories.

    All peer instances must conform to this interface.

    """

@interfaceutil.implementer(ipeerbase)

class peer:

    """Base class for peer repositories."""

    limitedarguments = False

    def __init__(self, ui, path=None, remotehidden=False):

        self.ui = ui

        self.path = path

    def capable(self, name):

        # TODO: this class should maybe subclass ipeercommands too, otherwise it

        #  is assuming whatever uses this as a mixin also has this interface.

        caps = self.capabilities()  # pytype: disable=attribute-error

        if name in caps:

            return True

        name = b'%s=' % name

        for cap in caps:

            if cap.startswith(name):

                return cap[len(name) :]

        return False

    def requirecap(self, name, purpose):

        if self.capable(name):

            return

        raise error.CapabilityError(

            _(

                b'cannot %s; remote repository does not support the '

                b'\'%s\' capability'

            )

            % (purpose, name)

        )

class iverifyproblem(Protocol):

    """Represents a problem with the integrity of the repository.

    Instances of this interface are emitted to describe an integrity issue

    with a repository (e.g. corrupt storage, missing data, etc).

    Instances are essentially messages associated with severity.

    """

    warning = interfaceutil.Attribute(

        """Message indicating a non-fatal problem."""

    )

    error = interfaceutil.Attribute("""Message indicating a fatal problem.""")

    node = interfaceutil.Attribute(

        """Revision encountering the problem.

        ``None`` means the problem doesn't apply to a single revision.

        """

    )

class irevisiondelta(Protocol):

    """Represents a delta between one revision and another.

    Instances convey enough information to allow a revision to be exchanged

    with another repository.

    Instances represent the fulltext revision data or a delta against

    another revision. Therefore the ``revision`` and ``delta`` attributes

    are mutually exclusive.

    Typically used for changegroup generation.

    """

    node = interfaceutil.Attribute("""20 byte node of this revision.""")

    p1node = interfaceutil.Attribute(

        """20 byte node of 1st parent of this revision."""

    )

    p2node = interfaceutil.Attribute(