#!/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 help():

    ui.status("""\

 commands:

 add [files...]        add the given files in the next commit

 addremove             add all new files, delete all missing files

 annotate [files...]   show changeset number per file line

 branch <path>         create a branch of <path> in this directory

 checkout [changeset]  checkout the latest or given changeset

 commit                commit all changes to the repository

 diff [files...]       diff working directory (or selected files)

 dump <file> [rev]     dump the latest or given revision of a file

 dumpmanifest [rev]    dump the latest or given revision of the manifest

 export <rev>          dump the changeset header and diffs for a revision

 history               show changeset history

 init                  create a new repository in this directory

 log <file>            show revision history of a single file

 merge <path>          merge changes from <path> into local repository

 recover               rollback an interrupted transaction

 remove [files...]     remove the given files in the next commit

 serve                 export the repository via HTTP

 status                show new, missing, and changed files in working dir

 tags                  show current changeset tags

 undo                  undo the last transaction

""")

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

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

        a = [] # ignore unknown files in repo, by popular request

        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 == "add":

    repo.add(args)

elif cmd == "forget":

    repo.forget(args)

elif cmd == "remove" or cmd == "rm" or cmd == "del" or cmd == "delete":

    repo.remove(args)

elif cmd == "commit" or cmd == "checkin" or cmd == "ci":

    if 1:

        if len(args) > 0:

            repo.commit(repo.current, args)

        else:

            repo.commit(repo.current)

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(repo.current, files, text)

elif cmd == "diff":

    revs = []

    if args:

        doptions = {}

        opts = [('r', 'revision', [], 'revision')]

        args = fancyopts.fancyopts(args, opts, doptions,

                                   'hg diff [options] [files]')

        revs = map(lambda x: repo.lookup(x), doptions['revision'])

    if len(revs) > 2:

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

        sys.exit(1)

    if relpath:

        if not args: args = [ relpath ]

        else: args = [ os.path.join(relpath, x) for x in args ]

    diff(args, *revs)

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 == "addremove":

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

    repo.add(a)

    repo.remove(d)

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 == "tip":

    n = repo.changelog.tip()

    t = repo.changelog.rev(n)

    ui.status("%d:%s\n" % (t, hg.hex(n)))

elif cmd == "log":

    if len(args) == 1:

        if relpath:

            args[0] = os.path.join(relpath, args[0])

        r = repo.file(args[0])

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

            n = r.node(i)

            (p1, p2) = r.parents(n)

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

            (i1, i2) = map(r.rev, (p1, p2))

            cr = r.linkrev(n)

            cn = hg.hex(repo.changelog.node(cr))

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

            print "changeset: %4d:%s" % (cr, cn)

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

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

            changes = repo.changelog.read(repo.changelog.node(cr))

            print "user: %s" % changes[1]

            print "date: %s" % time.asctime(

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

            print "description:"

            print changes[4]

            print

    elif len(args) > 1:

        print "too many args"

    else:

        print "missing filename"

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