lib/wlrepo/mercurial_backend/document.py

   1 # -*- encoding: utf-8 -*-
   2
   3 __author__ = "Łukasz Rekucki"
   4 __date__ = "$2009-09-25 09:35:06$"
   5 __doc__ = "Module documentation."
   6
   7 import wlrepo
   8 import mercurial.error
   9
  10 class MercurialDocument(wlrepo.Document):
  11
  12     def data(self, entry):
  13         path = self._library._sanitize_string(self.id + u'.' + entry)
  14         try:
  15             return self._library._filectx(path, \
  16                 self._revision.hgrev()).data().decode('utf-8')
  17         except mercurial.error.LookupError, e:
  18             fl = [x.decode('utf-8') for x in self._revision._changectx]
  19             raise wlrepo.EntryNotFound(self._revision, path.decode('utf-8'), fl)
  20
  21     def quickwrite(self, entry, data, msg, user=None):
  22         user = user or self.owner
  23
  24         if isinstance(data, unicode):
  25             data = data.encode('utf-8')
  26
  27         user = self._library._sanitize_string(user)
  28         msg = self._library._sanitize_string(msg)
  29         entry = self._library._sanitize_string(entry)
  30
  31         if user is None:
  32             raise ValueError("Can't determine user.")
  33
  34         def write(l, r):
  35             f = l._fileopen(r(entry), "w+")
  36             f.write(data)
  37             f.close()
  38             l._fileadd(r(entry))
  39
  40         return self.invoke_and_commit(write, lambda d: (msg, \
  41                 self._library._sanitize_string(self.owner)) )
  42
  43     def invoke_and_commit(self, ops, commit_info):
  44         lock = self._library.lock()
  45         try:
  46             self._library._checkout(self._revision.hgrev())
  47
  48             def entry_path(entry):
  49                 return self._library._sanitize_string(self.id + u'.' + entry)
  50
  51             ops(self._library, entry_path)
  52             message, user = commit_info(self)
  53
  54             message = self._library._sanitize_string(message)
  55             user = self._library._sanitize_string(user)
  56
  57             self._library._commit(message, user)
  58             try:
  59                 return self._library.document(docid=self.id, user=user)
  60             except Exception, e:
  61                 # rollback the last commit
  62                 self._library._rollback()
  63                 raise e
  64         finally:
  65             lock.release()
  66
  67     # def commit(self, message, user):
  68     #    """Make a new commit."""
  69     #    self.invoke_and_commit(message, user, lambda *a: True)
  70
  71     def ismain(self):
  72         return self._revision.user_name is None
  73
  74     def shared(self):
  75         if self.ismain():
  76             return self
  77
  78         return self._library.document(docid=self.id)
  79
  80     def latest(self):
  81         return self._library.document(docid=self.id, user=self.owner)
  82
  83     def take(self, user):
  84         fullid = self._library.fulldocid(self.id, user)
  85
  86         def take_action(library, resolve):
  87             # branch from latest
  88             library._create_branch(fullid, parent=self._revision)
  89
  90         if not self._library.has_revision(fullid):
  91             self.invoke_and_commit(take_action, \
  92                 lambda d: ("$AUTO$ File checkout.", user) )
  93
  94         return self._library.document_for_rev(fullid)
  95
  96     def update(self, user):
  97         """Update parts of the document."""
  98         lock = self.library.lock()
  99         try:
 100             if self.ismain():
 101                 # main revision of the document
 102                 return (True, False)
 103
 104             if self._revision.has_children():
 105                 print 'Update failed: has children.'
 106                 # can't update non-latest revision
 107                 return (False, False)
 108
 109             sv = self.shared()
 110
 111             if self.parentof(sv):
 112                 return (True, False)
 113
 114             if sv.ancestorof(self):
 115                 return (True, False)
 116
 117
 118             return self._revision.merge_with(sv._revision, user=user,
 119                 message="$AUTO$ Personal branch update.")
 120         finally:
 121             lock.release()
 122
 123     def share(self, message):
 124         lock = self.library.lock()
 125         try:
 126             if self.ismain():
 127                 return (True, False) # always shared
 128
 129             user = self._revision.user_name
 130             main = self.shared()._revision
 131             local = self._revision
 132
 133             # Case 1:
 134             #         * local
 135             #         |
 136             #         * <- can also be here!
 137             #        /|
 138             #       / |
 139             # main *  *
 140             #      |  |
 141             # The local branch has been recently updated,
 142             # so we don't need to update yet again, but we need to
 143             # merge down to default branch, even if there was
 144             # no commit's since last update
 145
 146             if main.ancestorof(local):
 147                 print "case 1"
 148                 success, changed = main.merge_with(local, user=user, message=message)
 149             # Case 2:
 150             #
 151             # main *  * local
 152             #      |\ |
 153             #      | \|
 154             #      |  *
 155             #      |  |
 156             #
 157             # Default has no changes, to update from this branch
 158             # since the last merge of local to default.
 159             elif local.has_common_ancestor(main):
 160                 print "case 2"
 161                 if not local.parentof(main):
 162                     success, changed = main.merge_with(local, user=user, message=message)
 163
 164                 success = True
 165                 changed = False
 166
 167             # Case 3:
 168             # main *
 169             #      |
 170             #      * <- this case overlaps with previos one
 171             #      |\
 172             #      | \
 173             #      |  * local
 174             #      |  |
 175             #
 176             # There was a recent merge to the defaul branch and
 177             # no changes to local branch recently.
 178             #
 179             # Use the fact, that user is prepared to see changes, to
 180             # update his branch if there are any
 181             elif local.ancestorof(main):
 182                 print "case 3"
 183                 if not local.parentof(main):
 184                     success, changed = local.merge_with(main, user=user, \
 185                         message='$AUTO$ Local branch update during share.')
 186
 187                 success = True
 188                 changed = False
 189
 190             else:
 191                 print "case 4"
 192                 success, changed = local.merge_with(main, user=user, \
 193                         message='$AUTO$ Local branch update during share.')
 194
 195                 if not success:
 196                     return False
 197
 198                 if changed:
 199                     local = self.latest()._revision
 200
 201                 success, changed = main.merge_with(local, user=user,\
 202                     message=message)
 203
 204             return success, changed
 205         finally:
 206             lock.release()
 207
 208     def __unicode__(self):
 209         return u"Document(%s:%s)" % (self.name, self.owner)
 210
 211     def __str__(self):
 212         return self.__unicode__().encode('utf-8')
 213
 214     def __eq__(self, other):
 215         return (self._revision == other._revision) and (self.name == other.name)
 216