19
from gi.repository import Gdk
20
19
from gi.repository import Gtk
21
20
from gi.repository import GObject
22
21
from gi.repository import Pango
24
23
from bzrlib import (
28
revision as _mod_revision,
29
from bzrlib import bencode
31
from bzrlib.util import bencode
32
33
from bzrlib.plugins.gtk.dialog import question_dialog
33
from bzrlib.plugins.gtk.diff import DiffView
34
34
from bzrlib.plugins.gtk.errors import show_bzr_error
35
35
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
45
def pending_revisions(wt):
73
46
"""Return a list of pending merges or None if there are none of them.
88
61
last_revision = parents[0]
90
graph = branch.repository.get_graph()
91
other_revisions = [last_revision]
63
if last_revision is not None:
65
ignore = set(branch.repository.get_ancestry(last_revision,
67
except errors.NoSuchRevision:
68
# the last revision is a ghost : assume everything is new
70
ignore = set([None, last_revision])
94
75
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)
78
rev = branch.repository.get_revision(merge)
80
pm.append((rev, children))
82
# This does need to be topo sorted, so we search backwards
83
inner_merges = branch.repository.get_ancestry(merge)
84
assert inner_merges[0] is None
86
for mmerge in reversed(inner_merges):
89
rev = branch.repository.get_revision(mmerge)
93
except errors.NoSuchRevision:
94
print "DEBUG: NoSuchRevision:", merge
142
99
_newline_variants_re = re.compile(r'\r\n?')
143
100
def _sanitize_and_decode_message(utf8_message):
144
101
"""Turn a utf-8 message into a sanitized Unicode message."""
145
102
fixed_newline = _newline_variants_re.sub('\n', utf8_message)
146
return osutils.safe_unicode(fixed_newline)
103
return fixed_newline.decode('utf-8')
149
106
class CommitDialog(Gtk.Dialog):
150
107
"""Implementation of Commit."""
152
109
def __init__(self, wt, selected=None, parent=None):
153
super(CommitDialog, self).__init__(
154
title="Commit to %s" % wt.basedir, parent=parent, flags=0)
110
GObject.GObject.__init__(self, title="Commit to %s" % wt.basedir,
111
parent=parent, flags=0,)
155
112
self.connect('delete-event', self._on_delete_window)
156
113
self._question_dialog = question_dialog
334
280
"""Build up the dialog widgets."""
335
281
# The primary pane which splits it into left and right (adjustable)
337
self._hpane = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL)
283
self._hpane = Gtk.HPaned()
339
285
self._construct_left_pane()
340
286
self._construct_right_pane()
341
287
self._construct_action_pane()
343
self.get_content_area().pack_start(self._hpane, True, True, 0)
289
self.vbox.pack_start(self._hpane, True, True, 0)
344
290
self._hpane.show()
345
291
self.set_focus(self._global_message_text_view)
408
354
self._button_cancel = Gtk.Button(stock=Gtk.STOCK_CANCEL)
409
355
self._button_cancel.connect('clicked', self._on_cancel_clicked)
410
356
self._button_cancel.show()
411
self.get_action_area().pack_end(
412
self._button_cancel, True, True, 0)
357
self.action_area.pack_end(self._button_cancel)
413
358
self._button_commit = Gtk.Button(_i18n("Comm_it"), use_underline=True)
414
359
self._button_commit.connect('clicked', self._on_commit_clicked)
415
360
self._button_commit.set_can_default(True)
416
361
self._button_commit.show()
417
self.get_action_area().pack_end(
418
self._button_commit, True, True, 0)
362
self.action_area.pack_end(self._button_commit)
419
363
self._button_commit.grab_default()
421
365
def _add_to_right_table(self, widget, weight, expanding=False):
441
385
# file_label.show()
442
386
self._files_box.pack_start(file_label, False, True, 0)
444
self._commit_all_files_radio = Gtk.RadioButton.new_with_label(
388
self._commit_all_files_radio = Gtk.RadioButton(
445
389
None, _i18n("Commit all changes"))
446
390
self._files_box.pack_start(self._commit_all_files_radio, False, True, 0)
447
391
self._commit_all_files_radio.show()
448
392
self._commit_all_files_radio.connect('toggled',
449
393
self._toggle_commit_selection)
450
self._commit_selected_radio = Gtk.RadioButton.new_with_label_from_widget(
394
self._commit_selected_radio = Gtk.RadioButton(
451
395
self._commit_all_files_radio, _i18n("Only commit selected changes"))
452
396
self._files_box.pack_start(self._commit_selected_radio, False, True, 0)
453
397
self._commit_selected_radio.show()
528
473
pending_message = Gtk.Label()
529
474
pending_message.set_markup(
530
475
_i18n('<i>* Cannot select specific files when merging</i>'))
531
self._pending_box.pack_start(pending_message, False, True, 5)
476
self._pending_box.pack_start(pending_message, expand=False, padding=5)
532
477
pending_message.show()
534
479
pending_label = Gtk.Label(label=_i18n('Pending Revisions'))
535
self._pending_box.pack_start(pending_label, False, True, 0)
480
self._pending_box.pack_start(pending_label, expand=False, padding=0)
536
481
pending_label.show()
538
483
scroller = Gtk.ScrolledWindow()
592
534
self._file_message_text_view.modify_font(Pango.FontDescription("Monospace"))
593
535
self._file_message_text_view.set_wrap_mode(Gtk.WrapMode.WORD)
594
536
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
537
self._file_message_text_view.show()
599
self._file_message_expander = Gtk.Expander(
600
label=_i18n('File commit message'))
539
self._file_message_expander = Gtk.Expander(_i18n('File commit message'))
601
540
self._file_message_expander.set_expanded(True)
602
541
self._file_message_expander.add(scroller)
603
542
self._add_to_right_table(self._file_message_expander, 1, False)
630
569
def _on_treeview_files_cursor_changed(self, treeview):
631
570
treeselection = treeview.get_selection()
632
if treeselection is None:
633
# The treeview was probably destroyed as the dialog closes.
635
571
(model, selection) = treeselection.get_selected()
637
573
if selection is not None:
638
574
path, display_path = model.get(selection, 1, 3)
639
575
self._diff_label.set_text(_i18n('Diff for ') + display_path)
641
577
self._diff_view.show_diff(None)
643
self._diff_view.show_diff([osutils.safe_unicode(path)])
579
self._diff_view.show_diff([path.decode('UTF-8')])
644
580
self._update_per_file_info(selection)
646
582
def _on_accel_next(self, accel_group, window, keyval, modifier):
657
593
# We have either made it to the end of the list, or nothing was
658
594
# selected. Either way, select All Files, and jump to the global
659
595
# commit message.
660
self._treeview_files.set_cursor(
661
Gtk.TreePath(path=0), "", False)
596
self._treeview_files.set_cursor((0,))
662
597
self._global_message_text_view.grab_focus()
664
599
# 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)
601
self._treeview_files.set_cursor(model.get_path(next))
667
602
self._file_message_text_view.grab_focus()
669
604
def _save_current_file_message(self):
707
642
records = iter(self._files_store)
708
643
rec = records.next() # Skip the All Files record
709
assert rec[0] == "", "Are we skipping the wrong record?"
644
assert rec[0] is None, "Are we skipping the wrong record?"
712
647
for record in records:
713
648
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
649
file_id = record[0] # [0] file_id
650
path = record[1] # [1] real path
716
651
# [5] commit message
717
652
file_message = _sanitize_and_decode_message(record[5])
718
653
files.append(path.decode('UTF-8'))
854
789
rev_dict['revision_id'] = rev.revision_id
793
class SavedCommitMessagesManager:
794
"""Save glogal and per-file commit messages.
796
Saves global commit message and utf-8 file_id->message dictionary
797
of per-file commit messages on disk. Re-reads them later for re-using.
800
def __init__(self, tree=None, branch=None):
801
"""If branch is None, builds empty messages, otherwise reads them
802
from branch's disk storage. 'tree' argument is for the future."""
804
self.global_message = u''
805
self.file_messages = {}
807
config = branch.get_config()
808
self.global_message = config.get_user_option(
809
'gtk_global_commit_message')
810
if self.global_message is None:
811
self.global_message = u''
812
file_messages = config.get_user_option('gtk_file_commit_messages')
813
if file_messages: # unicode and B-encoded:
814
self.file_messages = bencode.bdecode(
815
file_messages.encode('UTF-8'))
817
self.file_messages = {}
820
return self.global_message, self.file_messages
822
def is_not_empty(self):
823
return bool(self.global_message or self.file_messages)
825
def insert(self, global_message, file_info):
826
"""Formats per-file commit messages (list of dictionaries, one per file)
827
into one utf-8 file_id->message dictionary and merges this with
828
previously existing dictionary. Merges global commit message too."""
831
file_message = fi['message']
833
file_messages[fi['file_id']] = file_message # utf-8 strings
834
for k,v in file_messages.iteritems():
836
self.file_messages[k] = v + '\n******\n' + self.file_messages[k]
838
self.file_messages[k] = v
839
if self.global_message:
840
self.global_message = global_message + '\n******\n' \
841
+ self.global_message
843
self.global_message = global_message
845
def save(self, tree, branch):
846
# We store in branch's config, which can be a problem if two gcommit
847
# are done in two checkouts of one single branch (comments overwrite
848
# each other). Ideally should be in working tree. But uncommit does
849
# not always have a working tree, though it always has a branch.
850
# 'tree' argument is for the future
851
config = branch.get_config()
852
# should it be named "gtk_" or some more neutral name ("gui_" ?) to
853
# be compatible with qbzr in the future?
854
config.set_user_option('gtk_global_commit_message', self.global_message)
855
# bencode() does not know unicode objects but set_user_option()
857
config.set_user_option(
858
'gtk_file_commit_messages',
859
bencode.bencode(self.file_messages).decode('UTF-8'))
862
def save_commit_messages(local, master, old_revno, old_revid,
863
new_revno, new_revid):
867
mgr = SavedCommitMessagesManager(None, b)
868
revid_iterator = b.repository.iter_reverse_revision_history(old_revid)
869
cur_revno = old_revno
870
new_revision_id = old_revid
871
graph = b.repository.get_graph()
872
for rev_id in revid_iterator:
873
if cur_revno == new_revno:
876
rev = b.repository.get_revision(rev_id)
877
file_info = rev.properties.get('file-info', None)
878
if file_info is None:
881
file_info = bencode.bdecode(file_info.encode('UTF-8'))
882
global_message = osutils.safe_unicode(rev.message)
883
# Concatenate comment of the uncommitted revision
884
mgr.insert(global_message, file_info)
886
parents = graph.get_parent_map([rev_id]).get(rev_id, None)