# 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
        
        self.ui_right = gtk.UIManager()
        
        self.actiongroup_right = gtk.ActionGroup('context_right')
        self.actiongroup_right.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)])
        
        self.ui_right.insert_action_group(self.actiongroup_right, 0)
        self.ui_right.add_ui_from_file(self.uifile)
        
        self.cmenu_right = self.ui_right.get_widget('/context_right')

    def right_context_menu(self):
        return self.cmenu_right
    
    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.')