# HG changeset patch
# User Idan Kamara <idankk86@gmail.com>
# Date 1324573967 -7200
# Node ID 4da6bb8abfcc1b47c50fd24bd65a886ed657674f
# Parent  a4fcece7dd8eeeb65bbdfcc679e7f1822d88ad44
context: initial implementation of changectx

tries to mimic Mercurial's changectx

diff -r a4fcece7dd8e -r 4da6bb8abfcc hglib/context.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hglib/context.py	Thu Dec 22 19:12:47 2011 +0200
@@ -0,0 +1,208 @@
+import client, util, templates
+
+_nullcset = ['-1', '000000000000000000000000000000000000000', '', '', '', '']
+
+class changectx(object):
+    """A changecontext object makes access to data related to a particular
+    changeset convenient."""
+    def __init__(self, repo, changeid=''):
+        """changeid is a revision number, node, or tag"""
+        if changeid == '':
+            changeid = '.'
+        self._repo = repo
+        if isinstance(changeid, client.revision):
+            cset = changeid
+        elif changeid == -1:
+            cset = _nullcset
+        else:
+            if isinstance(changeid, (long, int)):
+                changeid = 'rev(%d)' % changeid
+
+            cset = self._repo.log(changeid)
+            if not len(cset):
+                raise ValueError('changeid %r not found in repo' % changeid)
+            if len(cset) > 1:
+                raise ValueError('changeid must yield a single changeset')
+            cset = cset[0]
+
+        self._rev, self._node, self._tags = cset[:3]
+        self._branch, self._author, self._description = cset[3:]
+
+        self._rev = int(self._rev)
+
+        self._tags = self._tags.split()
+        try:
+            self._tags.remove('tip')
+        except ValueError:
+            pass
+
+        self._ignored = None
+        self._clean = None
+
+    def __str__(self):
+        return self._node[:12]
+
+    def __int__(self):
+        return self._rev
+
+    def __repr__(self):
+        return "<changectx %s>" % str(self)
+
+    def __hash__(self):
+        try:
+            return hash(self._rev)
+        except AttributeError:
+            return id(self)
+
+    def __eq__(self, other):
+        try:
+            return self._rev == other._rev
+        except AttributeError:
+            return False
+
+    def __ne__(self, other):
+        return not (self == other)
+
+    def __nonzero__(self):
+        return self._rev != -1
+
+    def __contains__(self, key):
+        return key in self._manifest
+
+    def __iter__(self):
+        for f in sorted(self._manifest):
+            yield f
+
+    @util.propertycache
+    def _status(self):
+        return self._parsestatus(self._repo.status(change=self))[:4]
+
+    def _parsestatus(self, stat):
+        d = dict((c, []) for c in 'MAR!?IC ')
+        for k, path in stat:
+            d[k].append(path)
+        return d['M'], d['A'], d['R'], d['!'], d['?'], d['I'], d['C']
+
+    def status(self, ignored=False, clean=False):
+        """Explicit status query
+        Unless this method is used to query the working copy status, the
+        _status property will implicitly read the status using its default
+        arguments."""
+        stat = self._parsestatus(self._repo.status(change=self, ignored=ignored,
+                                                   clean=clean))
+        self._unknown = self._ignored = self._clean = None
+        if ignored:
+            self._ignored = stat[5]
+        if clean:
+            self._clean = stat[6]
+        self._status = stat[:4]
+        return stat
+
+    def rev(self):
+        return self._rev
+
+    def node(self):
+        return self._node
+
+    def tags(self):
+        return self._tags
+
+    def branch(self):
+        return self._branch
+
+    def author(self):
+        return self._author
+
+    def user(self):
+        return self._author
+
+    def date(self):
+        return self._date
+
+    def description(self):
+        return self._description
+
+    def files(self):
+        return sorted(self._status[0] + self._status[1] + self._status[2])
+
+    def modified(self):
+        return self._status[0]
+
+    def added(self):
+        return self._status[1]
+
+    def removed(self):
+        return self._status[2]
+
+    def ignored(self):
+        if self._ignored is None:
+            self.status(ignored=True)
+        return self._ignored
+
+    def clean(self):
+        if self._clean is None:
+            self.status(clean=True)
+        return self._clean
+
+    @util.propertycache
+    def _manifest(self):
+        d = {}
+        for node, p, e, s, path in self._repo.manifest(rev=self):
+            d[path] = node
+        return d
+
+    def manifest(self):
+        return self._manifest
+
+    def hex(self):
+        return hex(self._node)
+
+    @util.propertycache
+    def _parents(self):
+        """return contexts for each parent changeset"""
+        par = self._repo.parents(rev=self)
+        if not par:
+            return [changectx(self._repo, -1)]
+        return [changectx(self._repo, int(cset.rev)) for cset in par]
+
+    def parents(self):
+        return self._parents
+
+    def p1(self):
+        return self._parents[0]
+
+    def p2(self):
+        if len(self._parents) == 2:
+            return self._parents[1]
+        return changectx(self._repo, -1)
+
+    @util.propertycache
+    def _bookmarks(self):
+        books = [bm for bm in self._repo.bookmarks()[0] if bm[1] == self._rev]
+
+        bms = []
+        for name, r, n in books:
+            bms.append(name)
+        return bms
+
+    def bookmarks(self):
+        return self._bookmarks
+
+    def children(self):
+        """return contexts for each child changeset"""
+        for c in self._repo.log('children(%s)' % self._node):
+            yield changectx(self._repo, c)
+
+    def ancestors(self):
+        for a in self._repo.log('ancestors(%s)' % self._node):
+            yield changectx(self._repo, a)
+
+    def descendants(self):
+        for d in self._repo.log('descendants(%s)' % self._node):
+            yield changectx(self._repo, d)
+
+    def ancestor(self, c2):
+        """
+        return the ancestor context of self and c2
+        """
+        return changectx(self._repo, 'ancestor(%s, %s)' % (self, n2))
diff -r a4fcece7dd8e -r 4da6bb8abfcc tests/test-context.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test-context.py	Thu Dec 22 19:12:47 2011 +0200
@@ -0,0 +1,68 @@
+import common, hglib
+from hglib import context
+
+class test_context(common.basetest):
+    def test_non_existent(self):
+        self.assertRaises(ValueError, context.changectx, self.client, 'foo')
+
+    def test_basic(self):
+        self.append('a', 'a')
+        self.append('b', 'b')
+        rev0, node0 = self.client.commit('first', addremove=True)
+
+        self.append('c', 'c')
+        rev1, node1 = self.client.commit('second', addremove=True)
+
+        ctx = context.changectx(self.client, node0)
+
+        self.assertEquals(ctx.description(), 'first')
+        self.assertEquals(str(ctx), node0[:12])
+        self.assertEquals(ctx.node(), node0)
+        self.assertEquals(int(ctx), rev0)
+        self.assertEquals(ctx.rev(), rev0)
+        self.assertEquals(ctx.branch(), 'default')
+
+        self.assertTrue(ctx)
+
+        self.assertTrue('a' in ctx and 'b' in ctx)
+        self.assertFalse('c' in ctx)
+        self.assertEquals(list(ctx), ['a', 'b'])
+        self.assertEquals(ctx.files(), ['a', 'b'])
+
+        self.assertEquals(ctx.modified(), [])
+        self.assertEquals(ctx.added(), ['a', 'b'])
+        self.assertEquals(ctx.removed(), [])
+        self.assertEquals(ctx.ignored(), [])
+        self.assertEquals(ctx.clean(), [])
+
+        man = {'a' : '047b75c6d7a3ef6a2243bd0e99f94f6ea6683597',
+               'b' : '62452855512f5b81522aa3895892760bb8da9f3f'}
+        self.assertEquals(ctx.manifest(), man)
+
+        self.assertEquals([int(c) for c in ctx.parents()], [-1])
+        self.assertEquals(int(ctx.p1()), -1)
+        self.assertEquals(int(ctx.p2()), -1)
+
+        self.assertEquals([int(c) for c in ctx.children()], [1])
+        self.assertEquals([int(c) for c in ctx.descendants()], [0, 1])
+        self.assertEquals([int(c) for c in ctx.ancestors()], [0])
+
+        self.client.bookmark('bookmark', inactive=True, rev=node0)
+        self.assertEquals(ctx.bookmarks(), ['bookmark'])
+
+        self.client.tag('tag', rev=node0)
+        # tags are read on construction
+        self.assertEquals(context.changectx(self.client, node0).tags(), ['tag'])
+
+    def test_construction(self):
+        self.append('a', 'a')
+        rev0, node0 = self.client.commit('first', addremove=True)
+        tip = self.client.tip()
+
+        # from client.revision
+        ctx = context.changectx(self.client, tip)
+        self.assertEquals(ctx.node(), tip.node)
+
+        # from revset
+        ctx = context.changectx(self.client, 'all()')
+        self.assertEquals(ctx.node(), tip.node)