1
# Trivial Bazaar plugin for Nautilus
3
# Copyright (C) 2006 Jeff Bailey
4
# Copyright (C) 2006 Wouter van Heyst
5
# Copyright (C) 2006-2011 Jelmer Vernooij <jelmer@samba.org>
7
# This program is free software; you can redistribute it and/or modify
8
# it under the terms of the GNU General Public License as published by
9
# the Free Software Foundation; either version 3 of the License, or
10
# (at your option) any later version.
12
# This program is distributed in the hope that it will be useful,
13
# but WITHOUT ANY WARRANTY; without even the implied warranty of
14
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15
# GNU General Public License for more details.
17
# You should have received a copy of the GNU General Public License
18
# along with this program; if not, write to the Free Software
19
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22
# setup.py can install nautilus-bzr to the right system folder, if pkg-config
25
# You can also install nautilus-bzr manually by copying it (or linking it from)
26
# ~/.local/share/nautilus-python/extensions/nautilus-bzr.py
28
from gi.repository import Gtk, GObject, Nautilus
29
from bzrlib.controldir import ControlDir
30
from bzrlib.errors import (
34
from bzrlib.ignores import tree_ignores_add_patterns
35
from bzrlib.tree import InterTree
37
from bzrlib.plugin import load_plugins
41
class BazaarExtension(Nautilus.MenuProvider, Nautilus.ColumnProvider,
42
Nautilus.InfoProvider, Nautilus.PropertyPageProvider,
43
Nautilus.LocationWidgetProvider, GObject.GObject):
44
"""Nautilus extension providing Bazaar integration."""
50
def _open_bzrdir(cls, vfs_file):
51
uri = vfs_file.get_uri()
52
controldir, path = ControlDir.open_containing(uri)
53
return controldir, path
56
def _open_tree(cls, vfs_file):
57
controldir, path = cls._open_bzrdir(vfs_file)
58
return controldir.open_workingtree(), path
60
def add_cb(self, menu, tree, path):
63
def ignore_cb(self, menu, tree, path):
64
# We can only cope with local files
65
tree_ignores_add_patterns(tree, [path])
66
#FIXME: Add path to ignore file
68
def unignore_cb(self, menu, tree, path):
70
# We can only cope with local files
73
def diff_cb(self, menu, tree, path=None):
74
from bzrlib.plugins.gtk.diff import DiffWindow
76
window.set_diff(tree.branch._get_nick(local=True), tree,
77
tree.branch.basis_tree())
80
def newtree_cb(self, menu, vfs_file):
81
controldir, path = self._open_bzrdir(vfs_file)
82
controldir.create_workingtree()
84
def remove_cb(self, menu, tree, path):
87
def annotate_cb(self, menu, tree, path, file_id):
88
from bzrlib.plugins.gtk.annotate.gannotate import GAnnotateWindow
89
win = GAnnotateWindow()
91
win.annotate(tree, tree.branch, file_id)
94
def clone_cb(self, menu, vfs_file=None):
95
from bzrlib.plugins.gtk.branch import BranchDialog
96
controldir, path = self._open_bzrdir(vfs_file)
98
dialog = BranchDialog(vfs_file.get_name())
99
response = dialog.run()
100
if response != Gtk.ResponseType.NONE:
104
def commit_cb(self, menu, tree, path=None):
105
from bzrlib.plugins.gtk.commit import CommitDialog
106
dialog = CommitDialog(tree, path)
107
response = dialog.run()
108
if response != Gtk.ResponseType.NONE:
112
def log_cb(self, menu, controldir, path=None):
113
from bzrlib.plugins.gtk.viz import BranchWindow
114
branch = controldir.open_branch()
115
pp = BranchWindow(branch, [branch.last_revision()], None)
119
def pull_cb(self, menu, controldir, path=None):
120
from bzrlib.plugins.gtk.pull import PullDialog
121
dialog = PullDialog(controldir.open_workingtree(), path)
125
def merge_cb(self, menu, tree, path=None):
126
from bzrlib.plugins.gtk.merge import MergeDialog
127
dialog = MergeDialog(tree, path)
131
def create_tree_cb(self, menu, controldir):
132
controldir.create_workingtree()
134
def get_background_items(self, window, vfs_file):
136
controldir, path = self._open_bzrdir(vfs_file)
137
except NotBranchError:
140
branch = controldir.open_branch()
141
except NotBranchError:
143
item = Nautilus.MenuItem(name='BzrNautilus::newtree',
144
label='Make directory versioned',
145
tip='Create new Bazaar tree in this folder',
147
item.connect('activate', self.newtree_cb, vfs_file)
150
item = Nautilus.MenuItem(name='BzrNautilus::clone',
151
label='Checkout Bazaar branch ...',
152
tip='Checkout Existing Bazaar Branch',
154
item.connect('activate', self.clone_cb, vfs_file)
160
nautilus_integration = self.check_branch_enabled(branch)
161
if not nautilus_integration:
162
item = Nautilus.MenuItem(name='BzrNautilus::enable',
163
label='Enable Bazaar Plugin for this Branch',
164
tip='Enable Bazaar plugin for nautilus',
166
item.connect('activate', self.toggle_integration, True, branch)
169
item = Nautilus.MenuItem(name='BzrNautilus::disable',
170
label='Disable Bazaar Plugin this Branch',
171
tip='Disable Bazaar plugin for nautilus',
173
item.connect('activate', self.toggle_integration, False, branch)
176
item = Nautilus.MenuItem(name='BzrNautilus::log',
178
tip='Show Bazaar history',
180
item.connect('activate', self.log_cb, controldir)
183
item = Nautilus.MenuItem(name='BzrNautilus::pull',
185
tip='Pull from another branch',
187
item.connect('activate', self.pull_cb, controldir)
191
tree = controldir.open_workingtree()
192
except NoWorkingTree:
193
item = Nautilus.MenuItem(name='BzrNautilus::create_tree',
194
label='Create working tree...',
195
tip='Create a working tree for this branch',
197
item.connect('activate', self.create_tree_cb, controldir)
200
item = Nautilus.MenuItem(name='BzrNautilus::merge',
202
tip='Merge from another branch',
204
item.connect('activate', self.merge_cb, tree, path)
207
item = Nautilus.MenuItem(name='BzrNautilus::commit',
209
tip='Commit Changes',
211
item.connect('activate', self.commit_cb, tree, path)
216
def _get_file_menuitems(self, tree, intertree, path):
217
file_id = tree.path2id(path)
219
item = Nautilus.MenuItem(name='BzrNautilus::add',
221
tip='Add as versioned file',
223
item.connect('activate', self.add_cb, tree, path)
226
item = Nautilus.MenuItem(name='BzrNautilus::ignore',
228
tip='Ignore file for versioning',
230
item.connect('activate', self.ignore_cb, tree, path)
232
elif tree.is_ignored(path):
233
item = Nautilus.MenuItem(name='BzrNautilus::unignore',
235
tip='Unignore file for versioning',
237
item.connect('activate', self.unignore_cb, tree, path)
240
item = Nautilus.MenuItem(name='BzrNautilus::log',
244
item.connect('activate', self.log_cb, tree.bzrdir, path)
247
if not intertree.file_content_matches(file_id, file_id):
248
item = Nautilus.MenuItem(name='BzrNautilus::diff',
249
label='View Changes ...',
250
tip='Show differences',
252
item.connect('activate', self.diff_cb, tree, path)
255
item = Nautilus.MenuItem(name='BzrNautilus::commit',
257
tip='Commit Changes',
259
item.connect('activate', self.commit_cb, tree, path)
262
item = Nautilus.MenuItem(name='BzrNautilus::remove',
264
tip='Remove this file from versioning',
266
item.connect('activate', self.remove_cb, tree, path)
269
item = Nautilus.MenuItem(name='BzrNautilus::annotate',
270
label='Annotate ...',
271
tip='Annotate File Data',
273
item.connect('activate', self.annotate_cb, tree, path, file_id)
276
def get_file_items(self, window, files):
281
for vfs_file in files:
282
controldir, path = self._open_bzrdir(vfs_file)
285
tree = trees[controldir.user_url]
288
tree = controldir.open_workingtree()
289
except NoWorkingTree:
291
trees[controldir.user_url] = tree
294
nautilus_integration = self.check_branch_enabled(tree.branch)
295
if not nautilus_integration:
298
intertree = InterTree.get(tree.basis_tree(), tree)
299
items.extend(list(self._get_file_menuitems(tree, intertree, path)))
301
for tree in trees.itervalues():
306
def get_columns(self):
308
Nautilus.Column(name="BzrNautilus::bzr_status",
309
attribute="bzr_status",
311
description="Version control status"),
312
Nautilus.Column(name="BzrNautilus::bzr_revision",
313
attribute="bzr_revision",
315
description="Last change revision"),
318
def _file_summary(self, tree, basis_tree, intertree, path):
322
file_id = tree.path2id(path)
324
if tree.is_ignored(path):
326
emblem = 'bzr-ignored'
328
status = 'unversioned'
329
file_revision = "N/A"
330
elif tree.has_filename(path): # Still present
331
if not intertree.file_content_matches(file_id, file_id):
332
if not basis_tree.has_id(file_id):
335
file_revision = "new file"
336
elif basis_tree.path2id(file_id) != path:
337
status = 'bzr-renamed'
338
status = 'renamed from %s' % basis_tree.path2id(file_id)
340
emblem = 'bzr-modified'
343
emblem = 'bzr-controlled'
345
elif basis_tree.has_filename(path):
346
emblem = 'bzr-removed'
349
# FIXME: Check for ignored files
350
status = 'unversioned'
351
return (status, emblem, file_revision)
353
def update_file_info(self, vfs_file):
355
controldir, path = self._open_bzrdir(vfs_file)
356
except NotBranchError:
360
tree = controldir.open_workingtree()
361
except NoWorkingTree:
366
nautilus_integration = self.check_branch_enabled(tree.branch)
367
if not nautilus_integration:
370
basis_tree = tree.basis_tree()
371
intertree = InterTree.get(basis_tree, tree)
373
basis_tree.lock_read()
375
(status, emblem, file_revision) = self._file_summary(tree, basis_tree, intertree, path)
378
if emblem is not None:
379
vfs_file.add_emblem(emblem)
380
vfs_file.add_string_attribute('bzr_status', status)
381
vfs_file.add_string_attribute('bzr_revision', file_revision)
385
def check_branch_enabled(self, branch):
386
# Supports global disable, but there is currently no UI to do this
387
config = branch.get_config_stack()
388
return config.get("nautilus_integration")
390
def toggle_integration(self, menu, action, branch):
391
config = branch.get_config_stack()
392
config.set("nautilus_integration", action)
394
def get_property_pages(self, files):
396
for vfs_file in files:
398
controldir, path = self._open_bzrdir(vfs_file)
399
except NotBranchError:
403
tree = controldir.open_workingtree()
404
except NoWorkingTree:
409
file_id = tree.path2id(path)
410
pages.append(PropertyPageFile(tree, file_id, path))
411
pages.append(PropertyPageBranch(tree.branch))
416
def get_widget(self, uri, window):
417
controldir, path = ControlDir.open_containing(uri)
419
tree = controldir.open_workingtree()
420
except NoWorkingTree:
422
ret = Gtk.HBox(False, 4)
423
text = 'This is a Bazaar working tree. '
424
get_shelf_manager = getattr(tree, 'get_shelf_manager', None)
425
if get_shelf_manager is not None:
426
manager = get_shelf_manager()
427
shelves = manager.active_shelves()
428
if len(shelves) == 0:
430
elif len(shelves) == 1:
431
text += '1 shelf exists. '
433
text += '%d shelf exists. ' % len(shelves)
434
label = Gtk.Label(text)
436
ret.pack_start(label, True, True, 0)
441
class PropertyPageFile(Nautilus.PropertyPage):
443
def __init__(self, tree, file_id, path):
445
self.file_id = file_id
447
label = Gtk.Label('File Version')
450
table = self._create_table()
452
super(PropertyPageFile, self).__init__(label=label,
453
name="BzrNautilus::file_page", page=table)
455
def _create_table(self):
456
table = Gtk.Table(homogeneous=False, columns=2, rows=3)
458
table.attach(Gtk.Label('File id:'), 0, 1, 0, 1)
459
table.attach(Gtk.Label(self.file_id), 1, 2, 0, 1)
461
table.attach(Gtk.Label('SHA1Sum:'), 0, 1, 1, 2)
462
table.attach(Gtk.Label(self.tree.get_file_sha1(self.file_id, self.path)), 1, 1, 1, 2)
464
basis_tree = self.tree.revision_tree(self.tree.last_revision())
465
last_revision = basis_tree.get_file_revision(self.file_id)
467
table.attach(Gtk.Label('Last Change Revision:'), 0, 1, 2, 3)
468
revno = ".".join([str(x) for x in
469
self.tree.branch.revision_id_to_dotted_revno(last_revision)])
470
table.attach(Gtk.Label(revno), 1, 1, 2, 3)
472
table.attach(Gtk.Label('Last Change Author:'), 0, 1, 3, 4)
473
rev = self.tree.branch.repository.get_revision(last_revision)
474
table.attach(Gtk.Label("\n".join(rev.get_apparent_authors())), 1, 1, 3, 4)
480
class PropertyPageBranch(Nautilus.PropertyPage):
482
def __init__(self, branch):
484
label = Gtk.Label('Branch')
487
table = self._create_table()
489
super(PropertyPageBranch, self).__init__(label=label,
490
name="BzrNautilus::branch_page", page=table)
492
def _create_location_entry(self, get_location, set_location):
493
location = get_location()
495
if location is not None:
496
ret.set_text(location)
499
def _create_table(self):
500
table = Gtk.Table(homogeneous=False, columns=2, rows=6)
502
self._push_location_entry = self._create_location_entry(
503
self.branch.get_push_location, self.branch.set_push_location)
504
self._parent_location_entry = self._create_location_entry(
505
self.branch.get_parent, self.branch.set_parent)
506
self._bound_location_entry = self._create_location_entry(
507
self.branch.get_bound_location, self.branch.set_bound_location)
508
self._public_location_entry = self._create_location_entry(
509
self.branch.get_public_branch, self.branch.set_public_branch)
510
self._submit_location_entry = self._create_location_entry(
511
self.branch.get_submit_branch, self.branch.set_submit_branch)
513
table.attach(Gtk.Label('Push location:'), 0, 1, 0, 1)
514
table.attach(self._push_location_entry, 1, 2, 0, 1)
516
table.attach(Gtk.Label('Parent location:'), 0, 1, 1, 2)
517
table.attach(self._parent_location_entry, 1, 1, 1, 2)
519
table.attach(Gtk.Label('Bound location:'), 0, 1, 2, 3)
520
table.attach(self._bound_location_entry, 1, 1, 2, 3)
522
table.attach(Gtk.Label('Public location:'), 0, 1, 3, 4)
523
table.attach(self._public_location_entry, 1, 1, 3, 4)
525
table.attach(Gtk.Label('Submit location:'), 0, 1, 4, 5)
526
table.attach(self._submit_location_entry, 1, 1, 4, 5)
528
self._append_revisions_only = Gtk.CheckButton('Append revisions only')
529
value = self.branch.get_append_revisions_only()
532
self._append_revisions_only.set_active(value)
533
table.attach(self._append_revisions_only, 0, 2, 5, 6)