1
# Trivial Bazaar plugin for Nautilus
3
# Copyright (C) 2006 Jeff Bailey
4
# Copyright (C) 2006 Wouter van Heyst
5
# Copyright (C) 2006-2008 Jelmer Vernooij <jelmer@samba.org>
7
# Published under the GNU GPL
12
from bzrlib.branch import Branch
13
from bzrlib.bzrdir import BzrDir
14
from bzrlib.errors import NotBranchError, NoWorkingTree, UnsupportedProtocol
15
from bzrlib.tree import file_status
16
from bzrlib.workingtree import WorkingTree
17
from bzrlib.config import GlobalConfig
19
from bzrlib.plugin import load_plugins
22
from bzrlib.plugins.gtk import _i18n, cmd_visualise, cmd_gannotate
24
class BzrExtension(nautilus.MenuProvider, nautilus.ColumnProvider, nautilus.InfoProvider):
28
def add_cb(self, menu, vfs_file):
29
# We can only cope with local files
30
if vfs_file.get_uri_scheme() != 'file':
33
file = vfs_file.get_uri()
35
tree, path = WorkingTree.open_containing(file)
36
except NotBranchError:
43
def ignore_cb(self, menu, vfs_file):
44
# We can only cope with local files
45
if vfs_file.get_uri_scheme() != 'file':
48
file = vfs_file.get_uri()
50
tree, path = WorkingTree.open_containing(file)
51
except NotBranchError:
58
def unignore_cb(self, menu, vfs_file):
59
# We can only cope with local files
60
if vfs_file.get_uri_scheme() != 'file':
63
file = vfs_file.get_uri()
65
tree, path = WorkingTree.open_containing(file)
66
except NotBranchError:
73
def diff_cb(self, menu, vfs_file):
74
# We can only cope with local files
75
if vfs_file.get_uri_scheme() != 'file':
78
file = vfs_file.get_uri()
80
tree, path = WorkingTree.open_containing(file)
81
except NotBranchError:
84
from bzrlib.plugins.gtk.diff import DiffWindow
86
window.set_diff(tree.branch.nick, tree, tree.branch.basis_tree())
91
def newtree_cb(self, menu, vfs_file):
92
# We can only cope with local files
93
if vfs_file.get_uri_scheme() != 'file':
96
file = vfs_file.get_uri()
98
# We only want to continue here if we get a NotBranchError
100
tree, path = WorkingTree.open_containing(file)
101
except NotBranchError:
102
BzrDir.create_standalone_workingtree(file)
104
def remove_cb(self, menu, vfs_file):
105
# We can only cope with local files
106
if vfs_file.get_uri_scheme() != 'file':
109
file = vfs_file.get_uri()
111
tree, path = WorkingTree.open_containing(file)
112
except NotBranchError:
117
def annotate_cb(self, menu, vfs_file):
118
# We can only cope with local files
119
if vfs_file.get_uri_scheme() != 'file':
122
file = vfs_file.get_uri()
124
vis = cmd_gannotate()
127
def clone_cb(self, menu, vfs_file=None):
128
# We can only cope with local files
129
if vfs_file.get_uri_scheme() != 'file':
132
from bzrlib.plugins.gtk.branch import BranchDialog
134
dialog = BranchDialog(vfs_file.get_name())
135
response = dialog.run()
136
if response != gtk.RESPONSE_NONE:
140
def commit_cb(self, menu, vfs_file=None):
141
# We can only cope with local files
142
if vfs_file.get_uri_scheme() != 'file':
145
file = vfs_file.get_uri()
149
tree, path = WorkingTree.open_containing(file)
151
except NotBranchError, e:
154
except NoWorkingTree, e:
157
(branch, path) = Branch.open_containing(path)
158
except NotBranchError, e:
161
from bzrlib.plugins.gtk.commit import CommitDialog
162
dialog = CommitDialog(tree, path)
163
response = dialog.run()
164
if response != gtk.RESPONSE_NONE:
168
def log_cb(self, menu, vfs_file):
169
# We can only cope with local files
170
if vfs_file.get_uri_scheme() != 'file':
173
file = vfs_file.get_uri()
175
# We only want to continue here if we get a NotBranchError
177
tree, path = WorkingTree.open_containing(file)
178
except NotBranchError:
181
vis = cmd_visualise()
186
def pull_cb(self, menu, vfs_file):
187
# We can only cope with local files
188
if vfs_file.get_uri_scheme() != 'file':
191
file = vfs_file.get_uri()
193
# We only want to continue here if we get a NotBranchError
195
tree, path = WorkingTree.open_containing(file)
196
except NotBranchError:
199
from bzrlib.plugins.gtk.pull import PullDialog
200
dialog = PullDialog(tree, path)
204
def merge_cb(self, menu, vfs_file):
205
# We can only cope with local files
206
if vfs_file.get_uri_scheme() != 'file':
209
file = vfs_file.get_uri()
211
# We only want to continue here if we get a NotBranchError
213
tree, path = WorkingTree.open_containing(file)
214
except NotBranchError:
217
from bzrlib.plugins.gtk.merge import MergeDialog
218
dialog = MergeDialog(tree, path)
222
def get_background_items(self, window, vfs_file):
224
file = vfs_file.get_uri()
227
tree, path = WorkingTree.open_containing(file)
228
disabled_flag = self.check_branch_enabled(tree.branch)
229
except UnsupportedProtocol:
231
except NotBranchError:
232
disabled_flag = self.check_branch_enabled()
233
item = nautilus.MenuItem('BzrNautilus::newtree',
234
'Make directory versioned',
235
'Create new Bazaar tree in this folder')
236
item.connect('activate', self.newtree_cb, vfs_file)
239
item = nautilus.MenuItem('BzrNautilus::clone',
240
'Checkout Bazaar branch',
241
'Checkout Existing Bazaar Branch')
242
item.connect('activate', self.clone_cb, vfs_file)
246
except NoWorkingTree:
249
if disabled_flag == 'False':
250
item = nautilus.MenuItem('BzrNautilus::enable',
251
'Enable Bazaar Plugin for this Branch',
252
'Enable Bazaar plugin for nautilus')
253
item.connect('activate', self.toggle_integration, 'True', vfs_file)
256
item = nautilus.MenuItem('BzrNautilus::disable',
257
'Disable Bazaar Plugin for the Branch',
258
'Disable Bazaar plugin for nautilus')
259
item.connect('activate', self.toggle_integration, 'False', vfs_file)
262
item = nautilus.MenuItem('BzrNautilus::log',
264
'Show Bazaar history')
265
item.connect('activate', self.log_cb, vfs_file)
268
item = nautilus.MenuItem('BzrNautilus::pull',
270
'Pull from another branch')
271
item.connect('activate', self.pull_cb, vfs_file)
274
item = nautilus.MenuItem('BzrNautilus::merge',
276
'Merge from another branch')
277
item.connect('activate', self.merge_cb, vfs_file)
280
item = nautilus.MenuItem('BzrNautilus::commit',
283
item.connect('activate', self.commit_cb, vfs_file)
288
def get_file_items(self, window, files):
292
for vfs_file in files:
293
# We can only cope with local files
294
if vfs_file.get_uri_scheme() != 'file':
297
file = vfs_file.get_uri()
299
tree, path = WorkingTree.open_containing(file)
300
disabled_flag = self.check_branch_enabled(tree.branch)
301
except NotBranchError:
302
disabled_flag = self.check_branch_enabled()
303
if not vfs_file.is_directory():
306
if disabled_flag == 'False':
309
item = nautilus.MenuItem('BzrNautilus::newtree',
310
'Make directory versioned',
311
'Create new Bazaar tree in %s' % vfs_file.get_name())
312
item.connect('activate', self.newtree_cb, vfs_file)
314
except NoWorkingTree:
316
# Refresh the list of filestatuses in the working tree
317
if path not in wtfiles.keys():
319
for rpath, file_class, kind, id, entry in tree.list_files():
320
wtfiles[rpath] = file_class
324
if wtfiles[path] == '?':
325
item = nautilus.MenuItem('BzrNautilus::add',
327
'Add as versioned file')
328
item.connect('activate', self.add_cb, vfs_file)
331
item = nautilus.MenuItem('BzrNautilus::ignore',
333
'Ignore file for versioning')
334
item.connect('activate', self.ignore_cb, vfs_file)
336
elif wtfiles[path] == 'I':
337
item = nautilus.MenuItem('BzrNautilus::unignore',
339
'Unignore file for versioning')
340
item.connect('activate', self.unignore_cb, vfs_file)
342
elif wtfiles[path] == 'V':
343
item = nautilus.MenuItem('BzrNautilus::log',
346
item.connect('activate', self.log_cb, vfs_file)
349
item = nautilus.MenuItem('BzrNautilus::diff',
352
item.connect('activate', self.diff_cb, vfs_file)
355
item = nautilus.MenuItem('BzrNautilus::remove',
357
'Remove this file from versioning')
358
item.connect('activate', self.remove_cb, vfs_file)
361
item = nautilus.MenuItem('BzrNautilus::annotate',
363
'Annotate File Data')
364
item.connect('activate', self.annotate_cb, vfs_file)
367
item = nautilus.MenuItem('BzrNautilus::commit',
370
item.connect('activate', self.commit_cb, vfs_file)
375
def get_columns(self):
376
return nautilus.Column("BzrNautilus::bzr_status",
379
"Version control status"),
381
def update_file_info(self, file):
383
if file.get_uri_scheme() != 'file':
387
tree, path = WorkingTree.open_containing(file.get_uri())
388
except NotBranchError:
390
except NoWorkingTree:
393
disabled_flag = self.check_branch_enabled(tree.branch)
394
if disabled_flag == 'False':
400
if tree.has_filename(path):
401
emblem = 'bzr-controlled'
403
id = tree.path2id(path)
405
delta = tree.changes_from(tree.branch.basis_tree())
406
if delta.touches_file_id(id):
407
emblem = 'bzr-modified'
409
for f, _, _ in delta.added:
414
for of, f, _, _, _, _ in delta.renamed:
416
status = 'renamed from %s' % f
418
elif tree.branch.basis_tree().has_filename(path):
419
emblem = 'bzr-removed'
422
# FIXME: Check for ignored files
423
status = 'unversioned'
425
if emblem is not None:
426
file.add_emblem(emblem)
427
file.add_string_attribute('bzr_status', status)
429
def check_branch_enabled(self, branch=None):
430
# Supports global disable, but there is currently no UI to do this
431
config = GlobalConfig()
432
disabled_flag = config.get_user_option('nautilus_integration')
433
if disabled_flag != 'False':
434
if branch is not None:
435
config = branch.get_config()
436
disabled_flag = config.get_user_option('nautilus_integration')
439
def toggle_integration(self, menu, action, vfs_file=None):
441
tree, path = WorkingTree.open_containing(vfs_file.get_uri())
442
except NotBranchError:
444
except NoWorkingTree:
448
config = GlobalConfig()
450
config = branch.get_config()
451
config.set_user_option('nautilus_integration', action)
added
removed
nautilus-bzr.py
