1
# -*- coding: UTF-8 -*-
2
1
"""Difference window.
4
3
This module contains the code to manage the diff window which shows
5
4
the changes made between two revisions on a branch.
8
__copyright__ = "Copyright © 2005 Canonical Ltd."
9
__author__ = "Scott James Remnant <scott@ubuntu.com>"
7
__copyright__ = "Copyright 2005 Canonical Ltd."
8
__author__ = "Scott James Remnant <scott@ubuntu.com>"
12
11
from cStringIO import StringIO
16
from gi.repository import Gtk
17
from gi.repository import Pango
19
from gi.repository import GtkSource
24
20
have_gtksourceview = True
25
21
except ImportError:
26
22
have_gtksourceview = False
33
24
from bzrlib import (
34
26
merge as _mod_merge,
40
from bzrlib.diff import show_diff_trees, internal_diff
41
from bzrlib.errors import NoSuchFile
31
from bzrlib.diff import show_diff_trees
42
32
from bzrlib.patches import parse_patches
43
from bzrlib.trace import warning
44
from bzrlib.plugins.gtk import _i18n
33
from bzrlib.plugins.gtk.dialog import (
38
from bzrlib.plugins.gtk.i18n import _i18n
45
39
from bzrlib.plugins.gtk.window import Window
46
from dialog import error_dialog, info_dialog, warning_dialog
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():
49
50
class SelectCancelled(Exception):
54
class DiffFileView(gtk.ScrolledWindow):
55
class DiffFileView(Gtk.ScrolledWindow):
55
56
"""Window for displaying diffs from a diff file"""
57
60
def __init__(self):
58
gtk.ScrolledWindow.__init__(self)
61
super(DiffFileView, self).__init__()
62
65
def construct(self):
63
self.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
64
self.set_shadow_type(gtk.SHADOW_IN)
66
self.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
67
self.set_shadow_type(Gtk.ShadowType.IN)
66
69
if have_gtksourceview:
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
sourceview = gtksourceview.SourceView(self.buffer)
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)
78
self.buffer = gtk.TextBuffer()
79
sourceview = gtk.TextView(self.buffer)
81
sourceview.set_editable(False)
82
sourceview.modify_font(pango.FontDescription("Monospace"))
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()
77
self.buffer = Gtk.TextBuffer()
78
self.sourceview = Gtk.TextView(self.buffer)
80
self.sourceview.set_editable(False)
81
self.sourceview.override_font(Pango.FontDescription("Monospace"))
82
self.add(self.sourceview)
84
self.sourceview.show()
222
86
def set_trees(self, rev_tree, parent_tree):
223
87
self.rev_tree = rev_tree
292
158
self.buffer.set_text(decoded.encode('UTF-8'))
295
class DiffWidget(gtk.HPaned):
161
class DiffWidget(Gtk.HPaned):
299
168
def __init__(self):
300
169
super(DiffWidget, self).__init__()
302
171
# The file hierarchy: a scrollable treeview
303
scrollwin = gtk.ScrolledWindow()
304
scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
305
scrollwin.set_shadow_type(gtk.SHADOW_IN)
172
scrollwin = Gtk.ScrolledWindow()
173
scrollwin.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
174
scrollwin.set_shadow_type(Gtk.ShadowType.IN)
306
175
self.pack1(scrollwin)
176
if self.SHOW_WIDGETS:
309
self.model = gtk.TreeStore(str, str)
310
self.treeview = gtk.TreeView(self.model)
179
self.model = Gtk.TreeStore(str, str)
180
self.treeview = Gtk.TreeView(model=self.model)
311
181
self.treeview.set_headers_visible(False)
312
182
self.treeview.set_search_column(1)
313
183
self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
314
184
scrollwin.add(self.treeview)
185
if self.SHOW_WIDGETS:
317
cell = gtk.CellRendererText()
188
cell = Gtk.CellRendererText()
318
189
cell.set_property("width-chars", 20)
319
column = gtk.TreeViewColumn()
320
column.pack_start(cell, expand=True)
190
column = Gtk.TreeViewColumn()
191
column.pack_start(cell, True)
321
192
column.add_attribute(cell, "text", 0)
322
193
self.treeview.append_column(column)
356
231
self.model.clear()
357
232
delta = self.rev_tree.changes_from(self.parent_tree)
359
self.model.append(None, [ "Complete Diff", "" ])
234
self.model.append(None, ["Complete Diff", ""])
361
236
if len(delta.added):
362
titer = self.model.append(None, [ "Added", None ])
237
titer = self.model.append(None, ["Added", None])
363
238
for path, id, kind in delta.added:
364
self.model.append(titer, [ path, path ])
239
self.model.append(titer, [path, path])
366
241
if len(delta.removed):
367
titer = self.model.append(None, [ "Removed", None ])
242
titer = self.model.append(None, ["Removed", None])
368
243
for path, id, kind in delta.removed:
369
self.model.append(titer, [ path, path ])
244
self.model.append(titer, [path, path])
371
246
if len(delta.renamed):
372
titer = self.model.append(None, [ "Renamed", None ])
247
titer = self.model.append(None, ["Renamed", None])
373
248
for oldpath, newpath, id, kind, text_modified, meta_modified \
374
249
in delta.renamed:
375
self.model.append(titer, [ oldpath, newpath ])
250
self.model.append(titer, [oldpath, newpath])
377
252
if len(delta.modified):
378
titer = self.model.append(None, [ "Modified", None ])
253
titer = self.model.append(None, ["Modified", None])
379
254
for path, id, kind, text_modified, meta_modified in delta.modified:
380
self.model.append(titer, [ path, path ])
255
self.model.append(titer, [path, path])
382
257
self.treeview.expand_all()
258
self.diff_view.show_diff(None)
384
260
def set_file(self, file_path):
385
261
"""Select the current file to display"""
390
266
tv_path = child.path
392
268
if tv_path is None:
393
raise NoSuchFile(file_path)
394
self.treeview.set_cursor(tv_path)
269
raise errors.NoSuchFile(file_path)
270
self.treeview.set_cursor(tv_path, None, False)
395
271
self.treeview.scroll_to_cell(tv_path)
397
273
def _treeview_cursor_cb(self, *args):
398
274
"""Callback for when the treeview cursor changes."""
399
275
(path, col) = self.treeview.get_cursor()
400
specific_files = [ self.model[path][1] ]
401
if specific_files == [ None ]:
403
elif specific_files == [ "" ]:
278
specific_files = [self.model[path][1]]
279
if specific_files == [None]:
281
elif specific_files == [""]:
404
282
specific_files = None
406
284
self.diff_view.show_diff(specific_files)
286
def _on_wraplines_toggled(self, widget=None, wrap=False):
287
"""Callback for when the wrap lines checkbutton is toggled"""
288
if wrap or widget.get_active():
289
self.diff_view.sourceview.set_wrap_mode(Gtk.WrapMode.WORD)
291
self.diff_view.sourceview.set_wrap_mode(Gtk.WrapMode.NONE)
409
294
class DiffWindow(Window):
429
316
def construct(self, operations):
430
317
"""Construct the window contents."""
431
self.vbox = gtk.VBox()
318
self.vbox = Gtk.VBox()
432
319
self.add(self.vbox)
320
if self.SHOW_WIDGETS:
322
self.diff = DiffWidget()
323
self.vbox.pack_end(self.diff, True, True, 0)
324
if self.SHOW_WIDGETS:
326
# Build after DiffWidget to connect signals
327
menubar = self._get_menu_bar()
328
self.vbox.pack_start(menubar, False, False, 0)
434
329
hbox = self._get_button_bar(operations)
435
330
if hbox is not None:
436
self.vbox.pack_start(hbox, expand=False, fill=True)
437
self.diff = DiffWidget()
438
self.vbox.add(self.diff)
331
self.vbox.pack_start(hbox, False, True, 0)
333
def _get_menu_bar(self):
334
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"))
340
mb_view_wrapsource.connect('activate', self.diff._on_wraplines_toggled)
341
mb_view_menu.append(mb_view_wrapsource)
342
mb_view.set_submenu(mb_view_menu)
343
menubar.append(mb_view)
344
if self.SHOW_WIDGETS:
441
348
def _get_button_bar(self, operations):
442
349
"""Return a button bar to use.
446
353
if operations is None:
448
hbox = gtk.HButtonBox()
449
hbox.set_layout(gtk.BUTTONBOX_START)
355
hbox = Gtk.HButtonBox()
356
hbox.set_layout(Gtk.ButtonBoxStyle.START)
450
357
for title, method in operations:
451
merge_button = gtk.Button(title)
453
merge_button.set_relief(gtk.RELIEF_NONE)
358
merge_button = Gtk.Button(title)
359
if self.SHOW_WIDGETS:
361
merge_button.set_relief(Gtk.ReliefStyle.NONE)
454
362
merge_button.connect("clicked", method)
455
hbox.pack_start(merge_button, expand=False, fill=True)
363
hbox.pack_start(merge_button, False, True, 0)
364
if self.SHOW_WIDGETS:
459
368
def _get_merge_target(self):
460
d = gtk.FileChooserDialog('Merge branch', self,
461
gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER,
462
buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
464
gtk.RESPONSE_CANCEL,))
369
d = Gtk.FileChooserDialog('Merge branch', self,
370
Gtk.FileChooserAction.SELECT_FOLDER,
371
buttons=(Gtk.STOCK_OK, Gtk.ResponseType.OK,
373
Gtk.ResponseType.CANCEL,))
467
if result != gtk.RESPONSE_OK:
376
if result != Gtk.ResponseType.OK:
468
377
raise SelectCancelled()
469
378
return d.get_current_folder_uri()