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

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(ui, cwd, pats = [], opts = {}):

    head = ''

    if opts.get('rootless'): head = '(?:.*/|)'

    def reify(name, tail):

        if name.startswith('re:'):

            return name[3:]

        elif name.startswith('glob:'):

            return head + util.globre(name[5:], '', tail)

        elif name.startswith('path:'):

            return '^' + re.escape(name[5:]) + '$'

        return head + util.globre(name, '', tail)

    cwdsep = cwd + os.sep

    def under(fn):

        if not cwd or fn.startswith(cwdsep): return True

    def matchfn(pats, tail, ifempty = util.always):

        if not pats: return ifempty

        pat = '(?:%s)' % '|'.join([reify(p, tail) for p in pats])

        if cwd: pat = re.escape(cwd + os.sep) + pat

        ui.debug('regexp: %s\n' % pat)

        return re.compile(pat).match

    patmatch = matchfn(pats, '$')

    incmatch = matchfn(opts.get('include'), '(?:/|$)', under)

    excmatch = matchfn(opts.get('exclude'), '(?:/|$)', util.never)

    return lambda fn: (incmatch(fn) and not excmatch(fn) and

                       (fn.endswith('/') or patmatch(fn)))

def walk(repo, pats, opts):

    cwd = repo.getcwd()

    if cwd: c = len(cwd) + 1

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

        }

    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)

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

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