# 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.

import os, re, sys, signal

import fancyopts, ui, hg, util

from demandload import *

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

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

class UnknownCommand(Exception): pass

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):

    if os.getcwd() != repo.root:

        p = os.getcwd()[len(repo.root) + 1: ]

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

    return files

def relpath(repo, args):

    if os.getcwd() != repo.root:

        p = os.getcwd()[len(repo.root) + 1: ]

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

                 for x in args ]

    return args

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:

                    ui.warn('abort: invalid revision identifier %s\n' % val)

                    sys.exit(1)

        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 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])

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

        date2 = date(change)

    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(parent), hg.hex(parent))

               for parent in log.parents(node)

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

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

        parents = []

    if filelog:

        ui.write("revision:    %d:%s\n" % (filerev, hg.hex(filenode)))

        for parent in parents:

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

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

    else:

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

        for tag in repo.nodetags(changenode):

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

        for parent in parents:

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

        ui.note("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, file, *files):

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

    repo.add(relpath(repo, (file,) + files))

def addremove(ui, repo, *files):

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

    if files:

        files = relpath(repo, files)

        d = []

        u = []

        for f in files:

            p = repo.wjoin(f)

            s = repo.dirstate.state(f)

            isfile = os.path.isfile(p)

            if s != 'r' and not isfile:

                d.append(f)

            elif s not in 'nmai' and isfile:

                u.append(f)

    else:

        (c, a, d, u) = repo.changes(None, None)

    repo.add(u)

    repo.remove(d)

def annotate(u, repo, file, *files, **ops):

    """show changeset information per file line"""

    def getnode(rev):

        return hg.short(repo.changelog.node(rev))

    def getname(rev):

        try:

            return bcache[rev]

        except KeyError:

            cl = repo.changelog.read(repo.changelog.node(rev))

            name = cl[1]

            f = name.find('@')

            if f >= 0:

                name = name[:f]

            f = name.find('<')

            if f >= 0:

                name = name[f+1:]

            bcache[rev] = name

            return name

    bcache = {}

    opmap = [['user', getname], ['number', str], ['changeset', getnode]]

    if not ops['user'] and not ops['changeset']:

        ops['number'] = 1

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

    if ops['revision']:

        node = repo.changelog.lookup(ops['revision'])

    change = repo.changelog.read(node)

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

    for f in relpath(repo, (file,) + files):

        lines = repo.file(f).annotate(mmap[f])

        pieces = []

        for o, f in opmap:

            if ops[o]:

                l = [ f(n) for n,t in lines ]

                m = max(map(len, l))

                pieces.append([ "%*s" % (m, x) for x in l])

        for p,l in zip(zip(*pieces), lines):

            u.write(" ".join(p) + ": " + l[1])

def cat(ui, repo, file, rev = []):

    """output the latest or given revision of a file"""

    r = repo.file(relpath(repo, [file])[0])

    n = r.tip()

    if rev: n = r.lookup(rev)

    sys.stdout.write(r.read(n))

def clone(ui, source, dest = None, **opts):

    """make a copy of an existing repository"""

    source = ui.expandpath(source)

    if dest is None:

        dest = os.path.basename(os.path.normpath(source))

    if os.path.exists(dest):

        ui.warn("abort: destination '%s' already exists\n" % dest)

        return 1

    class dircleanup:

        def __init__(self, dir):

            self.dir = dir

            os.mkdir(dir)

        def close(self):

            self.dir = None

        def __del__(self):

            if self.dir:

                import shutil

                shutil.rmtree(self.dir, True)

    d = dircleanup(dest)

    link = 0

    abspath = source

    if not (source.startswith("http://") or

            source.startswith("hg://") or

            source.startswith("old-http://")):

        abspath = os.path.abspath(source)

        d1 = os.stat(dest).st_dev

        d2 = os.stat(source).st_dev

        if d1 == d2: link = 1

    if link:

        ui.note("copying by hardlink\n")

        util.system("cp -al '%s'/.hg '%s'/.hg" % (source, dest))

        try:

            os.remove(os.path.join(dest, ".hg", "dirstate"))

        except: pass

        repo = hg.repository(ui, dest)

    else:

        repo = hg.repository(ui, dest, create=1)

        other = hg.repository(ui, source)

        fetch = repo.findincoming(other)

        if fetch:

            cg = other.changegroup(fetch)

            repo.addchangegroup(cg)

    f = repo.opener("hgrc", "w")

    f.write("[paths]\n")

    f.write("default = %s\n" % abspath)

    if not opts['noupdate']:

        update(ui, repo)

    d.close()

def commit(ui, repo, *files, **opts):

    """commit the specified files or all outstanding changes"""

    text = opts['text']

    if not text and opts['logfile']:

        try: text = open(opts['logfile']).read()

        except IOError: pass

    if opts['addremove']:

        addremove(ui, repo, *files)

    repo.commit(relpath(repo, files), text, opts['user'], opts['date'])

def copy(ui, repo, source, dest):

    """mark a file as copied or renamed for the next commit"""

    return repo.copy(*relpath(repo, (source, dest)))

def debugcheckstate(ui, repo):

    """validate the correctness of the current dirstate"""

    parent1, parent2 = repo.dirstate.parents()

    repo.dirstate.read()

    dc = repo.dirstate.map

    keys = dc.keys()

    keys.sort()

    m1n = repo.changelog.read(parent1)[0]

    m2n = repo.changelog.read(parent2)[0]

    m1 = repo.manifest.read(m1n)

    m2 = repo.manifest.read(m2n)

    errors = 0

    for f in dc:

        state = repo.dirstate.state(f)

        if state in "nr" and f not in m1:

            ui.warn("%s in state %s, but not in manifest1\n" % (f, state))

            errors += 1

        if state in "a" and f in m1:

            ui.warn("%s in state %s, but also in manifest1\n" % (f, state))

            errors += 1

        if state in "m" and f not in m1 and f not in m2:

            ui.warn("%s in state %s, but not in either manifest\n" %

                    (f, state))

            errors += 1

    for f in m1:

        state = repo.dirstate.state(f)

        if state not in "nrm":

            ui.warn("%s in manifest1, but listed as state %s" % (f, state))

            errors += 1

    if errors:

        ui.warn(".hg/dirstate inconsistent with current parent's manifest\n")

        sys.exit(1)

def debugstate(ui, repo):

    """show the contents of the current dirstate"""

    repo.dirstate.read()

    dc = repo.dirstate.map

    keys = dc.keys()

    keys.sort()

    for file in keys:

        ui.write("%c %s\n" % (dc[file][0], file))

def debugindex(ui, file):

    """dump the contents of an index file"""

    r = hg.revlog(hg.opener(""), file, "")

    ui.write("   rev    offset  length   base linkrev" +

             " p1           p2           nodeid\n")

    for i in range(r.count()):

        e = r.index[i]

        ui.write("% 6d % 9d % 7d % 6d % 7d %s.. %s.. %s..\n" % (

                i, e[0], e[1], e[2], e[3],

            hg.hex(e[4][:5]), hg.hex(e[5][:5]), hg.hex(e[6][:5])))

def debugindexdot(ui, file):

    """dump an index DAG as a .dot file"""

    r = hg.revlog(hg.opener(""), file, "")

    ui.write("digraph G {\n")

    for i in range(r.count()):

        e = r.index[i]

        ui.write("\t%d -> %d\n" % (r.rev(e[4]), i))

        if e[5] != hg.nullid:

            ui.write("\t%d -> %d\n" % (r.rev(e[5]), i))

    ui.write("}\n")

def diff(ui, repo, *files, **opts):

    """diff working directory (or selected files)"""

    revs = []

    if opts['rev']:

        revs = map(lambda x: repo.lookup(x), opts['rev'])

    if len(revs) > 2:

        ui.warn("too many revisions to diff\n")

        sys.exit(1)

    if files:

        files = relpath(repo, files)

    else:

        files = relpath(repo, [""])

    dodiff(sys.stdout, ui, repo, files, *revs)

def doexport(ui, repo, changeset, seqno, total, revwidth, opts):

    node = repo.lookup(changeset)

    prev, other = repo.changelog.parents(node)

    change = repo.changelog.read(node)

    def expand(name):

        expansions = {

            '%': lambda: '%',

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

            'N': lambda: str(total),