/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 dir.py

Set and verify testament.

Show diffs side-by-side

added added

removed removed

Lines of Context:
dir.py
1
1
# Copyright (C) 2007 Canonical Ltd
 
2
# Copyright (C) 2010 Jelmer Vernooij
2
3
#
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
14
15
# along with this program; if not, write to the Free Software
15
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
17
 
17
 
"""An adapter between a Git control dir and a Bazaar BzrDir."""
18
 
 
19
 
import os
20
 
 
21
 
import bzrlib
 
18
"""An adapter between a Git control dir and a Bazaar ControlDir."""
 
19
 
22
20
from bzrlib import (
23
 
    bzrdir,
 
21
    errors as bzr_errors,
24
22
    lockable_files,
25
23
    urlutils,
 
24
    version_info as bzrlib_version,
26
25
    )
27
26
 
28
27
LockWarner = getattr(lockable_files, "_LockWarner", None)
29
28
 
30
29
from bzrlib.plugins.git import (
31
 
    LocalGitBzrDirFormat,
32
 
    branch,
33
 
    errors,
34
 
    get_rich_root_format,
35
 
    repository,
36
 
    workingtree,
 
30
    LocalGitControlDirFormat,
37
31
    )
 
32
try:
 
33
    from bzrlib.controldir import (
 
34
        ControlDir,
 
35
        format_registry,
 
36
        )
 
37
except ImportError:
 
38
    # bzr < 2.3
 
39
    from bzrlib.bzrdir import (
 
40
        BzrDir,
 
41
        format_registry,
 
42
        )
 
43
    ControlDir = BzrDir
38
44
 
39
45
 
40
46
class GitLock(object):
55
61
    def validate_token(self, token):
56
62
        pass
57
63
 
 
64
    def break_lock(self):
 
65
        pass
 
66
 
58
67
 
59
68
class GitLockableFiles(lockable_files.LockableFiles):
60
69
    """Git specific lockable files abstraction."""
71
80
            self._lock_warner = LockWarner(repr(self))
72
81
 
73
82
 
74
 
class GitDir(bzrdir.BzrDir):
 
83
class GitDir(ControlDir):
75
84
    """An adapter to the '.git' dir used by git."""
76
85
 
77
86
    def is_supported(self):
78
87
        return True
79
88
 
 
89
    def can_convert_format(self):
 
90
        return False
 
91
 
80
92
    def cloning_metadir(self, stacked=False):
81
 
        return get_rich_root_format(stacked)
 
93
        return format_registry.make_bzrdir("default")
 
94
 
 
95
    def _branch_name_to_ref(self, name):
 
96
        raise NotImplementedError(self._branch_name_to_ref)
 
97
 
 
98
    if bzrlib_version >= (2, 2):
 
99
        def open_branch(self, name=None, unsupported=False, 
 
100
            ignore_fallbacks=None):
 
101
            return self._open_branch(name=name,
 
102
                ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
 
103
    else:
 
104
        def open_branch(self, ignore_fallbacks=None, unsupported=False):
 
105
            return self._open_branch(name=None,
 
106
                ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
82
107
 
83
108
 
84
109
class LocalGitDir(GitDir):
85
110
    """An adapter to the '.git' dir used by git."""
86
111
 
87
 
    _gitrepository_class = repository.LocalGitRepository
 
112
    def _get_gitrepository_class(self):
 
113
        from bzrlib.plugins.git.repository import LocalGitRepository
 
114
        return LocalGitRepository
 
115
 
 
116
    _gitrepository_class = property(_get_gitrepository_class)
 
117
 
 
118
    @property
 
119
    def user_transport(self):
 
120
        return self.root_transport
 
121
 
 
122
    @property
 
123
    def control_transport(self):
 
124
        return self.transport
88
125
 
89
126
    def __init__(self, transport, lockfiles, gitrepo, format):
90
127
        self._format = format
91
128
        self.root_transport = transport
 
129
        self._mode_check_done = False
92
130
        self._git = gitrepo
93
131
        if gitrepo.bare:
94
132
            self.transport = transport
97
135
        self._lockfiles = lockfiles
98
136
        self._mode_check_done = None
99
137
 
 
138
    def _branch_name_to_ref(self, name):
 
139
        from bzrlib.plugins.git.refs import branch_name_to_ref
 
140
        ref = branch_name_to_ref(name, None)
 
141
        if ref == "HEAD":
 
142
            from dulwich.repo import SYMREF
 
143
            refcontents = self._git.refs.read_ref(ref)
 
144
            if refcontents.startswith(SYMREF):
 
145
                ref = refcontents[len(SYMREF):]
 
146
        return ref
 
147
 
100
148
    def is_control_filename(self, filename):
101
149
        return filename == '.git' or filename.startswith('.git/')
102
150
 
103
 
    def get_branch_transport(self, branch_format):
 
151
    def get_branch_transport(self, branch_format, name=None):
104
152
        if branch_format is None:
105
153
            return self.transport
106
 
        if isinstance(branch_format, LocalGitBzrDirFormat):
107
 
            return self.transport
108
 
        raise errors.bzr_errors.IncompatibleFormat(branch_format, self._format)
109
 
 
110
 
    get_repository_transport = get_branch_transport
111
 
    get_workingtree_transport = get_branch_transport
112
 
 
113
 
    def open_branch(self, ignore_fallbacks=None):
 
154
        if isinstance(branch_format, LocalGitControlDirFormat):
 
155
            return self.transport
 
156
        raise bzr_errors.IncompatibleFormat(branch_format, self._format)
 
157
 
 
158
    def get_repository_transport(self, format):
 
159
        if format is None:
 
160
            return self.transport
 
161
        if isinstance(format, LocalGitControlDirFormat):
 
162
            return self.transport
 
163
        raise bzr_errors.IncompatibleFormat(format, self._format)
 
164
 
 
165
    def get_workingtree_transport(self, format):
 
166
        if format is None:
 
167
            return self.transport
 
168
        if isinstance(format, LocalGitControlDirFormat):
 
169
            return self.transport
 
170
        raise bzr_errors.IncompatibleFormat(format, self._format)
 
171
 
 
172
    def _open_branch(self, name=None, ignore_fallbacks=None, unsupported=False):
114
173
        """'create' a branch for this dir."""
115
174
        repo = self.open_repository()
116
 
        return branch.LocalGitBranch(self, repo, "HEAD", self._lockfiles)
 
175
        from bzrlib.plugins.git.branch import LocalGitBranch
 
176
        return LocalGitBranch(self, repo, self._branch_name_to_ref(name),
 
177
            self._lockfiles)
 
178
 
 
179
    def destroy_branch(self, name=None):
 
180
        refname = self._branch_name_to_ref(name)
 
181
        if not refname in self._git.refs:
 
182
            raise bzr_errors.NotBranchError(self.root_transport.base,
 
183
                    bzrdir=self)
 
184
        del self._git.refs[refname]
 
185
 
 
186
    def destroy_repository(self):
 
187
        raise bzr_errors.UnsupportedOperation(self.destroy_repository, self)
 
188
 
 
189
    def destroy_workingtree(self):
 
190
        raise bzr_errors.UnsupportedOperation(self.destroy_workingtree, self)
 
191
 
 
192
    def needs_format_conversion(self, format=None):
 
193
        return not isinstance(self._format, format.__class__)
 
194
 
 
195
    def list_branches(self):
 
196
        ret = []
 
197
        for name in self._git.get_refs():
 
198
            if name.startswith("refs/heads/"):
 
199
                ret.append(self.open_branch(name=name))
 
200
        return ret
117
201
 
118
202
    def open_repository(self, shared=False):
119
203
        """'open' a repository for this dir."""
120
204
        return self._gitrepository_class(self, self._lockfiles)
121
205
 
122
206
    def open_workingtree(self, recommend_upgrade=True):
123
 
        if not self._git.bare and self._git.has_index():
124
 
            return workingtree.GitWorkingTree(self, self.open_repository(), 
125
 
                                                  self.open_branch())
 
207
        if not self._git.bare:
 
208
            from dulwich.errors import NoIndexPresent
 
209
            repo = self.open_repository()
 
210
            try:
 
211
                index = repo._git.open_index()
 
212
            except NoIndexPresent:
 
213
                pass
 
214
            else:
 
215
                from bzrlib.plugins.git.workingtree import GitWorkingTree
 
216
                try:
 
217
                    branch = self.open_branch()
 
218
                except bzr_errors.NotBranchError:
 
219
                    pass
 
220
                else:
 
221
                    return GitWorkingTree(self, repo, branch, index)
126
222
        loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
127
 
        raise errors.bzr_errors.NoWorkingTree(loc)
 
223
        raise bzr_errors.NoWorkingTree(loc)
128
224
 
129
225
    def create_repository(self, shared=False):
130
226
        return self.open_repository()
131
227
 
132
 
    def create_branch(self):
133
 
        return self.open_branch()
 
228
    def create_branch(self, name=None):
 
229
        refname = self._branch_name_to_ref(name)
 
230
        from dulwich.protocol import ZERO_SHA
 
231
        self._git.refs[refname or "HEAD"] = ZERO_SHA
 
232
        return self.open_branch(name)
134
233
 
135
234
    def backup_bzrdir(self):
136
235
        if self._git.bare:
138
237
            return (self.root_transport.abspath(".git"),
139
238
                    self.root_transport.abspath(".git.backup"))
140
239
        else:
141
 
            raise errors.bzr_errors.BzrError("Unable to backup bare repositories")
 
240
            raise bzr_errors.BzrError("Unable to backup bare repositories")
142
241
 
143
242
    def create_workingtree(self, revision_id=None, from_branch=None,
144
243
        accelerator_tree=None, hardlink=False):
145
244
        if self._git.bare:
146
 
            raise errors.bzr_errors.BzrError("Can't create working tree in a bare repo")
 
245
            raise bzr_errors.BzrError("Can't create working tree in a bare repo")
147
246
        from dulwich.index import write_index
148
 
        write_index(self.root_transport.abspath(".git/index"), [])
 
247
        from dulwich.pack import SHA1Writer
 
248
        f = open(self.transport.local_abspath("index"), 'w+')
 
249
        try:
 
250
            f = SHA1Writer(f)
 
251
            write_index(f, [])
 
252
        finally:
 
253
            f.close()
149
254
        return self.open_workingtree()
 
255
 
 
256
    def find_repository(self):
 
257
        """Find the repository that should be used.
 
258
 
 
259
        This does not require a branch as we use it to find the repo for
 
260
        new branches as well as to hook existing branches up to their
 
261
        repository.
 
262
        """
 
263
        return self.open_repository()
 
264
 
 
265
    def _find_creation_modes(self):
 
266
        """Determine the appropriate modes for files and directories.
 
267
 
 
268
        They're always set to be consistent with the base directory,
 
269
        assuming that this transport allows setting modes.
 
270
        """
 
271
        # TODO: Do we need or want an option (maybe a config setting) to turn
 
272
        # this off or override it for particular locations? -- mbp 20080512
 
273
        if self._mode_check_done:
 
274
            return
 
275
        self._mode_check_done = True
 
276
        try:
 
277
            st = self.transport.stat('.')
 
278
        except TransportNotPossible:
 
279
            self._dir_mode = None
 
280
            self._file_mode = None
 
281
        else:
 
282
            # Check the directory mode, but also make sure the created
 
283
            # directories and files are read-write for this user. This is
 
284
            # mostly a workaround for filesystems which lie about being able to
 
285
            # write to a directory (cygwin & win32)
 
286
            if (st.st_mode & 07777 == 00000):
 
287
                # FTP allows stat but does not return dir/file modes
 
288
                self._dir_mode = None
 
289
                self._file_mode = None
 
290
            else:
 
291
                self._dir_mode = (st.st_mode & 07777) | 00700
 
292
                # Remove the sticky and execute bits for files
 
293
                self._file_mode = self._dir_mode & ~07111
 
294
 
 
295
    def _get_file_mode(self):
 
296
        """Return Unix mode for newly created files, or None.
 
297
        """
 
298
        if not self._mode_check_done:
 
299
            self._find_creation_modes()
 
300
        return self._file_mode
 
301
 
 
302
    def _get_dir_mode(self):
 
303
        """Return Unix mode for newly created directories, or None.
 
304
        """
 
305
        if not self._mode_check_done:
 
306
            self._find_creation_modes()
 
307
        return self._dir_mode
 
308
 
 
309