1
# Copyright (C) 2005-2010 Canonical Ltd
1
# Copyright (C) 2005, 2006, 2007 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
5
5
# the Free Software Foundation; either version 2 of the License, or
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
from __future__ import absolute_import
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
17
__all__ = ['show_bzrdir_info']
21
from cStringIO import StringIO
25
23
from bzrlib import (
33
30
from bzrlib.errors import (NoWorkingTree, NotBranchError,
34
31
NoRepositoryPresent, NotLocalUrl)
35
32
from bzrlib.missing import find_unmerged
33
from bzrlib.symbol_versioning import (deprecated_function,
38
37
def plural(n, base='', pl=None):
79
78
return [" %*s: %s\n" % (max_len, l, u) for l, u in self.locs ]
82
def gather_location_info(repository=None, branch=None, working=None,
81
def gather_location_info(repository, branch=None, working=None):
83
repository_path = repository.bzrdir.root_transport.base
85
84
if branch is not None:
86
branch_path = branch.user_url
85
branch_path = branch.bzrdir.root_transport.base
87
86
master_path = branch.get_bound_location()
88
87
if master_path is None:
89
88
master_path = branch_path
94
if control is not None and control.get_branch_reference():
95
locs['checkout of branch'] = control.get_branch_reference()
96
except NotBranchError:
99
working_path = working.user_url
93
working_path = working.bzrdir.root_transport.base
100
94
if working_path != branch_path:
101
95
locs['light checkout root'] = working_path
102
96
if master_path != branch_path:
113
107
locs['branch root'] = branch_path
115
109
working_path = None
116
if repository is not None and repository.is_shared():
110
if repository.is_shared():
117
111
# lightweight checkout of branch in shared repository
118
112
if branch_path is not None:
119
113
locs['repository branch'] = branch_path
120
114
elif branch_path is not None:
122
116
locs['branch root'] = branch_path
123
elif repository is not None:
124
locs['repository'] = repository.user_url
125
elif control is not None:
126
locs['control directory'] = control.user_url
117
if master_path != branch_path:
118
locs['bound to branch'] = master_path
128
# Really, at least a control directory should be
129
# passed in for this method to be useful.
131
if master_path != branch_path:
132
locs['bound to branch'] = master_path
133
if repository is not None and repository.is_shared():
120
locs['repository'] = repository_path
121
if repository.is_shared():
134
122
# lightweight checkout of branch in shared repository
135
locs['shared repository'] = repository.user_url
136
order = ['control directory', 'light checkout root',
137
'repository checkout root', 'checkout root',
138
'checkout of branch', 'shared repository',
123
locs['shared repository'] = repository_path
124
order = ['light checkout root', 'repository checkout root',
125
'checkout root', 'checkout of branch', 'shared repository',
139
126
'repository', 'repository branch', 'branch root',
140
127
'bound to branch']
141
128
return [(n, locs[n]) for n in order if n in locs]
156
143
locs.add_url('push branch', branch.get_push_location())
157
144
locs.add_url('parent branch', branch.get_parent())
158
145
locs.add_url('submit branch', branch.get_submit_branch())
160
locs.add_url('stacked on', branch.get_stacked_on_url())
161
except (errors.UnstackableBranchFormat, errors.UnstackableRepositoryFormat,
173
155
outfile.writelines(locs.get_lines())
176
def _show_control_dir_info(control, outfile):
177
"""Show control dir information."""
178
if control._format.colocated_branches:
180
outfile.write('Control directory:\n')
181
outfile.write(' %d branches\n' % len(control.list_branches()))
184
158
def _show_format_info(control=None, repository=None, branch=None,
185
159
working=None, outfile=None):
186
160
"""Show known formats for control, working, branch and repository."""
200
174
repository._format.get_format_description())
203
def _show_locking_info(repository=None, branch=None, working=None,
177
def _show_locking_info(repository, branch=None, working=None, outfile=None):
205
178
"""Show locking status of working, branch and repository."""
206
if (repository and repository.get_physical_lock_status() or
179
if (repository.get_physical_lock_status() or
207
180
(branch and branch.get_physical_lock_status()) or
208
181
(working and working.get_physical_lock_status())):
209
182
outfile.write('\n')
245
218
"""Show missing revisions in working tree."""
246
219
branch = working.branch
247
220
basis = working.basis_tree()
249
branch_revno, branch_last_revision = branch.last_revision_info()
250
except errors.UnsupportedOperation:
221
work_inv = working.inventory
222
branch_revno, branch_last_revision = branch.last_revision_info()
253
224
tree_last_id = working.get_parent_ids()[0]
254
225
except IndexError:
265
236
def _show_working_stats(working, outfile):
266
237
"""Show statistics about a working tree."""
267
238
basis = working.basis_tree()
239
work_inv = working.inventory
268
240
delta = working.changes_from(basis, want_unchanged=True)
270
242
outfile.write('\n')
285
257
outfile.write(' %8d ignored\n' % ignore_cnt)
288
root_id = working.get_root_id()
289
for path, entry in working.iter_entries_by_dir():
290
if entry.kind == 'directory' and entry.file_id != root_id:
260
for file_id in work_inv:
261
if (work_inv.get_file_kind(file_id) == 'directory' and
262
not work_inv.is_root(file_id)):
292
264
outfile.write(' %8d versioned %s\n' % (dir_cnt,
293
265
plural(dir_cnt, 'subdirectory', 'subdirectories')))
296
268
def _show_branch_stats(branch, verbose, outfile):
297
269
"""Show statistics about a branch."""
299
revno, head = branch.last_revision_info()
300
except errors.UnsupportedOperation:
270
revno, head = branch.last_revision_info()
302
271
outfile.write('\n')
303
272
outfile.write('Branch history:\n')
304
273
outfile.write(' %8d revision%s\n' % (revno, plural(revno)))
327
296
'the repository.\n')
330
def _show_repository_stats(repository, stats, outfile):
299
def _show_repository_stats(stats, outfile):
331
300
"""Show statistics about a repository."""
301
if 'revisions' in stats or 'size' in stats:
303
outfile.write('Repository:\n')
333
304
if 'revisions' in stats:
334
305
revisions = stats['revisions']
335
f.write(' %8d revision%s\n' % (revisions, plural(revisions)))
306
outfile.write(' %8d revision%s\n' % (revisions, plural(revisions)))
336
307
if 'size' in stats:
337
f.write(' %8d KiB\n' % (stats['size']/1024))
338
for hook in hooks['repository']:
339
hook(repository, stats, f)
340
if f.getvalue() != "":
342
outfile.write('Repository:\n')
343
outfile.write(f.getvalue())
308
outfile.write(' %8d KiB\n' % (stats['size']/1024))
346
311
def show_bzrdir_info(a_bzrdir, verbose=False, outfile=None):
351
316
tree = a_bzrdir.open_workingtree(
352
317
recommend_upgrade=False)
353
except (NoWorkingTree, NotLocalUrl, NotBranchError):
318
except (NoWorkingTree, NotLocalUrl):
356
branch = a_bzrdir.open_branch(name="")
321
branch = a_bzrdir.open_branch()
357
322
except NotBranchError:
360
325
repository = a_bzrdir.open_repository()
361
326
except NoRepositoryPresent:
327
# Return silently; cmd_info already returned NotBranchError
328
# if no bzrdir could be opened.
365
331
lockable = repository
371
337
repository = branch.repository
374
if lockable is not None:
377
342
show_component_info(a_bzrdir, repository, branch, tree, verbose,
380
if lockable is not None:
384
348
def show_component_info(control, repository, branch=None, working=None,
391
355
if verbose is True:
393
layout = describe_layout(repository, branch, working, control)
357
layout = describe_layout(repository, branch, working)
394
358
format = describe_format(control, repository, branch, working)
395
359
outfile.write("%s (format: %s)\n" % (layout, format))
397
gather_location_info(control=control, repository=repository,
398
branch=branch, working=working),
360
_show_location_info(gather_location_info(repository, branch, working),
400
362
if branch is not None:
401
363
_show_related_info(branch, outfile)
404
366
_show_format_info(control, repository, branch, working, outfile)
405
367
_show_locking_info(repository, branch, working, outfile)
406
_show_control_dir_info(control, outfile)
407
368
if branch is not None:
408
369
_show_missing_revisions_branch(branch, outfile)
409
370
if working is not None:
412
373
elif branch is not None:
413
374
_show_missing_revisions_branch(branch, outfile)
414
375
if branch is not None:
415
show_committers = verbose >= 2
416
stats = _show_branch_stats(branch, show_committers, outfile)
417
elif repository is not None:
376
stats = _show_branch_stats(branch, verbose==2, outfile)
418
378
stats = repository.gather_stats()
419
if branch is None and working is None and repository is not None:
379
if branch is None and working is None:
420
380
_show_repository_info(repository, outfile)
421
if repository is not None:
422
_show_repository_stats(repository, stats, outfile)
425
def describe_layout(repository=None, branch=None, tree=None, control=None):
381
_show_repository_stats(stats, outfile)
384
def describe_layout(repository=None, branch=None, tree=None):
426
385
"""Convert a control directory layout into a user-understandable term
428
387
Common outputs include "Standalone tree", "Repository branch" and
429
388
"Checkout". Uncommon outputs include "Unshared repository with trees"
430
389
and "Empty control directory"
432
if branch is None and control is not None:
434
branch_reference = control.get_branch_reference()
435
except NotBranchError:
438
if branch_reference is not None:
439
return "Dangling branch reference"
440
391
if repository is None:
441
392
return 'Empty control directory'
442
393
if branch is None and tree is None:
444
395
phrase = 'Shared repository'
446
397
phrase = 'Unshared repository'
448
398
if repository.make_working_trees():
449
extra.append('trees')
450
if len(control.get_branches()) > 0:
451
extra.append('colocated branches')
453
phrase += ' with ' + " and ".join(extra)
399
phrase += ' with trees'
456
402
if repository.is_shared():
464
410
if branch is None and tree is not None:
465
411
phrase = "branchless tree"
467
if (tree is not None and tree.user_url !=
413
if (tree is not None and tree.bzrdir.root_transport.base !=
414
branch.bzrdir.root_transport.base):
469
415
independence = ''
470
416
phrase = "Lightweight checkout"
471
417
elif branch.get_bound_location() is not None:
492
438
if (branch is not None and tree is not None and
493
branch.user_url != tree.user_url):
439
branch.bzrdir.root_transport.base !=
440
tree.bzrdir.root_transport.base):
495
442
repository = None
496
non_aliases = set(controldir.format_registry.keys())
497
non_aliases.difference_update(controldir.format_registry.aliases())
498
for key in non_aliases:
499
format = controldir.format_registry.make_bzrdir(key)
443
for key in bzrdir.format_registry.keys():
444
format = bzrdir.format_registry.make_bzrdir(key)
500
445
if isinstance(format, bzrdir.BzrDirMetaFormat1):
501
446
if (tree and format.workingtree_format !=
512
457
candidates.append(key)
513
458
if len(candidates) == 0:
460
new_candidates = [c for c in candidates if c != 'default']
461
if len(new_candidates) > 0:
462
candidates = new_candidates
516
463
new_candidates = [c for c in candidates if not
517
controldir.format_registry.get_info(c).hidden]
464
bzrdir.format_registry.get_info(c).hidden]
518
465
if len(new_candidates) > 0:
519
# If there are any non-hidden formats that match, only return those to
520
# avoid listing hidden formats except when only a hidden format will
522
466
candidates = new_candidates
523
467
return ' or '.join(candidates)
526
class InfoHooks(_mod_hooks.Hooks):
527
"""Hooks for the info command."""
530
super(InfoHooks, self).__init__("bzrlib.info", "hooks")
531
self.add_hook('repository',
532
"Invoked when displaying the statistics for a repository. "
533
"repository is called with a statistics dictionary as returned "
534
"by the repository and a file-like object to write to.", (1, 15))