from mercurial import fancyopts, ui, hg

class UnknownCommand(Exception): pass

def filterfiles(filters, files):

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

    for t in filters:

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

        l += [ x for x in files if x.startswith(t) ]

    return l

def relfilter(repo, files):

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

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

        return filterfiles(p, files)

    return files

def relpath(repo, args):

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

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

        return [ os.path.normpath(os.path.join(p, x)) for x in args ]

    return args

def dodiff(repo, 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()

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

        if not node1:

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

        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(files, x), (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))

def help(ui, cmd=None):

    '''show help'''

    if cmd:

        try:

            i = find(cmd)

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

            ui.write(i[0].__doc__, "\n")

        except UnknownCommand:

            ui.warn("unknown command %s", cmd)

        sys.exit(0)

    ui.status("""\

 hg 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 add(ui, repo, file, *files):

    '''add the specified files on the next commit'''

    repo.add(relpath(repo, (file,) + files))

def addremove(ui, repo):

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

    repo.add(a)

    repo.remove(d)

def annotate(u, repo, file, *files, **ops):

    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]

            bcache[rev] = name

            return name

    bcache = {}

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

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

        ops['number'] = 1

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

    if ops['revision']:

        node = repo.changelog.lookup(ops['revision'])

    change = repo.changelog.read(node)

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

    maxuserlen = 0

    maxchangelen = 0

    for f in relpath(repo, (file,) + files):

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

        pieces = []

        for o, f in opmap:

            if ops[o]:

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

                m = max(map(len, l))

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

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

            u.write(" ".join(p) + ": " + l[1])

def branch(ui, path):

    '''branch from a local repository'''

    # this should eventually support remote repos

    os.system("cp -al %s/.hg .hg" % path)

def cat(ui, repo, file, rev = []):

    r = repo.file(file)

    n = r.tip()

    if rev: n = r.lookup(rev)

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

def checkout(ui, repo, changeset=None):

    '''checkout a given changeset or the current tip'''

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

    if c or a or d:

        ui.warn("aborting (outstanding changes in working directory)\n")

        sys.exit(1)

    node = repo.changelog.tip()

    if changeset:

        node = repo.lookup(changeset)

    repo.checkout(node)

def commit(ui, repo, *files):

    """commit the specified files or all outstanding changes"""

    repo.commit(relpath(repo, files))

def debugaddchangegroup(ui, repo):

    data = sys.stdin.read()

    repo.addchangegroup(data)

def debugchangegroup(ui, repo, roots):

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

    for chunk in repo.changegroup(newer):

        sys.stdout.write(chunk)

def debugindex(ui, file):

    r = hg.revlog(open, file, "")

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

def debugindexdot(ui, file):

    r = hg.revlog(open, file, "")

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

def diff(ui, repo, *files, **opts):

    revs = []

    if opts['rev']:

        revs = map(lambda x: repo.lookup(x), opts['rev'])

    if len(revs) > 2:

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

        sys.exit(1)

    if files:

        files = relpath(repo, files)

    else:

        files = relpath(repo, [""])

    dodiff(repo, files, *revs)

def export(ui, repo, changeset):

    node = repo.lookup(changeset)

    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].rstrip()

    print

    dodiff(repo, None, prev, node)

def forget(ui, repo, file, *files):

    """don't add the specified files on the next commit"""

    repo.forget(relpath(repo, (file,) + files))

def heads(ui, repo):

    '''show current repository heads'''

    for n in repo.changelog.heads():

        i = repo.changelog.rev(n)

        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]

def history(ui, repo):

    """show the changelog 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]

def patch(ui, repo, patches, opts):

    """import an ordered set of patches"""

    try:

        import psyco

        psyco.full()

    except:

        pass

    d = opts["base"]

    strip = opts["strip"]

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

    for patch in patches:

        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)

def init(ui):

    """create a repository"""

    hg.repository(ui, ".", create=1)

def log(ui, repo, f):

    f = relpath(repo, [f])[0]

    r = repo.file(f)

    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].rstrip()

        print

def manifest(ui, repo, rev = []):

    n = repo.manifest.tip()

    if rev:

        n = repo.manifest.lookup(rev)

    m = repo.manifest.read(n)

    files = m.keys()

    files.sort()

    for f in files:

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

def parents(ui, repo, node = None):

    '''show the parents of the current working dir'''

    if node:

        p = repo.changelog.parents(repo.lookup(hg.bin(node)))

    else:

        p = repo.dirstate.parents()

    for n in p:

        if n != hg.nullid:

            ui.write("%d:%s\n" % (repo.changelog.rev(n), hg.hex(n)))

def pull(ui, repo, source):

    """pull changes from the specified source"""

    paths = {}

    try:

        pf = os.path.expanduser("~/.hgpaths")

        for l in file(pf):

            name, path = l.split()

            paths[name] = path

    except IOError:

        pass

    if source in paths: source = paths[source]

    other = hg.repository(ui, source)

    cg = repo.getchangegroup(other)

    repo.addchangegroup(cg)

def rawcommit(ui, repo, files, rc):

    "raw commit interface"

    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"

        return 1

    files = relpath(repo, files)

    if rc['files']:

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

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

def recover(ui, repo):

    repo.recover()

def remove(ui, repo, file, *files):

    """remove the specified files on the next commit"""

    repo.remove(relpath(repo, (file,) + files))

def resolve(ui, repo, node=None):

    '''merge a given node or the current tip into the working dir'''

    if not node:

        node = repo.changelog.tip()

    else:

        node = repo.lookup(node)

    repo.resolve(node)

def serve(ui, repo, **opts):

    from mercurial import hgweb

    hgweb.server(repo.root, opts["name"], opts["templates"],

                 opts["address"], opts["port"])

def status(ui, repo):

    '''show changed files in the working directory

    C = changed

    A = added

    R = removed

    ? = not tracked'''

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

    (c, a, d, u) = map(lambda x: relfilter(repo, x), (c, a, d, u))

    for f in c: print "C", f

    for f in a: print "A", f

    for f in d: print "R", f

    for f in u: print "?", f

def tags(ui, repo):

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

def tip(ui, repo):

    n = repo.changelog.tip()

    t = repo.changelog.rev(n)

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

def undo(ui, repo):

    repo.undo()

def verify(ui, repo):

    """verify the integrity of the repository"""

    return repo.verify()

table = {

    "add": (add, [], "hg add [files]"),

    "addremove": (addremove, [], "hg addremove"),

    "ann|annotate": (annotate,

                     [('r', 'revision', '', 'revision'),

                      ('u', 'user', None, 'show user'),

                      ('n', 'number', None, 'show revision number'),

                      ('c', 'changeset', None, 'show changeset')],

                     'hg annotate [-u] [-c] [-n] [-r id] [files]'),

    "branch|clone": (branch, [], 'hg branch [path]'),

    "cat|dump": (cat, [], 'hg cat <file> [rev]'),

    "checkout|co": (checkout, [], 'hg checkout [changeset]'),

    "commit|ci": (commit, [], 'hg commit [files]'),

    "debugaddchangegroup": (debugaddchangegroup, [], 'debugaddchangegroup'),

    "debugchangegroup": (debugchangegroup, [], 'debugchangegroup [roots]'),

    "debugindex": (debugindex, [], 'debugindex <file>'),

    "debugindexdot": (debugindexdot, [], 'debugindexdot <file>'),

    "diff": (diff, [('r', 'rev', [], 'revision')],

             'hg diff [-r A] [-r B] [files]'),

    "export": (export, [], "hg export <changeset>"),

    "forget": (forget, [], "hg forget [files]"),

    "heads": (heads, [], 'hg heads'),

    "history": (history, [], 'hg history'),

    "help": (help, [], 'hg help [command]'),

    "init": (init, [], 'hg init'),

    "log": (log, [], 'hg log <file>'),

    "manifest|dumpmanifest": (manifest, [], 'hg manifest [rev]'),

    "parents": (parents, [], 'hg parents [node]'),

    "patch|import": (patch,

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

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

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

                     "hg import [options] patches"),

    "pull|merge": (pull, [], 'hg pull [source]'),

    "rawcommit": (rawcommit,

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

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