← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge_directive.py

  • Committer: Vincent Ladeuil
  • Date: 2008-09-11 19:36:38 UTC
  • mfrom: (3703 +trunk)
  • mto: (3705.1.1 trunk2)
  • mto: This revision was merged to the branch mainline in revision 3708.
  • Revision ID: v.ladeuil+lp@free.fr-20080911193638-wtjyc1kcmacc6t1f
merge bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/merge_directive.py
1
 
# Copyright (C) 2007-2011 Canonical Ltd
 
1
# Copyright (C) 2007 Canonical Ltd
2
2
#
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
12
12
#
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
16
16
 
17
17
 
18
18
from StringIO import StringIO
19
19
import re
20
20
 
21
 
from bzrlib import lazy_import
22
 
lazy_import.lazy_import(globals(), """
23
21
from bzrlib import (
24
22
    branch as _mod_branch,
25
23
    diff,
26
 
    email_message,
27
24
    errors,
28
25
    gpg,
29
 
    hooks,
30
26
    registry,
31
27
    revision as _mod_revision,
32
28
    rio,
33
29
    testament,
34
30
    timestamp,
35
 
    trace,
36
31
    )
37
32
from bzrlib.bundle import (
38
33
    serializer as bundle_serializer,
39
34
    )
40
 
""")
41
 
 
42
 
 
43
 
class MergeRequestBodyParams(object):
44
 
    """Parameter object for the merge_request_body hook."""
45
 
 
46
 
    def __init__(self, body, orig_body, directive, to, basename, subject,
47
 
                 branch, tree=None):
48
 
        self.body = body
49
 
        self.orig_body = orig_body
50
 
        self.directive = directive
51
 
        self.branch = branch
52
 
        self.tree = tree
53
 
        self.to = to
54
 
        self.basename = basename
55
 
        self.subject = subject
56
 
 
57
 
 
58
 
class MergeDirectiveHooks(hooks.Hooks):
59
 
    """Hooks for MergeDirective classes."""
60
 
 
61
 
    def __init__(self):
62
 
        hooks.Hooks.__init__(self, "bzrlib.merge_directive", "BaseMergeDirective.hooks")
63
 
        self.add_hook('merge_request_body',
64
 
            "Called with a MergeRequestBodyParams when a body is needed for"
65
 
            " a merge request.  Callbacks must return a body.  If more"
66
 
            " than one callback is registered, the output of one callback is"
67
 
            " provided to the next.", (1, 15, 0))
68
 
 
69
 
 
70
 
class BaseMergeDirective(object):
71
 
    """A request to perform a merge into a branch.
72
 
 
73
 
    This is the base class that all merge directive implementations 
74
 
    should derive from.
75
 
 
76
 
    :cvar multiple_output_files: Whether or not this merge directive 
77
 
        stores a set of revisions in more than one file
78
 
    """
79
 
 
80
 
    hooks = MergeDirectiveHooks()
81
 
 
82
 
    multiple_output_files = False
 
35
from bzrlib.email_message import EmailMessage
 
36
 
 
37
 
 
38
class _BaseMergeDirective(object):
83
39
 
84
40
    def __init__(self, revision_id, testament_sha1, time, timezone,
85
41
                 target_branch, patch=None, source_branch=None, message=None,
105
61
        self.source_branch = source_branch
106
62
        self.message = message
107
63
 
108
 
    def to_lines(self):
109
 
        """Serialize as a list of lines
110
 
 
111
 
        :return: a list of lines
112
 
        """
113
 
        raise NotImplementedError(self.to_lines)
114
 
 
115
 
    def to_files(self):
116
 
        """Serialize as a set of files.
117
 
 
118
 
        :return: List of tuples with filename and contents as lines
119
 
        """
120
 
        raise NotImplementedError(self.to_files)
121
 
 
122
 
    def get_raw_bundle(self):
123
 
        """Return the bundle for this merge directive.
124
 
 
125
 
        :return: bundle text or None if there is no bundle
126
 
        """
127
 
        return None
128
 
 
129
64
    def _to_lines(self, base_revision=False):
130
65
        """Serialize as a list of lines
131
66
 
145
80
        lines.append('# \n')
146
81
        return lines
147
82
 
148
 
    def write_to_directory(self, path):
149
 
        """Write this merge directive to a series of files in a directory.
150
 
 
151
 
        :param path: Filesystem path to write to
152
 
        """
153
 
        raise NotImplementedError(self.write_to_directory)
154
 
 
155
83
    @classmethod
156
84
    def from_objects(klass, repository, revision_id, time, timezone,
157
85
                 target_branch, patch_type='bundle',
266
194
            body = self.to_signed(branch)
267
195
        else:
268
196
            body = ''.join(self.to_lines())
269
 
        message = email_message.EmailMessage(mail_from, mail_to, subject,
270
 
            body)
 
197
        message = EmailMessage(mail_from, mail_to, subject, body)
271
198
        return message
272
199
 
273
200
    def install_revisions(self, target_repo):
313
240
                target_repo.fetch(source_branch.repository, self.revision_id)
314
241
        return self.revision_id
315
242
 
316
 
    def compose_merge_request(self, mail_client, to, body, branch, tree=None):
317
 
        """Compose a request to merge this directive.
318
 
 
319
 
        :param mail_client: The mail client to use for composing this request.
320
 
        :param to: The address to compose the request to.
321
 
        :param branch: The Branch that was used to produce this directive.
322
 
        :param tree: The Tree (if any) for the Branch used to produce this
323
 
            directive.
324
 
        """
325
 
        basename = self.get_disk_name(branch)
326
 
        subject = '[MERGE] '
327
 
        if self.message is not None:
328
 
            subject += self.message
329
 
        else:
330
 
            revision = branch.repository.get_revision(self.revision_id)
331
 
            subject += revision.get_summary()
332
 
        if getattr(mail_client, 'supports_body', False):
333
 
            orig_body = body
334
 
            for hook in self.hooks['merge_request_body']:
335
 
                params = MergeRequestBodyParams(body, orig_body, self,
336
 
                                                to, basename, subject, branch,
337
 
                                                tree)
338
 
                body = hook(params)
339
 
        elif len(self.hooks['merge_request_body']) > 0:
340
 
            trace.warning('Cannot run merge_request_body hooks because mail'
341
 
                          ' client %s does not support message bodies.',
342
 
                        mail_client.__class__.__name__)
343
 
        mail_client.compose_merge_request(to, subject,
344
 
                                          ''.join(self.to_lines()),
345
 
                                          basename, body)
346
 
 
347
 
 
348
 
class MergeDirective(BaseMergeDirective):
 
243
 
 
244
class MergeDirective(_BaseMergeDirective):
349
245
 
350
246
    """A request to perform a merge into a branch.
351
247
 
380
276
        :param source_branch: A public location to merge the revision from
381
277
        :param message: The message to use when committing this merge
382
278
        """
383
 
        BaseMergeDirective.__init__(self, revision_id, testament_sha1, time,
 
279
        _BaseMergeDirective.__init__(self, revision_id, testament_sha1, time,
384
280
            timezone, target_branch, patch, source_branch, message)
385
281
        if patch_type not in (None, 'diff', 'bundle'):
386
282
            raise ValueError(patch_type)
413
309
        :return: a MergeRequest
414
310
        """
415
311
        line_iter = iter(lines)
416
 
        firstline = ""
417
312
        for line in line_iter:
418
313
            if line.startswith('# Bazaar merge directive format '):
419
 
                return _format_registry.get(line[2:].rstrip())._from_lines(
420
 
                    line_iter)
421
 
            firstline = firstline or line.strip()
422
 
        raise errors.NotAMergeDirective(firstline)
 
314
                break
 
315
        else:
 
316
            if len(lines) > 0:
 
317
                raise errors.NotAMergeDirective(lines[0])
 
318
            else:
 
319
                raise errors.NotAMergeDirective('')
 
320
        return _format_registry.get(line[2:].rstrip())._from_lines(line_iter)
423
321
 
424
322
    @classmethod
425
323
    def _from_lines(klass, line_iter):
470
368
        return None, self.revision_id, 'inapplicable'
471
369
 
472
370
 
473
 
class MergeDirective2(BaseMergeDirective):
 
371
class MergeDirective2(_BaseMergeDirective):
474
372
 
475
373
    _format_string = 'Bazaar merge directive format 2 (Bazaar 0.90)'
476
374
 
479
377
                 bundle=None, base_revision_id=None):
480
378
        if source_branch is None and bundle is None:
481
379
            raise errors.NoMergeSource()
482
 
        BaseMergeDirective.__init__(self, revision_id, testament_sha1, time,
 
380
        _BaseMergeDirective.__init__(self, revision_id, testament_sha1, time,
483
381
            timezone, target_branch, patch, source_branch, message)
484
382
        self.bundle = bundle
485
383
        self.base_revision_id = base_revision_id
622
520
                    revision_id):
623
521
                    raise errors.PublicBranchOutOfDate(public_branch,
624
522
                                                       revision_id)
625
 
            testament_sha1 = t.as_sha1()
626
523
        finally:
627
524
            for entry in reversed(locked):
628
525
                entry.unlock()
629
 
        return klass(revision_id, testament_sha1, time, timezone,
630
 
            target_branch, patch, public_branch, message, bundle,
631
 
            base_revision_id)
 
526
        return klass(revision_id, t.as_sha1(), time, timezone, target_branch,
 
527
            patch, public_branch, message, bundle, base_revision_id)
632
528
 
633
529
    def _verify_patch(self, repository):
634
530
        calculated_patch = self._generate_diff(repository, self.revision_id,
670
566
_format_registry = MergeDirectiveFormatRegistry()
671
567
_format_registry.register(MergeDirective)
672
568
_format_registry.register(MergeDirective2)
673
 
# 0.19 never existed.  It got renamed to 0.90.  But by that point, there were
674
 
# already merge directives in the wild that used 0.19. Registering with the old
675
 
# format string to retain compatibility with those merge directives.
676
569
_format_registry.register(MergeDirective2,
677
570
                          'Bazaar merge directive format 2 (Bazaar 0.19)')