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
18
from xml.etree.ElementTree import (
23
Element, SubElement, tostring # Hush PEP8 redefinition.
25
from elementtree.ElementTree import (
31
from gi.repository import Gtk
32
from gi.repository import Pango
34
from gi.repository import GtkSource
24
35
have_gtksourceview = True
25
36
except ImportError:
26
37
have_gtksourceview = False
39
from gi.repository import GConf
30
41
except ImportError:
33
44
from bzrlib import (
34
46
merge as _mod_merge,
40
from bzrlib.diff import show_diff_trees, internal_diff
41
from bzrlib.errors import NoSuchFile
51
from bzrlib.diff import show_diff_trees
42
52
from bzrlib.patches import parse_patches
43
53
from bzrlib.trace import warning
44
from bzrlib.plugins.gtk import _i18n
54
from bzrlib.plugins.gtk.dialog import (
59
from bzrlib.plugins.gtk.i18n import _i18n
45
60
from bzrlib.plugins.gtk.window import Window
46
from dialog import error_dialog, info_dialog, warning_dialog
63
def fallback_guess_language(slm, content_type):
64
for lang_id in slm.get_language_ids():
65
lang = slm.get_language(lang_id)
66
if "text/x-patch" in lang.get_mime_types():
49
71
class SelectCancelled(Exception):
54
class DiffFileView(gtk.ScrolledWindow):
76
class DiffFileView(Gtk.ScrolledWindow):
55
77
"""Window for displaying diffs from a diff file"""
57
79
def __init__(self):
58
gtk.ScrolledWindow.__init__(self)
80
super(DiffFileView, self).__init__()
62
84
def construct(self):
63
self.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
64
self.set_shadow_type(gtk.SHADOW_IN)
85
self.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
86
self.set_shadow_type(Gtk.ShadowType.IN)
66
88
if have_gtksourceview:
67
self.buffer = gtksourceview.SourceBuffer()
68
slm = gtksourceview.SourceLanguagesManager()
69
gsl = slm.get_language_from_mime_type("text/x-patch")
89
self.buffer = GtkSource.Buffer()
90
lang_manager = GtkSource.LanguageManager.get_default()
91
language = lang_manager.guess_language(None, "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)
93
self.apply_gedit_colors(self.buffer)
94
self.apply_colordiff_colors(self.buffer)
95
self.buffer.set_language(language)
96
self.buffer.set_highlight_syntax(True)
76
sourceview = gtksourceview.SourceView(self.buffer)
98
self.sourceview = GtkSource.View(buffer=self.buffer)
78
self.buffer = gtk.TextBuffer()
79
sourceview = gtk.TextView(self.buffer)
100
self.buffer = Gtk.TextBuffer()
101
self.sourceview = Gtk.TextView(self.buffer)
81
sourceview.set_editable(False)
82
sourceview.modify_font(pango.FontDescription("Monospace"))
103
self.sourceview.set_editable(False)
104
self.sourceview.modify_font(Pango.FontDescription("Monospace"))
105
self.add(self.sourceview)
106
self.sourceview.show()
87
def apply_gedit_colors(lang):
88
"""Set style for lang to that specified in gedit configuration.
109
def apply_gedit_colors(buf):
110
"""Set style to that specified in gedit configuration.
90
112
This method needs the gconf module.
92
:param lang: a gtksourceview.SourceLanguage object.
114
:param buf: a GtkSource.Buffer 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)
116
GEDIT_SCHEME_PATH = '/apps/gedit-2/preferences/editor/colors/scheme'
117
GEDIT_USER_STYLES_PATH = os.path.expanduser('~/.gnome2/gedit/styles')
119
client = GConf.Client.get_default()
120
style_scheme_name = client.get_string(GEDIT_SCHEME_PATH)
121
if style_scheme_name is not None:
122
style_scheme_mgr = GtkSource.StyleSchemeManager()
123
style_scheme_mgr.append_search_path(GEDIT_USER_STYLES_PATH)
125
style_scheme = style_scheme_mgr.get_scheme(style_scheme_name)
127
if style_scheme is not None:
128
buf.set_style_scheme(style_scheme)
150
def apply_colordiff_colors(klass, lang):
131
def apply_colordiff_colors(klass, buf):
151
132
"""Set style colors for lang using the colordiff configuration file.
153
134
Both ~/.colordiffrc and ~/.colordiffrc.bzr-gtk are read.
155
:param lang: a "Diff" gtksourceview.SourceLanguage object.
136
:param buf: a "Diff" GtkSource.Buffer 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
138
scheme_manager = GtkSource.StyleSchemeManager()
139
style_scheme = scheme_manager.get_scheme('colordiff')
141
# if style scheme not found, we'll generate it from colordiffrc
142
# TODO: reload if colordiffrc has changed.
143
if style_scheme is None:
146
for f in ('~/.colordiffrc', '~/.colordiffrc.bzr-gtk'):
147
f = os.path.expanduser(f)
148
if os.path.exists(f):
152
warning('could not open file %s: %s' % (f, str(e)))
154
colors.update(klass.parse_colordiffrc(f))
158
# ~/.colordiffrc does not exist
162
# map GtkSourceView2 scheme styles to colordiff names
176
163
# since GSV is richer, accept new names for extra bits,
177
164
# 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)
165
'diff:added-line': ['newtext'],
166
'diff:removed-line': ['oldtext'],
167
'diff:location': ['location', 'diffstuff'],
168
'diff:file': ['file', 'diffstuff'],
169
'diff:special-case': ['specialcase', 'diffstuff'],
172
converted_colors = {}
173
for name, values in mapping.items():
176
color = colors.get(value, None)
177
if color is not None:
181
converted_colors[name] = color
183
# some xml magic to produce needed style scheme description
184
e_style_scheme = Element('style-scheme')
185
e_style_scheme.set('id', 'colordiff')
186
e_style_scheme.set('_name', 'ColorDiff')
187
e_style_scheme.set('version', '1.0')
188
for name, color in converted_colors.items():
189
style = SubElement(e_style_scheme, 'style')
190
style.set('name', name)
191
style.set('foreground', '#%s' % color)
193
scheme_xml = tostring(e_style_scheme, 'UTF-8')
194
if not os.path.exists(os.path.expanduser(
195
'~/.local/share/gtksourceview-2.0/styles')):
196
os.makedirs(os.path.expanduser(
197
'~/.local/share/gtksourceview-2.0/styles'))
198
file(os.path.expanduser(
199
'~/.local/share/gtksourceview-2.0/styles/colordiff.xml'),
200
'w').write(scheme_xml)
202
scheme_manager.force_rescan()
203
style_scheme = scheme_manager.get_scheme('colordiff')
205
buf.set_style_scheme(style_scheme)
207
208
def parse_colordiffrc(fileobj):
356
368
self.model.clear()
357
369
delta = self.rev_tree.changes_from(self.parent_tree)
359
self.model.append(None, [ "Complete Diff", "" ])
371
self.model.append(None, ["Complete Diff", ""])
361
373
if len(delta.added):
362
titer = self.model.append(None, [ "Added", None ])
374
titer = self.model.append(None, ["Added", None])
363
375
for path, id, kind in delta.added:
364
self.model.append(titer, [ path, path ])
376
self.model.append(titer, [path, path])
366
378
if len(delta.removed):
367
titer = self.model.append(None, [ "Removed", None ])
379
titer = self.model.append(None, ["Removed", None])
368
380
for path, id, kind in delta.removed:
369
self.model.append(titer, [ path, path ])
381
self.model.append(titer, [path, path])
371
383
if len(delta.renamed):
372
titer = self.model.append(None, [ "Renamed", None ])
384
titer = self.model.append(None, ["Renamed", None])
373
385
for oldpath, newpath, id, kind, text_modified, meta_modified \
374
386
in delta.renamed:
375
self.model.append(titer, [ oldpath, newpath ])
387
self.model.append(titer, [oldpath, newpath])
377
389
if len(delta.modified):
378
titer = self.model.append(None, [ "Modified", None ])
390
titer = self.model.append(None, ["Modified", None])
379
391
for path, id, kind, text_modified, meta_modified in delta.modified:
380
self.model.append(titer, [ path, path ])
392
self.model.append(titer, [path, path])
382
394
self.treeview.expand_all()
395
self.diff_view.show_diff(None)
384
397
def set_file(self, file_path):
385
398
"""Select the current file to display"""