# 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 or ['.'], opts.get('include'),

                        opts.get('exclude'), head)

def pathto(n1, n2):

    '''return the relative path from one place to another'''

    if not n1: return n2

    a, b = n1.split(os.sep), n2.split(os.sep)

    a.reverse(), b.reverse()

    while a and b and a[-1] == b[-1]:

        a.pop(), b.pop()

    b.reverse()

    return os.sep.join((['..'] * len(a)) + b)

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

    cwd = repo.getcwd()

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

    def walk():

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

            yield src, fn, pathto(cwd, fn)

    return files, matchfn, walk()

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

    files, matchfn, results = makewalk(repo, pats, opts, head)

    for r in results: yield r

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, match=util.always):

    def date(c):

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

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

    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 Distributed SCM (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 and cmd != 'shortlist':

        key, i = find(cmd)

        # synopsis

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

        # description

        doc = i[0].__doc__

        if ui.quiet:

            doc = doc.splitlines(0)[0]

        ui.write("%s\n" % doc.rstrip())

        # aliases

        if not ui.quiet:

            aliases = ', '.join(key.split('|')[1:])

            if aliases:

                ui.write("\naliases: %s\n" % aliases)

        # options

        if not ui.quiet and i[1]:

            ui.write("\noptions:\n\n")

            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)

    else:

        # program name

        if ui.verbose:

            show_version(ui)

        else:

            ui.status("Mercurial Distributed SCM\n")

        ui.status('\n')

        # list of commands

        if cmd == "shortlist":

            ui.status('basic commands (use "hg help" '

                      'for the full list or option "-v" for details):\n\n')

        elif ui.verbose:

            ui.status('list of commands:\n\n')

        else:

            ui.status('list of commands (use "hg help -v" '

                      'to show aliases and global options):\n\n')

        h = {}

        cmds = {}

        for c, e in table.items():

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

            if cmd == "shortlist" and not f.startswith("^"):

                continue

            f = f.lstrip("^")

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

                continue

            d = ""

            if e[0].__doc__:

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

            h[f] = d

            cmds[f]=c.lstrip("^")

        fns = h.keys()

        fns.sort()

        m = max(map(len, fns))

        for f in fns:

            if ui.verbose:

                commands = cmds[f].replace("|",", ")

                ui.write(" %s:\n      %s\n"%(commands,h[f]))

            else:

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

    # global options

    if ui.verbose:

        ui.write("\nglobal options:\n\n")

        for s, l, d, c in globalopts:

            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)

# 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 = [], []

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

        if src == 'f':

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

                add.append(abs)

                if rel not in q: ui.status('adding ', rel, '\n')

        elif repo.dirstate.state(abs) != 'r' and not os.path.exists(rel):

            remove.append(abs)

            if rel not in q: ui.status('removing ', rel, '\n')

    repo.add(add)

    repo.remove(remove)

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

    """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

    if not pats:

        raise Abort('at least one file name or pattern required')

    bcache = {}

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

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

        opts['number'] = 1

    if opts['rev']:

        node = repo.changelog.lookup(opts['rev'])

    else:

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

    change = repo.changelog.read(node)

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

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

        if abs not in mmap:

            ui.warn("warning: %s is not in the repository!\n" % rel)

            continue

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

        pieces = []

        for o, f in opmap:

            if opts[o]:

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

                if l:

                    m = max(map(len, l))

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

        if pieces:

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

                ui.write("%s: %s" % (" ".join(p), l[1]))

def cat(ui, repo, file1, rev=None, **opts):

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

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

    if rev:

        n = r.lookup(rev)

    else:

        n = r.tip()

    fp = make_file(repo, r, opts['output'], node=n)

    fp.write(r.read(n))

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

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

    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.rmtree = shutil.rmtree

            self.dir_ = dir_