3
3
# Copyright (C) 2006 Jeff Bailey
4
4
# Copyright (C) 2006 Wouter van Heyst
5
# Copyright (C) 2006 Jelmer Vernooij
7
# Published under the GNU GPL
11
from bzrlib.bzrdir import BzrDir
12
from bzrlib.errors import NotBranchError
13
from bzrlib.workingtree import WorkingTree
14
from bzrlib.tree import file_status
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
16
37
from bzrlib.plugin import load_plugins
19
from bzrlib.plugins.gtk import cmd_visualise, cmd_gannotate
21
class BzrExtension(nautilus.MenuProvider, nautilus.ColumnProvider, nautilus.InfoProvider):
40
from bzrlib.plugins.gtk.i18n import _i18n
43
class BazaarExtension(Nautilus.MenuProvider, Nautilus.ColumnProvider,
44
Nautilus.InfoProvider, Nautilus.PropertyPageProvider,
45
Nautilus.LocationWidgetProvider, GObject.GObject):
46
"""Nautilus extension providing Bazaar integration."""
22
48
def __init__(self):
25
def add_cb(self, menu, vfs_file):
26
# We can only cope with local files
27
if vfs_file.get_uri_scheme() != 'file':
30
file = vfs_file.get_uri()
32
tree, path = WorkingTree.open_containing(file)
33
except NotBranchError:
52
def _open_bzrdir(cls, vfs_file):
53
uri = vfs_file.get_uri()
54
controldir, path = ControlDir.open_containing(uri)
55
return controldir, path
58
def _open_tree(cls, vfs_file):
59
controldir, path = cls._open_bzrdir(vfs_file)
60
return controldir.open_workingtree(), path
62
def add_cb(self, menu, tree, path):
40
def ignore_cb(self, menu, vfs_file):
41
# We can only cope with local files
42
if vfs_file.get_uri_scheme() != 'file':
45
file = vfs_file.get_uri()
47
tree, path = WorkingTree.open_containing(file)
48
except NotBranchError:
55
def unignore_cb(self, menu, vfs_file):
56
# We can only cope with local files
57
if vfs_file.get_uri_scheme() != 'file':
60
file = vfs_file.get_uri()
62
tree, path = WorkingTree.open_containing(file)
63
except NotBranchError:
70
def diff_cb(self, menu, vfs_file):
71
# We can only cope with local files
72
if vfs_file.get_uri_scheme() != 'file':
75
file = vfs_file.get_uri()
77
tree, path = WorkingTree.open_containing(file)
78
except NotBranchError:
81
from bzrlib.plugins.gtk.viz.diff import DiffWindow
65
def ignore_cb(self, menu, tree, path):
66
# We can only cope with local files
67
tree_ignores_add_patterns(tree, [path])
68
#FIXME: Add path to ignore file
70
def unignore_cb(self, menu, tree, path):
72
# We can only cope with local files
75
def diff_cb(self, menu, tree, path=None):
76
from bzrlib.plugins.gtk.diff import DiffWindow
82
77
window = DiffWindow()
83
window.set_diff(tree.branch.nick, tree, tree.branch.basis_tree())
78
window.set_diff(tree.branch._get_nick(local=True), tree,
79
tree.branch.basis_tree())
88
82
def newtree_cb(self, menu, vfs_file):
89
# We can only cope with local files
90
if vfs_file.get_uri_scheme() != 'file':
93
file = vfs_file.get_uri()
95
# We only want to continue here if we get a NotBranchError
97
tree, path = WorkingTree.open_containing(file)
98
except NotBranchError:
99
BzrDir.create_branch_and_repo(file)
101
def remove_cb(self, menu, vfs_file):
102
# We can only cope with local files
103
if vfs_file.get_uri_scheme() != 'file':
106
file = vfs_file.get_uri()
108
tree, path = WorkingTree.open_containing(file)
109
except NotBranchError:
83
controldir, path = self._open_bzrdir(vfs_file)
84
controldir.create_workingtree()
86
def remove_cb(self, menu, tree, path):
114
def annotate_cb(self, menu, vfs_file):
115
# We can only cope with local files
116
if vfs_file.get_uri_scheme() != 'file':
119
file = vfs_file.get_uri()
121
vis = cmd_gannotate()
89
def annotate_cb(self, menu, tree, path, file_id):
90
from bzrlib.plugins.gtk.annotate.gannotate import GAnnotateWindow
91
win = GAnnotateWindow()
93
win.annotate(tree, tree.branch, file_id)
124
96
def clone_cb(self, menu, vfs_file=None):
125
# We can only cope with local files
126
if vfs_file.get_uri_scheme() != 'file':
129
97
from bzrlib.plugins.gtk.branch import BranchDialog
98
controldir, path = self._open_bzrdir(vfs_file)
131
100
dialog = BranchDialog(vfs_file.get_name())
134
def commit_cb(self, menu, vfs_file=None):
135
# We can only cope with local files
136
if vfs_file.get_uri_scheme() != 'file':
139
file = vfs_file.get_uri()
141
tree, path = WorkingTree.open_containing(file)
142
except NotBranchError:
101
response = dialog.run()
102
if response != Gtk.ResponseType.NONE:
106
def commit_cb(self, menu, tree, path=None):
145
107
from bzrlib.plugins.gtk.commit import CommitDialog
146
108
dialog = CommitDialog(tree, path)
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()
168
def pull_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:
109
response = dialog.run()
110
if response != Gtk.ResponseType.NONE:
114
def log_cb(self, menu, controldir, path=None):
115
from bzrlib.plugins.gtk.viz import BranchWindow
116
branch = controldir.open_branch()
117
pp = BranchWindow(branch, [branch.last_revision()], None)
121
def pull_cb(self, menu, controldir, path=None):
181
122
from bzrlib.plugins.gtk.pull import PullDialog
182
dialog = PullDialog(tree, path)
186
def merge_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:
123
dialog = PullDialog(controldir.open_workingtree(), path)
127
def push_cb(self, menu, controldir, path=None):
128
from bzrlib.plugins.gtk.push import PushDialog
129
dialog = PushDialog(branch=controldir.open_workingtree().branch)
133
def merge_cb(self, menu, tree, path=None):
199
134
from bzrlib.plugins.gtk.merge import MergeDialog
200
135
dialog = MergeDialog(tree, path)
139
def create_tree_cb(self, menu, controldir):
140
controldir.create_workingtree()
204
142
def get_background_items(self, window, vfs_file):
206
file = vfs_file.get_uri()
208
tree, path = WorkingTree.open_containing(file)
209
except NotBranchError:
210
item = nautilus.MenuItem('BzrNautilus::newtree',
211
'Make directory versioned',
212
'Create new Bazaar tree in this folder')
144
controldir, path = self._open_bzrdir(vfs_file)
145
except NotBranchError:
148
branch = controldir.open_branch()
149
except NotBranchError:
151
item = Nautilus.MenuItem(name='BzrNautilus::newtree',
152
label='Make directory versioned',
153
tip='Create new Bazaar tree in this folder',
213
155
item.connect('activate', self.newtree_cb, vfs_file)
214
156
items.append(item)
216
item = nautilus.MenuItem('BzrNautilus::clone',
217
'Checkout Bazaar branch',
218
'Checkout Existing Bazaar Branch')
158
item = Nautilus.MenuItem(name='BzrNautilus::clone',
159
label='Checkout Bazaar branch ...',
160
tip='Checkout Existing Bazaar Branch',
219
162
item.connect('activate', self.clone_cb, vfs_file)
220
163
items.append(item)
224
item = nautilus.MenuItem('BzrNautilus::log',
226
'Show Bazaar history')
227
item.connect('activate', self.log_cb, vfs_file)
230
item = nautilus.MenuItem('BzrNautilus::pull',
232
'Pull from another branch')
233
item.connect('activate', self.pull_cb, vfs_file)
236
item = nautilus.MenuItem('BzrNautilus::merge',
238
'Merge from another branch')
239
item.connect('activate', self.merge_cb, vfs_file)
242
item = nautilus.MenuItem('BzrNautilus::commit',
245
item.connect('activate', self.commit_cb, vfs_file)
168
nautilus_integration = self.check_branch_enabled(branch)
169
if not nautilus_integration:
170
item = Nautilus.MenuItem(name='BzrNautilus::enable',
171
label='Enable Bazaar Plugin for this Branch',
172
tip='Enable Bazaar plugin for nautilus',
174
item.connect('activate', self.toggle_integration, True, branch)
177
item = Nautilus.MenuItem(name='BzrNautilus::disable',
178
label='Disable Bazaar Plugin this Branch',
179
tip='Disable Bazaar plugin for nautilus',
181
item.connect('activate', self.toggle_integration, False, branch)
184
item = Nautilus.MenuItem(name='BzrNautilus::log',
186
tip='Show Bazaar history',
188
item.connect('activate', self.log_cb, controldir)
191
item = Nautilus.MenuItem(name='BzrNautilus::pull',
193
tip='Pull from another branch',
195
item.connect('activate', self.pull_cb, controldir)
199
tree = controldir.open_workingtree()
200
except NoWorkingTree:
201
item = Nautilus.MenuItem(name='BzrNautilus::create_tree',
202
label='Create working tree...',
203
tip='Create a working tree for this branch',
205
item.connect('activate', self.create_tree_cb, controldir)
208
item = Nautilus.MenuItem(name='BzrNautilus::merge',
210
tip='Merge from another branch',
212
item.connect('activate', self.merge_cb, tree, path)
215
item = Nautilus.MenuItem(name='BzrNautilus::commit',
217
tip='Commit Changes',
219
item.connect('activate', self.commit_cb, tree, path)
224
def _get_file_menuitems(self, tree, intertree, path):
225
file_id = tree.path2id(path)
227
item = Nautilus.MenuItem(name='BzrNautilus::add',
229
tip='Add as versioned file',
231
item.connect('activate', self.add_cb, tree, path)
234
item = Nautilus.MenuItem(name='BzrNautilus::ignore',
236
tip='Ignore file for versioning',
238
item.connect('activate', self.ignore_cb, tree, path)
240
elif tree.is_ignored(path):
241
item = Nautilus.MenuItem(name='BzrNautilus::unignore',
243
tip='Unignore file for versioning',
245
item.connect('activate', self.unignore_cb, tree, path)
248
kind = tree.kind(file_id)
249
item = Nautilus.MenuItem(name='BzrNautilus::log',
253
item.connect('activate', self.log_cb, tree.bzrdir, path)
256
if not intertree.file_content_matches(file_id, file_id):
257
item = Nautilus.MenuItem(name='BzrNautilus::diff',
258
label='View Changes ...',
259
tip='Show differences',
261
item.connect('activate', self.diff_cb, tree, path)
264
item = Nautilus.MenuItem(name='BzrNautilus::commit',
266
tip='Commit Changes',
268
item.connect('activate', self.commit_cb, tree, path)
271
item = Nautilus.MenuItem(name='BzrNautilus::remove',
273
tip='Remove this file from versioning',
275
item.connect('activate', self.remove_cb, tree, path)
279
item = Nautilus.MenuItem(name='BzrNautilus::annotate',
280
label='Annotate ...',
281
tip='Annotate File Data',
283
item.connect('activate', self.annotate_cb, tree, path, file_id)
251
286
def get_file_items(self, window, files):
254
for vfs_file in files:
255
# We can only cope with local files
256
if vfs_file.get_uri_scheme() != 'file':
259
file = vfs_file.get_uri()
261
tree, path = WorkingTree.open_containing(file)
262
except NotBranchError:
263
if not vfs_file.is_directory():
265
item = nautilus.MenuItem('BzrNautilus::newtree',
266
'Make directory versioned',
267
'Create new Bazaar tree in %s' % vfs_file.get_name())
268
item.connect('activate', self.newtree_cb, vfs_file)
271
file_class = tree.file_class(path)
273
if file_class == '?':
274
item = nautilus.MenuItem('BzrNautilus::add',
276
'Add as versioned file')
277
item.connect('activate', self.add_cb, vfs_file)
280
item = nautilus.MenuItem('BzrNautilus::ignore',
282
'Ignore file for versioning')
283
item.connect('activate', self.ignore_cb, vfs_file)
285
elif file_class == 'I':
286
item = nautilus.MenuItem('BzrNautilus::unignore',
288
'Unignore file for versioning')
289
item.connect('activate', self.unignore_cb, vfs_file)
291
elif file_class == 'V':
292
item = nautilus.MenuItem('BzrNautilus::log',
295
item.connect('activate', self.log_cb, vfs_file)
298
item = nautilus.MenuItem('BzrNautilus::diff',
301
item.connect('activate', self.diff_cb, vfs_file)
304
item = nautilus.MenuItem('BzrNautilus::remove',
306
'Remove this file from versioning')
307
item.connect('activate', self.remove_cb, vfs_file)
310
item = nautilus.MenuItem('BzrNautilus::annotate',
312
'Annotate File Data')
313
item.connect('activate', self.annotate_cb, vfs_file)
316
item = nautilus.MenuItem('BzrNautilus::commit',
319
item.connect('activate', self.commit_cb, vfs_file)
291
for vfs_file in files:
292
controldir, path = self._open_bzrdir(vfs_file)
295
tree = trees[controldir.user_url]
298
tree = controldir.open_workingtree()
299
except NoWorkingTree:
301
trees[controldir.user_url] = tree
304
nautilus_integration = self.check_branch_enabled(tree.branch)
305
if not nautilus_integration:
308
intertree = InterTree.get(tree.basis_tree(), tree)
309
items.extend(list(self._get_file_menuitems(tree, intertree, path)))
311
for tree in trees.itervalues():
324
316
def get_columns(self):
325
return nautilus.Column("BzrNautilus::bzr_status",
328
"Version control status"),
330
def update_file_info(self, file):
331
if file.get_uri_scheme() != 'file':
335
tree, path = WorkingTree.open_containing(file.get_uri())
336
except NotBranchError:
318
Nautilus.Column(name="BzrNautilus::bzr_status",
319
attribute="bzr_status",
321
description="Version control status"),
322
Nautilus.Column(name="BzrNautilus::bzr_revision",
323
attribute="bzr_revision",
325
description="Last change revision"),
328
def _file_summary(self, tree, basis_tree, intertree, path):
342
if tree.has_filename(path):
343
emblem = 'cvs-controlled'
345
id = tree.path2id(path)
347
delta = tree.changes_from(tree.branch.basis_tree())
348
if delta.touches_file_id(id):
349
emblem = 'cvs-modified'
351
for f, _, _ in delta.added:
332
file_id = tree.path2id(path)
334
if tree.is_ignored(path):
336
emblem = 'bzr-ignored'
338
status = 'unversioned'
339
file_revision = "N/A"
340
elif tree.has_filename(path): # Still present
341
if not intertree.file_content_matches(file_id, file_id):
342
if not basis_tree.has_id(file_id):
356
for of, f, _, _, _, _ in delta.renamed:
358
status = 'renamed from %s' % f
360
elif tree.branch.basis_tree().has_filename(path):
361
emblem = 'cvs-removed'
345
file_revision = "new file"
346
elif basis_tree.path2id(file_id) != path:
347
status = 'bzr-renamed'
348
status = 'renamed from %s' % basis_tree.path2id(file_id)
350
emblem = 'bzr-modified'
353
emblem = 'bzr-controlled'
355
elif basis_tree.has_filename(path):
356
emblem = 'bzr-removed'
362
357
status = 'removed'
364
359
# FIXME: Check for ignored files
365
360
status = 'unversioned'
367
if emblem is not None:
368
file.add_emblem(emblem)
369
file.add_string_attribute('bzr_status', status)
361
return (status, emblem, file_revision)
363
def update_file_info(self, vfs_file):
365
controldir, path = self._open_bzrdir(vfs_file)
366
except NotBranchError:
370
tree = controldir.open_workingtree()
371
except NoWorkingTree:
376
nautilus_integration = self.check_branch_enabled(tree.branch)
377
if not nautilus_integration:
380
basis_tree = tree.basis_tree()
381
intertree = InterTree.get(basis_tree, tree)
383
basis_tree.lock_read()
385
(status, emblem, file_revision) = self._file_summary(tree, basis_tree, intertree, path)
388
if emblem is not None:
389
vfs_file.add_emblem(emblem)
390
vfs_file.add_string_attribute('bzr_status', status)
391
vfs_file.add_string_attribute('bzr_revision', file_revision)
395
def check_branch_enabled(self, branch):
396
# Supports global disable, but there is currently no UI to do this
397
config = branch.get_config_stack()
398
return config.get("nautilus_integration")
400
def toggle_integration(self, menu, action, branch):
401
config = branch.get_config_stack()
402
config.set("nautilus_integration", action)
404
def get_property_pages(self, files):
406
for vfs_file in files:
408
controldir, path = self._open_bzrdir(vfs_file)
409
except NotBranchError:
413
tree = controldir.open_workingtree()
414
except NoWorkingTree:
419
file_id = tree.path2id(path)
420
pages.append(PropertyPageFile(tree, file_id, path))
421
pages.append(PropertyPageBranch(tree.branch))
426
def get_widget(self, uri, window):
427
controldir, path = ControlDir.open_containing(uri)
429
tree = controldir.open_workingtree()
430
except NoWorkingTree:
432
ret = Gtk.HBox(False, 4)
433
text = 'This is a Bazaar working tree. '
434
get_shelf_manager = getattr(tree, 'get_shelf_manager', None)
435
if get_shelf_manager is not None:
436
manager = get_shelf_manager()
437
shelves = manager.active_shelves()
438
if len(shelves) == 0:
440
elif len(shelves) == 1:
441
text += '1 shelf exists. '
443
text += '%d shelf exists. ' % len(shelves)
444
label = Gtk.Label(text)
446
ret.pack_start(label, True, True, 0)
451
class PropertyPageFile(Nautilus.PropertyPage):
453
def __init__(self, tree, file_id, path):
455
self.file_id = file_id
457
label = Gtk.Label('File Version')
460
table = self._create_table()
462
super(PropertyPageFile, self).__init__(label=label,
463
name="BzrNautilus::file_page", page=table)
465
def _create_table(self):
466
table = Gtk.Table(homogeneous=False, columns=2, rows=3)
468
table.attach(Gtk.Label(_i18n('File id:')), 0, 1, 0, 1)
469
table.attach(Gtk.Label(self.file_id), 1, 2, 0, 1)
471
table.attach(Gtk.Label(_i18n('SHA1Sum:')), 0, 1, 1, 2)
472
table.attach(Gtk.Label(self.tree.get_file_sha1(self.file_id, self.path)), 1, 1, 1, 2)
474
basis_tree = self.tree.revision_tree(self.tree.last_revision())
475
last_revision = basis_tree.get_file_revision(self.file_id)
477
table.attach(Gtk.Label(_i18n('Last Change Revision:')), 0, 1, 2, 3)
478
revno = ".".join([str(x) for x in
479
self.tree.branch.revision_id_to_dotted_revno(last_revision)])
480
table.attach(Gtk.Label(revno), 1, 1, 2, 3)
482
table.attach(Gtk.Label(_i18n('Last Change Author:')), 0, 1, 3, 4)
483
rev = self.tree.branch.repository.get_revision(last_revision)
484
table.attach(Gtk.Label("\n".join(rev.get_apparent_authors())), 1, 1, 3, 4)
490
class PropertyPageBranch(Nautilus.PropertyPage):
492
def __init__(self, branch):
494
label = Gtk.Label('Branch')
497
table = self._create_table()
499
super(PropertyPageBranch, self).__init__(label=label,
500
name="BzrNautilus::branch_page", page=table)
502
def _create_location_entry(self, get_location, set_location):
503
location = get_location()
505
if location is not None:
506
ret.set_text(location)
509
def _create_table(self):
510
table = Gtk.Table(homogeneous=False, columns=2, rows=6)
512
self._push_location_entry = self._create_location_entry(
513
self.branch.get_push_location, self.branch.set_push_location)
514
self._parent_location_entry = self._create_location_entry(
515
self.branch.get_parent, self.branch.set_parent)
516
self._bound_location_entry = self._create_location_entry(
517
self.branch.get_bound_location, self.branch.set_bound_location)
518
self._public_location_entry = self._create_location_entry(
519
self.branch.get_public_branch, self.branch.set_public_branch)
520
self._submit_location_entry = self._create_location_entry(
521
self.branch.get_submit_branch, self.branch.set_submit_branch)
523
table.attach(Gtk.Label(_i18n('Push location:')), 0, 1, 0, 1)
524
table.attach(self._push_location_entry, 1, 2, 0, 1)
526
table.attach(Gtk.Label(_i18n('Parent location:')), 0, 1, 1, 2)
527
table.attach(self._parent_location_entry, 1, 1, 1, 2)
529
table.attach(Gtk.Label(_i18n('Bound location:')), 0, 1, 2, 3)
530
table.attach(self._bound_location_entry, 1, 1, 2, 3)
532
table.attach(Gtk.Label(_i18n('Public location:')), 0, 1, 3, 4)
533
table.attach(self._public_location_entry, 1, 1, 3, 4)
535
table.attach(Gtk.Label(_i18n('Submit location:')), 0, 1, 4, 5)
536
table.attach(self._submit_location_entry, 1, 1, 4, 5)
538
self._append_revisions_only = Gtk.CheckButton(_i18n('Append revisions only'))
539
value = self.branch.get_append_revisions_only()
542
self._append_revisions_only.set_active(value)
543
table.attach(self._append_revisions_only, 0, 2, 5, 6)