657fa6e3949de9f60391ac2dd19826a6939f4863
[redakcja.git] / 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
9 class MercurialDocument(wlrepo.Document):
10
11     def data(self, entry):
12         path = self._revision._docname + '.' + entry            
13         return self._library._filectx(path, \
14             self._revision.hgrev()).data()   
15
16     def quickwrite(self, entry, data, msg, user=None):
17         user = user or self.owner
18         if user is None:
19             raise ValueError("Can't determine user.")
20         
21         def write(l, r):
22             f = l._fileopen(r(entry), "w+")
23             f.write(data)
24             f.close()
25             l._fileadd(r(entry))
26
27         return self.invoke_and_commit(write, lambda d: (msg, user))
28
29     def invoke_and_commit(self, ops, before_commit):
30         lock = self._library._lock()
31         try:            
32             self._library._checkout(self._revision.hgrev())
33
34             def entry_path(entry):
35                 return self.id + '.' + entry
36             
37             ops(self._library, entry_path)
38             message, user = before_commit(self)            
39             self._library._commit(message, user)
40
41             return self._library.document(docid=self.id, user=self.owner)
42         finally:
43             lock.release()            
44
45     # def commit(self, message, user):
46     #    """Make a new commit."""
47     #    self.invoke_and_commit(message, user, lambda *a: True)
48
49     def ismain(self):
50         return self._revision.user_name() is None
51
52     def shared(self):
53         if self.ismain():
54             return self
55         return self._library.document(docid=self._revision.document_name())
56
57     def take(self, user):
58         fullid = self._library.fulldocid(self.id, user)
59
60         def take_action(library, resolve):
61             # branch from latest 
62             library._create_branch(fullid, parent=self._revision)            
63
64         if not self._library.has_revision(fullid):
65             self.invoke_and_commit(take_action, \
66                 lambda d: ("$AUTO$ File checkout.", user) )
67
68         return self._library.document_for_rev(fullid)
69             
70     def update(self, user):
71         """Update parts of the document."""
72         lock = self.library._lock()
73         try:
74             if self.ismain():
75                 # main revision of the document
76                 return True
77             
78             if self._revision.has_children():
79                 # can't update non-latest revision
80                 return False
81
82             sv = self.shared()
83             
84             if not sv.ancestorof(self) and not self.parentof(sv):
85                 self._revision.merge_with(sv._revision, user=user)
86
87             return True
88         finally:
89             lock.release()  
90
91     def share(self, message):
92         lock = self.library._lock()
93         try:            
94             if self.ismain():
95                 return True # always shared          
96
97             user = self._revision.user_name()
98
99             main = self.shared()._revision
100             local = self._revision
101
102             no_changes = True
103
104             # Case 1:
105             #         * local
106             #         |
107             #         * <- can also be here!
108             #        /|
109             #       / |
110             # main *  *
111             #      |  |
112             # The local branch has been recently updated,
113             # so we don't need to update yet again, but we need to
114             # merge down to default branch, even if there was
115             # no commit's since last update
116
117             if main.ancestorof(local):
118                 print "case 1"
119                 main.merge_with(local, user=user, message=message)
120                 no_changes = False
121             # Case 2:
122             #
123             # main *  * local
124             #      |\ |
125             #      | \|
126             #      |  *
127             #      |  |
128             #
129             # Default has no changes, to update from this branch
130             # since the last merge of local to default.
131             elif local.has_common_ancestor(main):
132                 print "case 2"
133                 if not local.parentof(main):
134                     main.merge_with(local, user=user, message=message)
135                     no_changes = False
136
137             # Case 3:
138             # main *
139             #      |
140             #      * <- this case overlaps with previos one
141             #      |\
142             #      | \
143             #      |  * local
144             #      |  |
145             #
146             # There was a recent merge to the defaul branch and
147             # no changes to local branch recently.
148             #
149             # Use the fact, that user is prepared to see changes, to
150             # update his branch if there are any
151             elif local.ancestorof(main):
152                 print "case 3"
153                 if not local.parentof(main):
154                     local.merge_with(main, user=user, message='Local branch update.')
155                     no_changes = False
156             else:
157                 print "case 4"
158                 local.merge_with(main, user=user, message='Local branch update.')
159                 local = self.shelf()
160                 main.merge_with(local, user=user, message=message)
161
162             print "no_changes: ", no_changes
163             return no_changes
164         finally:
165             lock.release()     
166
167     def __str__(self):
168         return u"Document(%s:%s)" % (self.name, self.owner)
169
170     def __eq__(self, other):
171         return (self._revision == other._revision) and (self.name == other.name)
172