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

More work on roundtrip push support.

Show diffs side-by-side

added added

removed removed

Lines of Context:
workingtree.py
25
25
from dulwich.objects import (
26
26
    Blob,
27
27
    )
 
28
from dulwich.protocol import (
 
29
    ZERO_SHA,
 
30
    )
28
31
import os
29
32
import stat
30
33
 
36
39
    osutils,
37
40
    transport,
38
41
    tree,
39
 
    urlutils,
40
42
    workingtree,
41
43
    )
42
44
from bzrlib.decorators import (
51
53
    changes_from_git_changes,
52
54
    tree_delta_from_git_changes,
53
55
    )
54
 
 
 
56
from bzrlib.plugins.git.mapping import (
 
57
    GitFileIdMap,
 
58
    )
55
59
 
56
60
IGNORE_FILENAME = ".gitignore"
57
61
 
59
63
class GitWorkingTree(workingtree.WorkingTree):
60
64
    """A Git working tree."""
61
65
 
62
 
    def __init__(self, bzrdir, repo, branch):
 
66
    def __init__(self, bzrdir, repo, branch, index):
63
67
        self.basedir = bzrdir.root_transport.local_abspath('.')
64
68
        self.bzrdir = bzrdir
65
69
        self.repository = repo
67
71
        self._branch = branch
68
72
        self._transport = bzrdir.transport
69
73
 
70
 
        self.controldir = urlutils.join(self.repository._git._controldir, 'bzr')
 
74
        self.controldir = self.bzrdir.transport.local_abspath('bzr')
71
75
 
72
76
        try:
73
77
            os.makedirs(self.controldir)
78
82
        self._control_files = lockable_files.LockableFiles(
79
83
            transport.get_transport(self.controldir), 'lock', lockdir.LockDir)
80
84
        self._format = GitWorkingTreeFormat()
81
 
        self.index = self.repository._git.open_index()
 
85
        self.index = index
82
86
        self.views = self._make_views()
83
87
        self._detect_case_handling()
84
88
 
133
137
                    from posix import stat_result
134
138
                    stat_val = stat_result((stat.S_IFLNK, 0, 0, 0, 0, 0, 0, 0, 0, 0))
135
139
                blob.set_raw_string(entry.symlink_target)
 
140
            else:
 
141
                raise AssertionError("unknown kind '%s'" % entry.kind)
136
142
            # Add object to the repository if it didn't exist yet
137
143
            if not blob.id in self.repository._git.object_store:
138
144
                self.repository._git.object_store.add_object(blob)
144
150
        # TODO: Maybe this should only write on dirty ?
145
151
        if self._control_files._lock_mode != 'w':
146
152
            raise errors.NotWriteLocked(self)
147
 
        self._rewrite_index()           
 
153
        self._rewrite_index()
148
154
        self.index.write()
149
155
        self._inventory_is_modified = False
150
156
 
170
176
 
171
177
    def _reset_data(self):
172
178
        self._inventory_is_modified = False
173
 
        basis_inv = self.repository.get_inventory(self.mapping.revision_id_foreign_to_bzr(self.repository._git.head()))
174
 
        result = GitIndexInventory(basis_inv, self.mapping, self.index,
175
 
            self.repository._git.object_store)
 
179
        try:
 
180
            head = self.repository._git.head()
 
181
        except KeyError, name:
 
182
            raise errors.NotBranchError("branch %s at %s" % (name, self.repository.base))
 
183
        basis_inv = self.repository.get_inventory(self.branch.lookup_foreign_revision_id(head))
 
184
        store = self.repository._git.object_store
 
185
        if head == ZERO_SHA:
 
186
            fileid_map = GitFileIdMap({}, self.mapping)
 
187
            basis_inv = None
 
188
        else:
 
189
            fileid_map = self.mapping.get_fileid_map(store.__getitem__,
 
190
                store[head].tree)
 
191
        result = GitIndexInventory(basis_inv, fileid_map, self.index, store)
176
192
        self._set_inventory(result, dirty=False)
177
193
 
178
194
    @needs_read_lock
197
213
 
198
214
class GitWorkingTreeFormat(workingtree.WorkingTreeFormat):
199
215
 
 
216
    @property
 
217
    def _matchingbzrdir(self):
 
218
        from bzrlib.plugins.git import LocalGitControlDirFormat
 
219
        return LocalGitControlDirFormat()
 
220
 
200
221
    def get_format_description(self):
201
222
        return "Git Working Tree"
202
223
 
220
241
        changes = self._index.changes_from_tree(
221
242
            self.source._repository._git.object_store, self.source.tree, 
222
243
            want_unchanged=want_unchanged)
223
 
        ret = tree_delta_from_git_changes(changes, self.target.mapping, 
 
244
        source_fileid_map = self.source.mapping.get_fileid_map(
 
245
            self.source._repository._git.object_store.__getitem__,
 
246
            self.source.tree)
 
247
        if self.target.mapping.BZR_FILE_IDS_FILE is not None:
 
248
            file_id = self.target.path2id(
 
249
                self.target.mapping.BZR_FILE_IDS_FILE)
 
250
            if file_id is None:
 
251
                target_fileid_map = {}
 
252
            else:
 
253
                target_fileid_map = self.target.mapping.import_fileid_map(Blob.from_string(self.target.get_file_text(file_id)))
 
254
        else:
 
255
            target_fileid_map = {}
 
256
        target_fileid_map = GitFileIdMap(target_fileid_map, self.target.mapping)
 
257
        ret = tree_delta_from_git_changes(changes, self.target.mapping,
 
258
            (source_fileid_map, target_fileid_map),
224
259
            specific_file=specific_files, require_versioned=require_versioned)
225
260
        if want_unversioned:
226
261
            for e in self.target.extras():