14
15
# along with this program; if not, write to the Free Software
 
15
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
21
20
    pygtk.require("2.0")
 
31
 
import olive.backend.commit as commit
 
32
 
import olive.backend.errors as errors
 
33
 
import olive.backend.info as info
 
36
 
    """ Display Push dialog and perform the needed actions. """
 
37
 
    def __init__(self, gladefile, comm, dialog):
 
 
26
from errors import show_bzr_error
 
 
28
# FIXME: This needs to be public JRV 20070714
 
 
29
from bzrlib.builtins import _create_prefix
 
 
30
from bzrlib.config import LocationConfig
 
 
31
import bzrlib.errors as errors
 
 
33
from dialog import error_dialog, info_dialog, question_dialog
 
 
35
from history import UrlHistory
 
 
37
class PushDialog(gtk.Dialog):
 
 
38
    """ New implementation of the Push dialog. """
 
 
39
    def __init__(self, repository, revid, branch=None, parent=None):
 
38
40
        """ Initialize the Push dialog. """
 
39
 
        self.gladefile = gladefile
 
40
 
        self.glade = gtk.glade.XML(self.gladefile, 'window_push')
 
42
 
        # Communication object
 
47
 
        self.window = self.glade.get_widget('window_push')
 
49
 
        # Dictionary for signal_autoconnect
 
50
 
        dic = { "on_button_push_push_clicked": self.push,
 
51
 
                "on_button_push_cancel_clicked": self.close,
 
52
 
                "on_button_push_test_clicked": self.test,
 
53
 
                "on_radiobutton_push_stored_toggled": self.stored_toggled,
 
54
 
                "on_radiobutton_push_specific_toggled": self.specific_toggled, }
 
56
 
        # Connect the signals to the handlers
 
57
 
        self.glade.signal_autoconnect(dic)
 
59
 
        # Get some useful widgets
 
60
 
        self.radio_stored = self.glade.get_widget('radiobutton_push_stored')
 
61
 
        self.radio_specific = self.glade.get_widget('radiobutton_push_specific')
 
62
 
        self.entry_stored = self.glade.get_widget('entry_push_stored')
 
63
 
        self.entry_location = self.glade.get_widget('entry_push_location')
 
64
 
        self.check_remember = self.glade.get_widget('checkbutton_push_remember')
 
65
 
        self.check_overwrite = self.glade.get_widget('checkbutton_push_overwrite')
 
66
 
        self.check_create = self.glade.get_widget('checkbutton_push_create')
 
67
 
        self.label_test = self.glade.get_widget('label_push_test')
 
68
 
        self.image_test = self.glade.get_widget('image_push_test')
 
71
 
        self.entry_location.set_sensitive(0)
 
72
 
        self.check_remember.set_sensitive(0)
 
73
 
        self.check_create.set_sensitive(0)
 
76
 
        self.notbranch = False
 
 
41
        gtk.Dialog.__init__(self, title="Push - Olive",
 
 
44
                                  buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
 
 
47
        self.repository = repository
 
 
52
        self._label_location = gtk.Label(_("Location:"))
 
 
53
        self._combo = gtk.ComboBoxEntry()
 
 
54
        self._button_push = gtk.Button(_("_Push"), use_underline=True)
 
 
55
        self._hbox_location = gtk.HBox()
 
 
58
        self._button_push.connect('clicked', self._on_push_clicked)
 
 
61
        self._label_location.set_alignment(0, 0.5)
 
 
62
        self._hbox_location.set_spacing(3)
 
 
63
        self.vbox.set_spacing(3)
 
 
66
        self._hbox_location.pack_start(self._label_location, False, False)
 
 
67
        self._hbox_location.pack_start(self._combo, True, True)
 
 
68
        self.vbox.pack_start(self._hbox_location)
 
 
69
        self.action_area.pack_end(self._button_push)
 
 
74
        # Build location history
 
 
75
        self._history = UrlHistory(self.branch.get_config(), 'push_history')
 
 
78
    def _build_history(self):
 
 
79
        """ Build up the location history. """
 
 
80
        self._combo_model = gtk.ListStore(str)
 
 
81
        for item in self._history.get_entries():
 
 
82
            self._combo_model.append([ item ])
 
 
83
        self._combo.set_model(self._combo_model)
 
 
84
        self._combo.set_text_column(0)
 
 
86
        if self.branch is not None:
 
 
87
            location = self.branch.get_push_location()
 
 
88
            if location is not None:
 
 
89
                self._combo.get_child().set_text(location)
 
 
92
    def _on_push_clicked(self, widget):
 
 
93
        """ Push button clicked handler. """
 
 
94
        location = self._combo.get_child().get_text()
 
 
96
        if self.branch is not None and self.branch.get_push_location() is None:
 
 
97
            response = question_dialog(_('Set default push location'),
 
 
98
                                       _('There is no default push location set.\nSet %r as default now?') % location)
 
 
99
            if response == gtk.REPONSE_OK:
 
 
100
                self.branch.set_push_location(location)
 
78
 
            loc = info.get_push_location(self.comm.get_path())
 
79
 
        except errors.NotBranchError:
 
 
103
            revs = do_push(self.branch, location=location, overwrite=False)
 
 
104
        except errors.DivergedBranches:
 
 
105
            response = question_dialog(_('Branches have been diverged'),
 
 
106
                                       _('You cannot push if branches have diverged.\nOverwrite?'))
 
 
107
            if response == gtk.RESPONSE_OK:
 
 
108
                revs = do_push(self.branch, location=location, overwrite=True)
 
84
 
            self.entry_stored.set_text(loc)
 
87
 
        """ Display the Push dialog. """
 
89
 
            self.dialog.error_dialog(_('Directory is not a branch'),
 
90
 
                                     _('You can perform this action only in a branch.'))
 
94
 
            self.width, self.height = self.window.get_size()
 
96
 
    def stored_toggled(self, widget):
 
97
 
        if widget.get_active():
 
98
 
            self.entry_stored.set_sensitive(1)
 
99
 
            self.entry_location.set_sensitive(0)
 
100
 
            self.check_remember.set_sensitive(0)
 
101
 
            self.check_create.set_sensitive(0)
 
103
 
            self.entry_stored.set_sensitive(0)
 
104
 
            self.entry_location.set_sensitive(1)
 
105
 
            self.check_remember.set_sensitive(1)
 
106
 
            self.check_create.set_sensitive(1)
 
108
 
    def specific_toggled(self, widget):
 
109
 
        if widget.get_active():
 
110
 
            self.entry_stored.set_sensitive(0)
 
111
 
            self.entry_location.set_sensitive(1)
 
112
 
            self.check_remember.set_sensitive(1)
 
113
 
            self.check_create.set_sensitive(1)
 
115
 
            self.entry_stored.set_sensitive(1)
 
116
 
            self.entry_location.set_sensitive(0)
 
117
 
            self.check_remember.set_sensitive(0)
 
118
 
            self.check_create.set_sensitive(0)
 
120
 
    def push(self, widget):
 
122
 
        self.comm.set_busy(self.window)
 
123
 
        if self.radio_stored.get_active():
 
125
 
                revs = commit.push(self.comm.get_path(),
 
126
 
                                   overwrite=self.check_overwrite.get_active())
 
127
 
            except errors.NotBranchError:
 
128
 
                self.dialog.error_dialog(_('Directory is not a branch'),
 
129
 
                                         _('You can perform this action only in a branch.'))
 
131
 
            except errors.NoLocationKnown:
 
132
 
                self.dialog.error_dialog(_('Push location is unknown'),
 
133
 
                                         _('Please specify a location manually.'))
 
135
 
            except errors.NonExistingParent, errmsg:
 
136
 
                self.dialog.error_dialog(_('Non existing parent directory'),
 
137
 
                                         _("The parent directory (%s)\ndoesn't exist.") % errmsg)
 
139
 
            except errors.DivergedBranchesError:
 
140
 
                self.dialog.error_dialog(_('Branches have been diverged'),
 
141
 
                                         _('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
 
145
 
        elif self.radio_specific.get_active():
 
146
 
            location = self.entry_location.get_text()
 
148
 
                self.dialog.error_dialog(_('No location specified'),
 
149
 
                                         _('Please specify a location or use the default.'))
 
153
 
                revs = commit.push(self.comm.get_path(), location,
 
154
 
                                   self.check_remember.get_active(),
 
155
 
                                   self.check_overwrite.get_active(),
 
156
 
                                   self.check_create.get_active())
 
157
 
            except errors.NotBranchError:
 
158
 
                self.dialog.error_dialog(_('Directory is not a branch'),
 
159
 
                                         _('You can perform this action only in a branch.'))
 
160
 
                self.comm.set_busy(self.window, False)
 
162
 
            except errors.NonExistingParent, errmsg:
 
163
 
                self.dialog.error_dialog(_('Non existing parent directory'),
 
164
 
                                         _("The parent directory (%s)\ndoesn't exist.") % errmsg)
 
165
 
                self.comm.set_busy(self.window, False)
 
167
 
            except errors.DivergedBranchesError:
 
168
 
                self.dialog.error_dialog(_('Branches have been diverged'),
 
169
 
                                         _('You cannot push if branches have diverged. Use the\noverwrite option if you want to push anyway.'))
 
170
 
                self.comm.set_busy(self.window, False)
 
172
 
            except errors.PathPrefixNotCreated:
 
173
 
                self.dialog.error_dialog(_('Path prefix not created'),
 
174
 
                                         _("The path leading up to the specified location couldn't\nbe created."))
 
175
 
                self.comm.set_busy(self.window, False)
 
180
 
            # This should really never happen
 
184
 
        self.dialog.info_dialog(_('Push successful'),
 
185
 
                                _('%d revision(s) pushed.') % revs)
 
187
 
    def test(self, widget):
 
188
 
        """ Test if write access possible. """
 
190
 
        _urlRE = re.compile(r'^(?P<proto>[^:/\\]+)://(?P<path>.*)$')
 
192
 
        if self.radio_stored.get_active():
 
193
 
            url = self.entry_stored.get_text()
 
194
 
        elif self.radio_specific.get_active():
 
195
 
            url = self.entry_location.get_text()
 
197
 
        m = _urlRE.match(url)
 
199
 
            proto = m.groupdict()['proto']
 
200
 
            if (proto == 'sftp') or (proto == 'file') or (proto == 'ftp'):
 
201
 
                # have write acces (most probably)
 
202
 
                self.image_test.set_from_stock(gtk.STOCK_YES, 4)
 
203
 
                self.label_test.set_markup(_('<b>Write access is probably available</b>'))
 
 
111
        self._history.add_entry(location)
 
 
112
        info_dialog(_('Push successful'),
 
 
113
                    _("%d revision(s) pushed.") % revs)
 
 
115
        self.response(gtk.RESPONSE_OK)
 
 
117
def do_push(br_from, location, overwrite):
 
 
118
    """ Update a mirror of a branch.
 
 
120
    :param br_from: the source branch
 
 
122
    :param location: the location of the branch that you'd like to update
 
 
124
    :param overwrite: overwrite target location if it diverged
 
 
126
    :return: number of revisions pushed
 
 
128
    from bzrlib.bzrdir import BzrDir
 
 
129
    from bzrlib.transport import get_transport
 
 
131
    transport = get_transport(location)
 
 
132
    location_url = transport.base
 
 
137
        dir_to = BzrDir.open(location_url)
 
 
138
        br_to = dir_to.open_branch()
 
 
139
    except errors.NotBranchError:
 
 
141
        transport = transport.clone('..')
 
 
143
            relurl = transport.relpath(location_url)
 
 
144
            transport.mkdir(relurl)
 
 
145
        except errors.NoSuchFile:
 
 
146
            response = question_dialog(_('Non existing parent directory'),
 
 
147
                         _("The parent directory (%s)\ndoesn't exist. Create?") % location)
 
 
148
            if response == gtk.RESPONSE_OK:
 
 
149
                _create_prefix(transport)
 
206
 
                self.image_test.set_from_stock(gtk.STOCK_NO, 4)
 
207
 
                self.label_test.set_markup(_('<b>No write access</b>'))
 
 
152
        dir_to = br_from.bzrdir.clone(location_url,
 
 
153
            revision_id=br_from.last_revision())
 
 
154
        br_to = dir_to.open_branch()
 
 
155
        count = len(br_to.revision_history())
 
 
157
        old_rh = br_to.revision_history()
 
 
159
            tree_to = dir_to.open_workingtree()
 
 
160
        except errors.NotLocalUrl:
 
 
161
            # FIXME - what to do here? how should we warn the user?
 
 
162
            count = br_to.pull(br_from, overwrite)
 
 
163
        except errors.NoWorkingTree:
 
 
164
            count = br_to.pull(br_from, overwrite)
 
210
 
            self.image_test.set_from_stock(gtk.STOCK_DIALOG_QUESTION, 4)
 
211
 
            self.label_test.set_markup(_('<b>Could not determine</b>'))
 
213
 
    def close(self, widget=None):
 
214
 
        self.window.destroy()
 
 
166
            count = tree_to.pull(br_from, overwrite)