comparison hgext/largefiles/lfutil.py @ 15224:7c604d8c7e83

largefiles: remove pre-1.9 code from extension first bundled with 1.9

15223:fc035e5370ca

'''largefiles utility code: must not import other modules in this package.'''

import os
import errno
import inspect
import shutil
import stat
import hashlib

from mercurial import cmdutil, dirstate, httpconnection, match as match_, \
        url as url_, util
from mercurial.i18n import _

try:
    from mercurial import scmutil
except ImportError:

longname = 'largefiles'


# -- Portability wrappers ----------------------------------------------

if 'subrepos' in inspect.getargspec(dirstate.dirstate.status)[0]:
    # for Mercurial >= 1.5
    def dirstate_walk(dirstate, matcher, unknown=False, ignored=False):
        return dirstate.walk(matcher, [], unknown, ignored)
else:
    # for Mercurial <= 1.4
    def dirstate_walk(dirstate, matcher, unknown=False, ignored=False):
        return dirstate.walk(matcher, unknown, ignored)

def repo_add(repo, list):
    try:
        # Mercurial <= 1.5
        add = repo.add
    except AttributeError:
        # Mercurial >= 1.6
        add = repo[None].add
    return add(list)

def repo_remove(repo, list, unlink=False):
    try:
        # Mercurial <= 1.5
        remove = repo.remove
    except AttributeError:
        # Mercurial >= 1.6
        try:
            # Mercurial <= 1.8
            remove = repo[None].remove
        except AttributeError:
            # Mercurial >= 1.9
            def remove(list, unlink):
                wlock = repo.wlock()
                try:
                    if unlink:
                        for f in list:
                            try:
                                util.unlinkpath(repo.wjoin(f))
                            except OSError, inst:
                                if inst.errno != errno.ENOENT:
                                    raise
                    repo[None].forget(list)
                finally:
                    wlock.release()

    return remove(list, unlink=unlink)

def repo_forget(repo, list):
    try:
        # Mercurial <= 1.5
        forget = repo.forget
    except AttributeError:
        # Mercurial >= 1.6
        forget = repo[None].forget
    return forget(list)

def findoutgoing(repo, remote, force):
    # First attempt is for Mercurial <= 1.5 second is for >= 1.6
    try:
        return repo.findoutgoing(remote)
    except AttributeError:
        from mercurial import discovery
        try:
            # Mercurial <= 1.8
            return discovery.findoutgoing(repo, remote, force=force)
        except AttributeError:
            # Mercurial >= 1.9
            common, _anyinc, _heads = discovery.findcommonincoming(repo,
                remote, force=force)
            return repo.changelog.findmissing(common)

# -- Private worker functions ------------------------------------------

def link(src, dest):
    try:

    '''
    Return a dirstate object that tracks big files: i.e. its root is the
    repo root, but it is saved in .hg/largefiles/dirstate.
    '''
    admin = repo.join(longname)
    try:
        # Mercurial >= 1.9
        opener = scmutil.opener(admin)
    except ImportError:
        # Mercurial <= 1.8
        opener = util.opener(admin)
    if util.safehasattr(repo.dirstate, '_validate'):
        lfdirstate = largefiles_dirstate(opener, ui, repo.root,
            repo.dirstate._validate)
    else:
        lfdirstate = largefiles_dirstate(opener, ui, repo.root)

def getmatcher(repo, pats=[], opts={}, showbad=True):
    '''Wrapper around scmutil.match() that adds showbad: if false, neuter
    the match object\'s bad() method so it does not print any warnings
    about missing files or directories.'''
    try:
        # Mercurial >= 1.9
        match = scmutil.match(repo[None], pats, opts)
    except ImportError:
        # Mercurial <= 1.8
        match = cmdutil.match(repo, pats, opts)

    if not showbad:
        match.bad = lambda f, msg: None
    return match

    for chunk in util.filechunkiter(data):
        h.update(chunk)
    return h.hexdigest()

def httpsendfile(ui, filename):
    try:
        # Mercurial >= 1.9
        return httpconnection.httpsendfile(ui, filename, 'rb')
    except ImportError:
        if 'ui' in inspect.getargspec(url_.httpsendfile.__init__)[0]:
            # Mercurial == 1.8
            return url_.httpsendfile(ui, filename, 'rb')
        else:
            # Mercurial <= 1.7
            return url_.httpsendfile(filename, 'rb')

# Convert a path to a unix style path. This is used to give a
# canonical path to the lfdirstate.
def unixpath(path):
    return os.path.normpath(path).replace(os.sep, '/')

15224:7c604d8c7e83

'''largefiles utility code: must not import other modules in this package.'''

import os
import errno
import shutil
import stat
import hashlib

from mercurial import dirstate, httpconnection, match as match_, util
from mercurial.i18n import _

try:
    from mercurial import scmutil
except ImportError:

longname = 'largefiles'


# -- Portability wrappers ----------------------------------------------

def dirstate_walk(dirstate, matcher, unknown=False, ignored=False):
    return dirstate.walk(matcher, [], unknown, ignored)

def repo_add(repo, list):
    add = repo[None].add
    return add(list)

def repo_remove(repo, list, unlink=False):
    def remove(list, unlink):
        wlock = repo.wlock()
        try:
            if unlink:
                for f in list:
                    try:
                        util.unlinkpath(repo.wjoin(f))
                    except OSError, inst:
                        if inst.errno != errno.ENOENT:
                            raise
            repo[None].forget(list)
        finally:
            wlock.release()
    return remove(list, unlink=unlink)

def repo_forget(repo, list):
    forget = repo[None].forget
    return forget(list)

def findoutgoing(repo, remote, force):
    from mercurial import discovery
    common, _anyinc, _heads = discovery.findcommonincoming(repo,
        remote, force=force)
    return repo.changelog.findmissing(common)

# -- Private worker functions ------------------------------------------

def link(src, dest):
    try:

    '''
    Return a dirstate object that tracks big files: i.e. its root is the
    repo root, but it is saved in .hg/largefiles/dirstate.
    '''
    admin = repo.join(longname)
    opener = scmutil.opener(admin)
    if util.safehasattr(repo.dirstate, '_validate'):
        lfdirstate = largefiles_dirstate(opener, ui, repo.root,
            repo.dirstate._validate)
    else:
        lfdirstate = largefiles_dirstate(opener, ui, repo.root)

def getmatcher(repo, pats=[], opts={}, showbad=True):
    '''Wrapper around scmutil.match() that adds showbad: if false, neuter
    the match object\'s bad() method so it does not print any warnings
    about missing files or directories.'''
    match = scmutil.match(repo[None], pats, opts)

    if not showbad:
        match.bad = lambda f, msg: None
    return match

    for chunk in util.filechunkiter(data):
        h.update(chunk)
    return h.hexdigest()

def httpsendfile(ui, filename):
    return httpconnection.httpsendfile(ui, filename, 'rb')

# Convert a path to a unix style path. This is used to give a
# canonical path to the lfdirstate.
def unixpath(path):
    return os.path.normpath(path).replace(os.sep, '/')