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
3
from bzrlib.bzrdir import BzrDir
4
from bzrlib.errors import NotBranchError
5
from bzrlib.workingtree import WorkingTree
37
7
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."""
10
from bzrlib.plugins.gtk import cmd_visualise, cmd_gannotate
12
class BzrExtension(nautilus.MenuProvider):
46
13
def __init__(self):
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):
16
def add_cb(self, menu, vfs_file):
17
# We can only cope with local files
18
if vfs_file.get_uri_scheme() != 'file':
21
file = vfs_file.get_uri()
23
tree, path = WorkingTree.open_containing(file)
24
except NotBranchError:
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
31
def ignore_cb(self, menu, vfs_file):
32
# We can only cope with local files
33
if vfs_file.get_uri_scheme() != 'file':
36
file = vfs_file.get_uri()
38
tree, path = WorkingTree.open_containing(file)
39
except NotBranchError:
46
def unignore_cb(self, menu, vfs_file):
47
# We can only cope with local files
48
if vfs_file.get_uri_scheme() != 'file':
51
file = vfs_file.get_uri()
53
tree, path = WorkingTree.open_containing(file)
54
except NotBranchError:
61
def diff_cb(self, menu, vfs_file):
62
# We can only cope with local files
63
if vfs_file.get_uri_scheme() != 'file':
66
file = vfs_file.get_uri()
68
tree, path = WorkingTree.open_containing(file)
69
except NotBranchError:
72
from bzrlib.plugins.gtk.viz.diffwin import DiffWindow
75
73
window = DiffWindow()
76
window.set_diff(tree.branch._get_nick(local=True), tree,
77
tree.branch.basis_tree())
74
window.set_diff(tree.branch, tree, tree.branch.revision_tree())
80
79
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):
80
# We can only cope with local files
81
if vfs_file.get_uri_scheme() != 'file':
84
file = vfs_file.get_uri()
86
# We only want to continue here if we get a NotBranchError
88
tree, path = WorkingTree.open_containing(file)
89
except NotBranchError:
90
BzrDir.create_branch_and_repo(file)
92
def remove_cb(self, menu, vfs_file):
93
# We can only cope with local files
94
if vfs_file.get_uri_scheme() != 'file':
97
file = vfs_file.get_uri()
99
tree, path = WorkingTree.open_containing(file)
100
except NotBranchError:
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)
105
def annotate_cb(self, menu, vfs_file):
106
# We can only cope with local files
107
if vfs_file.get_uri_scheme() != 'file':
110
file = vfs_file.get_uri()
112
vis = cmd_gannotate()
94
115
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()
116
# We can only cope with local files
117
if vfs_file.get_uri_scheme() != 'file':
120
file = vfs_file.get_uri()
122
tree, path = WorkingTree.open_containing(file)
123
except NotBranchError:
126
from bzrlib.plugins.gtk.clone import CloneDialog
127
dialog = CloneDialog(file)
128
if dialog.run() != gtk.RESPONSE_CANCEL:
129
bzrdir = BzrDir.open(dialog.url)
130
bzrdir.sprout(dialog.dest_path)
132
def commit_cb(self, menu, vfs_file=None):
133
# We can only cope with local files
134
if vfs_file.get_uri_scheme() != 'file':
137
file = vfs_file.get_uri()
139
tree, path = WorkingTree.open_containing(file)
140
except NotBranchError:
143
from bzrlib.plugins.gtk.commit import GCommitDialog
144
dialog = GCommitDialog(tree)
145
dialog.set_title(path + " - Commit")
146
if dialog.run() != gtk.RESPONSE_CANCEL:
147
Commit().commit(working_tree=wt,message=dialog.message,
148
specific_files=dialog.specific_files)
150
def log_cb(self, menu, vfs_file):
151
# We can only cope with local files
152
if vfs_file.get_uri_scheme() != 'file':
155
file = vfs_file.get_uri()
157
# We only want to continue here if we get a NotBranchError
159
tree, path = WorkingTree.open_containing(file)
160
except NotBranchError:
163
vis = cmd_visualise()
134
168
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',
169
file = vfs_file.get_uri()
171
tree, path = WorkingTree.open_containing(file)
172
except NotBranchError:
173
item = nautilus.MenuItem('BzrNautilus::newtree',
174
'Create new Bazaar tree',
175
'Create new Bazaar tree in this folder')
147
176
item.connect('activate', self.newtree_cb, vfs_file)
148
177
items.append(item)
150
item = Nautilus.MenuItem(name='BzrNautilus::clone',
151
label='Checkout Bazaar branch ...',
152
tip='Checkout Existing Bazaar Branch',
179
item = nautilus.MenuItem('BzrNautilus::clone',
181
'Checkout Existing Bazaar Branch')
154
182
item.connect('activate', self.clone_cb, vfs_file)
155
183
items.append(item)
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)
188
item = nautilus.MenuItem('BzrNautilus::log',
190
'Show Bazaar history')
191
item.connect('activate', self.log_cb, vfs_file)
194
item = nautilus.MenuItem('BzrNautilus::commit',
197
item.connect('activate', self.commit_cb, vfs_file)
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
203
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:
206
for vfs_file in files:
207
# We can only cope with local files
208
if vfs_file.get_uri_scheme() != 'file':
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)
211
file = vfs_file.get_uri()
213
tree, path = WorkingTree.open_containing(file)
399
214
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)
215
if not vfs_file.is_directory():
217
item = nautilus.MenuItem('BzrNautilus::newtree',
218
'Create new Bazaar tree',
219
'Create new Bazaar tree in %s' % vfs_file.get_name())
220
item.connect('activate', self.newtree_cb, vfs_file)
223
file_class = tree.file_class(path)
225
if file_class == '?':
226
item = nautilus.MenuItem('BzrNautilus::add',
228
'Add as versioned file')
229
item.connect('activate', self.add_cb, vfs_file)
232
item = nautilus.MenuItem('BzrNautilus::ignore',
234
'Ignore file for versioning')
235
item.connect('activate', self.ignore_cb, vfs_file)
237
elif file_class == 'I':
238
item = nautilus.MenuItem('BzrNautilus::unignore',
240
'Unignore file for versioning')
241
item.connect('activate', self.unignore_cb, vfs_file)
243
elif file_class == 'V':
244
item = nautilus.MenuItem('BzrNautilus::log',
247
item.connect('activate', self.log_cb, vfs_file)
250
item = nautilus.MenuItem('BzrNautilus::diff',
253
item.connect('activate', self.diff_cb, vfs_file)
256
item = nautilus.MenuItem('BzrNautilus::remove',
258
'Remove this file from versioning')
259
item.connect('activate', self.remove_cb, vfs_file)
262
item = nautilus.MenuItem('BzrNautilus::annotate',
264
'Annotate File Data')
265
item.connect('activate', self.annotate_cb, vfs_file)
268
item = nautilus.MenuItem('BzrNautilus::commit',
271
item.connect('activate', self.commit_cb, vfs_file)