22
22
    pygtk.require("2.0")
 
31
 
import olive.backend.fileops as fileops
 
32
 
import olive.backend.errors as errors
 
 
28
import bzrlib.errors as errors
 
 
29
from bzrlib.workingtree import WorkingTree
 
 
31
from bzrlib.plugins.gtk.dialog import error_dialog, info_dialog, warning_dialog
 
 
32
from errors import show_bzr_error
 
34
33
from launch import launch
 
 
34
from olive import Preferences, DiffWindow
 
37
37
    """ This class is responsible for building the context menus. """
 
38
 
    def __init__(self, gladefile, comm, dialog):
 
 
38
    def __init__(self, path, selected, app=None):
 
40
 
        if sys.platform == 'win32':
 
41
 
            self.uifile = os.path.dirname(sys.executable) + "/share/olive/cmenu.ui"
 
43
 
            self.uifile = "/usr/share/olive/cmenu.ui"
 
45
 
        if not os.path.exists(self.uifile):
 
46
 
            # Load from current directory if not installed
 
47
 
            self.uifile = "cmenu.ui"
 
49
 
            if not os.path.exists(self.uifile):
 
51
 
                print _('UI description file cannot be found.')
 
54
 
        self.gladefile = gladefile
 
 
40
        from guifiles import UIFILENAME
 
 
42
        self.uifile = UIFILENAME
 
 
45
        self.pref = Preferences()
 
 
49
        self.selected = selected
 
58
52
        # Create the file list context menu
 
59
53
        self.ui = gtk.UIManager()
 
 
113
115
        self.toolbar_diff = self.ui.get_widget('/toolbar_diff')
 
116
 
        commit_menu = self.ui.get_widget('/context_right/commit')
 
117
 
        commit_image = self.comm.menuitem_branch_commit.get_image()
 
118
 
        commit_pixbuf = commit_image.get_pixbuf()
 
119
 
        commit_icon = gtk.Image()
 
120
 
        commit_icon.set_from_pixbuf(commit_pixbuf)
 
121
 
        commit_menu.set_image(commit_icon)
 
122
 
        diff_menu = self.ui.get_widget('/context_right/diff')
 
123
 
        diff_image = self.comm.menuitem_stats_diff.get_image()
 
124
 
        diff_pixbuf = diff_image.get_pixbuf()
 
125
 
        diff_icon = gtk.Image()
 
126
 
        diff_icon.set_from_pixbuf(diff_pixbuf)
 
127
 
        diff_menu.set_image(diff_icon)
 
 
118
        # TODO: do it without using deprecated comm
 
 
119
        #commit_menu = self.ui.get_widget('/context_right/commit')
 
 
120
        #commit_image = self.comm.menuitem_branch_commit.get_image()
 
 
121
        #commit_pixbuf = commit_image.get_pixbuf()
 
 
122
        #commit_icon = gtk.Image()
 
 
123
        #commit_icon.set_from_pixbuf(commit_pixbuf)
 
 
124
        #commit_menu.set_image(commit_icon)
 
 
125
        #diff_menu = self.ui.get_widget('/context_right/diff')
 
 
126
        #diff_image = self.comm.menuitem_stats_diff.get_image()
 
 
127
        #diff_pixbuf = diff_image.get_pixbuf()
 
 
128
        #diff_icon = gtk.Image()
 
 
129
        #diff_icon.set_from_pixbuf(diff_pixbuf)
 
 
130
        #diff_menu.set_image(diff_icon)
 
129
132
    def right_context_menu(self):
 
130
133
        return self.cmenu_right
 
 
132
135
    def left_context_menu(self):
 
133
136
        return self.cmenu_left
 
135
139
    def add_file(self, action):
 
136
140
        """ Right context menu -> Add """
 
137
143
        # Add only the selected file
 
138
 
        directory = self.comm.get_path()
 
139
 
        filename = self.comm.get_selected_right()
 
 
144
        directory = self.path
 
 
145
        filename = self.selected
 
141
147
        if filename is None:
 
142
 
            self.dialog.error_dialog(_('No file was selected'),
 
143
 
                                     _('Please select a file from the list,\nor choose the other option.'))
 
147
 
            fileops.add([directory + '/' + filename])
 
148
 
        except errors.NotBranchError:
 
149
 
            self.dialog.error_dialog(_('Directory is not a branch'),
 
150
 
                                     _('You can perform this action only in a branch.'))
 
155
 
        self.comm.refresh_right()
 
 
148
            error_dialog(_('No file was selected'),
 
 
149
                         _('Please select a file from the list,\nor choose the other option.'))
 
 
152
        bzrlib.add.smart_add([os.path.join(directory, filename)])
 
157
155
    def remove_file(self, action):
 
158
156
        """ Right context menu -> Remove """
 
159
157
        # Remove only the selected file
 
160
 
        directory = self.comm.get_path()
 
161
 
        filename = self.comm.get_selected_right()
 
 
158
        directory = self.path
 
 
159
        filename = self.selected
 
163
161
        if filename is None:
 
164
 
            self.dialog.error_dialog(_('No file was selected'),
 
165
 
                                     _('Please select a file from the list,\nor choose the other option.'))
 
169
 
            fileops.remove([directory + '/' + filename])
 
170
 
        except errors.NotBranchError:
 
171
 
            self.dialog.error_dialog(_('Directory is not a branch'),
 
172
 
                                     _('You can perform this action only in a branch.'))
 
174
 
        except errors.NotVersionedError:
 
175
 
            self.dialog.error_dialog(_('File not versioned'),
 
176
 
                                     _('The selected file is not versioned.'))
 
181
 
        self.comm.refresh_right()
 
 
162
            error_dialog(_('No file was selected'),
 
 
163
                         _('Please select a file from the list,\nor choose the other option.'))
 
 
166
        wt, path = WorkingTree.open_containing(os.path.join(directory, filename))
 
 
168
        self.app.set_path(self.path)
 
 
169
        self.app.refresh_right()
 
 
171
    def rename_file(self, action):
 
 
172
        """ Right context menu -> Rename """
 
 
173
        from rename import OliveRename
 
 
174
        wt = WorkingTree.open_containing(self.path + os.sep + self.selected)[0]
 
 
175
        rename = OliveRename(wt, wt.relpath(self.path), self.selected)
 
183
178
    def open_file(self, action):
 
184
179
        """ Right context menu -> Open """
 
185
180
        # Open only the selected file
 
186
 
        filename = self.comm.get_selected_right()
 
 
181
        filename = self.selected
 
188
183
        if filename is None:
 
189
 
            self.dialog.error_dialog(_('No file was selected'),
 
190
 
                                     _('Please select a file from the list,\nor choose the other option.'))
 
 
184
            error_dialog(_('No file was selected'),
 
 
185
                         _('Please select a file from the list,\nor choose the other option.'))
 
193
188
        if filename == '..':
 
194
 
            self.comm.set_path(os.path.split(self.comm.get_path())[0])
 
 
189
            # TODO: how to enter a directory?
 
196
 
            fullpath = self.comm.get_path() + os.sep + filename
 
 
192
            fullpath = self.path + os.sep + filename
 
197
193
            if os.path.isdir(fullpath):
 
198
194
                # selected item is an existant directory
 
199
 
                self.comm.set_path(fullpath)
 
 
195
                # TODO: how to enter a directory?
 
203
 
        self.comm.refresh_right()
 
 
200
    def revert(self, action):
 
 
201
        """ Right context menu -> Revert """
 
 
202
        wt, path = WorkingTree.open_containing(self.path)
 
 
203
        ret = wt.revert([os.path.join(path, self.selected)])
 
 
205
            warning_dialog(_('Conflicts detected'),
 
 
206
                           _('Please have a look at the working tree before continuing.'))
 
 
208
            info_dialog(_('Revert successful'),
 
 
209
                        _('All files reverted to last revision.'))
 
 
210
        self.app.refresh_right()       
 
205
212
    def commit(self, action):
 
206
213
        """ Right context menu -> Commit """
 
207
 
        from commit import OliveCommit
 
208
 
        commit = OliveCommit(self.gladefile, self.comm, self.dialog)
 
 
214
        from commit import CommitDialog
 
 
217
            wt, path = WorkingTree.open_containing(self.path)
 
 
219
        except NotBranchError, e:
 
 
222
        commit = CommitDialog(wt, path, not branch, self.selected)
 
 
223
        response = commit.run()
 
 
224
        if response != gtk.RESPONSE_NONE:
 
 
227
            if response == gtk.RESPONSE_OK:
 
 
228
                self.app.refresh_right()
 
211
233
    def diff(self, action):
 
212
234
        """ Right context menu -> Diff """
 
213
 
        from diff import OliveDiff
 
214
 
        diff = OliveDiff(self.gladefile, self.comm, self.dialog)
 
 
235
        wt = WorkingTree.open_containing(self.path)[0]
 
 
236
        window = DiffWindow()
 
 
237
        parent_tree = wt.branch.repository.revision_tree(wt.branch.last_revision())
 
 
238
        window.set_diff(wt.branch.nick, wt, parent_tree)
 
 
239
        window.set_file(wt.relpath(self.path + os.sep + self.selected))
 
217
242
    def bookmark(self, action):
 
218
243
        """ Right context menu -> Bookmark """
 
219
 
        if self.comm.pref.add_bookmark(self.comm.get_path()):
 
220
 
            self.dialog.info_dialog(_('Bookmark successfully added'),
 
221
 
                                    _('The current directory was bookmarked. You can reach\nit by selecting it from the left panel.'))
 
 
244
        if self.pref.add_bookmark(self.path):
 
 
245
            info_dialog(_('Bookmark successfully added'),
 
 
246
                        _('The current directory was bookmarked. You can reach\nit by selecting it from the left panel.'))
 
223
 
            self.dialog.warning_dialog(_('Location already bookmarked'),
 
224
 
                                       _('The current directory is already bookmarked.\nSee the left panel for reference.'))
 
 
249
            warning_dialog(_('Location already bookmarked'),
 
 
250
                           _('The current directory is already bookmarked.\nSee the left panel for reference.'))
 
226
 
        self.comm.refresh_left()
 
 
252
        self.app.refresh_left()
 
228
254
    def edit_bookmark(self, action):
 
229
255
        """ Left context menu -> Edit """
 
230
 
        from bookmark import OliveBookmark
 
232
 
        if self.comm.get_selected_left() != None:
 
233
 
            bookmark = OliveBookmark(self.gladefile, self.comm, self.dialog)
 
 
256
        from bookmark import BookmarkDialog
 
 
258
        if self.selected != None:
 
 
259
            bookmark = BookmarkDialog(self.selected, self.app.window)
 
 
260
            response = bookmark.run()
 
 
262
            if response != gtk.RESPONSE_NONE:
 
 
265
                if response == gtk.RESPONSE_OK:
 
 
266
                    self.app.refresh_left()
 
236
270
    def remove_bookmark(self, action):
 
237
271
        """ Left context menu -> Remove """
 
239
 
        if self.comm.get_selected_left() != None:
 
240
 
            self.comm.pref.remove_bookmark(self.comm.get_selected_left())
 
241
 
            self.comm.refresh_left()
 
 
273
        if self.selected != None:
 
 
274
            self.pref.remove_bookmark(self.selected)
 
 
277
        self.app.refresh_left()
 
243
279
    def open_folder(self, action):
 
244
280
        """ Left context menu -> Open Folder """
 
245
 
        path = self.comm.get_selected_left()