/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: Szilveszter Farkas (Phanatic)
  • Date: 2006-07-16 16:27:59 UTC
  • mto: (0.14.1 main) (93.1.1 win32.bialix)
  • mto: This revision was merged to the branch mainline in revision 83.
  • Revision ID: Szilveszter.Farkas@gmail.com-20060716162759-f3e8596921edc374
2006-07-16  Szilveszter Farkas <Szilveszter.Farkas@gmail.com>

    * Done some directory reorganization.
    * setup.py: added some basic install script
    * Began to implement the GTK UI.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
try:
18
 
    import pygtk
19
 
    pygtk.require("2.0")
20
 
except:
21
 
    pass
22
 
 
23
 
import gtk
24
 
import gobject
25
 
import pango
26
 
 
27
 
import os.path
28
 
import re
29
 
 
30
 
from bzrlib import errors, osutils
31
 
from bzrlib.trace import mutter
32
 
from bzrlib.util import bencode
33
 
 
34
 
from bzrlib.plugins.gtk import _i18n
35
 
from dialog import error_dialog, question_dialog
36
 
from errors import show_bzr_error
37
 
 
38
 
try:
39
 
    import dbus
40
 
    import dbus.glib
41
 
    have_dbus = True
42
 
except ImportError:
43
 
    have_dbus = False
44
 
 
45
 
 
46
 
def pending_revisions(wt):
47
 
    """Return a list of pending merges or None if there are none of them.
48
 
 
49
 
    Arguably this should be a core function, and
50
 
    ``bzrlib.status.show_pending_merges`` should be built on top of it.
51
 
 
52
 
    :return: [(rev, [children])]
53
 
    """
54
 
    parents = wt.get_parent_ids()
55
 
    if len(parents) < 2:
56
 
        return None
57
 
 
58
 
    # The basic pending merge algorithm uses the same algorithm as
59
 
    # bzrlib.status.show_pending_merges
60
 
    pending = parents[1:]
61
 
    branch = wt.branch
62
 
    last_revision = parents[0]
63
 
 
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])
74
 
 
75
 
    pm = []
76
 
    for merge in pending:
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
98
 
 
99
 
 
100
 
class CommitDialog(gtk.Dialog):
101
 
    """Implementation of Commit."""
102
 
 
103
 
    def __init__(self, wt, selected=None, parent=None):
104
 
        gtk.Dialog.__init__(self, title="Commit - Olive",
105
 
                                  parent=parent,
106
 
                                  flags=0,
107
 
                                  buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
108
 
        self._question_dialog = question_dialog
109
 
 
110
 
        self._wt = wt
111
 
        # TODO: Do something with this value, it is used by Olive
112
 
        #       It used to set all changes but this one to False
113
 
        self._selected = selected
114
 
        self._enable_per_file_commits = True
115
 
        self._commit_all_changes = True
116
 
        self.committed_revision_id = None # Nothing has been committed yet
117
 
 
118
 
        self.setup_params()
119
 
        self.construct()
120
 
        self.fill_in_data()
121
 
 
122
 
    def setup_params(self):
123
 
        """Setup the member variables for state."""
124
 
        self._basis_tree = self._wt.basis_tree()
125
 
        self._delta = None
126
 
        self._pending = pending_revisions(self._wt)
127
 
 
128
 
        self._is_checkout = (self._wt.branch.get_bound_location() is not None)
129
 
 
130
 
    def fill_in_data(self):
131
 
        # Now that we are built, handle changes to the view based on the state
132
 
        self._fill_in_pending()
133
 
        self._fill_in_diff()
134
 
        self._fill_in_files()
135
 
        self._fill_in_checkout()
136
 
        self._fill_in_per_file_info()
137
 
 
138
 
    def _fill_in_pending(self):
139
 
        if not self._pending:
140
 
            self._pending_box.hide()
141
 
            return
142
 
 
143
 
        # TODO: We'd really prefer this to be a nested list
144
 
        for rev, children in self._pending:
145
 
            rev_info = self._rev_to_pending_info(rev)
146
 
            self._pending_store.append([
147
 
                rev_info['revision_id'],
148
 
                rev_info['date'],
149
 
                rev_info['committer'],
150
 
                rev_info['summary'],
151
 
                ])
152
 
            for child in children:
153
 
                rev_info = self._rev_to_pending_info(child)
154
 
                self._pending_store.append([
155
 
                    rev_info['revision_id'],
156
 
                    rev_info['date'],
157
 
                    rev_info['committer'],
158
 
                    rev_info['summary'],
159
 
                    ])
160
 
        self._pending_box.show()
161
 
 
162
 
    def _fill_in_files(self):
163
 
        # We should really use add a progress bar of some kind.
164
 
        # While we fill in the view, hide the store
165
 
        store = self._files_store
166
 
        self._treeview_files.set_model(None)
167
 
 
168
 
        added = _i18n('added')
169
 
        removed = _i18n('removed')
170
 
        renamed = _i18n('renamed')
171
 
        renamed_and_modified = _i18n('renamed and modified')
172
 
        modified = _i18n('modified')
173
 
        kind_changed = _i18n('kind changed')
174
 
 
175
 
        # The store holds:
176
 
        # [file_id, real path, checkbox, display path, changes type, message]
177
 
        # iter_changes returns:
178
 
        # (file_id, (path_in_source, path_in_target),
179
 
        #  changed_content, versioned, parent, name, kind,
180
 
        #  executable)
181
 
 
182
 
        all_enabled = (self._selected is None)
183
 
        # The first entry is always the 'whole tree'
184
 
        all_iter = store.append([None, None, all_enabled, 'All Files', '', ''])
185
 
        initial_cursor = store.get_path(all_iter)
186
 
        # should we pass specific_files?
187
 
        self._wt.lock_read()
188
 
        self._basis_tree.lock_read()
189
 
        try:
190
 
            from diff import iter_changes_to_status
191
 
            for (file_id, real_path, change_type, display_path
192
 
                ) in iter_changes_to_status(self._basis_tree, self._wt):
193
 
                if self._selected and real_path != self._selected:
194
 
                    enabled = False
195
 
                else:
196
 
                    enabled = True
197
 
                item_iter = store.append([
198
 
                    file_id,
199
 
                    real_path.encode('UTF-8'),
200
 
                    enabled,
201
 
                    display_path.encode('UTF-8'),
202
 
                    change_type,
203
 
                    '', # Initial comment
204
 
                    ])
205
 
                if self._selected and enabled:
206
 
                    initial_cursor = store.get_path(item_iter)
207
 
        finally:
208
 
            self._basis_tree.unlock()
209
 
            self._wt.unlock()
210
 
 
211
 
        self._treeview_files.set_model(store)
212
 
        self._last_selected_file = None
213
 
        # This sets the cursor, which causes the expander to close, which
214
 
        # causes the _file_message_text_view to never get realized. So we have
215
 
        # to give it a little kick, or it warns when we try to grab the focus
216
 
        self._treeview_files.set_cursor(initial_cursor)
217
 
 
218
 
        def _realize_file_message_tree_view(*args):
219
 
            self._file_message_text_view.realize()
220
 
        self.connect_after('realize', _realize_file_message_tree_view)
221
 
 
222
 
    def _fill_in_diff(self):
223
 
        self._diff_view.set_trees(self._wt, self._basis_tree)
224
 
 
225
 
    def _fill_in_checkout(self):
226
 
        if not self._is_checkout:
227
 
            self._check_local.hide()
228
 
            return
229
 
        if have_dbus:
230
 
            bus = dbus.SystemBus()
231
 
            try:
232
 
                proxy_obj = bus.get_object('org.freedesktop.NetworkManager',
233
 
                                           '/org/freedesktop/NetworkManager')
234
 
            except dbus.DBusException:
235
 
                mutter("networkmanager not available.")
236
 
                self._check_local.show()
237
 
                return
238
 
            
239
 
            dbus_iface = dbus.Interface(proxy_obj,
240
 
                                        'org.freedesktop.NetworkManager')
241
 
            try:
242
 
                # 3 is the enum value for STATE_CONNECTED
243
 
                self._check_local.set_active(dbus_iface.state() != 3)
244
 
            except dbus.DBusException, e:
245
 
                # Silently drop errors. While DBus may be
246
 
                # available, NetworkManager doesn't necessarily have to be
247
 
                mutter("unable to get networkmanager state: %r" % e)
248
 
        self._check_local.show()
249
 
 
250
 
    def _fill_in_per_file_info(self):
251
 
        config = self._wt.branch.get_config()
252
 
        enable_per_file_commits = config.get_user_option('per_file_commits')
253
 
        if (enable_per_file_commits is None
254
 
            or enable_per_file_commits.lower()
255
 
                not in ('y', 'yes', 'on', 'enable', '1', 't', 'true')):
256
 
            self._enable_per_file_commits = False
257
 
        else:
258
 
            self._enable_per_file_commits = True
259
 
        if not self._enable_per_file_commits:
260
 
            self._file_message_expander.hide()
261
 
            self._global_message_label.set_markup(_i18n('<b>Commit Message</b>'))
262
 
 
263
 
    def _compute_delta(self):
264
 
        self._delta = self._wt.changes_from(self._basis_tree)
265
 
 
266
 
    def construct(self):
267
 
        """Build up the dialog widgets."""
268
 
        # The primary pane which splits it into left and right (adjustable)
269
 
        # sections.
270
 
        self._hpane = gtk.HPaned()
271
 
 
272
 
        self._construct_left_pane()
273
 
        self._construct_right_pane()
274
 
        self._construct_action_pane()
275
 
 
276
 
        self.vbox.pack_start(self._hpane)
277
 
        self._hpane.show()
278
 
        self.set_focus(self._global_message_text_view)
279
 
 
280
 
        self._construct_accelerators()
281
 
        self._set_sizes()
282
 
 
283
 
    def _set_sizes(self):
284
 
        # This seems like a reasonable default, we might like it to
285
 
        # be a bit wider, so that by default we can fit an 80-line diff in the
286
 
        # diff window.
287
 
        # Alternatively, we should be saving the last position/size rather than
288
 
        # setting it to a fixed value every time we start up.
289
 
        screen = self.get_screen()
290
 
        monitor = 0 # We would like it to be the monitor we are going to
291
 
                    # display on, but I don't know how to figure that out
292
 
                    # Only really useful for freaks like me that run dual
293
 
                    # monitor, with different sizes on the monitors
294
 
        monitor_rect = screen.get_monitor_geometry(monitor)
295
 
        width = int(monitor_rect.width * 0.66)
296
 
        height = int(monitor_rect.height * 0.66)
297
 
        self.set_default_size(width, height)
298
 
        self._hpane.set_position(300)
299
 
 
300
 
    def _construct_accelerators(self):
301
 
        group = gtk.AccelGroup()
302
 
        group.connect_group(gtk.gdk.keyval_from_name('N'),
303
 
                            gtk.gdk.CONTROL_MASK, 0, self._on_accel_next)
304
 
        self.add_accel_group(group)
305
 
 
306
 
    def _construct_left_pane(self):
307
 
        self._left_pane_box = gtk.VBox(homogeneous=False, spacing=5)
308
 
        self._construct_file_list()
309
 
        self._construct_pending_list()
310
 
 
311
 
        self._check_local = gtk.CheckButton(_i18n("_Only commit locally"),
312
 
                                            use_underline=True)
313
 
        self._left_pane_box.pack_end(self._check_local, False, False)
314
 
        self._check_local.set_active(False)
315
 
 
316
 
        self._hpane.pack1(self._left_pane_box, resize=False, shrink=False)
317
 
        self._left_pane_box.show()
318
 
 
319
 
    def _construct_right_pane(self):
320
 
        # TODO: I really want to make it so the diff view gets more space than
321
 
        # the global commit message, and the per-file commit message gets even
322
 
        # less. When I did it with wxGlade, I set it to 4 for diff, 2 for
323
 
        # commit, and 1 for file commit, and it looked good. But I don't seem
324
 
        # to have a way to do that with the gtk boxes... :( (Which is extra
325
 
        # weird since wx uses gtk on Linux...)
326
 
        self._right_pane_table = gtk.Table(rows=10, columns=1, homogeneous=False)
327
 
        self._right_pane_table.set_row_spacings(5)
328
 
        self._right_pane_table.set_col_spacings(5)
329
 
        self._right_pane_table_row = 0
330
 
        self._construct_diff_view()
331
 
        self._construct_file_message()
332
 
        self._construct_global_message()
333
 
 
334
 
        self._right_pane_table.show()
335
 
        self._hpane.pack2(self._right_pane_table, resize=True, shrink=True)
336
 
 
337
 
    def _construct_action_pane(self):
338
 
        self._button_commit = gtk.Button(_i18n("Comm_it"), use_underline=True)
339
 
        self._button_commit.connect('clicked', self._on_commit_clicked)
340
 
        self._button_commit.set_flags(gtk.CAN_DEFAULT)
341
 
        self._button_commit.show()
342
 
        self.action_area.pack_end(self._button_commit)
343
 
        self._button_commit.grab_default()
344
 
 
345
 
    def _add_to_right_table(self, widget, weight, expanding=False):
346
 
        """Add another widget to the table
347
 
 
348
 
        :param widget: The object to add
349
 
        :param weight: How many rows does this widget get to request
350
 
        :param expanding: Should expand|fill|shrink be set?
351
 
        """
352
 
        end_row = self._right_pane_table_row + weight
353
 
        options = 0
354
 
        expand_opts = gtk.EXPAND | gtk.FILL | gtk.SHRINK
355
 
        if expanding:
356
 
            options = expand_opts
357
 
        self._right_pane_table.attach(widget, 0, 1,
358
 
                                      self._right_pane_table_row, end_row,
359
 
                                      xoptions=expand_opts, yoptions=options)
360
 
        self._right_pane_table_row = end_row
361
 
 
362
 
    def _construct_file_list(self):
363
 
        self._files_box = gtk.VBox(homogeneous=False, spacing=0)
364
 
        file_label = gtk.Label(_i18n('Files'))
365
 
        # file_label.show()
366
 
        self._files_box.pack_start(file_label, expand=False)
367
 
 
368
 
        self._commit_all_files_radio = gtk.RadioButton(
369
 
            None, _i18n("Commit all changes"))
370
 
        self._files_box.pack_start(self._commit_all_files_radio, expand=False)
371
 
        self._commit_all_files_radio.show()
372
 
        self._commit_all_files_radio.connect('toggled',
373
 
            self._toggle_commit_selection)
374
 
        self._commit_selected_radio = gtk.RadioButton(
375
 
            self._commit_all_files_radio, _i18n("Only commit selected changes"))
376
 
        self._files_box.pack_start(self._commit_selected_radio, expand=False)
377
 
        self._commit_selected_radio.show()
378
 
        self._commit_selected_radio.connect('toggled',
379
 
            self._toggle_commit_selection)
380
 
        if self._pending:
381
 
            self._commit_all_files_radio.set_label(_i18n('Commit all changes*'))
382
 
            self._commit_all_files_radio.set_sensitive(False)
383
 
            self._commit_selected_radio.set_sensitive(False)
384
 
 
385
 
        scroller = gtk.ScrolledWindow()
386
 
        scroller.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
387
 
        self._treeview_files = gtk.TreeView()
388
 
        self._treeview_files.show()
389
 
        scroller.add(self._treeview_files)
390
 
        scroller.set_shadow_type(gtk.SHADOW_IN)
391
 
        scroller.show()
392
 
        self._files_box.pack_start(scroller,
393
 
                                   expand=True, fill=True)
394
 
        self._files_box.show()
395
 
        self._left_pane_box.pack_start(self._files_box)
396
 
 
397
 
        # Keep note that all strings stored in a ListStore must be UTF-8
398
 
        # strings. GTK does not support directly setting and restoring Unicode
399
 
        # objects.
400
 
        liststore = gtk.ListStore(
401
 
            gobject.TYPE_STRING,  # [0] file_id
402
 
            gobject.TYPE_STRING,  # [1] real path
403
 
            gobject.TYPE_BOOLEAN, # [2] checkbox
404
 
            gobject.TYPE_STRING,  # [3] display path
405
 
            gobject.TYPE_STRING,  # [4] changes type
406
 
            gobject.TYPE_STRING,  # [5] commit message
407
 
            )
408
 
        self._files_store = liststore
409
 
        self._treeview_files.set_model(liststore)
410
 
        crt = gtk.CellRendererToggle()
411
 
        crt.set_property('activatable', not bool(self._pending))
412
 
        crt.connect("toggled", self._toggle_commit, self._files_store)
413
 
        if self._pending:
414
 
            name = _i18n('Commit*')
415
 
        else:
416
 
            name = _i18n('Commit')
417
 
        commit_col = gtk.TreeViewColumn(name, crt, active=2)
418
 
        commit_col.set_visible(False)
419
 
        self._treeview_files.append_column(commit_col)
420
 
        self._treeview_files.append_column(gtk.TreeViewColumn(_i18n('Path'),
421
 
                                           gtk.CellRendererText(), text=3))
422
 
        self._treeview_files.append_column(gtk.TreeViewColumn(_i18n('Type'),
423
 
                                           gtk.CellRendererText(), text=4))
424
 
        self._treeview_files.connect('cursor-changed',
425
 
                                     self._on_treeview_files_cursor_changed)
426
 
 
427
 
    def _toggle_commit(self, cell, path, model):
428
 
        if model[path][0] is None: # No file_id means 'All Files'
429
 
            new_val = not model[path][2]
430
 
            for node in model:
431
 
                node[2] = new_val
432
 
        else:
433
 
            model[path][2] = not model[path][2]
434
 
 
435
 
    def _toggle_commit_selection(self, button):
436
 
        all_files = self._commit_all_files_radio.get_active()
437
 
        if self._commit_all_changes != all_files:
438
 
            checked_col = self._treeview_files.get_column(0)
439
 
            self._commit_all_changes = all_files
440
 
            if all_files:
441
 
                checked_col.set_visible(False)
442
 
            else:
443
 
                checked_col.set_visible(True)
444
 
            renderer = checked_col.get_cell_renderers()[0]
445
 
            renderer.set_property('activatable', not all_files)
446
 
 
447
 
    def _construct_pending_list(self):
448
 
        # Pending information defaults to hidden, we put it all in 1 box, so
449
 
        # that we can show/hide all of them at once
450
 
        self._pending_box = gtk.VBox()
451
 
        self._pending_box.hide()
452
 
 
453
 
        pending_message = gtk.Label()
454
 
        pending_message.set_markup(
455
 
            _i18n('<i>* Cannot select specific files when merging</i>'))
456
 
        self._pending_box.pack_start(pending_message, expand=False, padding=5)
457
 
        pending_message.show()
458
 
 
459
 
        pending_label = gtk.Label(_i18n('Pending Revisions'))
460
 
        self._pending_box.pack_start(pending_label, expand=False, padding=0)
461
 
        pending_label.show()
462
 
 
463
 
        scroller = gtk.ScrolledWindow()
464
 
        scroller.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
465
 
        self._treeview_pending = gtk.TreeView()
466
 
        scroller.add(self._treeview_pending)
467
 
        scroller.set_shadow_type(gtk.SHADOW_IN)
468
 
        scroller.show()
469
 
        self._pending_box.pack_start(scroller,
470
 
                                     expand=True, fill=True, padding=5)
471
 
        self._treeview_pending.show()
472
 
        self._left_pane_box.pack_start(self._pending_box)
473
 
 
474
 
        liststore = gtk.ListStore(gobject.TYPE_STRING, # revision_id
475
 
                                  gobject.TYPE_STRING, # date
476
 
                                  gobject.TYPE_STRING, # committer
477
 
                                  gobject.TYPE_STRING, # summary
478
 
                                 )
479
 
        self._pending_store = liststore
480
 
        self._treeview_pending.set_model(liststore)
481
 
        self._treeview_pending.append_column(gtk.TreeViewColumn(_i18n('Date'),
482
 
                                             gtk.CellRendererText(), text=1))
483
 
        self._treeview_pending.append_column(gtk.TreeViewColumn(_i18n('Committer'),
484
 
                                             gtk.CellRendererText(), text=2))
485
 
        self._treeview_pending.append_column(gtk.TreeViewColumn(_i18n('Summary'),
486
 
                                             gtk.CellRendererText(), text=3))
487
 
 
488
 
    def _construct_diff_view(self):
489
 
        from diff import DiffView
490
 
 
491
 
        # TODO: jam 2007-10-30 The diff label is currently disabled. If we
492
 
        #       decide that we really don't ever want to display it, we should
493
 
        #       actually remove it, and other references to it, along with the
494
 
        #       tests that it is set properly.
495
 
        self._diff_label = gtk.Label(_i18n('Diff for whole tree'))
496
 
        self._diff_label.set_alignment(0, 0)
497
 
        self._right_pane_table.set_row_spacing(self._right_pane_table_row, 0)
498
 
        self._add_to_right_table(self._diff_label, 1, False)
499
 
        # self._diff_label.show()
500
 
 
501
 
        self._diff_view = DiffView()
502
 
        self._add_to_right_table(self._diff_view, 4, True)
503
 
        self._diff_view.show()
504
 
 
505
 
    def _construct_file_message(self):
506
 
        scroller = gtk.ScrolledWindow()
507
 
        scroller.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
508
 
 
509
 
        self._file_message_text_view = gtk.TextView()
510
 
        scroller.add(self._file_message_text_view)
511
 
        scroller.set_shadow_type(gtk.SHADOW_IN)
512
 
        scroller.show()
513
 
 
514
 
        self._file_message_text_view.modify_font(pango.FontDescription("Monospace"))
515
 
        self._file_message_text_view.set_wrap_mode(gtk.WRAP_WORD)
516
 
        self._file_message_text_view.set_accepts_tab(False)
517
 
        self._file_message_text_view.show()
518
 
 
519
 
        self._file_message_expander = gtk.Expander(_i18n('File commit message'))
520
 
        self._file_message_expander.set_expanded(True)
521
 
        self._file_message_expander.add(scroller)
522
 
        self._add_to_right_table(self._file_message_expander, 1, False)
523
 
        self._file_message_expander.show()
524
 
 
525
 
    def _construct_global_message(self):
526
 
        self._global_message_label = gtk.Label(_i18n('Global Commit Message'))
527
 
        self._global_message_label.set_markup(
528
 
            _i18n('<b>Global Commit Message</b>'))
529
 
        self._global_message_label.set_alignment(0, 0)
530
 
        self._right_pane_table.set_row_spacing(self._right_pane_table_row, 0)
531
 
        self._add_to_right_table(self._global_message_label, 1, False)
532
 
        # Can we remove the spacing between the label and the box?
533
 
        self._global_message_label.show()
534
 
 
535
 
        scroller = gtk.ScrolledWindow()
536
 
        scroller.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
537
 
 
538
 
        self._global_message_text_view = gtk.TextView()
539
 
        self._global_message_text_view.modify_font(pango.FontDescription("Monospace"))
540
 
        scroller.add(self._global_message_text_view)
541
 
        scroller.set_shadow_type(gtk.SHADOW_IN)
542
 
        scroller.show()
543
 
        self._add_to_right_table(scroller, 2, True)
544
 
        self._file_message_text_view.set_wrap_mode(gtk.WRAP_WORD)
545
 
        self._file_message_text_view.set_accepts_tab(False)
546
 
        self._global_message_text_view.show()
547
 
 
548
 
    def _on_treeview_files_cursor_changed(self, treeview):
549
 
        treeselection = treeview.get_selection()
550
 
        (model, selection) = treeselection.get_selected()
551
 
 
552
 
        if selection is not None:
553
 
            path, display_path = model.get(selection, 1, 3)
554
 
            self._diff_label.set_text(_i18n('Diff for ') + display_path)
555
 
            if path is None:
556
 
                self._diff_view.show_diff(None)
557
 
            else:
558
 
                self._diff_view.show_diff([path.decode('UTF-8')])
559
 
            self._update_per_file_info(selection)
560
 
 
561
 
    def _on_accel_next(self, accel_group, window, keyval, modifier):
562
 
        # We don't really care about any of the parameters, because we know
563
 
        # where this message came from
564
 
        tree_selection = self._treeview_files.get_selection()
565
 
        (model, selection) = tree_selection.get_selected()
566
 
        if selection is None:
567
 
            next = None
568
 
        else:
569
 
            next = model.iter_next(selection)
570
 
 
571
 
        if next is None:
572
 
            # We have either made it to the end of the list, or nothing was
573
 
            # selected. Either way, select All Files, and jump to the global
574
 
            # commit message.
575
 
            self._treeview_files.set_cursor((0,))
576
 
            self._global_message_text_view.grab_focus()
577
 
        else:
578
 
            # Set the cursor to this entry, and jump to the per-file commit
579
 
            # message
580
 
            self._treeview_files.set_cursor(model.get_path(next))
581
 
            self._file_message_text_view.grab_focus()
582
 
 
583
 
    def _save_current_file_message(self):
584
 
        if self._last_selected_file is None:
585
 
            return # Nothing to save
586
 
        text_buffer = self._file_message_text_view.get_buffer()
587
 
        cur_text = text_buffer.get_text(text_buffer.get_start_iter(),
588
 
                                        text_buffer.get_end_iter())
589
 
        last_selected = self._files_store.get_iter(self._last_selected_file)
590
 
        self._files_store.set_value(last_selected, 5, cur_text)
591
 
 
592
 
    def _update_per_file_info(self, selection):
593
 
        # The node is changing, so cache the current message
594
 
        if not self._enable_per_file_commits:
595
 
            return
596
 
 
597
 
        self._save_current_file_message()
598
 
        text_buffer = self._file_message_text_view.get_buffer()
599
 
        file_id, display_path, message = self._files_store.get(selection, 0, 3, 5)
600
 
        if file_id is None: # Whole tree
601
 
            self._file_message_expander.set_label(_i18n('File commit message'))
602
 
            self._file_message_expander.set_expanded(False)
603
 
            self._file_message_expander.set_sensitive(False)
604
 
            text_buffer.set_text('')
605
 
            self._last_selected_file = None
606
 
        else:
607
 
            self._file_message_expander.set_label(_i18n('Commit message for ')
608
 
                                                  + display_path)
609
 
            self._file_message_expander.set_expanded(True)
610
 
            self._file_message_expander.set_sensitive(True)
611
 
            text_buffer.set_text(message)
612
 
            self._last_selected_file = self._files_store.get_path(selection)
613
 
 
614
 
    def _get_specific_files(self):
615
 
        """Return the list of selected paths, and file info.
616
 
 
617
 
        :return: ([unicode paths], [{utf-8 file info}]
618
 
        """
619
 
        self._save_current_file_message()
620
 
        files = []
621
 
        records = iter(self._files_store)
622
 
        rec = records.next() # Skip the All Files record
623
 
        assert rec[0] is None, "Are we skipping the wrong record?"
624
 
 
625
 
        file_info = []
626
 
        for record in records:
627
 
            if self._commit_all_changes or record[2]:# [2] checkbox
628
 
                file_id = record[0] # [0] file_id
629
 
                path = record[1]    # [1] real path
630
 
                file_message = record[5] # [5] commit message
631
 
                files.append(path.decode('UTF-8'))
632
 
                if self._enable_per_file_commits and file_message:
633
 
                    # All of this needs to be utf-8 information
634
 
                    file_info.append({'path':path, 'file_id':file_id,
635
 
                                     'message':file_message})
636
 
        file_info.sort(key=lambda x:(x['path'], x['file_id']))
637
 
        if self._enable_per_file_commits:
638
 
            return files, file_info
639
 
        else:
640
 
            return files, []
641
 
 
642
 
    @show_bzr_error
643
 
    def _on_commit_clicked(self, button):
644
 
        """ Commit button clicked handler. """
645
 
        self._do_commit()
646
 
 
647
 
    def _do_commit(self):
648
 
        message = self._get_global_commit_message()
649
 
 
650
 
        if message == '':
651
 
            response = self._question_dialog(
652
 
                _i18n('Commit with an empty message?'),
653
 
                _i18n('You can describe your commit intent in the message.'))
654
 
            if response == gtk.RESPONSE_NO:
655
 
                # Kindly give focus to message area
656
 
                self._global_message_text_view.grab_focus()
657
 
                return
658
 
 
659
 
        specific_files, file_info = self._get_specific_files()
660
 
        if self._pending:
661
 
            specific_files = None
662
 
 
663
 
        local = self._check_local.get_active()
664
 
 
665
 
        # All we care about is if there is a single unknown, so if this loop is
666
 
        # entered, then there are unknown files.
667
 
        # TODO: jam 20071002 It seems like this should cancel the dialog
668
 
        #       entirely, since there isn't a way for them to add the unknown
669
 
        #       files at this point.
670
 
        for path in self._wt.unknowns():
671
 
            response = self._question_dialog(
672
 
                _i18n("Commit with unknowns?"),
673
 
                _i18n("Unknown files exist in the working tree. Commit anyway?"))
674
 
            if response == gtk.RESPONSE_NO:
675
 
                return
676
 
            break
677
 
 
678
 
        rev_id = None
679
 
        revprops = {}
680
 
        if file_info:
681
 
            revprops['file-info'] = bencode.bencode(file_info).decode('UTF-8')
682
 
        try:
683
 
            rev_id = self._wt.commit(message,
684
 
                       allow_pointless=False,
685
 
                       strict=False,
686
 
                       local=local,
687
 
                       specific_files=specific_files,
688
 
                       revprops=revprops)
689
 
        except errors.PointlessCommit:
690
 
            response = self._question_dialog(
691
 
                _i18n('Commit with no changes?'),
692
 
                _i18n('There are no changes in the working tree.'
693
 
                      ' Do you want to commit anyway?'))
694
 
            if response == gtk.RESPONSE_YES:
695
 
                rev_id = self._wt.commit(message,
696
 
                               allow_pointless=True,
697
 
                               strict=False,
698
 
                               local=local,
699
 
                               specific_files=specific_files,
700
 
                               revprops=revprops)
701
 
        self.committed_revision_id = rev_id
702
 
        self.response(gtk.RESPONSE_OK)
703
 
 
704
 
    def _get_global_commit_message(self):
705
 
        buf = self._global_message_text_view.get_buffer()
706
 
        start, end = buf.get_bounds()
707
 
        return buf.get_text(start, end).decode('utf-8')
708
 
 
709
 
    def _set_global_commit_message(self, message):
710
 
        """Just a helper for the test suite."""
711
 
        if isinstance(message, unicode):
712
 
            message = message.encode('UTF-8')
713
 
        self._global_message_text_view.get_buffer().set_text(message)
714
 
 
715
 
    def _set_file_commit_message(self, message):
716
 
        """Helper for the test suite."""
717
 
        if isinstance(message, unicode):
718
 
            message = message.encode('UTF-8')
719
 
        self._file_message_text_view.get_buffer().set_text(message)
720
 
 
721
 
    @staticmethod
722
 
    def _rev_to_pending_info(rev):
723
 
        """Get the information from a pending merge."""
724
 
        from bzrlib.osutils import format_date
725
 
        rev_dict = {}
726
 
        rev_dict['committer'] = re.sub('<.*@.*>', '', rev.committer).strip(' ')
727
 
        rev_dict['summary'] = rev.get_summary()
728
 
        rev_dict['date'] = format_date(rev.timestamp,
729
 
                                       rev.timezone or 0,
730
 
                                       'original', date_fmt="%Y-%m-%d",
731
 
                                       show_offset=False)
732
 
        rev_dict['revision_id'] = rev.revision_id
733
 
        return rev_dict