1
 
# Copyright (C) 2005, 2006, 2009 Canonical Ltd
 
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
3
3
# This program is free software; you can redistribute it and/or modify
 
4
4
# it under the terms of the GNU General Public License as published by
 
 
13
13
# You should have received a copy of the GNU General Public License
 
14
14
# along with this program; if not, write to the Free Software
 
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
17
"""Serializer factory for reading and writing bundles.
 
 
22
from bzrlib import errors
 
26
23
from bzrlib.bundle.serializer import (BundleSerializer,
 
29
26
from bzrlib.bundle.serializer import binary_diff
 
30
27
from bzrlib.bundle.bundle_data import (RevisionInfo, BundleInfo, BundleTree)
 
31
28
from bzrlib.diff import internal_diff
 
32
29
from bzrlib.osutils import pathjoin
 
 
30
from bzrlib.progress import DummyProgress
 
33
31
from bzrlib.revision import NULL_REVISION
 
34
33
from bzrlib.testament import StrictTestament
 
35
34
from bzrlib.timestamp import (
 
36
35
    format_highres_date,
 
 
120
119
        source.lock_read()
 
122
121
            self._write_main_header()
 
123
 
            pb = ui.ui_factory.nested_progress_bar()
 
125
124
                self._write_revisions(pb)
 
131
 
    def write_bundle(self, repository, target, base, fileobj):
 
132
 
        return self._write_bundle(repository, target, base, fileobj)
 
134
131
    def _write_main_header(self):
 
135
132
        """Write the header for the changes"""
 
137
 
        f.write(_get_bundle_header('0.8'))
 
 
134
        f.write(BUNDLE_HEADER)
 
140
 
    def _write(self, key, value, indent=1, trailing_space_when_empty=False):
 
141
 
        """Write out meta information, with proper indenting, etc.
 
143
 
        :param trailing_space_when_empty: To work around a bug in earlier
 
144
 
            bundle readers, when writing an empty property, we use "prop: \n"
 
145
 
            rather than writing "prop:\n".
 
146
 
            If this parameter is True, and value is the empty string, we will
 
147
 
            write an extra space.
 
150
 
            raise ValueError('indentation must be greater than 0')
 
 
138
    def _write(self, key, value, indent=1):
 
 
139
        """Write out meta information, with proper indenting, etc"""
 
 
140
        assert indent > 0, 'indentation must be greater than 0'
 
152
142
        f.write('#' + (' ' * indent))
 
153
143
        f.write(key.encode('utf-8'))
 
155
 
            if trailing_space_when_empty and value == '':
 
159
146
        elif isinstance(value, str):
 
 
184
171
        i_max = len(self.revision_ids)
 
185
172
        for i, rev_id in enumerate(self.revision_ids):
 
186
 
            pb.update("Generating revision data", i, i_max)
 
 
173
            pb.update("Generating revsion data", i, i_max)
 
187
174
            rev = self.source.get_revision(rev_id)
 
188
175
            if rev_id == last_rev_id:
 
189
176
                rev_tree = last_rev_tree
 
 
207
194
                base_tree = self.source.revision_tree(base_id)
 
208
195
            force_binary = (i != 0)
 
209
 
            self._write_revision(rev, rev_tree, base_id, base_tree,
 
 
196
            self._write_revision(rev, rev_tree, base_id, base_tree, 
 
210
197
                                 explicit_base, force_binary)
 
212
199
            last_rev_id = base_id
 
213
200
            last_rev_tree = base_tree
 
215
202
    def _testament_sha1(self, revision_id):
 
216
 
        return StrictTestament.from_revision(self.source,
 
 
203
        return StrictTestament.from_revision(self.source, 
 
217
204
                                             revision_id).as_sha1()
 
219
 
    def _write_revision(self, rev, rev_tree, base_rev, base_tree,
 
 
206
    def _write_revision(self, rev, rev_tree, base_rev, base_tree, 
 
220
207
                        explicit_base, force_binary):
 
221
208
        """Write out the information for a revision."""
 
222
209
        def w(key, value):
 
 
238
225
            w('base id', base_rev)
 
239
226
        if rev.properties:
 
240
227
            self._write('properties', None, indent=1)
 
241
 
            for name, value in sorted(rev.properties.items()):
 
242
 
                self._write(name, value, indent=3,
 
243
 
                            trailing_space_when_empty=True)
 
 
228
            for name, value in rev.properties.items():
 
 
229
                self._write(name, value, indent=3)
 
245
231
        # Add an extra blank space at the end
 
246
232
        self.to_file.write('\n')
 
 
254
240
        self.to_file.write(' // '.join(p_texts).encode('utf-8'))
 
255
241
        self.to_file.write('\n')
 
257
 
    def _write_delta(self, new_tree, old_tree, default_revision_id,
 
 
243
    def _write_delta(self, new_tree, old_tree, default_revision_id, 
 
259
245
        """Write out the changes between the trees."""
 
260
246
        DEVNULL = '/dev/null'
 
 
277
263
                old_lines = tree_lines(old_tree, require_text=True)
 
278
264
                new_lines = tree_lines(new_tree, require_text=True)
 
279
265
                action.write(self.to_file)
 
280
 
                internal_diff(old_path, old_lines, new_path, new_lines,
 
 
266
                internal_diff(old_path, old_lines, new_path, new_lines, 
 
282
268
            except errors.BinaryFile:
 
283
269
                old_lines = tree_lines(old_tree, require_text=False)
 
284
270
                new_lines = tree_lines(new_tree, require_text=False)
 
285
271
                action.add_property('encoding', 'base64')
 
286
272
                action.write(self.to_file)
 
287
 
                binary_diff(old_path, old_lines, new_path, new_lines,
 
 
273
                binary_diff(old_path, old_lines, new_path, new_lines, 
 
290
276
        def finish_action(action, file_id, kind, meta_modified, text_modified,
 
 
309
295
        for path, file_id, kind in delta.added:
 
310
296
            action = Action('added', [kind, path], [('file-id', file_id)])
 
311
 
            meta_modified = (kind=='file' and
 
 
297
            meta_modified = (kind=='file' and 
 
312
298
                             new_tree.is_executable(file_id))
 
313
299
            finish_action(action, file_id, kind, meta_modified, True,
 
 
333
319
            old_rev = getattr(old_tree.inventory[ie.file_id], 'revision', None)
 
334
320
            if new_rev != old_rev:
 
335
 
                action = Action('modified', [ie.kind,
 
 
321
                action = Action('modified', [ie.kind, 
 
336
322
                                             new_tree.id2path(ie.file_id)])
 
337
323
                action.add_utf8_property('last-changed', ie.revision)
 
338
324
                action.write(self.to_file)
 
 
350
336
        object.__init__(self)
 
351
337
        self.from_file = iter(from_file)
 
352
338
        self._next_line = None
 
354
340
        self.info = self._get_info()
 
355
341
        # We put the actual inventory ids in the footer, so that the patch
 
356
342
        # is easier to read for humans.
 
 
462
448
            # What do we do with a key we don't recognize
 
463
449
            raise errors.MalformedHeader('Unknown Key: "%s"' % key)
 
465
451
    def _read_many(self, indent):
 
466
452
        """If a line ends with no entry, that means that it should be
 
467
453
        followed with multiple lines of values.
 
 
504
490
            elif line.startswith('... '):
 
505
491
                action += line[len('... '):-1].decode('utf-8')
 
507
 
            if (self._next_line is not None and
 
 
493
            if (self._next_line is not None and 
 
508
494
                self._next_line.startswith('===')):
 
509
495
                return action, lines, True
 
510
496
            elif self._next_line is None or self._next_line.startswith('#'):
 
 
524
510
            action, lines, do_continue = self._read_one_patch()
 
525
511
            if action is not None:
 
526
512
                revision_actions.append((action, lines))
 
527
 
        if self.info.revisions[-1].tree_actions is not None:
 
528
 
            raise AssertionError()
 
 
513
        assert self.info.revisions[-1].tree_actions is None
 
529
514
        self.info.revisions[-1].tree_actions = revision_actions
 
531
516
    def _read_footer(self):