/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

import os.path

import shutil

import sys

try:

    import pygtk

    pygtk.require("2.0")

except:

    pass

import gtk

import bzrlib.errors as errors

from bzrlib.workingtree import WorkingTree

from bzrlib.plugins.gtk import _i18n

from bzrlib.plugins.gtk.dialog import error_dialog, info_dialog, question_dialog, warning_dialog

from bzrlib.plugins.gtk.errors import show_bzr_error

from bzrlib.plugins.gtk.annotate.gannotate import GAnnotateWindow

from bzrlib.plugins.gtk.annotate.config import GAnnotateConfig

from bzrlib.plugins.gtk.diff import DiffWindow

from launch import launch

from bzrlib.plugins.gtk.olive import Preferences

class OliveMenu:

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

    def __init__(self, path, selected, app=None):

        # Load the UI file

        from guifiles import UIFILENAME

        self.uifile = UIFILENAME

        # Preferences handler

        self.pref = Preferences()

        # Set default values

        self.path = path

        self.selected = selected

        self.app = app

        # Create the file list context menu

        self.ui = gtk.UIManager()

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

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

                                       _i18n('Add'), None,

                                       _i18n('Add the selected file'),

                                       self.add_file),

                                      ('remove', gtk.STOCK_REMOVE,

                                       _i18n('Remove'), None,

                                       _i18n('Remove the selected file'),

                                       self.remove_file),

                                      ('remove_and_delete', gtk.STOCK_REMOVE,

                                       _i18n('Remove and delete'), None,

                                       _i18n('Remove the selected file/dir and delete from disk'),

                                       self.remove_and_delete_file),

                                      ('rename', None,

                                       _i18n('Rename'), None,

                                       _i18n('Rename the selected file'),

                                       self.rename_file),

                                      ('open', gtk.STOCK_OPEN,

                                       _i18n('Open'), None,

                                       _i18n('Open the selected file'),

                                       self.open_file),

                                      ('revert', None,

                                       _i18n('Revert'), None,

                                       _i18n('Revert the changes'),

                                       self.revert),

                                      ('commit', None,

                                       _i18n('Commit'), None,

                                       _i18n('Commit the changes'),

                                       self.commit),

                                      ('annotate', None,

                                       _i18n('Annotate'), None,

                                       _i18n('Annotate the selected file'),

                                       self.annotate),

                                      ('diff', None,

                                       _i18n('Diff'), None,

                                       _i18n('Show the diff of the file'),

                                       self.diff),

                                      ('bookmark', None,

                                       _i18n('Bookmark'), None,

                                       _i18n('Bookmark current location'),

                                       self.bookmark),

                                      ('edit_bookmark', gtk.STOCK_EDIT,

                                       _i18n('Edit'), None,

                                       _i18n('Edit the selected bookmark'),

                                       self.edit_bookmark),

                                      ('remove_bookmark', gtk.STOCK_REMOVE,

                                       _i18n('Remove'), None,

                                       _i18n('Remove the selected bookmark'),

                                       self.remove_bookmark),

                                      ('open_folder', gtk.STOCK_OPEN,

                                       _i18n('Open Folder'), None,

                                       _i18n('Open bookmark folder in Nautilus'),

                                       self.open_folder),

                                      ('diff_selected', None,

                                       _i18n('Selected...'), None,

                                       _i18n('Show the differences of the selected file'),

                                       self.diff_selected),

                                      ('diff_all', None,

                                       _i18n('All...'), None,

                                       _i18n('Show the differences of all files'),

                                       self.diff_all),

                                      ('view_remote', None,

                                       _i18n('View contents'), None,

                                       _i18n('View the contents of the file in a builtin viewer'),

                                       self.view_remote),

                                      ('diff_remote', None,

                                       _i18n('Show differences'), None,

                                       _i18n('Show the differences between two revisions of the file'),

                                       self.diff_remote),

                                      ('revert_remote', None,

                                       _i18n('Revert to this revision'), None,

                                       _i18n('Revert the selected file to the selected revision'),

                                       self.revert_remote)

                                     ])

        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')

        self.toolbar_diff = self.ui.get_widget('/toolbar_diff')

        self.cmenu_remote = self.ui.get_widget('/context_remote')

        # Set icons

        # TODO: do it without using deprecated comm

        #commit_menu = self.ui.get_widget('/context_right/commit')

        #commit_image = self.comm.menuitem_branch_commit.get_image()

        #commit_pixbuf = commit_image.get_pixbuf()

        #commit_icon = gtk.Image()

        #commit_icon.set_from_pixbuf(commit_pixbuf)

        #commit_menu.set_image(commit_icon)

        #diff_menu = self.ui.get_widget('/context_right/diff')

        #diff_image = self.comm.menuitem_stats_diff.get_image()

        #diff_pixbuf = diff_image.get_pixbuf()

        #diff_icon = gtk.Image()

        #diff_icon.set_from_pixbuf(diff_pixbuf)

        #diff_menu.set_image(diff_icon)

    def right_context_menu(self):

        return self.cmenu_right

    def left_context_menu(self):

        return self.cmenu_left

    def remote_context_menu(self):

        return self.cmenu_remote

    @show_bzr_error

    def add_file(self, action):

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

        import bzrlib.add

        # Add only the selected file

        directory = self.path

        filename = self.selected

        if filename is None:

            error_dialog(_i18n('No file was selected'),

                         _i18n('Please select a file from the list,\nor choose the other option.'))

            return

        bzrlib.add.smart_add([os.path.join(directory, filename)])

    @show_bzr_error

    def annotate(self, action):

        """ Right context menu -> Annotate """

        directory = self.path

        filename = self.selected

        if filename is None:

            error_dialog(_i18n('No file was selected'),

                         _i18n('Please select a file from the list.'))

            return

        wt, path = WorkingTree.open_containing(os.path.join(directory, filename))

        branch = wt.branch

        file_id = wt.path2id(wt.relpath(os.path.join(directory, filename)))

        window = GAnnotateWindow(all=False, plain=False, parent=self.app)

        window.set_title(os.path.join(directory, filename) + " - Annotate")

        config = GAnnotateConfig(window)

        window.show()

        branch.lock_read()

        try:

            window.annotate(wt, branch, file_id)

        finally:

            branch.unlock()

    @show_bzr_error

    def remove_file(self, action,delete_on_disk=0):

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

        # Remove only the selected file

        directory = self.path

        filename = self.selected

        if filename is None:

            error_dialog(_i18n('No file was selected'),

                         _i18n('Please select a file from the list,\nor choose the other option.'))

            return

        wt, path = WorkingTree.open_containing(os.path.join(directory, filename))

        wt.remove(path)

        if delete_on_disk:

            abs_filename = os.path.join(directory,filename)

            if os.path.isdir(abs_filename):

                response = question_dialog(_i18n('Delete directory with all directories below ?'), abs_filename )

                if response == gtk.RESPONSE_YES:

                    shutil.rmtree(abs_filename)

            else:

                os.remove(abs_filename)

        self.app.set_path(self.path)

        self.app.refresh_right()

    def remove_and_delete_file(self, action):

        """ Right context menu -> Remove and delete"""

        self.remove_file(action,delete_on_disk=1)

    def rename_file(self, action):

        """ Right context menu -> Rename """

        from rename import OliveRename

        wt = WorkingTree.open_containing(self.path + os.sep + self.selected)[0]

        rename = OliveRename(wt, wt.relpath(self.path), self.selected)

        rename.display()

    def open_file(self, action):

        """ Right context menu -> Open """

        # Open only the selected file

        filename = self.selected

        if filename is None:

            error_dialog(_i18n('No file was selected'),

                         _i18n('Please select a file from the list,\nor choose the other option.'))

            return

        if filename == '..':

            # TODO: how to enter a directory?

            return

        else:

            fullpath = self.path + os.sep + filename

            if os.path.isdir(fullpath):

                # selected item is an existant directory

                # TODO: how to enter a directory?

                return

            else:

                launch(fullpath) 

    def revert(self, action):

        """ Right context menu -> Revert """

        wt, path = WorkingTree.open_containing(self.path)

        ret = wt.revert([os.path.join(path, self.selected)])

        if ret:

            warning_dialog(_i18n('Conflicts detected'),

                           _i18n('Please have a look at the working tree before continuing.'))

        else:

            info_dialog(_i18n('Revert successful'),

                        _i18n('All files reverted to last revision.'))

        self.app.refresh_right()       

    def commit(self, action):

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

        from commit import CommitDialog

        branch = None

        try:

            wt, path = WorkingTree.open_containing(self.path)

            branch = wt.branch

        except NotBranchError, e:

            path = e.path

        if self.selected:

            selected = os.path.join(path, self.selected)

        else:

            selected = None

        commit = CommitDialog(wt=wt, selected=selected, parent=None)

        response = commit.run()

        if response != gtk.RESPONSE_NONE:

            commit.hide()

            if response == gtk.RESPONSE_OK:

                self.app.refresh_right()

            commit.destroy()

    @show_bzr_error

    def diff(self, action):

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

        wt = WorkingTree.open_containing(self.path)[0]

        window = DiffWindow(self.app)

        parent_tree = wt.branch.repository.revision_tree(wt.branch.last_revision())

        window.set_diff(wt.branch.nick, wt, parent_tree)

        window.set_file(wt.relpath(self.path + os.sep + self.selected))

        window.show()

    def bookmark(self, action):

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

        if self.pref.add_bookmark(self.path):

            info_dialog(_i18n('Bookmark successfully added'),

                        _i18n('The current directory was bookmarked. You can reach\nit by selecting it from the left panel.'))

            self.pref.write()

        else:

            warning_dialog(_i18n('Location already bookmarked'),

                           _i18n('The current directory is already bookmarked.\nSee the left panel for reference.'))

        self.app.refresh_left()

    def edit_bookmark(self, action):

        """ Left context menu -> Edit """

        from bookmark import BookmarkDialog

        if self.selected != None:

            bookmark = BookmarkDialog(self.selected, self.app.window)

            response = bookmark.run()

            if response != gtk.RESPONSE_NONE:

                bookmark.hide()

                if response == gtk.RESPONSE_OK:

                    self.app.refresh_left()

                bookmark.destroy()

    def remove_bookmark(self, action):

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

        if self.selected != None:

            self.pref.remove_bookmark(self.selected)

            self.pref.write()

        self.app.refresh_left()

    def open_folder(self, action):

        """ Left context menu -> Open Folder """

        path = self.selected

        if path != None:

            launch(path)

    def diff_selected(self, action):

        """ Diff toolbutton -> Selected... """

        print "DEBUG: not implemented."

    def diff_all(self, action):

        """ Diff toolbutton -> All... """

        from diff import OliveDiff

        diff = OliveDiff(self.comm)

        diff.display()

    def view_remote(self, action):

        """ Remote context menu -> View contents """

        print "DEBUG: view contents."

    def diff_remote(self, action):

        """ Remote context menu -> Show differences """

        print "DEBUG: show differences."

    def revert_remote(self, action):

        """ Remote context menu -> Revert to this revision """

        print "DEBUG: revert to this revision."