19
from gi.repository import Gdk
20
25
from gi.repository import Gtk
21
26
from gi.repository import GObject
22
27
from gi.repository import Pango
24
29
from bzrlib import (
28
revision as _mod_revision,
35
from bzrlib import bencode
37
from bzrlib.util import bencode
32
39
from bzrlib.plugins.gtk.dialog import question_dialog
33
from bzrlib.plugins.gtk.diff import DiffView
34
40
from bzrlib.plugins.gtk.errors import show_bzr_error
35
41
from bzrlib.plugins.gtk.i18n import _i18n
36
from bzrlib.plugins.gtk.commitmsgs import SavedCommitMessagesManager
46
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
47
"""Get an iterator which will return the revisions in merge sorted order.
49
This will build up a list of all nodes, such that only nodes in the list
50
are referenced. It then uses MergeSorter to return them in 'merge-sorted'
53
:param revision_ids: A set of revision_ids
54
:param parent_map: The parent information for each node. Revisions which
55
are considered ghosts should not be present in the map.
56
:return: iterator from MergeSorter.iter_topo_order()
58
# MergeSorter requires that all nodes be present in the graph, so get rid
59
# of any references pointing outside of this graph.
61
for revision_id in revision_ids:
62
if revision_id not in parent_map: # ghost
63
parent_graph[revision_id] = []
65
# Only include parents which are in this sub-graph
66
parent_graph[revision_id] = [p for p in parent_map[revision_id]
68
sorter = tsort.MergeSorter(parent_graph, tip_revision)
69
return sorter.iter_topo_order()
72
51
def pending_revisions(wt):
73
52
"""Return a list of pending merges or None if there are none of them.
88
67
last_revision = parents[0]
90
graph = branch.repository.get_graph()
91
other_revisions = [last_revision]
69
if last_revision is not None:
71
ignore = set(branch.repository.get_ancestry(last_revision,
73
except errors.NoSuchRevision:
74
# the last revision is a ghost : assume everything is new
76
ignore = set([None, last_revision])
94
81
for merge in pending:
96
merge_rev = branch.repository.get_revision(merge)
97
except errors.NoSuchRevision:
98
# If we are missing a revision, just print out the revision id
99
trace.mutter("ghost: %r", merge)
100
other_revisions.append(merge)
103
# Find all of the revisions in the merge source, which are not in the
104
# last committed revision.
105
merge_extra = graph.find_unique_ancestors(merge, other_revisions)
106
other_revisions.append(merge)
107
merge_extra.discard(_mod_revision.NULL_REVISION)
109
# Get a handle to all of the revisions we will need
111
revisions = dict((rev.revision_id, rev) for rev in
112
branch.repository.get_revisions(merge_extra))
113
except errors.NoSuchRevision:
114
# One of the sub nodes is a ghost, check each one
116
for revision_id in merge_extra:
118
rev = branch.repository.get_revisions([revision_id])[0]
119
except errors.NoSuchRevision:
120
revisions[revision_id] = None
122
revisions[revision_id] = rev
124
# Display the revisions brought in by this merge.
125
rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
126
branch.repository.get_parent_map(merge_extra))
127
# Skip the first node
128
num, first, depth, eom = rev_id_iterator.next()
130
raise AssertionError('Somehow we misunderstood how'
131
' iter_topo_order works %s != %s' % (first, merge))
133
for num, sub_merge, depth, eom in rev_id_iterator:
134
rev = revisions[sub_merge]
136
trace.warning("ghost: %r", sub_merge)
139
yield (merge_rev, children)
84
rev = branch.repository.get_revision(merge)
86
pm.append((rev, children))
88
# This does need to be topo sorted, so we search backwards
89
inner_merges = branch.repository.get_ancestry(merge)
90
assert inner_merges[0] is None
92
for mmerge in reversed(inner_merges):
95
rev = branch.repository.get_revision(mmerge)
99
except errors.NoSuchRevision:
100
print "DEBUG: NoSuchRevision:", merge
142
105
_newline_variants_re = re.compile(r'\r\n?')
143
106
def _sanitize_and_decode_message(utf8_message):
144
107
"""Turn a utf-8 message into a sanitized Unicode message."""
145
108
fixed_newline = _newline_variants_re.sub('\n', utf8_message)
146
return osutils.safe_unicode(fixed_newline)
109
return fixed_newline.decode('utf-8')
149
112
class CommitDialog(Gtk.Dialog):
150
113
"""Implementation of Commit."""
152
115
def __init__(self, wt, selected=None, parent=None):
153
super(CommitDialog, self).__init__(
154
title="Commit to %s" % wt.basedir, parent=parent, flags=0)
116
GObject.GObject.__init__(self, title="Commit to %s" % wt.basedir,
117
parent=parent, flags=0,)
155
118
self.connect('delete-event', self._on_delete_window)
156
119
self._question_dialog = question_dialog
334
286
"""Build up the dialog widgets."""
335
287
# The primary pane which splits it into left and right (adjustable)
337
self._hpane = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL)
289
self._hpane = Gtk.HPaned()
339
291
self._construct_left_pane()
340
292
self._construct_right_pane()
341
293
self._construct_action_pane()
343
self.get_content_area().pack_start(self._hpane, True, True, 0)
295
self.vbox.pack_start(self._hpane, True, True, 0)
344
296
self._hpane.show()
345
297
self.set_focus(self._global_message_text_view)
408
360
self._button_cancel = Gtk.Button(stock=Gtk.STOCK_CANCEL)
409
361
self._button_cancel.connect('clicked', self._on_cancel_clicked)
410
362
self._button_cancel.show()
411
self.get_action_area().pack_end(
412
self._button_cancel, True, True, 0)
363
self.action_area.pack_end(self._button_cancel)
413
364
self._button_commit = Gtk.Button(_i18n("Comm_it"), use_underline=True)
414
365
self._button_commit.connect('clicked', self._on_commit_clicked)
415
366
self._button_commit.set_can_default(True)
416
367
self._button_commit.show()
417
self.get_action_area().pack_end(
418
self._button_commit, True, True, 0)
368
self.action_area.pack_end(self._button_commit)
419
369
self._button_commit.grab_default()
421
371
def _add_to_right_table(self, widget, weight, expanding=False):
441
391
# file_label.show()
442
392
self._files_box.pack_start(file_label, False, True, 0)
444
self._commit_all_files_radio = Gtk.RadioButton.new_with_label(
394
self._commit_all_files_radio = Gtk.RadioButton(
445
395
None, _i18n("Commit all changes"))
446
396
self._files_box.pack_start(self._commit_all_files_radio, False, True, 0)
447
397
self._commit_all_files_radio.show()
448
398
self._commit_all_files_radio.connect('toggled',
449
399
self._toggle_commit_selection)
450
self._commit_selected_radio = Gtk.RadioButton.new_with_label_from_widget(
400
self._commit_selected_radio = Gtk.RadioButton(
451
401
self._commit_all_files_radio, _i18n("Only commit selected changes"))
452
402
self._files_box.pack_start(self._commit_selected_radio, False, True, 0)
453
403
self._commit_selected_radio.show()
528
479
pending_message = Gtk.Label()
529
480
pending_message.set_markup(
530
481
_i18n('<i>* Cannot select specific files when merging</i>'))
531
self._pending_box.pack_start(pending_message, False, True, 5)
482
self._pending_box.pack_start(pending_message, expand=False, padding=5)
532
483
pending_message.show()
534
485
pending_label = Gtk.Label(label=_i18n('Pending Revisions'))
535
self._pending_box.pack_start(pending_label, False, True, 0)
486
self._pending_box.pack_start(pending_label, expand=False, padding=0)
536
487
pending_label.show()
538
489
scroller = Gtk.ScrolledWindow()
592
540
self._file_message_text_view.modify_font(Pango.FontDescription("Monospace"))
593
541
self._file_message_text_view.set_wrap_mode(Gtk.WrapMode.WORD)
594
542
self._file_message_text_view.set_accepts_tab(False)
595
line_height = self.get_line_height(self._file_message_text_view)
596
self._file_message_text_view.set_size_request(-1, line_height * 2)
597
543
self._file_message_text_view.show()
599
self._file_message_expander = Gtk.Expander(
600
label=_i18n('File commit message'))
545
self._file_message_expander = Gtk.Expander(_i18n('File commit message'))
601
546
self._file_message_expander.set_expanded(True)
602
547
self._file_message_expander.add(scroller)
603
548
self._add_to_right_table(self._file_message_expander, 1, False)
630
575
def _on_treeview_files_cursor_changed(self, treeview):
631
576
treeselection = treeview.get_selection()
632
if treeselection is None:
633
# The treeview was probably destroyed as the dialog closes.
635
577
(model, selection) = treeselection.get_selected()
637
579
if selection is not None:
638
580
path, display_path = model.get(selection, 1, 3)
639
581
self._diff_label.set_text(_i18n('Diff for ') + display_path)
641
583
self._diff_view.show_diff(None)
643
self._diff_view.show_diff([osutils.safe_unicode(path)])
585
self._diff_view.show_diff([path.decode('UTF-8')])
644
586
self._update_per_file_info(selection)
646
588
def _on_accel_next(self, accel_group, window, keyval, modifier):
657
599
# We have either made it to the end of the list, or nothing was
658
600
# selected. Either way, select All Files, and jump to the global
659
601
# commit message.
660
self._treeview_files.set_cursor(
661
Gtk.TreePath(path=0), "", False)
602
self._treeview_files.set_cursor((0,))
662
603
self._global_message_text_view.grab_focus()
664
605
# Set the cursor to this entry, and jump to the per-file commit
666
self._treeview_files.set_cursor(model.get_path(next), None, False)
607
self._treeview_files.set_cursor(model.get_path(next))
667
608
self._file_message_text_view.grab_focus()
669
610
def _save_current_file_message(self):
707
648
records = iter(self._files_store)
708
649
rec = records.next() # Skip the All Files record
709
assert rec[0] == "", "Are we skipping the wrong record?"
650
assert rec[0] is None, "Are we skipping the wrong record?"
712
653
for record in records:
713
654
if self._commit_all_changes or record[2]:# [2] checkbox
714
file_id = osutils.safe_utf8(record[0]) # [0] file_id
715
path = osutils.safe_utf8(record[1]) # [1] real path
655
file_id = record[0] # [0] file_id
656
path = record[1] # [1] real path
716
657
# [5] commit message
717
658
file_message = _sanitize_and_decode_message(record[5])
718
659
files.append(path.decode('UTF-8'))
854
795
rev_dict['revision_id'] = rev.revision_id
799
class SavedCommitMessagesManager:
800
"""Save glogal and per-file commit messages.
802
Saves global commit message and utf-8 file_id->message dictionary
803
of per-file commit messages on disk. Re-reads them later for re-using.
806
def __init__(self, tree=None, branch=None):
807
"""If branch is None, builds empty messages, otherwise reads them
808
from branch's disk storage. 'tree' argument is for the future."""
810
self.global_message = u''
811
self.file_messages = {}
813
config = branch.get_config()
814
self.global_message = config.get_user_option(
815
'gtk_global_commit_message')
816
if self.global_message is None:
817
self.global_message = u''
818
file_messages = config.get_user_option('gtk_file_commit_messages')
819
if file_messages: # unicode and B-encoded:
820
self.file_messages = bencode.bdecode(
821
file_messages.encode('UTF-8'))
823
self.file_messages = {}
826
return self.global_message, self.file_messages
828
def is_not_empty(self):
829
return bool(self.global_message or self.file_messages)
831
def insert(self, global_message, file_info):
832
"""Formats per-file commit messages (list of dictionaries, one per file)
833
into one utf-8 file_id->message dictionary and merges this with
834
previously existing dictionary. Merges global commit message too."""
837
file_message = fi['message']
839
file_messages[fi['file_id']] = file_message # utf-8 strings
840
for k,v in file_messages.iteritems():
842
self.file_messages[k] = v + '\n******\n' + self.file_messages[k]
844
self.file_messages[k] = v
845
if self.global_message:
846
self.global_message = global_message + '\n******\n' \
847
+ self.global_message
849
self.global_message = global_message
851
def save(self, tree, branch):
852
# We store in branch's config, which can be a problem if two gcommit
853
# are done in two checkouts of one single branch (comments overwrite
854
# each other). Ideally should be in working tree. But uncommit does
855
# not always have a working tree, though it always has a branch.
856
# 'tree' argument is for the future
857
config = branch.get_config()
858
# should it be named "gtk_" or some more neutral name ("gui_" ?) to
859
# be compatible with qbzr in the future?
860
config.set_user_option('gtk_global_commit_message', self.global_message)
861
# bencode() does not know unicode objects but set_user_option()
863
config.set_user_option(
864
'gtk_file_commit_messages',
865
bencode.bencode(self.file_messages).decode('UTF-8'))
868
def save_commit_messages(local, master, old_revno, old_revid,
869
new_revno, new_revid):
873
mgr = SavedCommitMessagesManager(None, b)
874
revid_iterator = b.repository.iter_reverse_revision_history(old_revid)
875
cur_revno = old_revno
876
new_revision_id = old_revid
877
graph = b.repository.get_graph()
878
for rev_id in revid_iterator:
879
if cur_revno == new_revno:
882
rev = b.repository.get_revision(rev_id)
883
file_info = rev.properties.get('file-info', None)
884
if file_info is None:
887
file_info = bencode.bdecode(file_info.encode('UTF-8'))
888
global_message = osutils.safe_unicode(rev.message)
889
# Concatenate comment of the uncommitted revision
890
mgr.insert(global_message, file_info)
892
parents = graph.get_parent_map([rev_id]).get(rev_id, None)