1
# -*- coding: UTF-8 -*-
1
2
"""Difference window.
3
4
This module contains the code to manage the diff window which shows
4
5
the changes made between two revisions on a branch.
7
__copyright__ = "Copyright 2005 Canonical Ltd."
8
__author__ = "Scott James Remnant <scott@ubuntu.com>"
8
__copyright__ = "Copyright © 2005 Canonical Ltd."
9
__author__ = "Scott James Remnant <scott@ubuntu.com>"
11
12
from cStringIO import StringIO
16
from gi.repository import Gtk
17
from gi.repository import Pango
19
from gi.repository import GtkSource
20
24
have_gtksourceview = True
21
25
except ImportError:
22
26
have_gtksourceview = False
24
33
from bzrlib import (
26
34
merge as _mod_merge,
31
from bzrlib.diff import show_diff_trees
40
from bzrlib.diff import show_diff_trees, internal_diff
41
from bzrlib.errors import NoSuchFile
32
42
from bzrlib.patches import parse_patches
33
from bzrlib.plugins.gtk.dialog import (
38
from bzrlib.plugins.gtk.i18n import _i18n
43
from bzrlib.trace import warning
44
from bzrlib.plugins.gtk import _i18n
39
45
from bzrlib.plugins.gtk.window import Window
42
def fallback_guess_language(slm, content_type):
43
for lang_id in slm.get_language_ids():
44
lang = slm.get_language(lang_id)
45
if "text/x-patch" in lang.get_mime_types():
46
from dialog import error_dialog, info_dialog, warning_dialog
50
49
class SelectCancelled(Exception):
55
class DiffFileView(Gtk.ScrolledWindow):
54
class DiffFileView(gtk.ScrolledWindow):
56
55
"""Window for displaying diffs from a diff file"""
60
57
def __init__(self):
61
super(DiffFileView, self).__init__()
58
gtk.ScrolledWindow.__init__(self)
65
62
def construct(self):
66
self.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
67
self.set_shadow_type(Gtk.ShadowType.IN)
63
self.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
64
self.set_shadow_type(gtk.SHADOW_IN)
69
66
if have_gtksourceview:
70
self.buffer = GtkSource.Buffer()
71
lang_manager = GtkSource.LanguageManager.get_default()
72
language = lang_manager.guess_language(None, "text/x-patch")
73
self.buffer.set_language(language)
74
self.buffer.set_highlight_syntax(True)
75
self.sourceview = GtkSource.View(buffer=self.buffer)
67
self.buffer = gtksourceview.SourceBuffer()
68
slm = gtksourceview.SourceLanguagesManager()
69
gsl = slm.get_language_from_mime_type("text/x-patch")
71
self.apply_gedit_colors(gsl)
72
self.apply_colordiff_colors(gsl)
73
self.buffer.set_language(gsl)
74
self.buffer.set_highlight(True)
76
self.sourceview = gtksourceview.SourceView(self.buffer)
77
self.buffer = Gtk.TextBuffer()
78
self.sourceview = Gtk.TextView(self.buffer)
78
self.buffer = gtk.TextBuffer()
79
self.sourceview = gtk.TextView(self.buffer)
80
81
self.sourceview.set_editable(False)
81
self.sourceview.override_font(Pango.FontDescription("Monospace"))
82
self.sourceview.modify_font(pango.FontDescription("Monospace"))
82
83
self.add(self.sourceview)
84
self.sourceview.show()
84
self.sourceview.show()
87
def apply_gedit_colors(lang):
88
"""Set style for lang to that specified in gedit configuration.
90
This method needs the gconf module.
92
:param lang: a gtksourceview.SourceLanguage object.
94
GEDIT_SYNTAX_PATH = '/apps/gedit-2/preferences/syntax_highlighting'
95
GEDIT_LANG_PATH = GEDIT_SYNTAX_PATH + '/' + lang.get_id()
97
client = gconf.client_get_default()
98
client.add_dir(GEDIT_LANG_PATH, gconf.CLIENT_PRELOAD_NONE)
100
for tag in lang.get_tags():
101
tag_id = tag.get_id()
102
gconf_key = GEDIT_LANG_PATH + '/' + tag_id
103
style_string = client.get_string(gconf_key)
105
if style_string is None:
108
# function to get a bool from a string that's either '0' or '1'
109
string_bool = lambda x: bool(int(x))
111
# style_string is a string like "2/#FFCCAA/#000000/0/1/0/0"
112
# values are: mask, fg, bg, italic, bold, underline, strike
113
# this packs them into (str_value, attr_name, conv_func) tuples
114
items = zip(style_string.split('/'), ['mask', 'foreground',
115
'background', 'italic', 'bold', 'underline', 'strikethrough' ],
116
[ int, gtk.gdk.color_parse, gtk.gdk.color_parse, string_bool,
117
string_bool, string_bool, string_bool ]
120
style = gtksourceview.SourceTagStyle()
122
# XXX The mask attribute controls whether the present values of
123
# foreground and background color should in fact be used. Ideally
124
# (and that's what gedit does), one could set all three attributes,
125
# and let the TagStyle object figure out which colors to use.
126
# However, in the GtkSourceview python bindings, the mask attribute
127
# is read-only, and it's derived instead from the colors being
128
# set or not. This means that we have to sometimes refrain from
129
# setting fg or bg colors, depending on the value of the mask.
130
# This code could go away if mask were writable.
131
mask = int(items[0][0])
132
if not (mask & 1): # GTK_SOURCE_TAG_STYLE_USE_BACKGROUND
134
if not (mask & 2): # GTK_SOURCE_TAG_STYLE_USE_FOREGROUND
136
items[0:1] = [] # skip the mask unconditionally
138
for value, attr, func in items:
142
warning('gconf key %s contains an invalid value: %s'
145
setattr(style, attr, value)
147
lang.set_tag_style(tag_id, style)
150
def apply_colordiff_colors(klass, lang):
151
"""Set style colors for lang using the colordiff configuration file.
153
Both ~/.colordiffrc and ~/.colordiffrc.bzr-gtk are read.
155
:param lang: a "Diff" gtksourceview.SourceLanguage object.
159
for f in ('~/.colordiffrc', '~/.colordiffrc.bzr-gtk'):
160
f = os.path.expanduser(f)
161
if os.path.exists(f):
165
warning('could not open file %s: %s' % (f, str(e)))
167
colors.update(klass.parse_colordiffrc(f))
171
# ~/.colordiffrc does not exist
175
# map GtkSourceView tags to colordiff names
176
# since GSV is richer, accept new names for extra bits,
177
# defaulting to old names if they're not present
178
'Added@32@line': ['newtext'],
179
'Removed@32@line': ['oldtext'],
180
'Location': ['location', 'diffstuff'],
181
'Diff@32@file': ['file', 'diffstuff'],
182
'Special@32@case': ['specialcase', 'diffstuff'],
185
for tag in lang.get_tags():
186
tag_id = tag.get_id()
187
keys = mapping.get(tag_id, [])
191
color = colors.get(key, None)
192
if color is not None:
198
style = gtksourceview.SourceTagStyle()
200
style.foreground = gtk.gdk.color_parse(color)
202
warning('not a valid color: %s' % color)
204
lang.set_tag_style(tag_id, style)
207
def parse_colordiffrc(fileobj):
208
"""Parse fileobj as a colordiff configuration file.
210
:return: A dict with the key -> value pairs.
214
if re.match(r'^\s*#', line):
218
key, val = line.split('=', 1)
219
colors[key.strip()] = val.strip()
86
222
def set_trees(self, rev_tree, parent_tree):
87
223
self.rev_tree = rev_tree
158
292
self.buffer.set_text(decoded.encode('UTF-8'))
161
class DiffWidget(Gtk.HPaned):
295
class DiffWidget(gtk.HPaned):
168
299
def __init__(self):
169
300
super(DiffWidget, self).__init__()
171
302
# The file hierarchy: a scrollable treeview
172
scrollwin = Gtk.ScrolledWindow()
173
scrollwin.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
174
scrollwin.set_shadow_type(Gtk.ShadowType.IN)
303
scrollwin = gtk.ScrolledWindow()
304
scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
305
scrollwin.set_shadow_type(gtk.SHADOW_IN)
175
306
self.pack1(scrollwin)
176
if self.SHOW_WIDGETS:
179
self.model = Gtk.TreeStore(str, str)
180
self.treeview = Gtk.TreeView(model=self.model)
309
self.model = gtk.TreeStore(str, str)
310
self.treeview = gtk.TreeView(self.model)
181
311
self.treeview.set_headers_visible(False)
182
312
self.treeview.set_search_column(1)
183
313
self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
184
314
scrollwin.add(self.treeview)
185
if self.SHOW_WIDGETS:
188
cell = Gtk.CellRendererText()
317
cell = gtk.CellRendererText()
189
318
cell.set_property("width-chars", 20)
190
column = Gtk.TreeViewColumn()
191
column.pack_start(cell, True)
319
column = gtk.TreeViewColumn()
320
column.pack_start(cell, expand=True)
192
321
column.add_attribute(cell, "text", 0)
193
322
self.treeview.append_column(column)
231
358
self.model.clear()
232
359
delta = self.rev_tree.changes_from(self.parent_tree)
234
self.model.append(None, ["Complete Diff", ""])
361
self.model.append(None, [ "Complete Diff", "" ])
236
363
if len(delta.added):
237
titer = self.model.append(None, ["Added", None])
364
titer = self.model.append(None, [ "Added", None ])
238
365
for path, id, kind in delta.added:
239
self.model.append(titer, [path, path])
366
self.model.append(titer, [ path, path ])
241
368
if len(delta.removed):
242
titer = self.model.append(None, ["Removed", None])
369
titer = self.model.append(None, [ "Removed", None ])
243
370
for path, id, kind in delta.removed:
244
self.model.append(titer, [path, path])
371
self.model.append(titer, [ path, path ])
246
373
if len(delta.renamed):
247
titer = self.model.append(None, ["Renamed", None])
374
titer = self.model.append(None, [ "Renamed", None ])
248
375
for oldpath, newpath, id, kind, text_modified, meta_modified \
249
376
in delta.renamed:
250
self.model.append(titer, [oldpath, newpath])
377
self.model.append(titer, [ oldpath, newpath ])
252
379
if len(delta.modified):
253
titer = self.model.append(None, ["Modified", None])
380
titer = self.model.append(None, [ "Modified", None ])
254
381
for path, id, kind, text_modified, meta_modified in delta.modified:
255
self.model.append(titer, [path, path])
382
self.model.append(titer, [ path, path ])
257
384
self.treeview.expand_all()
258
385
self.diff_view.show_diff(None)
266
393
tv_path = child.path
268
395
if tv_path is None:
269
raise errors.NoSuchFile(file_path)
270
self.treeview.set_cursor(tv_path, None, False)
396
raise NoSuchFile(file_path)
397
self.treeview.set_cursor(tv_path)
271
398
self.treeview.scroll_to_cell(tv_path)
273
400
def _treeview_cursor_cb(self, *args):
274
401
"""Callback for when the treeview cursor changes."""
275
402
(path, col) = self.treeview.get_cursor()
278
specific_files = [self.model[path][1]]
279
if specific_files == [None]:
281
elif specific_files == [""]:
403
specific_files = [ self.model[path][1] ]
404
if specific_files == [ None ]:
406
elif specific_files == [ "" ]:
282
407
specific_files = None
284
409
self.diff_view.show_diff(specific_files)
286
411
def _on_wraplines_toggled(self, widget=None, wrap=False):
287
412
"""Callback for when the wrap lines checkbutton is toggled"""
288
413
if wrap or widget.get_active():
289
self.diff_view.sourceview.set_wrap_mode(Gtk.WrapMode.WORD)
414
self.diff_view.sourceview.set_wrap_mode(gtk.WRAP_WORD)
291
self.diff_view.sourceview.set_wrap_mode(Gtk.WrapMode.NONE)
416
self.diff_view.sourceview.set_wrap_mode(gtk.WRAP_NONE)
294
418
class DiffWindow(Window):
316
438
def construct(self, operations):
317
439
"""Construct the window contents."""
318
self.vbox = Gtk.VBox()
440
self.vbox = gtk.VBox()
319
441
self.add(self.vbox)
320
if self.SHOW_WIDGETS:
322
443
self.diff = DiffWidget()
323
444
self.vbox.pack_end(self.diff, True, True, 0)
324
if self.SHOW_WIDGETS:
326
446
# Build after DiffWidget to connect signals
327
447
menubar = self._get_menu_bar()
328
448
self.vbox.pack_start(menubar, False, False, 0)
329
449
hbox = self._get_button_bar(operations)
330
450
if hbox is not None:
331
451
self.vbox.pack_start(hbox, False, True, 0)
333
454
def _get_menu_bar(self):
334
menubar = Gtk.MenuBar()
455
menubar = gtk.MenuBar()
336
mb_view = Gtk.MenuItem.new_with_mnemonic(_i18n("_View"))
337
mb_view_menu = Gtk.Menu()
338
mb_view_wrapsource = Gtk.CheckMenuItem.new_with_mnemonic(
339
_i18n("Wrap _Long Lines"))
457
mb_view = gtk.MenuItem(_i18n("_View"))
458
mb_view_menu = gtk.Menu()
459
mb_view_wrapsource = gtk.CheckMenuItem(_i18n("Wrap _Long Lines"))
340
460
mb_view_wrapsource.connect('activate', self.diff._on_wraplines_toggled)
461
mb_view_wrapsource.show()
341
462
mb_view_menu.append(mb_view_wrapsource)
342
464
mb_view.set_submenu(mb_view_menu)
343
466
menubar.append(mb_view)
344
if self.SHOW_WIDGETS:
348
470
def _get_button_bar(self, operations):
349
471
"""Return a button bar to use.
353
475
if operations is None:
355
hbox = Gtk.HButtonBox()
356
hbox.set_layout(Gtk.ButtonBoxStyle.START)
477
hbox = gtk.HButtonBox()
478
hbox.set_layout(gtk.BUTTONBOX_START)
357
479
for title, method in operations:
358
merge_button = Gtk.Button(title)
359
if self.SHOW_WIDGETS:
361
merge_button.set_relief(Gtk.ReliefStyle.NONE)
480
merge_button = gtk.Button(title)
482
merge_button.set_relief(gtk.RELIEF_NONE)
362
483
merge_button.connect("clicked", method)
363
hbox.pack_start(merge_button, False, True, 0)
364
if self.SHOW_WIDGETS:
484
hbox.pack_start(merge_button, expand=False, fill=True)
368
488
def _get_merge_target(self):
369
d = Gtk.FileChooserDialog('Merge branch', self,
370
Gtk.FileChooserAction.SELECT_FOLDER,
371
buttons=(Gtk.STOCK_OK, Gtk.ResponseType.OK,
373
Gtk.ResponseType.CANCEL,))
489
d = gtk.FileChooserDialog('Merge branch', self,
490
gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER,
491
buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
493
gtk.RESPONSE_CANCEL,))
376
if result != Gtk.ResponseType.OK:
496
if result != gtk.RESPONSE_OK:
377
497
raise SelectCancelled()
378
498
return d.get_current_folder_uri()