# cmdutil.py - help for command processing in mercurial

#

# 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 copy as copymod

import errno

import functools

import os

import re

from typing import (

    Any,

    AnyStr,

    Dict,

    Iterable,

    Optional,

    cast,

)

from .i18n import _

from .node import (

    hex,

    nullrev,

    short,

)

from .pycompat import (

    open,

)

from .thirdparty import attr

from . import (

    bookmarks,

    changelog,

    copies,

    crecord as crecordmod,

    encoding,

    error,

    formatter,

    logcmdutil,

    match as matchmod,

    merge as mergemod,

    mergestate as mergestatemod,

    mergeutil,

    obsolete,

    patch,

    pathutil,

    phases,

    pycompat,

    repair,

    revlog,

    rewriteutil,

    scmutil,

    state as statemod,

    subrepoutil,

    templatekw,

    templater,

    util,

    vfs as vfsmod,

)

from .utils import (

    dateutil,

    stringutil,

)

from .revlogutils import (

    constants as revlog_constants,

)

if pycompat.TYPE_CHECKING:

    from . import (

        ui as uimod,

    )

stringio = util.stringio

# templates of common command options

dryrunopts = [

    (b'n', b'dry-run', None, _(b'do not perform actions, just print output')),

]

confirmopts = [

    (b'', b'confirm', None, _(b'ask before applying actions')),

]

remoteopts = [

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

    (

        b'',

        b'remotecmd',

        b'',

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

        _(b'CMD'),

    ),

    (

        b'',

        b'insecure',

        None,

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

    ),

]

walkopts = [

    (

        b'I',

        b'include',

        [],

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

        _(b'PATTERN'),

    ),

    (

        b'X',

        b'exclude',

        [],

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

        _(b'PATTERN'),

    ),

]

commitopts = [

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

    (b'l', b'logfile', b'', _(b'read commit message from file'), _(b'FILE')),

]

commitopts2 = [

    (

        b'd',

        b'date',

        b'',

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

        _(b'DATE'),

    ),

    (

        b'u',

        b'user',

        b'',

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

        _(b'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 = [

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

]

templateopts = [

    (

        b'',

        b'style',

        b'',

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

        _(b'STYLE'),

    ),

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

]

logopts = [

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

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

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

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

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

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

] + templateopts

diffopts = [

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

    (

        b'g',

        b'git',

        None,

        _(b'use git extended diff format (DEFAULT: diff.git)'),

    ),

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

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

]

diffwsopts = [

    (

        b'w',

        b'ignore-all-space',

        None,

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

    ),

    (

        b'b',

        b'ignore-space-change',

        None,

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

    ),

    (

        b'B',

        b'ignore-blank-lines',

        None,

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

    ),

    (

        b'Z',

        b'ignore-space-at-eol',

        None,

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

    ),

]

diffopts2 = (

    [

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

        (

            b'p',

            b'show-function',

            None,

            _(

                b'show which function each change is in (DEFAULT: diff.showfunc)'

            ),

        ),

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

    ]

    + diffwsopts

    + [

        (

            b'U',

            b'unified',

            b'',

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

            _(b'NUM'),

        ),

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

        (

            b'',

            b'root',

            b'',

            _(b'produce diffs relative to subdirectory'),

            _(b'DIR'),

        ),

    ]

)

mergetoolopts = [

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

]

similarityopts = [

    (

        b's',

        b'similarity',

        b'',

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

        _(b'SIMILARITY'),

    )

]

subrepoopts = [(b'S', b'subrepos', None, _(b'recurse into subrepositories'))]

debugrevlogopts = [

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

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

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

]

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

# editor text

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

def check_at_most_one_arg(

    opts: Dict[AnyStr, Any],

    *args: AnyStr,

) -> Optional[AnyStr]:

    """abort if more than one of the arguments are in opts

    Returns the unique argument or None if none of them were specified.

    """

    def to_display(name: AnyStr) -> bytes:

        return pycompat.sysbytes(name).replace(b'_', b'-')

    previous = None

    for x in args:

        if opts.get(x):

            if previous:

                raise error.InputError(

                    _(b'cannot specify both --%s and --%s')

                    % (to_display(previous), to_display(x))

                )

            previous = x

    return previous

def check_incompatible_arguments(

    opts: Dict[AnyStr, Any],

    first: AnyStr,

    others: Iterable[AnyStr],

) -> None:

    """abort if the first argument is given along with any of the others

    Unlike check_at_most_one_arg(), `others` are not mutually exclusive

    among themselves, and they're passed as a single collection.

    """

    for other in others:

        check_at_most_one_arg(opts, first, other)

def resolve_commit_options(ui: "uimod.ui", opts: Dict[str, Any]) -> bool:

    """modify commit options dict to handle related options

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

    the ``date`` option is set.

    """

    check_at_most_one_arg(opts, 'date', 'currentdate')

    check_at_most_one_arg(opts, 'user', 'currentuser')

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

    if opts.get('currentdate'):

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

    elif (

        not opts.get('date')

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

        and opts.get('currentdate') is None

    ):

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

        datemaydiffer = True

    if opts.get('currentuser'):

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

    return datemaydiffer

def check_note_size(opts: Dict[str, Any]) -> None:

    """make sure note is of valid format"""

    note = opts.get('note')

    if not note:

        return

    if len(note) > 255:

        raise error.InputError(_(b"cannot store a note of more than 255 bytes"))

    if b'\n' in note:

        raise error.InputError(_(b"note cannot contain a newline"))

def ishunk(x):

    hunkclasses = (crecordmod.uihunk, patch.recordhunk)

    return isinstance(x, hunkclasses)

def isheader(x):

    headerclasses = (crecordmod.uiheader, patch.header)

    return isinstance(x, headerclasses)

def newandmodified(chunks):

    newlyaddedandmodifiedfiles = set()

    alsorestore = set()

    for chunk in chunks:

        if isheader(chunk) and chunk.isnewfile():

            newlyaddedandmodifiedfiles.add(chunk.filename())

            alsorestore.update(set(chunk.files()) - {chunk.filename()})

    return newlyaddedandmodifiedfiles, alsorestore

def parsealiases(cmd):

    base_aliases = cmd.split(b"|")

    all_aliases = set(base_aliases)

    extra_aliases = []

    for alias in base_aliases:

        if b'-' in alias:

            folded_alias = alias.replace(b'-', b'')

            if folded_alias not in all_aliases:

                all_aliases.add(folded_alias)

                extra_aliases.append(folded_alias)

    base_aliases.extend(extra_aliases)

    return base_aliases

def setupwrapcolorwrite(ui):

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

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

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

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