# commands.py - command processing for mercurial

#

# Copyright 2005 Matt Mackall <mpm@selenic.com>

#

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

# of the GNU General Public License, incorporated herein by reference.

from demandload import demandload

demandload(globals(), "os re sys signal shutil")

demandload(globals(), "fancyopts ui hg util")

demandload(globals(), "fnmatch hgweb mdiff random signal time traceback")

demandload(globals(), "errno socket version struct atexit")

class UnknownCommand(Exception):

    """Exception raised if command is not in the command table."""

class Abort(Exception):

    """Raised if a command needs to print an error and exit."""

def filterfiles(filters, files):

    l = [x for x in files if x in filters]

    for t in filters:

        if t and t[-1] != "/":

            t += "/"

        l += [x for x in files if x.startswith(t)]

    return l

def relfilter(repo, files):

    cwd = repo.getcwd()

    if cwd:

        return filterfiles([util.pconvert(cwd)], files)

    return files

def relpath(repo, args):

    cwd = repo.getcwd()

    if cwd:

        return [util.pconvert(os.path.normpath(os.path.join(cwd, x)))

                for x in args]

    return args

def matchpats(cwd, pats = [], opts = {}, head = ''):

    return util.matcher(cwd, pats, opts.get('include'),

                        opts.get('exclude'), head)

def walk(repo, pats, opts, head = ''):

    cwd = repo.getcwd()

    c = 0

    if cwd: c = len(cwd) + 1

    files, matchfn = matchpats(cwd, pats, opts, head)

    for src, fn in repo.walk(files = files, match = matchfn):

        yield src, fn, fn[c:]

revrangesep = ':'

def revrange(ui, repo, revs, revlog=None):

    if revlog is None:

        revlog = repo.changelog

    revcount = revlog.count()

    def fix(val, defval):

        if not val:

            return defval

        try:

            num = int(val)

            if str(num) != val:

                raise ValueError

            if num < 0:

                num += revcount

            if not (0 <= num < revcount):

                raise ValueError

        except ValueError:

            try:

                num = repo.changelog.rev(repo.lookup(val))

            except KeyError:

                try:

                    num = revlog.rev(revlog.lookup(val))

                except KeyError:

                    raise Abort('invalid revision identifier %s', val)

        return num

    for spec in revs:

        if spec.find(revrangesep) >= 0:

            start, end = spec.split(revrangesep, 1)

            start = fix(start, 0)

            end = fix(end, revcount - 1)

            if end > start:

                end += 1

                step = 1

            else:

                end -= 1

                step = -1

            for rev in xrange(start, end, step):

                yield str(rev)

        else:

            yield spec

def make_filename(repo, r, pat, node=None,

                  total=None, seqno=None, revwidth=None):

    node_expander = {

        'H': lambda: hg.hex(node),

        'R': lambda: str(r.rev(node)),

        'h': lambda: hg.short(node),

        }

    expander = {

        '%': lambda: '%',

        'b': lambda: os.path.basename(repo.root),

        }

    try:

        if node:

            expander.update(node_expander)

        if node and revwidth is not None:

            expander['r'] = lambda: str(r.rev(node)).zfill(revwidth)

        if total is not None:

            expander['N'] = lambda: str(total)

        if seqno is not None:

            expander['n'] = lambda: str(seqno)

        if total is not None and seqno is not None:

            expander['n'] = lambda:str(seqno).zfill(len(str(total)))

        newname = []

        patlen = len(pat)

        i = 0

        while i < patlen:

            c = pat[i]

            if c == '%':

                i += 1

                c = pat[i]

                c = expander[c]()

            newname.append(c)

            i += 1

        return ''.join(newname)

    except KeyError, inst:

        raise Abort("invalid format spec '%%%s' in output file name",

                    inst.args[0])

def make_file(repo, r, pat, node=None,

              total=None, seqno=None, revwidth=None, mode='wb'):

    if not pat or pat == '-':

        if 'w' in mode: return sys.stdout

        else: return sys.stdin

    if hasattr(pat, 'write') and 'w' in mode:

        return pat

    if hasattr(pat, 'read') and 'r' in mode:

        return pat

    return open(make_filename(repo, r, pat, node, total, seqno, revwidth),

                mode)

def dodiff(fp, ui, repo, files=None, node1=None, node2=None):

    def date(c):

        return time.asctime(time.gmtime(float(c[2].split(' ')[0])))

    (c, a, d, u) = repo.changes(node1, node2, files)

    if files:

        c, a, d = map(lambda x: filterfiles(files, x), (c, a, d))

    if not c and not a and not d:

        return

    if node2:

        change = repo.changelog.read(node2)

        mmap2 = repo.manifest.read(change[0])

        date2 = date(change)

        def read(f):

            return repo.file(f).read(mmap2[f])

    else:

        date2 = time.asctime()

        if not node1:

            node1 = repo.dirstate.parents()[0]

        def read(f):

            return repo.wfile(f).read()

    if ui.quiet:

        r = None

    else:

        hexfunc = ui.verbose and hg.hex or hg.short

        r = [hexfunc(node) for node in [node1, node2] if node]

    change = repo.changelog.read(node1)

    mmap = repo.manifest.read(change[0])

    date1 = date(change)

    for f in c:

        to = None

        if f in mmap:

            to = repo.file(f).read(mmap[f])

        tn = read(f)

        fp.write(mdiff.unidiff(to, date1, tn, date2, f, r))

    for f in a:

        to = None

        tn = read(f)

        fp.write(mdiff.unidiff(to, date1, tn, date2, f, r))

    for f in d:

        to = repo.file(f).read(mmap[f])

        tn = None

        fp.write(mdiff.unidiff(to, date1, tn, date2, f, r))

def show_changeset(ui, repo, rev=0, changenode=None, filelog=None):

    """show a single changeset or file revision"""

    changelog = repo.changelog

    if filelog:

        log = filelog

        filerev = rev

        node = filenode = filelog.node(filerev)

        changerev = filelog.linkrev(filenode)

        changenode = changenode or changelog.node(changerev)

    else:

        log = changelog

        changerev = rev

        if changenode is None:

            changenode = changelog.node(changerev)

        elif not changerev:

            rev = changerev = changelog.rev(changenode)

        node = changenode

    if ui.quiet:

        ui.write("%d:%s\n" % (rev, hg.hex(node)))

        return

    changes = changelog.read(changenode)

    parents = [(log.rev(p), ui.verbose and hg.hex(p) or hg.short(p))

               for p in log.parents(node)

               if ui.debugflag or p != hg.nullid]

    if not ui.debugflag and len(parents) == 1 and parents[0][0] == rev-1:

        parents = []

    if ui.verbose:

        ui.write("changeset:   %d:%s\n" % (changerev, hg.hex(changenode)))

    else:

        ui.write("changeset:   %d:%s\n" % (changerev, hg.short(changenode)))

    for tag in repo.nodetags(changenode):

        ui.status("tag:         %s\n" % tag)

    for parent in parents:

        ui.write("parent:      %d:%s\n" % parent)

    if filelog:

        ui.debug("file rev:    %d:%s\n" % (filerev, hg.hex(filenode)))

    ui.debug("manifest:    %d:%s\n" % (repo.manifest.rev(changes[0]),

                                      hg.hex(changes[0])))

    ui.status("user:        %s\n" % changes[1])

    ui.status("date:        %s\n" % time.asctime(

        time.localtime(float(changes[2].split(' ')[0]))))

    if ui.debugflag:

        files = repo.changes(changelog.parents(changenode)[0], changenode)

        for key, value in zip(["files:", "files+:", "files-:"], files):

            if value:

                ui.note("%-12s %s\n" % (key, " ".join(value)))

    else:

        ui.note("files:       %s\n" % " ".join(changes[3]))

    description = changes[4].strip()

    if description:

        if ui.verbose:

            ui.status("description:\n")

            ui.status(description)

            ui.status("\n\n")

        else:

            ui.status("summary:     %s\n" % description.splitlines()[0])

    ui.status("\n")

def show_version(ui):

    """output version and copyright information"""

    ui.write("Mercurial version %s\n" % version.get_version())

    ui.status(

        "\nCopyright (C) 2005 Matt Mackall <mpm@selenic.com>\n"

        "This is free software; see the source for copying conditions. "

        "There is NO\nwarranty; "

        "not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"

    )

def help_(ui, cmd=None):

    """show help for a given command or all commands"""

    if cmd:

        try:

            i = find(cmd)

            ui.write("%s\n\n" % i[2])

            if i[1]:

                for s, l, d, c in i[1]:

                    opt = ' '

                    if s:

                        opt = opt + '-' + s + ' '

                    if l:

                        opt = opt + '--' + l + ' '

                    if d:

                        opt = opt + '(' + str(d) + ')'

                    ui.write(opt, "\n")

                    if c:

                        ui.write('   %s\n' % c)

                ui.write("\n")

            ui.write(i[0].__doc__, "\n")

        except UnknownCommand:

            ui.warn("hg: unknown command %s\n" % cmd)

        sys.exit(0)

    else:

        if ui.verbose:

            show_version(ui)

            ui.write('\n')

        if ui.verbose:

            ui.write('hg commands:\n\n')

        else:

            ui.write('basic hg commands (use "hg help -v" for more):\n\n')

        h = {}

        for c, e in table.items():

            f = c.split("|")[0]

            if not ui.verbose and not f.startswith("^"):

                continue

            if not ui.debugflag and f.startswith("debug"):

                continue

            f = f.lstrip("^")

            d = ""

            if e[0].__doc__:

                d = e[0].__doc__.splitlines(0)[0].rstrip()

            h[f] = d

        fns = h.keys()

        fns.sort()

        m = max(map(len, fns))

        for f in fns:

            ui.write(' %-*s   %s\n' % (m, f, h[f]))

# Commands start here, listed alphabetically

def add(ui, repo, *pats, **opts):

    '''add the specified files on the next commit'''

    names = []

    q = dict(zip(pats, pats))

    for src, abs, rel in walk(repo, pats, opts):

        if rel in q or abs in q:

            names.append(abs)

        elif repo.dirstate.state(abs) == '?':

            ui.status('adding %s\n' % rel)

            names.append(abs)

    repo.add(names)

def addremove(ui, repo, *pats, **opts):

    """add all new files, delete all missing files"""

    q = dict(zip(pats, pats))

    add, remove = [], []