~bzr-pqm/bzr/bzr.dev

1553.5.28 by Martin Pool
[merge] from bzr.dev before integration
1
# Copyright (C) 2005, 2006 Canonical Ltd
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
2
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
"""Command which looks for unsigned commits by the current user, and signs them.
17
"""
18
19
from bzrlib.commands import Command
20
import bzrlib.config
21
import bzrlib.errors as errors
22
import bzrlib.gpg
23
from bzrlib.option import Option
24
25
26
class cmd_sign_my_commits(Command):
27
    """Sign all commits by a given committer.
28
29
    If location is not specified the local tree is used.
30
    If committer is not specified the default committer is used.
31
32
    This does not sign commits that already have signatures.
33
    """
34
35
    takes_options = [Option('dry-run'
36
                            , help='Don\'t actually sign anything, just print'
37
                                   ' the revisions that would be signed')
38
                    ]
39
    takes_args = ['location?', 'committer?']
40
41
    def run(self, location=None, committer=None, dry_run=False):
42
        if location is None:
43
            from bzrlib.workingtree import WorkingTree
44
            # Open the containing directory
45
            wt = WorkingTree.open_containing('.')[0]
46
            b = wt.branch
47
        else:
48
            # Passed in locations should be exact
49
            from bzrlib.branch import Branch
50
            b = Branch.open(location)
51
        repo = getattr(b, 'repository', b)
52
53
        config = bzrlib.config.BranchConfig(b)
54
55
        if committer is None:
56
            committer = config.username()
57
58
        gpg_strategy = bzrlib.gpg.GPGStrategy(config)
59
60
        count = 0
1553.5.28 by Martin Pool
[merge] from bzr.dev before integration
61
        # return in partial topological order for the sake of reproducibility
62
        for rev_id in repo.all_revision_ids():
1563.2.29 by Robert Collins
Remove all but fetch references to repository.revision_store.
63
            if repo.has_signature_for_revision_id(rev_id):
1185.78.6 by John Arbash Meinel
Adding sign-my-commits as a builtin, along with some simple tests.
64
                continue
65
            
66
            rev = repo.get_revision(rev_id)
67
            if rev.committer != committer:
68
                continue
69
70
            # We have a revision without a signature who has a 
71
            # matching committer, start signing
72
            print rev_id
73
            count += 1
74
            if not dry_run:
75
                repo.sign_revision(rev_id, gpg_strategy)
76
        print 'Signed %d revisions' % (count,)
77
78