/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 sys

try:

    import pygtk

    pygtk.require("2.0")

except:

    pass

try:

    import gtk

    import gtk.glade

except:

    sys.exit(1)

import bzrlib.errors as errors

from bzrlib.branch import Branch

from bzrlib.workingtree import WorkingTree

from dialog import about, error_dialog, info_dialog

from menu import OliveMenu

from launch import launch

class OliveHandler:

    """ Signal handler class for Olive. """

    def __init__(self, gladefile, comm):

        self.gladefile = gladefile

        self.comm = comm

        self.menu = OliveMenu(self.gladefile, self.comm)

    def on_about_activate(self, widget):

        about()

    def on_menuitem_add_files_activate(self, widget):

        """ Add file(s)... menu handler. """

        from add import OliveAdd

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

        add = OliveAdd(self.gladefile, wt, path, 

                self.comm.get_selected_right())

        add.display()

    def on_menuitem_branch_get_activate(self, widget):

        """ Branch/Get... menu handler. """

        from branch import OliveBranch

        branch = OliveBranch(self.gladefile, self.comm)

        branch.display()

    def on_menuitem_branch_checkout_activate(self, widget):

        """ Branch/Checkout... menu handler. """

        from checkout import OliveCheckout

        checkout = OliveCheckout(self.gladefile, self.comm)

        checkout.display()

    def on_menuitem_branch_commit_activate(self, widget):

        """ Branch/Commit... menu handler. """

        from commit import OliveCommit

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

        commit = OliveCommit(self.gladefile, wt, path)

        commit.display()

    def on_menuitem_branch_missing_revisions_activate(self, widget):

        """ Branch/Missing revisions menu handler. """

        self.comm.set_busy(self.comm.window_main)

        try:

            import bzrlib

            try:

                local_branch = Branch.open_containing(self.comm.get_path())[0]

            except NotBranchError:

                error_dialog(_('Directory is not a branch'),

                                         _('You can perform this action only in a branch.'))

                return

            other_branch = local_branch.get_parent()

            if other_branch is None:

                error_dialog(_('Parent location is unknown'),

                                         _('Cannot determine missing revisions if no parent location is known.'))

                return

            remote_branch = Branch.open(other_branch)

            if remote_branch.base == local_branch.base:

                remote_branch = local_branch

            ret = len(local_branch.missing_revisions(remote_branch))

            if ret > 0:

                info_dialog(_('There are missing revisions'),

                                        _('%d revision(s) missing.') % ret)

            else:

                info_dialog(_('Local branch up to date'),

                                        _('There are no missing revisions.'))

        finally:

            self.comm.set_busy(self.comm.window_main, False)

    def on_menuitem_branch_pull_activate(self, widget):

        """ Branch/Pull menu handler. """

        self.comm.set_busy(self.comm.window_main)

        try:

            try:

                from bzrlib.workingtree import WorkingTree

                tree_to = WorkingTree.open_containing(self.comm.get_path())[0]

                branch_to = tree_to.branch

            except errors.NoWorkingTree:

                tree_to = None

                branch_to = Branch.open_containing(self.comm.get_path())[0]

            except errors.NotBranchError:

                 error_dialog(_('Directory is not a branch'),

                                         _('You can perform this action only in a branch.'))

            location = branch_to.get_parent()

            if location is None:

                error_dialog(_('Parent location is unknown'),

                                         _('Pulling is not possible until there is a parent location.'))

                return

            try:

                branch_from = Branch.open(location)

            except errors.NotBranchError:

                error_dialog(_('Directory is not a branch'),

                                         _('You can perform this action only in a branch.'))

            if branch_to.get_parent() is None:

                branch_to.set_parent(branch_from.base)

            old_rh = branch_to.revision_history()

            if tree_to is not None:

                tree_to.pull(branch_from)

            else:

                branch_to.pull(branch_from)

            info_dialog(_('Pull successful'),

                                    _('%d revision(s) pulled.') % ret)

        finally:

            self.comm.set_busy(self.comm.window_main, False)

    def on_menuitem_branch_push_activate(self, widget):

        """ Branch/Push... menu handler. """

        from push import OlivePush

        push = OlivePush(self.gladefile, self.comm)

        push.display()

    def on_menuitem_branch_status_activate(self, widget):

        """ Branch/Status... menu handler. """

        from status import OliveStatus

        wt, wtpath = WorkingTree.open_containing(self.comm.get_path())

        status = OliveStatus(self.gladefile, wt, wtpath)

        status.display()

    def on_menuitem_branch_initialize_activate(self, widget):

        """ Initialize current directory. """

        try:

            location = self.comm.get_path()

            from bzrlib.builtins import get_format_type

            format = get_format_type('default')

            if not os.path.exists(location):

                os.mkdir(location)

            try:

                existing_bzrdir = bzrdir.BzrDir.open(location)

            except NotBranchError:

                bzrdir.BzrDir.create_branch_convenience(location, format=format)

            else:

                if existing_bzrdir.has_branch():

                    if existing_bzrdir.has_workingtree():

                        raise AlreadyBranchError(location)

                    else:

                        raise BranchExistsWithoutWorkingTree(location)

                else:

                    existing_bzrdir.create_branch()

                    existing_bzrdir.create_workingtree()

        except errors.AlreadyBranchError, errmsg:

            error_dialog(_('Directory is already a branch'),

                                     _('The current directory (%s) is already a branch.\nYou can start using it, or initialize another directory.') % errmsg)

        except errors.BranchExistsWithoutWorkingTree, errmsg:

            error_dialog(_('Branch without a working tree'),

                                     _('The current directory (%s)\nis a branch without a working tree.') % errmsg)

        else:

            info_dialog(_('Initialize successful'),

                                    _('Directory successfully initialized.'))

            self.comm.refresh_right()

    def on_menuitem_file_make_directory_activate(self, widget):

        """ File/Make directory... menu handler. """

        from mkdir import OliveMkdir

        mkdir = OliveMkdir(self.gladefile, self.comm)

        mkdir.display()

    def on_menuitem_file_move_activate(self, widget):

        """ File/Move... menu handler. """

        from move import OliveMove

        move = OliveMove(self.gladefile, self.comm)

        move.display()

    def on_menuitem_file_rename_activate(self, widget):

        """ File/Rename... menu handler. """

        from rename import OliveRename

        rename = OliveRename(self.gladefile, self.comm)

        rename.display()

    def on_menuitem_remove_file_activate(self, widget):

        """ Remove (unversion) selected file. """

        from remove import OliveRemove

        remove = OliveRemove(self.gladefile, self.comm)

        remove.display()

    def on_menuitem_stats_diff_activate(self, widget):

        """ Statistics/Differences... menu handler. """

        from bzrlib.plugins.gtk.viz.diffwin import DiffWindow

        window = DiffWindow()

        wt = WorkingTree.open_containing(self.comm.get_path())[0]

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

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

        window.show()

    def on_menuitem_stats_infos_activate(self, widget):

        """ Statistics/Informations... menu handler. """

        from info import OliveInfo

        info = OliveInfo(self.gladefile, self.comm)

        info.display()

    def on_menuitem_stats_log_activate(self, widget):

        """ Statistics/Log... menu handler. """

        from log import OliveLog

        log = OliveLog(self.gladefile, self.comm)

        log.display()

    def on_menuitem_view_refresh_activate(self, widget):

        """ View/Refresh menu handler. """

        # Refresh the left pane

        self.comm.refresh_left()

        # Refresh the right pane

        self.comm.refresh_right()

    def on_menuitem_view_show_hidden_files_activate(self, widget):

        """ View/Show hidden files menu handler. """

        if widget.get_active():

            # Show hidden files

            self.comm.pref.set_preference('dotted_files', True)

            self.comm.pref.refresh()

            self.comm.refresh_right()

        else:

            # Do not show hidden files

            self.comm.pref.set_preference('dotted_files', False)

            self.comm.pref.refresh()

            self.comm.refresh_right()

    def on_treeview_left_button_press_event(self, widget, event):

        """ Occurs when somebody right-clicks in the bookmark list. """

        if event.button == 3:

            # Don't show context with nothing selected

            if self.comm.get_selected_left() == None:

                return

            self.menu.left_context_menu().popup(None, None, None, 0,

                                                event.time)

    def on_treeview_left_row_activated(self, treeview, path, view_column):

        """ Occurs when somebody double-clicks or enters an item in the

        bookmark list. """

        newdir = self.comm.get_selected_left()

        if newdir == None:

            return

        self.comm.set_busy(treeview)

        self.comm.set_path(newdir)

        self.comm.refresh_right()

        self.comm.set_busy(treeview, False)

    def on_treeview_right_button_press_event(self, widget, event):

        """ Occurs when somebody right-clicks in the file list. """

        if event.button == 3:

            # get the menu items

            m_add = self.menu.ui.get_widget('/context_right/add')

            m_remove = self.menu.ui.get_widget('/context_right/remove')

            m_commit = self.menu.ui.get_widget('/context_right/commit')

            m_diff = self.menu.ui.get_widget('/context_right/diff')

            # check if we're in a branch

            try:

                from bzrlib.branch import Branch

                Branch.open_containing(self.comm.get_path())

                m_add.set_sensitive(False)

                m_remove.set_sensitive(False)

                m_commit.set_sensitive(False)

                m_diff.set_sensitive(False)

            except errors.NotBranchError:

                m_add.set_sensitive(True)

                m_remove.set_sensitive(True)

                m_commit.set_sensitive(True)

                m_diff.set_sensitive(True)

            self.menu.right_context_menu().popup(None, None, None, 0,

                                                 event.time)

    def on_treeview_right_row_activated(self, treeview, path, view_column):

        """ Occurs when somebody double-clicks or enters an item in the

        file list. """

        import os.path

        newdir = self.comm.get_selected_right()

        if newdir == '..':

            self.comm.set_path(os.path.split(self.comm.get_path())[0])

        else:

            fullpath = self.comm.get_path() + os.sep + newdir

            if os.path.isdir(fullpath):

                # selected item is an existant directory

                self.comm.set_path(fullpath)

            else:

                launch(fullpath) 

        self.comm.refresh_right()

    def on_window_main_delete_event(self, widget, event=None):

        """ Do some stuff before exiting. """

        width, height = self.comm.window_main.get_size()

        self.comm.pref.set_preference('window_width', width)

        self.comm.pref.set_preference('window_height', height)

        x, y = self.comm.window_main.get_position()

        self.comm.pref.set_preference('window_x', x)

        self.comm.pref.set_preference('window_y', y)

        self.comm.pref.set_preference('paned_position',

                                      self.comm.hpaned_main.get_position())

        self.comm.pref.write()

        self.comm.window_main.destroy()