1
# Copyright (C) 2007-2010 Canonical Ltd
1
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
29
29
from bzrlib.lazy_import import lazy_import
30
30
lazy_import(globals(), """
31
from bisect import bisect_left
33
from copy import deepcopy
35
42
from bzrlib import (
45
conflicts as _mod_conflicts,
43
55
revision as _mod_revision,
49
65
import bzrlib.branch
66
from bzrlib.transport import get_transport
70
from bzrlib import symbol_versioning
53
71
from bzrlib.decorators import needs_read_lock, needs_write_lock
54
72
from bzrlib.filters import filtered_input_file, internal_size_sha_file_byname
55
from bzrlib.inventory import Inventory, ROOT_ID, entry_factory
73
from bzrlib.inventory import InventoryEntry, Inventory, ROOT_ID, entry_factory
74
import bzrlib.mutabletree
56
75
from bzrlib.mutabletree import needs_tree_write_lock
57
76
from bzrlib.osutils import (
64
from bzrlib.trace import mutter
86
from bzrlib.trace import mutter, note
65
87
from bzrlib.transport.local import LocalTransport
66
88
from bzrlib.tree import InterTree
89
from bzrlib.progress import DummyProgress, ProgressPhase
90
from bzrlib.revision import NULL_REVISION, CURRENT_REVISION
91
from bzrlib.rio import RioReader, rio_file, Stanza
92
from bzrlib.symbol_versioning import (deprecated_passed,
67
97
from bzrlib.tree import Tree
68
98
from bzrlib.workingtree import WorkingTree, WorkingTree3, WorkingTreeFormat3
434
464
return osutils.lexists(pathjoin(
435
465
self.basedir, row[0].decode('utf8'), row[1].decode('utf8')))
437
def has_or_had_id(self, file_id):
438
state = self.current_dirstate()
439
row, parents = self._get_entry(file_id=file_id)
440
return row is not None
443
468
def id2path(self, file_id):
444
469
"Convert a file-id to a path."
691
716
from_entry = self._get_entry(path=from_rel)
692
717
if from_entry == (None, None):
693
718
raise errors.BzrMoveFailedError(from_rel,to_dir,
694
errors.NotVersionedError(path=from_rel))
719
errors.NotVersionedError(path=str(from_rel)))
696
721
from_id = from_entry[0][2]
697
722
to_rel = pathjoin(to_dir, from_tail)
1026
1051
def set_last_revision(self, new_revision):
1027
1052
"""Change the last revision in the working tree."""
1028
1053
parents = self.get_parent_ids()
1029
if new_revision in (_mod_revision.NULL_REVISION, None):
1054
if new_revision in (NULL_REVISION, None):
1030
1055
if len(parents) >= 2:
1031
1056
raise AssertionError(
1032
1057
"setting the last parent to none with a pending merge is "
1199
1224
# just forget the whole block.
1200
1225
entry_index = 0
1201
1226
while entry_index < len(block[1]):
1227
# Mark this file id as having been removed
1202
1228
entry = block[1][entry_index]
1203
if entry[1][0][0] in 'ar':
1204
# don't remove absent or renamed entries
1229
ids_to_unversion.discard(entry[0][2])
1230
if (entry[1][0][0] in 'ar' # don't remove absent or renamed
1232
or not state._make_absent(entry)):
1205
1233
entry_index += 1
1207
# Mark this file id as having been removed
1208
ids_to_unversion.discard(entry[0][2])
1209
if not state._make_absent(entry):
1210
# The block has not shrunk.
1212
1234
# go to the next block. (At the moment we dont delete empty
1214
1236
block_index += 1
1266
1288
if self._dirty:
1267
1289
raise AssertionError("attempting to write an inventory when the "
1268
1290
"dirstate is dirty will lose pending changes")
1269
had_inventory = self._inventory is not None
1270
# Setting self._inventory = None forces the dirstate to regenerate the
1271
# working inventory. We do this because self.inventory may be inv, or
1272
# may have been modified, and either case would prevent a clean delta
1274
self._inventory = None
1276
delta = inv._make_delta(self.inventory)
1278
self.apply_inventory_delta(delta)
1291
self.current_dirstate().set_state_from_inventory(inv)
1292
self._make_dirty(reset_inventory=False)
1293
if self._inventory is not None:
1280
1294
self._inventory = inv
1307
1321
return statvalue, sha1
1310
class ContentFilteringDirStateWorkingTree(DirStateWorkingTree):
1311
"""Dirstate working tree that supports content filtering.
1313
The dirstate holds the hash and size of the canonical form of the file,
1314
and most methods must return that.
1317
def _file_content_summary(self, path, stat_result):
1318
# This is to support the somewhat obsolete path_content_summary method
1319
# with content filtering: see
1320
# <https://bugs.edge.launchpad.net/bzr/+bug/415508>.
1322
# If the dirstate cache is up to date and knows the hash and size,
1324
# Otherwise if there are no content filters, return the on-disk size
1325
# and leave the hash blank.
1326
# Otherwise, read and filter the on-disk file and use its size and
1329
# The dirstate doesn't store the size of the canonical form so we
1330
# can't trust it for content-filtered trees. We just return None.
1331
dirstate_sha1 = self._dirstate.sha1_from_stat(path, stat_result)
1332
executable = self._is_executable_from_path_and_stat(path, stat_result)
1333
return ('file', None, executable, dirstate_sha1)
1336
1324
class WorkingTree4(DirStateWorkingTree):
1337
1325
"""This is the Format 4 working tree.
1373
1361
class DirStateWorkingTreeFormat(WorkingTreeFormat3):
1375
1362
def initialize(self, a_bzrdir, revision_id=None, from_branch=None,
1376
1363
accelerator_tree=None, hardlink=False):
1377
1364
"""See WorkingTreeFormat.initialize().
1416
1403
wt.lock_tree_write()
1418
1405
self._init_custom_control_files(wt)
1419
if revision_id in (None, _mod_revision.NULL_REVISION):
1406
if revision_id in (None, NULL_REVISION):
1420
1407
if branch.repository.supports_rich_root():
1421
1408
wt._set_root_id(generate_ids.gen_root_id())
1434
1421
if basis is None:
1435
1422
basis = branch.repository.revision_tree(revision_id)
1436
if revision_id == _mod_revision.NULL_REVISION:
1423
if revision_id == NULL_REVISION:
1437
1424
parents_list = []
1439
1426
parents_list = [(revision_id, basis)]
1447
1434
if basis_root_id is not None:
1448
1435
wt._set_root_id(basis_root_id)
1437
# If content filtering is supported, do not use the accelerator
1438
# tree - the cost of transforming the content both ways and
1439
# checking for changed content can outweight the gains it gives.
1440
# Note: do NOT move this logic up higher - using the basis from
1441
# the accelerator tree is still desirable because that can save
1442
# a minute or more of processing on large trees!
1443
# The original tree may not have the same content filters
1444
# applied so we can't safely build the inventory delta from
1450
1446
if wt.supports_content_filtering():
1451
# The original tree may not have the same content filters
1452
# applied so we can't safely build the inventory delta from
1447
accelerator_tree = None
1454
1448
delta_from_tree = False
1456
1450
delta_from_tree = True
1575
1569
class DirStateRevisionTree(Tree):
1576
"""A revision tree pulling the inventory from a dirstate.
1578
Note that this is one of the historical (ie revision) trees cached in the
1579
dirstate for easy access, not the workingtree.
1570
"""A revision tree pulling the inventory from a dirstate."""
1582
1572
def __init__(self, dirstate, revision_id, repository):
1583
1573
self._dirstate = dirstate
1756
1746
parent_index = self._get_parent_index()
1757
1747
last_changed_revision = entry[1][parent_index][4]
1759
rev = self._repository.get_revision(last_changed_revision)
1760
except errors.NoSuchRevision:
1761
raise errors.FileTimestampUnavailable(self.id2path(file_id))
1762
return rev.timestamp
1748
return self._repository.get_revision(last_changed_revision).timestamp
1764
1750
def get_file_sha1(self, file_id, path=None, stat_value=None):
1765
1751
entry = self._get_entry(file_id=file_id, path=path)
1832
1818
entry = self._get_entry(file_id=file_id)[1]
1833
1819
if entry is None:
1834
1820
raise errors.NoSuchId(tree=self, file_id=file_id)
1835
parent_index = self._get_parent_index()
1836
return dirstate.DirState._minikind_to_kind[entry[parent_index][0]]
1821
return dirstate.DirState._minikind_to_kind[entry[1][0]]
1838
1823
def stored_kind(self, file_id):
1839
1824
"""See Tree.stored_kind"""
1860
1845
return ie.executable
1862
def list_files(self, include_root=False, from_dir=None, recursive=True):
1847
def list_files(self, include_root=False):
1863
1848
# We use a standard implementation, because DirStateRevisionTree is
1864
1849
# dealing with one of the parents of the current state
1865
1850
inv = self._get_inventory()
1866
if from_dir is None:
1869
from_dir_id = inv.path2id(from_dir)
1870
if from_dir_id is None:
1871
# Directory not versioned
1873
entries = inv.iter_entries(from_dir=from_dir_id, recursive=recursive)
1874
if inv.root is not None and not include_root and from_dir is None:
1851
entries = inv.iter_entries()
1852
if self.inventory.root is not None and not include_root:
1876
1854
for path, entry in entries:
1877
1855
yield path, 'V', entry.kind, entry.file_id, entry
1982
def make_source_parent_tree_compiled_dirstate(klass, test_case, source,
1960
def make_source_parent_tree_compiled_dirstate(klass, test_case, source, target):
1984
1961
from bzrlib.tests.test__dirstate_helpers import \
1985
compiled_dirstate_helpers_feature
1986
test_case.requireFeature(compiled_dirstate_helpers_feature)
1987
from bzrlib._dirstate_helpers_pyx import ProcessEntryC
1962
CompiledDirstateHelpersFeature
1963
if not CompiledDirstateHelpersFeature.available():
1964
from bzrlib.tests import UnavailableFeature
1965
raise UnavailableFeature(CompiledDirstateHelpersFeature)
1966
from bzrlib._dirstate_helpers_c import ProcessEntryC
1988
1967
result = klass.make_source_parent_tree(source, target)
1989
1968
result[1]._iter_changes = ProcessEntryC
2020
1999
output. An unversioned file is defined as one with (False, False)
2021
2000
for the versioned pair.
2002
# NB: show_status depends on being able to pass in non-versioned files
2003
# and report them as unknown
2023
2004
# TODO: handle extra trees in the dirstate.
2024
2005
if (extra_trees or specific_files == []):
2025
2006
# we can't fast-path these cases (yet)
2028
2009
require_versioned, want_unversioned=want_unversioned)
2029
2010
parent_ids = self.target.get_parent_ids()
2030
2011
if not (self.source._revision_id in parent_ids
2031
or self.source._revision_id == _mod_revision.NULL_REVISION):
2012
or self.source._revision_id == NULL_REVISION):
2032
2013
raise AssertionError(
2033
2014
"revision {%s} is not stored in {%s}, but %s "
2034
2015
"can only be used for trees stored in the dirstate"
2035
2016
% (self.source._revision_id, self.target, self.iter_changes))
2036
2017
target_index = 0
2037
if self.source._revision_id == _mod_revision.NULL_REVISION:
2018
if self.source._revision_id == NULL_REVISION:
2038
2019
source_index = None
2039
2020
indices = (target_index,)
2061
2042
state._read_dirblocks_if_needed()
2062
2043
if require_versioned:
2063
2044
# -- check all supplied paths are versioned in a search tree. --
2045
all_versioned = True
2065
2046
for path in specific_files:
2066
2047
path_entries = state._entries_for_path(path)
2067
2048
if not path_entries:
2068
2049
# this specified path is not present at all: error
2069
not_versioned.append(path)
2050
all_versioned = False
2071
2052
found_versioned = False
2072
2053
# for each id at this path
2073
2054
for entry in path_entries:
2080
2061
if not found_versioned:
2081
2062
# none of the indexes was not 'absent' at all ids for this
2083
not_versioned.append(path)
2084
if len(not_versioned) > 0:
2085
raise errors.PathsNotVersionedError(not_versioned)
2064
all_versioned = False
2066
if not all_versioned:
2067
raise errors.PathsNotVersionedError(specific_files)
2086
2068
# -- remove redundancy in supplied specific_files to prevent over-scanning --
2087
2069
search_specific_files = osutils.minimum_path_selection(specific_files)
2102
2084
(revisiontree.RevisionTree, DirStateRevisionTree)):
2104
2086
# the source revid must be in the target dirstate
2105
if not (source._revision_id == _mod_revision.NULL_REVISION or
2087
if not (source._revision_id == NULL_REVISION or
2106
2088
source._revision_id in target.get_parent_ids()):
2107
2089
# TODO: what about ghosts? it may well need to
2108
2090
# check for them explicitly.
added
removed
bzrlib/workingtree_4.py
