/b-gtk/fix-viz

# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import os.path

import sys

try:

    import pygtk

    pygtk.require("2.0")

except:

    pass

try:

    import gtk

    import gtk.glade

except:

    sys.exit(1)

import olive.backend.fileops as fileops

import olive.backend.errors as errors

from commit import OliveCommit

from diff import OliveDiff

class OliveMenu:

    """ This class is responsible for building the context menus. """

    def __init__(self, gladefile, comm, dialog):

        # Load the UI file

        self.uifile = "/usr/share/olive/cmenu.ui"

        if not os.path.exists(self.uifile):

            # Load from current directory if not installed

            self.uifile = "cmenu.ui"

        self.gladefile = gladefile

        self.comm = comm

        self.dialog = dialog

        # Create the file list context menu

        self.ui = gtk.UIManager()

        self.actiongroup = gtk.ActionGroup('context')

        self.actiongroup.add_actions([('add', gtk.STOCK_ADD,

                                       'Add', None,

                                       'Add the selected file',

                                       self.add_file),

                                      ('remove', gtk.STOCK_REMOVE,

                                       'Remove', None,

                                       'Remove the selected file',

                                       self.remove_file),

                                      ('commit', gtk.STOCK_REDO,

                                       'Commit', None,

                                       'Commit the changes',

                                       self.commit),

                                      ('diff', None,

                                       'Diff', None,

                                       'Show the diff of the file',

                                       self.diff),

                                      ('log', None,

                                       'Log', None,

                                       'Show the log of the file',

                                       self.log),

                                      ('bookmark', None,

                                       'Bookmark', None,

                                       'Bookmark current location',

                                       self.bookmark),

                                      ('remove_bookmark', gtk.STOCK_REMOVE,

                                       'Remove', None,

                                       'Remove the selected bookmark',

                                       self.remove_bookmark)

                                     ])

        self.ui.insert_action_group(self.actiongroup, 0)

        self.ui.add_ui_from_file(self.uifile)

        self.cmenu_right = self.ui.get_widget('/context_right')

        self.cmenu_left = self.ui.get_widget('/context_left')

    def right_context_menu(self):

        return self.cmenu_right

    def left_context_menu(self):

        return self.cmenu_left

    def add_file(self, action):

        """ Right context menu -> Add """

        # Add only the selected file

        directory = self.comm.get_path()

        filename = self.comm.get_selected_right()

        if filename is None:

            self.dialog.error_dialog('No file was selected.')

            return

        try:

            fileops.add([directory + '/' + filename])

        except errors.NotBranchError:

            self.dialog.error_dialog('The directory is not a branch.')

            return

        except:

            raise

        self.comm.refresh_right()

    def remove_file(self, action):

        """ Right context menu -> Remove """

        # Remove only the selected file

        directory = self.comm.get_path()

        filename = self.comm.get_selected_right()

        if filename is None:

            self.dialog.error_dialog('No file was selected.')

            return

        try:

            fileops.remove([directory + '/' + filename])

        except errors.NotBranchError:

            self.dialog.error_dialog('The directory is not a branch.')

            return

        except errors.NotVersionedError:

            self.dialog.error_dialog('Selected file is not versioned.')

            return

        except:

            raise

        self.comm.refresh_right()

    def commit(self, action):

        """ Right context menu -> Commit """

        commit = OliveCommit(self.gladefile, self.comm)

        commit.display()

    def diff(self, action):

        """ Right context menu -> Diff """

        diff = OliveDiff(self.gladefile, self.comm)

        diff.display()

    def log(self, action):

        """ Right context menu -> Log """

        self.dialog.error_dialog('This feature is not yet implemented.')

    def bookmark(self, action):

        """ Right context menu -> Bookmark """

        if self.comm.pref.add_bookmark(self.comm.get_path()):

            self.dialog.info_dialog('Bookmark successfully added.')

        else:

            self.dialog.warning_dialog('Location already bookmarked.')

        self.comm.refresh_left()

    def remove_bookmark(self, action):

        """ Left context menu -> Remove """

        self.comm.pref.remove_bookmark(self.comm.get_selected_left())

        self.comm.refresh_left()