#!/usr/bin/env python

#

# mercurial - a minimal scalable distributed SCM

# v0.5b "katje"

#

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

# the psyco compiler makes commits a bit faster

# and makes changegroup merge about 20 times slower!

# try:

#    import psyco

#    psyco.full()

# except:

#    pass

import sys, os, time

from mercurial import hg, mdiff, fancyopts, ui, commands

def filterfiles(list, files):

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

    for f in files:

        if f[-1] != os.sep: f += os.sep

        l += [ x for x in list if x.startswith(f) ]

    return l

def diff(files = None, node1 = None, node2 = None):

    def date(c):

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

    if node2:

        change = repo.changelog.read(node2)

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

        (c, a, d) = repo.diffrevs(node1, node2)

        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]

        (c, a, d, u) = repo.diffdir(repo.root, node1)

        def read(f): return file(os.path.join(repo.root, f)).read()

    change = repo.changelog.read(node1)

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

    date1 = date(change)

    if files:

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

    for f in c:

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

        tn = read(f)

        sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f))

    for f in a:

        to = ""

        tn = read(f)

        sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f))

    for f in d:

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

        tn = ""

        sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f))

try:

    sys.exit(commands.dispatch(sys.argv[1:]))

except commands.UnknownCommand:

    # fall through

    pass

options = {}

opts = [('v', 'verbose', None, 'verbose'),

        ('d', 'debug', None, 'debug'),

        ('q', 'quiet', None, 'quiet'),

        ('y', 'noninteractive', None, 'run non-interactively'),

        ]

args = fancyopts.fancyopts(sys.argv[1:], opts, options,

                           'hg [options] <command> [command options] [files]')

try:

    cmd = args[0]

    args = args[1:]

except:

    cmd = "help"

ui = ui.ui(options["verbose"], options["debug"], options["quiet"],

           not options["noninteractive"])

try:

    repo = hg.repository(ui=ui)

except IOError:

    ui.warn("Unable to open repository\n")

    sys.exit(0)

relpath = None

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

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

elif cmd == "rawcommit":

    "raw commit interface"

    rc = {}

    opts = [('p', 'parent', [], 'parent'),

            ('d', 'date', "", 'data'),

            ('u', 'user', "", 'user'),

            ('F', 'files', "", 'file list'),

            ('t', 'text', "", 'commit text'),

            ('l', 'logfile', "", 'commit text file')

            ]

    args = fancyopts.fancyopts(args, opts, rc,

                               "hg rawcommit [options] files")

    text = rc['text']

    if not text and rc['logfile']:

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

        except IOError: pass

    if not text and not rc['logfile']:

        print "missing commit text"

        sys.exit(0)

    if rc['files']:

        files = open(rc['files']).read().splitlines()

    else:

        files = args

    repo.rawcommit(files, text, rc['user'], rc['date'], *rc['parent'])

elif cmd == "import" or cmd == "patch":

    try:

        import psyco

        psyco.full()

    except:

        pass

    ioptions = {}

    opts = [('p', 'strip', 1, 'path strip'),

            ('b', 'base', "", 'base path'),

            ('q', 'quiet', "", 'silence diff')

            ]

    args = fancyopts.fancyopts(args, opts, ioptions,

                              'hg import [options] <patch names>')

    d = ioptions["base"]

    strip = ioptions["strip"]

    quiet = ioptions["quiet"] and "> /dev/null" or ""

    for patch in args:

        ui.status("applying %s\n" % patch)

        pf = os.path.join(d, patch)

        text = ""

        for l in file(pf):

            if l[:4] == "--- ": break

            text += l

        f = os.popen("lsdiff --strip %d %s" % (strip, pf))

        files = filter(None, map(lambda x: x.rstrip(), f.read().splitlines()))

        f.close()

        if files:

            if os.system("patch -p%d < %s %s" % (strip, pf, quiet)):

                raise "patch failed!"

        repo.commit(files, text)

elif cmd == "export":

    node = repo.lookup(args[0])

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

    change = repo.changelog.read(node)

    print "# HG changeset patch"

    print "# User %s" % change[1]

    print "# Node ID %s" % hg.hex(node)

    print "# Parent  %s" % hg.hex(prev)

    print

    if other != hg.nullid:

        print "# Parent  %s" % hg.hex(other)

    print change[4]

    diff(None, prev, node)

elif cmd == "debugchangegroup":

    newer = repo.newer(map(repo.lookup, args))

    for chunk in repo.changegroup(newer):

        sys.stdout.write(chunk)

elif cmd == "debugaddchangegroup":

    data = sys.stdin.read()

    repo.addchangegroup(data)

elif cmd == "history":

    for i in range(repo.changelog.count()):

        n = repo.changelog.node(i)

        changes = repo.changelog.read(n)

        (p1, p2) = repo.changelog.parents(n)

        (h, h1, h2) = map(hg.hex, (n, p1, p2))

        (i1, i2) = map(repo.changelog.rev, (p1, p2))

        print "rev:      %4d:%s" % (i, h)

        print "parents:  %4d:%s" % (i1, h1)

        if i2: print "          %4d:%s" % (i2, h2)

        print "manifest: %4d:%s" % (repo.manifest.rev(changes[0]),

                                    hg.hex(changes[0]))

        print "user:", changes[1]

        print "date:", time.asctime(

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

        if ui.verbose: print "files:", " ".join(changes[3])

        print "description:"

        print changes[4]

elif cmd == "dump":

    if args:

        r = repo.file(args[0])

        n = r.tip()

        if len(args) > 1: n = r.lookup(args[1])

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

    else:

        print "missing filename"

elif cmd == "dumpmanifest":

    n = repo.manifest.tip()

    if len(args) > 0:

        n = repo.manifest.lookup(args[0])

    m = repo.manifest.read(n)

    files = m.keys()

    files.sort()

    for f in files:

        print hg.hex(m[f]), f

elif cmd == "debugindex":

    if ".hg" not in args[0]:

        args[0] = ".hg/data/" + repo.file(args[0]).encodepath(args[0]) + "i"

    r = hg.revlog(open, args[0], "")

    print "   rev    offset  length   base linkrev"+\

          " p1           p2           nodeid"

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

        e = r.index[i]

        print "% 6d % 9d % 7d % 6d % 7d %s.. %s.. %s.." % (

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

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

elif cmd == "debugindexdot":

    if ".hg" not in args[0]:

        args[0] = ".hg/data/" + repo.file(args[0]).encodepath(args[0]) + "i"

    r = hg.revlog(open, args[0], "")

    print "digraph G {"

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

        e = r.index[i]

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

        if e[5] != hg.nullid:

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

    print "}"

elif cmd == "merge":

    if args:

        paths = {}

        try:

            pf = os.path.join(os.environ["HOME"], ".hgpaths")

            for l in file(pf):

                name, path = l.split()

                paths[name] = path

        except:

            pass

        if args[0] in paths: args[0] = paths[args[0]]

        other = hg.repository(ui, args[0])

        cg = repo.getchangegroup(other)

        repo.addchangegroup(cg)

    else:

        print "missing source repository"

elif cmd == "tags":

    repo.lookup(0) # prime the cache

    i = repo.tags.items()

    i.sort()

    for k, n in i:

        try:

            r = repo.changelog.rev(n)

        except KeyError:

            r = "?"

        print "%-30s %5d:%s" % (k, repo.changelog.rev(n), hg.hex(n))

elif cmd == "verify":

    filelinkrevs = {}

    filenodes = {}

    manifestchangeset = {}

    changesets = revisions = files = 0

    errors = 0

    ui.status("checking changesets\n")

    for i in range(repo.changelog.count()):

        changesets += 1

        n = repo.changelog.node(i)

        for p in repo.changelog.parents(n):

            if p not in repo.changelog.nodemap:

                ui.warn("changeset %s has unknown parent %s\n" %

                        (hg.short(n), hg.short(p)))

                errors += 1

        try:

            changes = repo.changelog.read(n)

        except Exception, inst:

            ui.warn("unpacking changeset %s: %s\n" % (short(n), inst))

            errors += 1

        manifestchangeset[changes[0]] = n

        for f in changes[3]:

            filelinkrevs.setdefault(f, []).append(i)

    ui.status("checking manifests\n")

    for i in range(repo.manifest.count()):

        n = repo.manifest.node(i)

        for p in repo.manifest.parents(n):

            if p not in repo.manifest.nodemap:

                ui.warn("manifest %s has unknown parent %s\n" %

                        (hg.short(n), hg.short(p)))

                errors += 1

        ca = repo.changelog.node(repo.manifest.linkrev(n))

        cc = manifestchangeset[n]

        if ca != cc:

            ui.warn("manifest %s points to %s, not %s\n" %

                    (hg.hex(n), hg.hex(ca), hg.hex(cc)))

            errors += 1

        try:

            delta = mdiff.patchtext(repo.manifest.delta(n))

        except KeyboardInterrupt:

            print "aborted"

            sys.exit(0)

        except Exception, inst:

            ui.warn("unpacking manifest %s: %s\n" % (hg.short(n), inst))

            errors += 1

        ff = [ l.split('\0') for l in delta.splitlines() ]

        for f, fn in ff:

            filenodes.setdefault(f, {})[hg.bin(fn)] = 1

    ui.status("crosschecking files in changesets and manifests\n")

    for f in filenodes:

        if f not in filelinkrevs:

            ui.warn("file %s in manifest but not in changesets\n" % f)

            errors += 1