/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
 
50
53
    changes_from_git_changes,
51
54
    tree_delta_from_git_changes,
52
55
    )
53
 
 
 
56
from bzrlib.plugins.git.mapping import (
 
57
    GitFileIdMap,
 
58
    )
54
59
 
55
60
IGNORE_FILENAME = ".gitignore"
56
61
 
58
63
class GitWorkingTree(workingtree.WorkingTree):
59
64
    """A Git working tree."""
60
65
 
61
 
    def __init__(self, bzrdir, repo, branch):
 
66
    def __init__(self, bzrdir, repo, branch, index):
62
67
        self.basedir = bzrdir.root_transport.local_abspath('.')
63
68
        self.bzrdir = bzrdir
64
69
        self.repository = repo
77
82
        self._control_files = lockable_files.LockableFiles(
78
83
            transport.get_transport(self.controldir), 'lock', lockdir.LockDir)
79
84
        self._format = GitWorkingTreeFormat()
80
 
        self.index = self.repository._git.open_index()
 
85
        self.index = index
81
86
        self.views = self._make_views()
82
87
        self._detect_case_handling()
83
88
 
145
150
        # TODO: Maybe this should only write on dirty ?
146
151
        if self._control_files._lock_mode != 'w':
147
152
            raise errors.NotWriteLocked(self)
148
 
        self._rewrite_index()           
 
153
        self._rewrite_index()
149
154
        self.index.write()
150
155
        self._inventory_is_modified = False
151
156
 
171
176
 
172
177
    def _reset_data(self):
173
178
        self._inventory_is_modified = False
174
 
        basis_inv = self.repository.get_inventory(self.mapping.revision_id_foreign_to_bzr(self.repository._git.head()))
175
 
        result = GitIndexInventory(basis_inv, self.mapping, self.index,
176
 
            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)
177
192
        self._set_inventory(result, dirty=False)
178
193
 
179
194
    @needs_read_lock
200
215
 
201
216
    @property
202
217
    def _matchingbzrdir(self):
203
 
        from bzrlib.plugins.git import LocalGitBzrDirFormat
204
 
        return LocalGitBzrDirFormat()
 
218
        from bzrlib.plugins.git import LocalGitControlDirFormat
 
219
        return LocalGitControlDirFormat()
205
220
 
206
221
    def get_format_description(self):
207
222
        return "Git Working Tree"
226
241
        changes = self._index.changes_from_tree(
227
242
            self.source._repository._git.object_store, self.source.tree, 
228
243
            want_unchanged=want_unchanged)
229
 
        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),
230
259
            specific_file=specific_files, require_versioned=require_versioned)
231
260
        if want_unversioned:
232
261
            for e in self.target.extras():