# hg.py - repository classes for mercurial

#

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

# Copyright 2006 Vadim Gelfer <vadim.gelfer@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.

import os

import posixpath

import shutil

import stat

import weakref

from .i18n import _

from .node import (

    hex,

    sha1nodeconstants,

    short,

)

from .pycompat import getattr

from . import (

    bookmarks,

    bundlerepo,

    cmdutil,

    destutil,

    discovery,

    error,

    exchange,

    extensions,

    graphmod,

    httppeer,

    localrepo,

    lock,

    logcmdutil,

    logexchange,

    merge as mergemod,

    mergestate as mergestatemod,

    narrowspec,

    phases,

    requirements,

    scmutil,

    sshpeer,

    statichttprepo,

    ui as uimod,

    unionrepo,

    url,

    util,

    verify as verifymod,

    vfs as vfsmod,

)

from .interfaces import repository as repositorymod

from .utils import (

    hashutil,

    stringutil,

    urlutil,

)

release = lock.release

# shared features

sharedbookmarks = b'bookmarks'

def addbranchrevs(lrepo, other, branches, revs):

    peer = other.peer()  # a courtesy to callers using a localrepo for other

    hashbranch, branches = branches

    if not hashbranch and not branches:

        x = revs or None

        if revs:

            y = revs[0]

        else:

            y = None

        return x, y

    if revs:

        revs = list(revs)

    else:

        revs = []

    if not peer.capable(b'branchmap'):

        if branches:

            raise error.Abort(_(b"remote branch lookup not supported"))

        revs.append(hashbranch)

        return revs, revs[0]

    with peer.commandexecutor() as e:

        branchmap = e.callcommand(b'branchmap', {}).result()

    def primary(branch):

        if branch == b'.':

            if not lrepo:

                raise error.Abort(_(b"dirstate branch not accessible"))

            branch = lrepo.dirstate.branch()

        if branch in branchmap:

            revs.extend(hex(r) for r in reversed(branchmap[branch]))

            return True

        else:

            return False

    for branch in branches:

        if not primary(branch):

            raise error.RepoLookupError(_(b"unknown branch '%s'") % branch)

    if hashbranch:

        if not primary(hashbranch):

            revs.append(hashbranch)

    return revs, revs[0]

def _isfile(path):

    try:

        # we use os.stat() directly here instead of os.path.isfile()

        # because the latter started returning `False` on invalid path

        # exceptions starting in 3.8 and we care about handling

        # invalid paths specially here.

        st = os.stat(path)

    except ValueError as e:

        msg = stringutil.forcebytestr(e)

        raise error.Abort(_(b'invalid path %s: %s') % (path, msg))

    except OSError:

        return False

    else:

        return stat.S_ISREG(st.st_mode)

class LocalFactory:

    """thin wrapper to dispatch between localrepo and bundle repo"""

    @staticmethod

    def islocal(path: bytes) -> bool:

        path = util.expandpath(urlutil.urllocalpath(path))

        return not _isfile(path)

    @staticmethod

    def instance(ui, path, *args, **kwargs):

        path = util.expandpath(urlutil.urllocalpath(path))

        if _isfile(path):

            cls = bundlerepo

        else:

            cls = localrepo

        return cls.instance(ui, path, *args, **kwargs)

repo_schemes = {

    b'bundle': bundlerepo,

    b'union': unionrepo,

    b'file': LocalFactory,

}

peer_schemes = {

    b'http': httppeer,

    b'https': httppeer,

    b'ssh': sshpeer,

    b'static-http': statichttprepo,

}

def _peerlookup(path):

    u = urlutil.url(path)

    scheme = u.scheme or b'file'

    if scheme in peer_schemes:

        return peer_schemes[scheme]

    if scheme in repo_schemes:

        return repo_schemes[scheme]

    return LocalFactory

def islocal(repo):

    '''return true if repo (or path pointing to repo) is local'''

    if isinstance(repo, bytes):

        cls = _peerlookup(repo)

        cls.instance  # make sure we load the module

        if util.safehasattr(cls, 'islocal'):

            return cls.islocal(repo)  # pytype: disable=module-attr

        return False

    repo.ui.deprecwarn(b"use obj.local() instead of islocal(obj)", b"6.4")

    return repo.local()

def openpath(ui, path, sendaccept=True):

    '''open path with open if local, url.open if remote'''

    pathurl = urlutil.url(path, parsequery=False, parsefragment=False)

    if pathurl.islocal():

        return util.posixfile(pathurl.localpath(), b'rb')

    else:

        return url.open(ui, path, sendaccept=sendaccept)

# a list of (ui, repo) functions called for wire peer initialization

wirepeersetupfuncs = []

def _setup_repo_or_peer(ui, obj, presetupfuncs=None):

    ui = getattr(obj, "ui", ui)

    for f in presetupfuncs or []:

        f(ui, obj)

    ui.log(b'extension', b'- executing reposetup hooks\n')

    with util.timedcm('all reposetup') as allreposetupstats:

        for name, module in extensions.extensions(ui):

            ui.log(b'extension', b'  - running reposetup for %s\n', name)

            hook = getattr(module, 'reposetup', None)

            if hook:

                with util.timedcm('reposetup %r', name) as stats:

                    hook(ui, obj)

                msg = b'  > reposetup for %s took %s\n'

                ui.log(b'extension', msg, name, stats)

    ui.log(b'extension', b'> all reposetup took %s\n', allreposetupstats)

    if not obj.local():

        for f in wirepeersetupfuncs:

            f(ui, obj)

def repository(

    ui,

    path=b'',

    create=False,

    presetupfuncs=None,

    intents=None,

    createopts=None,

):

    """return a repository object for the specified path"""

    scheme = urlutil.url(path).scheme

    if scheme is None:

        scheme = b'file'

    cls = repo_schemes.get(scheme)

    if cls is None:

        if scheme in peer_schemes:

            raise error.Abort(_(b"repository '%s' is not local") % path)

        cls = LocalFactory

    repo = cls.instance(

        ui,

        path,

        create,

        intents=intents,

        createopts=createopts,

    )

    _setup_repo_or_peer(ui, repo, presetupfuncs=presetupfuncs)

    return repo.filtered(b'visible')

def peer(uiorrepo, opts, path, create=False, intents=None, createopts=None):

    '''return a repository peer for the specified path'''

    rui = remoteui(uiorrepo, opts)

    if util.safehasattr(path, 'url'):

        # this is a urlutil.path object

        scheme = path.url.scheme  # pytype: disable=attribute-error

        # XXX for now we don't do anything more than that

        path = path.loc  # pytype: disable=attribute-error

    else:

        scheme = urlutil.url(path).scheme

    if scheme in peer_schemes:

        cls = peer_schemes[scheme]

        peer = cls.instance(

            rui,

            path,

            create,

            intents=intents,

            createopts=createopts,

        )

        _setup_repo_or_peer(rui, peer)

    else:

        # this is a repository

        repo = repository(

            rui,

            path,

            create,

            intents=intents,

            createopts=createopts,

        )

        peer = repo.peer()

    return peer

def defaultdest(source):

    """return default destination of clone if none is given

    >>> defaultdest(b'foo')

    'foo'

    >>> defaultdest(b'/foo/bar')

    'bar'

    >>> defaultdest(b'/')

    ''

    >>> defaultdest(b'')

    ''

    >>> defaultdest(b'http://example.org/')

    ''

    >>> defaultdest(b'http://example.org/foo/')

    'foo'

    """

    path = urlutil.url(source).path

    if not path:

        return b''

    return os.path.basename(os.path.normpath(path))

def sharedreposource(repo):

    """Returns repository object for source repository of a shared repo.

    If repo is not a shared repository, returns None.

    """

    if repo.sharedpath == repo.path:

        return None

    if util.safehasattr(repo, b'srcrepo') and repo.srcrepo:

        return repo.srcrepo

    # the sharedpath always ends in the .hg; we want the path to the repo

    source = repo.vfs.split(repo.sharedpath)[0]

    srcurl, branches = urlutil.parseurl(source)

    srcrepo = repository(repo.ui, srcurl)

    repo.srcrepo = srcrepo

    return srcrepo

def share(

    ui,

    source,

    dest=None,

    update=True,

    bookmarks=True,

    defaultpath=None,

    relative=False,

):

    '''create a shared repository'''

    not_local_msg = _(b'can only share local repositories')

    if util.safehasattr(source, 'local'):

        if source.local() is None:

            raise error.Abort(not_local_msg)

    elif not islocal(source):

        # XXX why are we getting bytes here ?

        raise error.Abort(not_local_msg)

    if not dest:

        dest = defaultdest(source)

    else:

        dest = urlutil.get_clone_path(ui, dest)[1]

    if isinstance(source, bytes):

        origsource, source, branches = urlutil.get_clone_path(ui, source)

        srcrepo = repository(ui, source)

        rev, checkout = addbranchrevs(srcrepo, srcrepo, branches, None)

    else:

        srcrepo = source.local()

        checkout = None

    shareditems = set()

    if bookmarks:

        shareditems.add(sharedbookmarks)

    r = repository(

        ui,

        dest,

        create=True,

        createopts={

            b'sharedrepo': srcrepo,

            b'sharedrelative': relative,

            b'shareditems': shareditems,

        },

    )

    postshare(srcrepo, r, defaultpath=defaultpath)

    r = repository(ui, dest)

    _postshareupdate(r, update, checkout=checkout)

    return r

def _prependsourcehgrc(repo):

    """copies the source repo config and prepend it in current repo .hg/hgrc

    on unshare. This is only done if the share was perfomed using share safe

    method where we share config of source in shares"""

    srcvfs = vfsmod.vfs(repo.sharedpath)

    dstvfs = vfsmod.vfs(repo.path)

    if not srcvfs.exists(b'hgrc'):

        return

    currentconfig = b''

    if dstvfs.exists(b'hgrc'):

        currentconfig = dstvfs.read(b'hgrc')

    with dstvfs(b'hgrc', b'wb') as fp:

        sourceconfig = srcvfs.read(b'hgrc')