# cmdutil.py - help for command processing in mercurial

#

# Copyright 2005-2007 Matt Mackall <mpm@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.

from __future__ import absolute_import

import copy as copymod

import errno

import os

import re

from .i18n import _

from .node import (

    hex,

    nullid,

    nullrev,

    short,

)

from . import (

    bookmarks,

    changelog,

    copies,

    crecord as crecordmod,

    dirstateguard,

    encoding,

    error,

    formatter,

    logcmdutil,

    match as matchmod,

    merge as mergemod,

    mergeutil,

    obsolete,

    patch,

    pathutil,

    phases,

    pycompat,

    repair,

    revlog,

    rewriteutil,

    scmutil,

    smartset,

    state as statemod,

    subrepoutil,

    templatekw,

    templater,

    util,

    vfs as vfsmod,

)

from .utils import (

    dateutil,

    stringutil,

)

stringio = util.stringio

# templates of common command options

dryrunopts = [

    ('n', 'dry-run', None,

     _('do not perform actions, just print output')),

]

confirmopts = [

    ('', 'confirm', None,

     _('ask before applying actions')),

]

remoteopts = [

    ('e', 'ssh', '',

     _('specify ssh command to use'), _('CMD')),

    ('', 'remotecmd', '',

     _('specify hg command to run on the remote side'), _('CMD')),

    ('', 'insecure', None,

     _('do not verify server certificate (ignoring web.cacerts config)')),

]

walkopts = [

    ('I', 'include', [],

     _('include names matching the given patterns'), _('PATTERN')),

    ('X', 'exclude', [],

     _('exclude names matching the given patterns'), _('PATTERN')),

]

commitopts = [

    ('m', 'message', '',

     _('use text as commit message'), _('TEXT')),

    ('l', 'logfile', '',

     _('read commit message from file'), _('FILE')),

]

commitopts2 = [

    ('d', 'date', '',

     _('record the specified date as commit date'), _('DATE')),

    ('u', 'user', '',

     _('record the specified user as committer'), _('USER')),

]

commitopts3 = [

    (b'D', b'currentdate', None,

     _(b'record the current date as commit date')),

    (b'U', b'currentuser', None,

     _(b'record the current user as committer')),

]

formatteropts = [

    ('T', 'template', '',

     _('display with template'), _('TEMPLATE')),

]

templateopts = [

    ('', 'style', '',

     _('display using template map file (DEPRECATED)'), _('STYLE')),

    ('T', 'template', '',

     _('display with template'), _('TEMPLATE')),

]

logopts = [

    ('p', 'patch', None, _('show patch')),

    ('g', 'git', None, _('use git extended diff format')),

    ('l', 'limit', '',

     _('limit number of changes displayed'), _('NUM')),

    ('M', 'no-merges', None, _('do not show merges')),

    ('', 'stat', None, _('output diffstat-style summary of changes')),

    ('G', 'graph', None, _("show the revision DAG")),

] + templateopts

diffopts = [

    ('a', 'text', None, _('treat all files as text')),

    ('g', 'git', None, _('use git extended diff format')),

    ('', 'binary', None, _('generate binary diffs in git mode (default)')),

    ('', 'nodates', None, _('omit dates from diff headers'))

]

diffwsopts = [

    ('w', 'ignore-all-space', None,

     _('ignore white space when comparing lines')),

    ('b', 'ignore-space-change', None,

     _('ignore changes in the amount of white space')),

    ('B', 'ignore-blank-lines', None,

     _('ignore changes whose lines are all blank')),

    ('Z', 'ignore-space-at-eol', None,

     _('ignore changes in whitespace at EOL')),

]

diffopts2 = [

    ('', 'noprefix', None, _('omit a/ and b/ prefixes from filenames')),

    ('p', 'show-function', None, _('show which function each change is in')),

    ('', 'reverse', None, _('produce a diff that undoes the changes')),

] + diffwsopts + [

    ('U', 'unified', '',

     _('number of lines of context to show'), _('NUM')),

    ('', 'stat', None, _('output diffstat-style summary of changes')),

    ('', 'root', '', _('produce diffs relative to subdirectory'), _('DIR')),

]

mergetoolopts = [

    ('t', 'tool', '', _('specify merge tool'), _('TOOL')),

]

similarityopts = [

    ('s', 'similarity', '',

     _('guess renamed files by similarity (0<=s<=100)'), _('SIMILARITY'))

]

subrepoopts = [

    ('S', 'subrepos', None,

     _('recurse into subrepositories'))

]

debugrevlogopts = [

    ('c', 'changelog', False, _('open changelog')),

    ('m', 'manifest', False, _('open manifest')),

    ('', 'dir', '', _('open directory manifest')),

]

# special string such that everything below this line will be ingored in the

# editor text

_linebelow = "^HG: ------------------------ >8 ------------------------$"

def resolvecommitoptions(ui, opts):

    """modify commit options dict to handle related options

    The return value indicates that ``rewrite.update-timestamp`` is the reason

    the ``date`` option is set.

    """

    if opts.get('date') and opts.get('currentdate'):

        raise error.Abort(_('--date and --currentdate are mutually '

                            'exclusive'))

    if opts.get(b'user') and opts.get(b'currentuser'):

        raise error.Abort(_('--user and --currentuser are mutually '

                            'exclusive'))

    datemaydiffer = False  # date-only change should be ignored?

    if opts.get(b'currentdate'):

        opts[b'date'] = b'%d %d' % dateutil.makedate()

    elif (not opts.get('date')

          and ui.configbool('rewrite', 'update-timestamp')

          and opts.get('currentdate') is None):

        opts[b'date'] = b'%d %d' % dateutil.makedate()

        datemaydiffer = True

    if opts.get(b'currentuser'):

        opts[b'user'] = ui.username()

    return datemaydiffer

def ishunk(x):

    hunkclasses = (crecordmod.uihunk, patch.recordhunk)

    return isinstance(x, hunkclasses)

def newandmodified(chunks, originalchunks):

    newlyaddedandmodifiedfiles = set()

    alsorestore = set()

    for chunk in chunks:

        if (ishunk(chunk) and chunk.header.isnewfile() and chunk not in

            originalchunks):

            newlyaddedandmodifiedfiles.add(chunk.header.filename())

            alsorestore.update(set(chunk.header.files()) -

                               {chunk.header.filename()})

    return newlyaddedandmodifiedfiles, alsorestore

def parsealiases(cmd):

    return cmd.split("|")

def setupwrapcolorwrite(ui):

    # wrap ui.write so diff output can be labeled/colorized

    def wrapwrite(orig, *args, **kw):

        label = kw.pop(r'label', '')

        for chunk, l in patch.difflabel(lambda: args):

            orig(chunk, label=label + l)

    oldwrite = ui.write

    def wrap(*args, **kwargs):

        return wrapwrite(oldwrite, *args, **kwargs)

    setattr(ui, 'write', wrap)

    return oldwrite

def filterchunks(ui, originalhunks, usecurses, testfile, match,

                 operation=None):

    try:

        if usecurses:

            if testfile:

                recordfn = crecordmod.testdecorator(

                    testfile, crecordmod.testchunkselector)

            else:

                recordfn = crecordmod.chunkselector

            return crecordmod.filterpatch(ui, originalhunks, recordfn,

                                          operation)

    except crecordmod.fallbackerror as e:

        ui.warn('%s\n' % e.message)

        ui.warn(_('falling back to text mode\n'))

    return patch.filterpatch(ui, originalhunks, match, operation)

def recordfilter(ui, originalhunks, match, operation=None):

    """ Prompts the user to filter the originalhunks and return a list of

    selected hunks.

    *operation* is used for to build ui messages to indicate the user what

    kind of filtering they are doing: reverting, committing, shelving, etc.

    (see patch.filterpatch).

    """

    usecurses = crecordmod.checkcurses(ui)

    testfile = ui.config('experimental', 'crecordtest')

    oldwrite = setupwrapcolorwrite(ui)

    try:

        newchunks, newopts = filterchunks(ui, originalhunks, usecurses,

                                          testfile, match, operation)

    finally:

        ui.write = oldwrite

    return newchunks, newopts

def dorecord(ui, repo, commitfunc, cmdsuggest, backupall,

            filterfn, *pats, **opts):

    opts = pycompat.byteskwargs(opts)

    if not ui.interactive():

        if cmdsuggest:

            msg = _('running non-interactively, use %s instead') % cmdsuggest

        else:

            msg = _('running non-interactively')

        raise error.Abort(msg)

    # make sure username is set before going interactive

    if not opts.get('user'):

        ui.username() # raise exception, username not provided

    def recordfunc(ui, repo, message, match, opts):

        """This is generic record driver.

        Its job is to interactively filter local changes, and

        accordingly prepare working directory into a state in which the

        job can be delegated to a non-interactive commit command such as

        'commit' or 'qrefresh'.

        After the actual job is done by non-interactive command, the

        working directory is restored to its original state.

        In the end we'll record interesting changes, and everything else

        will be left in place, so the user can continue working.

        """

        if not opts.get('interactive-unshelve'):

            checkunfinished(repo, commit=True)

        wctx = repo[None]

        merge = len(wctx.parents()) > 1

        if merge:

            raise error.Abort(_('cannot partially commit a merge '

                               '(use "hg commit" instead)'))

        def fail(f, msg):

            raise error.Abort('%s: %s' % (f, msg))

        force = opts.get('force')

        if not force:

            vdirs = []

            match = matchmod.badmatch(match, fail)

            match.explicitdir = vdirs.append

        status = repo.status(match=match)

        overrides = {(b'ui', b'commitsubrepos'): True}

        with repo.ui.configoverride(overrides, b'record'):

            # subrepoutil.precommit() modifies the status

            tmpstatus = scmutil.status(copymod.copy(status[0]),

                                       copymod.copy(status[1]),

                                       copymod.copy(status[2]),

                                       copymod.copy(status[3]),

                                       copymod.copy(status[4]),

                                       copymod.copy(status[5]),

                                       copymod.copy(status[6]))

            # Force allows -X subrepo to skip the subrepo.

            subs, commitsubs, newstate = subrepoutil.precommit(

                repo.ui, wctx, tmpstatus, match, force=True)

            for s in subs:

                if s in commitsubs:

                    dirtyreason = wctx.sub(s).dirtyreason(True)

                    raise error.Abort(dirtyreason)

        if not force:

            repo.checkcommitpatterns(wctx, vdirs, match, status, fail)

        diffopts = patch.difffeatureopts(ui, opts=opts, whitespace=True,

                                         section='commands',

                                         configprefix='commit.interactive.')

        diffopts.nodates = True

        diffopts.git = True

        diffopts.showfunc = True

        originaldiff = patch.diff(repo, changes=status, opts=diffopts)

        originalchunks = patch.parsepatch(originaldiff)

        match = scmutil.match(repo[None], pats)

        # 1. filter patch, since we are intending to apply subset of it

        try:

            chunks, newopts = filterfn(ui, originalchunks, match)

        except error.PatchError as err:

            raise error.Abort(_('error parsing patch: %s') % err)

        opts.update(newopts)