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

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

    for src, fn in repo.walk(match = matchpats(cwd, pats, opts, head)):

        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(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 = []

    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)

    if filelog:

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

    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, *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, *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()

    repo.add(u)

    repo.remove(d)

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

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

        pieces = []

        for o, f in opmap:

            if opts[o]:

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

                m = max(map(len, l))

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

        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_

            os.mkdir(dir_)

        def close(self):

            self.dir_ = None

        def __del__(self):

            if self.dir_:

                self.rmtree(self.dir_, True)

    d = Dircleanup(dest)

    abspath = source

    source = ui.expandpath(source)

    other = hg.repository(ui, source)

    if other.dev() != -1:

        abspath = os.path.abspath(source)

        copyfile = (os.stat(dest).st_dev == other.dev()

                    and getattr(os, 'link', None) or shutil.copy2)

        if copyfile is not shutil.copy2:

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

        util.copytree(os.path.join(source, ".hg"), os.path.join(dest, ".hg"),

                      copyfile)

        try:

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

        except IOError:

            pass

        repo = hg.repository(ui, dest)

    else:

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

        repo.pull(other)

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

    if opts['text']:

        ui.warn("Warning: -t and --text is deprecated,"

                " please use -m or --message instead.\n")

    message = opts['message'] or opts['text']

    logfile = opts['logfile']

    if not message and logfile:

        try:

            message = open(logfile).read()

        except IOError, why:

            ui.warn("Can't read commit message %s: %s\n" % (logfile, why))

    if opts['addremove']:

        addremove(ui, repo, *files)

    repo.commit(relpath(repo, files), message, 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()