/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to server.py

  • Committer: Andrew Bennetts
  • Date: 2009-10-21 11:13:40 UTC
  • mto: This revision was merged to the branch mainline in revision 4762.
  • Revision ID: andrew.bennetts@canonical.com-20091021111340-w7x4d5yf83qwjncc
Add test that WSGI glue allows request handlers to access paths above that request's. backing transport, so long as it is within the WSGI app's backing transport.

Show diffs side-by-side

added added

removed removed

Lines of Context:
server.py
1
 
# Copyright (C) 2008 Canonical Ltd
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
 
 
17
 
import os
18
 
import stat
19
 
import tempfile
20
 
 
21
 
from bzrlib.bzrdir import (
22
 
    BzrDir,
23
 
    )
24
 
from bzrlib.inventory import (
25
 
    InventoryDirectory,
26
 
    InventoryFile,
27
 
    )
28
 
from bzrlib.osutils import (
29
 
    splitpath,
30
 
    )
31
 
from bzrlib.repository import (
32
 
    Repository,
33
 
    )
34
 
 
35
 
from bzrlib.plugins.git.fetch import (
36
 
    import_git_objects,
37
 
    )
38
 
from bzrlib.plugins.git.mapping import (
39
 
    default_mapping,
40
 
    inventory_to_tree_and_blobs,
41
 
    revision_to_commit,
42
 
    )
43
 
 
44
 
from dulwich.server import (
45
 
    Backend,
46
 
    )
47
 
from dulwich.pack import (
48
 
    Pack,
49
 
    PackData,
50
 
    write_pack_index_v2,
51
 
    )
52
 
from dulwich.objects import (
53
 
    Blob,
54
 
    Commit,
55
 
    ShaFile,
56
 
    Tree,
57
 
    )
58
 
 
59
 
S_IFGITLINK = 0160000
60
 
 
61
 
#S_IFREG | 0664 # *Might* see this; would fail fsck --strict
62
 
 
63
 
 
64
 
class BzrBackend(Backend):
65
 
 
66
 
    def __init__(self, directory):
67
 
        self.directory = directory
68
 
        self.mapping = default_mapping
69
 
 
70
 
    def get_refs(self):
71
 
        """ return a dict of all tags and branches in repository (and shas) """
72
 
        ret = {}
73
 
        repo_dir = BzrDir.open(self.directory)
74
 
        repo = repo_dir.open_repository()
75
 
        branch = None
76
 
        for branch in repo.find_branches(using=True):
77
 
            #FIXME: Look for 'master' or 'trunk' in here, and set HEAD accordingly...
78
 
            #FIXME: Need to get branch path relative to its repository and use this instead of nick
79
 
            ret["refs/heads/"+branch.nick] = self.mapping.revision_id_bzr_to_foreign(branch.last_revision())[0]
80
 
        if 'HEAD' not in ret and branch:
81
 
            ret['HEAD'] = self.mapping.revision_id_bzr_to_foreign(branch.last_revision())[0]
82
 
        return ret
83
 
 
84
 
    def apply_pack(self, refs, read):
85
 
        """ apply pack from client to current repository """
86
 
 
87
 
        fd, path = tempfile.mkstemp(suffix=".pack")
88
 
        f = os.fdopen(fd, 'w')
89
 
        f.write(read())
90
 
        f.close()
91
 
 
92
 
        p = PackData(path)
93
 
        entries = p.sorted_entries()
94
 
        write_pack_index_v2(path[:-5]+".idx", entries, p.calculate_checksum())
95
 
 
96
 
        def get_objects():
97
 
            pack = Pack(path[:-5])
98
 
            for obj in pack.iterobjects():
99
 
                yield obj
100
 
 
101
 
        target = Repository.open(self.directory)
102
 
 
103
 
        target.lock_write()
104
 
        try:
105
 
            target.start_write_group()
106
 
            try:
107
 
                import_git_objects(target, self.mapping, iter(get_objects()))
108
 
            finally:
109
 
                target.commit_write_group()
110
 
        finally:
111
 
            target.unlock()
112
 
 
113
 
        for oldsha, sha, ref in refs:
114
 
            if ref[:11] == 'refs/heads/':
115
 
                branch_nick = ref[11:]
116
 
 
117
 
                try:
118
 
                    target_dir = BzrDir.open(self.directory + "/" + branch_nick)
119
 
                except:
120
 
                    target_dir = BzrDir.create(self.directory + "/" + branch_nick)
121
 
 
122
 
                try:
123
 
                    target_branch = target_dir.open_branch()
124
 
                except:
125
 
                    target_branch = target_dir.create_branch()
126
 
 
127
 
                rev_id = self.mapping.revision_id_foreign_to_bzr(sha)
128
 
                target_branch.generate_revision_history(rev_id)
129
 
 
130
 
    def fetch_objects(self, determine_wants, graph_walker, progress):
131
 
        """ yield git objects to send to client """
132
 
        repo = Repository.open(self.directory)
133
 
 
134
 
        # If this is a Git repository, just use the existing fetch_objects implementation.
135
 
        if getattr(repo, "fetch_objects", None) is None:
136
 
            return repo.fetch_objects(determine_wants, graph_walker, None, progress)
137
 
 
138
 
        wants = determine_wants(self.get_refs())
139
 
        commits_to_send = set([self.mapping.revision_id_foreign_to_bzr(w) for w in wants])
140
 
        rev_done = set()
141
 
        obj_sent = set()
142
 
 
143
 
        objects = set()
144
 
 
145
 
        repo.lock_read()
146
 
        try:
147
 
            have = graph_walker.next()
148
 
            while have:
149
 
                rev_done.add(have)
150
 
                if repo.has_revision(self.mapping.revision_id_foreign_to_bzr(sha)):
151
 
                    graph_walker.ack(have)
152
 
                have = graph_walker.next()
153
 
 
154
 
            while commits_to_send:
155
 
                commit = commits_to_send.pop()
156
 
                if commit in rev_done:
157
 
                    continue
158
 
                rev_done.add(commit)
159
 
 
160
 
                rev = repo.get_revision(commit)
161
 
 
162
 
                commits_to_send.update([p for p in rev.parent_ids if not p in rev_done])
163
 
 
164
 
                for sha, obj, path in inventory_to_tree_and_blobs(repo.get_inventory(commit), repo.texts, self.mapping):
165
 
                    if sha not in obj_sent:
166
 
                        obj_sent.add(sha)
167
 
                        objects.add((obj, path))
168
 
 
169
 
                objects.add((revision_to_commit(rev, sha, self.mapping.revision_id_bzr_to_foreign), None))
170
 
 
171
 
        finally:
172
 
            repo.unlock()
173
 
 
174
 
        return (len(objects), iter(objects))
175