/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to commit.py

  • Committer: Jelmer Vernooij
  • Date: 2011-11-30 21:45:54 UTC
  • Revision ID: jelmer@samba.org-20111130214554-5kfx0b5y7t5zh033
Extend branch preferences, show location widget.

Show diffs side-by-side

added added

removed removed

Lines of Context:
commit.py
21
21
from gi.repository import GObject
22
22
from gi.repository import Pango
23
23
 
 
24
try:
 
25
    from bzrlib import bencode
 
26
except ImportError:
 
27
    from bzrlib.util import bencode
 
28
 
24
29
from bzrlib import (
25
 
    bencode,
26
30
    errors,
27
 
    osutils,
28
 
    revision as _mod_revision,
29
31
    trace,
30
 
    tsort,
31
32
    )
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
36
from bzrlib.plugins.gtk.commitmsgs import SavedCommitMessagesManager
43
43
    have_dbus = False
44
44
 
45
45
 
46
 
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
47
 
    """Get an iterator which will return the revisions in merge sorted order.
48
 
 
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'
51
 
    order.
52
 
 
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()
57
 
    """
58
 
    # MergeSorter requires that all nodes be present in the graph, so get rid
59
 
    # of any references pointing outside of this graph.
60
 
    parent_graph = {}
61
 
    for revision_id in revision_ids:
62
 
        if revision_id not in parent_map: # ghost
63
 
            parent_graph[revision_id] = []
64
 
        else:
65
 
            # Only include parents which are in this sub-graph
66
 
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
67
 
                                            if p in revision_ids]
68
 
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
69
 
    return sorter.iter_topo_order()
70
 
 
71
 
 
72
46
def pending_revisions(wt):
73
47
    """Return a list of pending merges or None if there are none of them.
74
48
 
79
53
    """
80
54
    parents = wt.get_parent_ids()
81
55
    if len(parents) < 2:
82
 
        return
 
56
        return None
83
57
 
84
58
    # The basic pending merge algorithm uses the same algorithm as
85
59
    # bzrlib.status.show_pending_merges
87
61
    branch = wt.branch
88
62
    last_revision = parents[0]
89
63
 
90
 
    graph = branch.repository.get_graph()
91
 
    other_revisions = [last_revision]
 
64
    if last_revision is not None:
 
65
        try:
 
66
            ignore = set(branch.repository.get_ancestry(last_revision,
 
67
                                                        topo_sorted=False))
 
68
        except errors.NoSuchRevision:
 
69
            # the last revision is a ghost : assume everything is new
 
70
            # except for it
 
71
            ignore = set([None, last_revision])
 
72
    else:
 
73
        ignore = set([None])
92
74
 
93
75
    pm = []
94
76
    for merge in pending:
95
 
        try:
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)
101
 
            continue
102
 
 
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)
108
 
 
109
 
        # Get a handle to all of the revisions we will need
110
 
        try:
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
115
 
            revisions = {}
116
 
            for revision_id in merge_extra:
117
 
                try:
118
 
                    rev = branch.repository.get_revisions([revision_id])[0]
119
 
                except errors.NoSuchRevision:
120
 
                    revisions[revision_id] = None
121
 
                else:
122
 
                    revisions[revision_id] = rev
123
 
 
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()
129
 
        if first != merge:
130
 
            raise AssertionError('Somehow we misunderstood how'
131
 
                ' iter_topo_order works %s != %s' % (first, merge))
132
 
        children = []
133
 
        for num, sub_merge, depth, eom in rev_id_iterator:
134
 
            rev = revisions[sub_merge]
135
 
            if rev is None:
136
 
                trace.warning("ghost: %r", sub_merge)
137
 
                continue
138
 
            children.append(rev)
139
 
        yield (merge_rev, children)
 
77
        ignore.add(merge)
 
78
        try:
 
79
            rev = branch.repository.get_revision(merge)
 
80
            children = []
 
81
            pm.append((rev, children))
 
82
 
 
83
            # This does need to be topo sorted, so we search backwards
 
84
            inner_merges = branch.repository.get_ancestry(merge)
 
85
            assert inner_merges[0] is None
 
86
            inner_merges.pop(0)
 
87
            for mmerge in reversed(inner_merges):
 
88
                if mmerge in ignore:
 
89
                    continue
 
90
                rev = branch.repository.get_revision(mmerge)
 
91
                children.append(rev)
 
92
 
 
93
                ignore.add(mmerge)
 
94
        except errors.NoSuchRevision:
 
95
            print "DEBUG: NoSuchRevision:", merge
 
96
 
 
97
    return pm
140
98
 
141
99
 
142
100
_newline_variants_re = re.compile(r'\r\n?')
143
101
def _sanitize_and_decode_message(utf8_message):
144
102
    """Turn a utf-8 message into a sanitized Unicode message."""
145
103
    fixed_newline = _newline_variants_re.sub('\n', utf8_message)
146
 
    return osutils.safe_unicode(fixed_newline)
 
104
    return fixed_newline.decode('utf-8')
147
105
 
148
106
 
149
107
class CommitDialog(Gtk.Dialog):
164
122
        self._enable_per_file_commits = True
165
123
        self._commit_all_changes = True
166
124
        self.committed_revision_id = None # Nothing has been committed yet
167
 
        self._last_selected_file = None
168
125
        self._saved_commit_messages_manager = SavedCommitMessagesManager(
169
126
            self._wt, self._wt.branch)
170
127
 
176
133
        """Setup the member variables for state."""
177
134
        self._basis_tree = self._wt.basis_tree()
178
135
        self._delta = None
179
 
        self._wt.lock_read()
180
 
        try:
181
 
            self._pending = list(pending_revisions(self._wt))
182
 
        finally:
183
 
            self._wt.unlock()
 
136
        self._pending = pending_revisions(self._wt)
184
137
 
185
138
        self._is_checkout = (self._wt.branch.get_bound_location() is not None)
186
139
 
238
191
 
239
192
        all_enabled = (self._selected is None)
240
193
        # The first entry is always the 'whole tree'
241
 
        all_iter = store.append(["", "", all_enabled, 'All Files', '', ''])
 
194
        all_iter = store.append([None, None, all_enabled, 'All Files', '', ''])
242
195
        initial_cursor = store.get_path(all_iter)
243
196
        # should we pass specific_files?
244
197
        self._wt.lock_read()
289
242
            self._check_local.hide()
290
243
            return
291
244
        if have_dbus:
292
 
            try:
293
 
                bus = dbus.SystemBus()
294
 
            except dbus.DBusException:
295
 
                trace.mutter("DBus system bus not available")
296
 
                self._check_local.show()
297
 
                return
 
245
            bus = dbus.SystemBus()
298
246
            try:
299
247
                proxy_obj = bus.get_object('org.freedesktop.NetworkManager',
300
248
                                           '/org/freedesktop/NetworkManager')
302
250
                trace.mutter("networkmanager not available.")
303
251
                self._check_local.show()
304
252
                return
305
 
 
 
253
            
306
254
            dbus_iface = dbus.Interface(proxy_obj,
307
255
                                        'org.freedesktop.NetworkManager')
308
256
            try:
334
282
        """Build up the dialog widgets."""
335
283
        # The primary pane which splits it into left and right (adjustable)
336
284
        # sections.
337
 
        self._hpane = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL)
 
285
        self._hpane = Gtk.HPaned()
338
286
 
339
287
        self._construct_left_pane()
340
288
        self._construct_right_pane()
500
448
                                     self._on_treeview_files_cursor_changed)
501
449
 
502
450
    def _toggle_commit(self, cell, path, model):
503
 
        if model[path][0] == "": # No file_id means 'All Files'
 
451
        if model[path][0] is None: # No file_id means 'All Files'
504
452
            new_val = not model[path][2]
505
453
            for node in model:
506
454
                node[2] = new_val
560
508
                                             Gtk.CellRendererText(), text=3))
561
509
 
562
510
    def _construct_diff_view(self):
 
511
        from bzrlib.plugins.gtk.diff import DiffView
 
512
 
563
513
        # TODO: jam 2007-10-30 The diff label is currently disabled. If we
564
514
        #       decide that we really don't ever want to display it, we should
565
515
        #       actually remove it, and other references to it, along with the
574
524
        self._add_to_right_table(self._diff_view, 4, True)
575
525
        self._diff_view.show()
576
526
 
577
 
    @staticmethod
578
 
    def get_line_height(widget):
579
 
        pango_layout = widget.create_pango_layout("X");
580
 
        ink_rectangle, logical_rectangle = pango_layout.get_pixel_extents()
581
 
        return logical_rectangle.height
582
 
 
583
527
    def _construct_file_message(self):
584
528
        scroller = Gtk.ScrolledWindow()
585
529
        scroller.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
592
536
        self._file_message_text_view.modify_font(Pango.FontDescription("Monospace"))
593
537
        self._file_message_text_view.set_wrap_mode(Gtk.WrapMode.WORD)
594
538
        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
539
        self._file_message_text_view.show()
598
540
 
599
541
        self._file_message_expander = Gtk.Expander(
629
571
 
630
572
    def _on_treeview_files_cursor_changed(self, treeview):
631
573
        treeselection = treeview.get_selection()
632
 
        if treeselection is None:
633
 
            # The treeview was probably destroyed as the dialog closes.
634
 
            return
635
574
        (model, selection) = treeselection.get_selected()
636
575
 
637
576
        if selection is not None:
638
577
            path, display_path = model.get(selection, 1, 3)
639
578
            self._diff_label.set_text(_i18n('Diff for ') + display_path)
640
 
            if path == "":
 
579
            if path is None:
641
580
                self._diff_view.show_diff(None)
642
581
            else:
643
 
                self._diff_view.show_diff([osutils.safe_unicode(path)])
 
582
                self._diff_view.show_diff([path.decode('UTF-8')])
644
583
            self._update_per_file_info(selection)
645
584
 
646
585
    def _on_accel_next(self, accel_group, window, keyval, modifier):
658
597
            # selected. Either way, select All Files, and jump to the global
659
598
            # commit message.
660
599
            self._treeview_files.set_cursor(
661
 
                Gtk.TreePath(path=0), "", False)
 
600
                Gtk.TreePath(path=0), None, False)
662
601
            self._global_message_text_view.grab_focus()
663
602
        else:
664
603
            # Set the cursor to this entry, and jump to the per-file commit
683
622
        self._save_current_file_message()
684
623
        text_buffer = self._file_message_text_view.get_buffer()
685
624
        file_id, display_path, message = self._files_store.get(selection, 0, 3, 5)
686
 
        if file_id == "": # Whole tree
 
625
        if file_id is None: # Whole tree
687
626
            self._file_message_expander.set_label(_i18n('File commit message'))
688
627
            self._file_message_expander.set_expanded(False)
689
628
            self._file_message_expander.set_sensitive(False)
706
645
        files = []
707
646
        records = iter(self._files_store)
708
647
        rec = records.next() # Skip the All Files record
709
 
        assert rec[0] == "", "Are we skipping the wrong record?"
 
648
        assert rec[0] is None, "Are we skipping the wrong record?"
710
649
 
711
650
        file_info = []
712
651
        for record in records:
713
652
            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
 
653
                file_id = record[0] # [0] file_id
 
654
                path = record[1]    # [1] real path
716
655
                # [5] commit message
717
656
                file_message = _sanitize_and_decode_message(record[5])
718
657
                files.append(path.decode('UTF-8'))