#!/usr/bin/python2.4

#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

# Copyright 2008 Sun Microsystems, Inc.  All rights reserved.

# Use is subject to license terms.

#

# Progress:

# Startup Progress has two phases:

# - Start phase:

#   The start phase should be fairly constant at around a few seconds and so is given 5%

#   of the total progress bar.

# - Package entry loading phase:

#   The package entry loading phase is given the remaining 95% of the bar for progress.

INITIAL_PROGRESS_TIME_INTERVAL = 0.5      # Time to update progress during start phase

INITIAL_PROGRESS_TIME_PERCENTAGE = 0.005  # Amount to update progress during start phase

INITIAL_PROGRESS_TOTAL_PERCENTAGE = 0.05  # Total progress for start phase

PACKAGE_PROGRESS_TOTAL_INCREMENTS = 95.0    # Total increments for loading phase

PACKAGE_PROGRESS_PERCENT_INCREMENT = 0.01 # Amount to update progress during loading phase

PACKAGE_PROGRESS_PERCENT_TOTAL = 1.0      # Total progress for loading phase

import getopt

import os

import sys

import time

import locale

import socket

import gettext

from threading import Thread

try:

        import gobject

        gobject.threads_init()

        import gtk

        import gtk.glade

        import pygtk

        pygtk.require("2.0")

except ImportError:

        sys.exit(1)

import pkg.client.image as image

import pkg.client.progress as progress

import pkg.misc as misc

import pkg.portable as portable

import pkg.gui.imageinfo as imageinfo

import pkg.gui.installupdate as installupdate

import pkg.gui.remove as remove

import pkg.gui.enumerations as enumerations

class PackageManager:

        def __init__(self):

                self.image_o = None

                socket.setdefaulttimeout(

                    int(os.environ.get("PKG_CLIENT_TIMEOUT", "30"))) # in seconds

                # Override default MAX_TIMEOUT_COUNT if a value has been specified

                # in the environment.

                timeout_max = misc.MAX_TIMEOUT_COUNT

                misc.MAX_TIMEOUT_COUNT = int(os.environ.get("PKG_TIMEOUT_MAX",

                    timeout_max))

                try:

                        self.application_dir = os.environ["PACKAGE_MANAGER_ROOT"]

                except KeyError:

                        self.application_dir = "/"

                locale.setlocale(locale.LC_ALL, '')

                for module in (gettext, gtk.glade):

                        module.bindtextdomain("packagemanager", self.application_dir + \

                            "/usr/share/locale")

                        module.textdomain("packagemanager")

                self._ = gettext.gettext

                main_window_title = self._('Package Manager - revision 0.1')

                self.user_rights = portable.is_admin()

                self.cancelled = False                    # For background processes

                self.image_directory = None

                self.description_thread_running = False   # For background processes

                self.pkginfo_thread = None                # For background processes

                gtk.rc_parse('~/.gtkrc-1.2-gnome2')       # Load gtk theme

                self.main_clipboard_text = None

                self.ipkg_fmri = "SUNWipkg"

                self.ipkggui_fmri = "SUNWipkg-gui"

                self.progress_stop_timer_thread = False

                self.progress_fraction_time_count = 0

                self.progress_canceled = False

                self.ips_uptodate = False

                self.image_dir_arg = None

                self.application_path = None

                self.application_list = \

                    gtk.ListStore(

                        gobject.TYPE_BOOLEAN,     # enumerations.MARK_COLUMN

                        gtk.gdk.Pixbuf,           # enumerations.STATUS_ICON_COLUMN

                        gtk.gdk.Pixbuf,           # enumerations.ICON_COLUMN

                        gobject.TYPE_STRING,      # enumerations.NAME_COLUMN

                        gobject.TYPE_STRING,      # enumerations.INSTALLED_VERSION_COLUMN

                        gobject.TYPE_PYOBJECT,    # enumerations.INSTALLED_OBJECT_COLUMN

                        gobject.TYPE_STRING,      # enumerations.LATEST_AVAILABLE_COLUMN

                        gobject.TYPE_INT,         # enumerations.RATING_COLUMN

                        gobject.TYPE_STRING,      # enumerations.DESCRIPTION_COLUMN

                        gobject.TYPE_PYOBJECT,    # enumerations.PACKAGE_OBJECT_COLUMN

                        gobject.TYPE_PYOBJECT,    # enumerations.IMAGE_OBJECT_COLUMN

                        gobject.TYPE_BOOLEAN,     # enumerations.IS_VISIBLE_COLUMN

                        gobject.TYPE_PYOBJECT     # enumerations.CATEGORY_LIST_OBJECT

                        )

                self.category_list = \

                    gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.CATEGORY_ID

                        gobject.TYPE_STRING,      # enumerations.CATEGORY_NAME

                        gobject.TYPE_STRING,      # enumerations.CATEGORY_DESCRIPTION

                        gtk.gdk.Pixbuf,           # enumerations.CATEGORY_ICON

                        gobject.TYPE_BOOLEAN,     # enumerations.CATEGORY_VISIBLE

                        gobject.TYPE_PYOBJECT,    # enumerations.SECTION_LIST_OBJECT

                        )

                self.section_list = \

                    gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.SECTION_ID

                        gobject.TYPE_STRING,      # enumerations.SECTION_NAME

                        )

                self.filter_list = \

                    gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.FILTER_ID

                        gobject.TYPE_STRING,      # enumerations.FILTER_NAME

                        )

                self.repositories_list = \

                    gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.REPOSITORY_ID

                        gobject.TYPE_STRING,      # enumerations.REPOSITORY_NAME

                        )

                self.application_list_filter = self.application_list.filter_new()

                self.pr = progress.NullProgressTracker()

                # Create Widgets and show gui

                self.gladefile = self.application_dir + \

                    "/usr/share/package-manager/packagemanager.glade"

                w_tree_main = gtk.glade.XML(self.gladefile, "mainwindow")

                w_tree_progress = gtk.glade.XML(self.gladefile, "progressdialog")

                self.w_main_window = w_tree_main.get_widget("mainwindow")

                self.w_application_treeview = \

                    w_tree_main.get_widget("applicationtreeview")

                self.w_categories_treeview = w_tree_main.get_widget("categoriestreeview")

                self.w_generalinfo_textview = \

                    w_tree_main.get_widget("generalinfotextview")

                self.w_installedfiles_textview = \

                    w_tree_main.get_widget("installedfilestextview")

                self.w_dependencies_textview = \

                    w_tree_main.get_widget("dependenciestextview")

                self.w_packagename_label = w_tree_main.get_widget("packagenamelabel")

                self.w_shortdescription_label = \

                    w_tree_main.get_widget("shortdescriptionlabel")

                self.w_searchentry_dialog = w_tree_main.get_widget("searchentry")

                self.w_installupdate_button = \

                    w_tree_main.get_widget("install_update_button")

                self.w_remove_button = w_tree_main.get_widget("remove_button")

                self.w_updateall_button = w_tree_main.get_widget("update_all_button")

                self.w_reload_button = w_tree_main.get_widget("reloadbutton")

                self.w_repository_combobox = w_tree_main.get_widget("repositorycombobox")

                self.w_sections_combobox = w_tree_main.get_widget("sectionscombobox")

                self.w_filter_combobox = w_tree_main.get_widget("filtercombobox")

                self.w_packageicon_image = w_tree_main.get_widget("packageimage")

                self.w_main_statusbar = w_tree_main.get_widget("statusbar")

                self.w_installupdate_menuitem = \

                    w_tree_main.get_widget("package_install_update")

                self.w_remove_menuitem = w_tree_main.get_widget("package_remove")

                self.w_updateall_menuitem = w_tree_main.get_widget("package_update_all")

                self.w_cut_menuitem = w_tree_main.get_widget("edit_cut")

                self.w_copy_menuitem = w_tree_main.get_widget("edit_copy")

                self.w_paste_menuitem = w_tree_main.get_widget("edit_paste")

                self.w_clear_menuitem = w_tree_main.get_widget("edit_clear")

                self.w_selectall_menuitem = w_tree_main.get_widget("edit_select_all")

                self.w_selectupdates_menuitem = \

                    w_tree_main.get_widget("edit_select_updates")

                self.w_deselect_menuitem = w_tree_main.get_widget("edit_deselect")

                self.w_main_clipboard = gtk.clipboard_get(gtk.gdk.SELECTION_CLIPBOARD)

                self.w_progress_dialog = w_tree_progress.get_widget("progressdialog")

                self.w_progress_dialog.set_title(self._("Update All"))

                self.w_progressinfo_label = w_tree_progress.get_widget("progressinfo")

                self.w_progressinfo_label.set_text(self._( \

                    "Checking SUNWipkg and SUNWipkg-gui versions\n\nPlease wait ..."))

                self.w_progressbar = w_tree_progress.get_widget("progressbar")

                self.w_progressbar.set_pulse_step(0.1)

                self.w_progress_cancel = w_tree_progress.get_widget("progresscancel")

                self.progress_canceled = False

                self.__update_reload_button()

                self.w_main_clipboard.request_text(self.__clipboard_text_received)

                self.w_main_window.set_title(main_window_title)

                try:

                        dic_mainwindow = \

                            {

                                "on_mainwindow_delete_event": \

                                    self.__on_mainwindow_delete_event,

                                "on_searchentry_changed":self.__on_searchentry_changed,

                                "on_searchentry_focus_in_event": \

                                    self.__on_searchentry_focus_in,

                                "on_searchentry_focus_out_event": \

                                    self.__on_searchentry_focus_out,

                                "on_searchentry_event":self.__on_searchentry_event,

                                "on_sectionscombobox_changed": \

                                    self.__on_sectionscombobox_changed,

                                "on_filtercombobox_changed": \

                                    self.__on_filtercombobox_changed,

                                "on_repositorycombobox_changed": \

                                    self.__on_repositorycombobox_changed,

                                #menu signals

                                "on_file_quit_activate":self.__on_file_quit_activate,

                                "on_package_install_update_activate": \

                                    self.__on_install_update,

                                "on_package_remove_activate":self.__on_remove,

                                "on_help_about_activate":self.__on_help_about,

                                "on_edit_paste_activate":self.__on_edit_paste,

                                "on_edit_clear_activate":self.__on_clear_paste,

                                "on_edit_copy_activate":self.__on_copy,

                                "on_edit_cut_activate":self.__on_cut,

                                "on_edit_select_all_activate":self.__on_select_all,

                                "on_edit_select_updates_activate": \

                                    self.__on_select_updates,

                                "on_edit_deselect_activate":self.__on_deselect,

                                # XXX disabled until new API

                                "on_package_update_all_activate":self.__on_update_all,

                                #toolbar signals

                                # XXX disabled until new API

                                "on_update_all_button_clicked":self.__on_update_all,

                                "on_reload_button_clicked":self.__on_reload,

                                "on_install_update_button_clicked": \

                                    self.__on_install_update,

                                "on_remove_button_clicked":self.__on_remove

                            }

                        dic_progress = \

                            {

                                "on_cancel_progressdialog_clicked": \

                                    self.__on_cancel_progressdialog_clicked,

                            }

                        w_tree_main.signal_autoconnect(dic_mainwindow)

                        w_tree_progress.signal_autoconnect(dic_progress)

                except AttributeError, error:

                        print self._( \

                            'GUI will not respond to any event! %s.' + \

                            'Check declare_signals()') \

                            % error

                self.package_selection = None

                self.category_list_filter = None

                self.in_setup = True

                self.w_main_window.show_all()

        def __init_tree_views(self):

                '''This function connects treeviews with their models and also applies

                filters'''

                ##APPLICATION MAIN TREEVIEW

                application_list_sort = \

                    gtk.TreeModelSort(self.application_list_filter)

                self.w_application_treeview.set_model(application_list_sort)

                model = self.w_application_treeview.get_model()

                toggle_renderer = gtk.CellRendererToggle()

                toggle_renderer.connect('toggled', self.__active_pane_toggle, model)

                column = gtk.TreeViewColumn("", toggle_renderer, \

                    active = enumerations.MARK_COLUMN)

                column.set_sort_column_id(enumerations.MARK_COLUMN)

                column.set_sort_indicator(True)

                column.set_cell_data_func(toggle_renderer, self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                column = gtk.TreeViewColumn()

                column.set_title("")

                #Commented, since there was funny jumping of the icons

                #column.set_sizing(gtk.TREE_VIEW_COLUMN_AUTOSIZE)

                render_pixbuf = gtk.CellRendererPixbuf()

                column.pack_start(render_pixbuf, expand = False)

                column.add_attribute(render_pixbuf, "pixbuf", \

                    enumerations.STATUS_ICON_COLUMN)

                column.set_fixed_width(32)

                column.set_cell_data_func(render_pixbuf, self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                column = gtk.TreeViewColumn()

                column.set_title("")

                #Commented, since there was funny jumping of the icons

                #column.set_sizing(gtk.TREE_VIEW_COLUMN_AUTOSIZE)

                render_pixbuf = gtk.CellRendererPixbuf()

                column.pack_start(render_pixbuf, expand = False)

                column.add_attribute(render_pixbuf, "pixbuf", enumerations.ICON_COLUMN)

                column.set_fixed_width(32)

                column.set_cell_data_func(render_pixbuf, self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                name_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._("Name"), name_renderer, \

                    text = enumerations.NAME_COLUMN)

                column.set_sort_column_id(enumerations.NAME_COLUMN)

                column.set_sort_indicator(True)

                column.set_cell_data_func(name_renderer, self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                installed_version_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._('Installed Version'), \

                    installed_version_renderer, \

                    text = enumerations.INSTALLED_VERSION_COLUMN)

                column.set_sort_column_id(enumerations.INSTALLED_VERSION_COLUMN)

                column.set_sort_indicator(True)

                column.set_cell_data_func(installed_version_renderer, \

                    self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                latest_available_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._('Latest Version'), \

                    latest_available_renderer, \

                    text = enumerations.LATEST_AVAILABLE_COLUMN)

                column.set_sort_column_id(enumerations.LATEST_AVAILABLE_COLUMN)

                column.set_sort_indicator(True)

                column.set_cell_data_func(latest_available_renderer, \

                    self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                rating_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._('Rating'), rating_renderer, \

                    text = enumerations.RATING_COLUMN)

                column.set_cell_data_func(rating_renderer, self.cell_data_function, None)

                description_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._('Description'), \

                    description_renderer, text = enumerations.DESCRIPTION_COLUMN)

                column.set_sort_column_id(enumerations.DESCRIPTION_COLUMN)

                column.set_sort_indicator(True)

                column.set_cell_data_func(description_renderer, \

                    self.cell_data_function, None)

                self.w_application_treeview.append_column(column)

                #Added selection listener

                self.package_selection = self.w_application_treeview.get_selection()

                self.package_selection.set_mode(gtk.SELECTION_SINGLE)

                self.package_selection.connect("changed", \

                    self.__on_package_selection_changed, None)

                ##CATEGORIES TREEVIEW

                #enumerations.CATEGORY_NAME

                self.category_list_filter = self.category_list.filter_new()

                self.category_list_filter.set_visible_func(self.category_filter)

                self.w_categories_treeview.set_model(self.category_list_filter)

                enumerations.CATEGORY_NAME_renderer = gtk.CellRendererText()

                column = gtk.TreeViewColumn(self._('Name'), \

                    enumerations.CATEGORY_NAME_renderer, \

                    text = enumerations.CATEGORY_NAME)

                self.w_categories_treeview.append_column(column)

                #Added selection listener

                category_selection = self.w_categories_treeview.get_selection()

                category_selection.set_mode(gtk.SELECTION_SINGLE)

                category_selection.connect("changed", \

                    self.__on_category_selection_changed, None)

                ##SECTION COMBOBOX

                #enumerations.SECTION_NAME

                self.w_sections_combobox.set_model(self.section_list)

                cell = gtk.CellRendererText()

                self.w_sections_combobox.pack_start(cell, True)

                self.w_sections_combobox.add_attribute(cell, 'text', \

                    enumerations.SECTION_NAME)

                self.w_sections_combobox.set_row_separator_func( \

                    self.combobox_id_separator)

                ##FILTER COMBOBOX

                #enumerations.FILTER_NAME

                self.w_filter_combobox.set_model(self.filter_list)

                cell = gtk.CellRendererText()

                self.w_filter_combobox.pack_start(cell, True)

                self.w_filter_combobox.add_attribute(cell, 'text', \

                    enumerations.FILTER_NAME)

                self.w_filter_combobox.set_row_separator_func(self.combobox_id_separator)

                ##FILTER COMBOBOX

                #enumerations.FILTER_NAME

                self.w_repository_combobox.set_model(self.repositories_list)

                cell = gtk.CellRendererText()

                self.w_repository_combobox.pack_start(cell, True)

                self.w_repository_combobox.add_attribute(cell, 'text', \

                    enumerations.REPOSITORY_NAME)

                self.w_repository_combobox.set_row_separator_func( \

                    self.combobox_id_separator)

                self.w_filter_combobox.set_active(0)

        def __init_sections(self):

                '''This function is for initializing sections combo box, also adds "All"

                Category. It sets active section combobox entry "All"'''

                self.section_list.append([0, self._('All'), ])

                self.section_list.append([-1, "", ])

                self.section_list.append([2, self._('Meta Packages'), ])

                self.section_list.append([3, self._('Applications Desktop'), ])

                self.section_list.append([4, self._('Applications Web-Based'), ])

                self.section_list.append([5, self._('Operating System'), ])

                self.section_list.append([6, self._('User Environment'), ])

                self.section_list.append([7, self._('Web Infrastructure'), ])

        def __init_show_filter(self):

                self.filter_list.append([0, self._('All Packages'), ])

                self.filter_list.append([-1, "", ])

                self.filter_list.append([2, self._('Installed Packages'), ])

                self.filter_list.append([3, self._('Updates'), ])

                self.filter_list.append([4, self._('Non-installed Packages'), ])

                self.filter_list.append([-1, "", ])

                # self.filter_list.append([self._('Locked Packages'), ])

                # self.filter_list.append(["", ])

                self.filter_list.append([6, self._('Selected Packages'), ])

        def __on_cancel_progressdialog_clicked(self, widget):

                self.progress_canceled = True

                self.progress_stop_timer_thread = True

        def __on_mainwindow_delete_event(self, widget, event):

                ''' handler for delete event of the main window '''

                if self.__check_if_something_was_changed() == True:

                        # XXX Change this to not quit and show dialog

                        # XXX if some changes were applied:

                        self.__main_application_quit()

                        return True

                else:

                        self.__main_application_quit()

        def __on_file_quit_activate(self, widget):

                ''' handler for quit menu event '''

                self.__on_mainwindow_delete_event(None, None)

        def __on_searchentry_changed(self, widget):

                '''On text search field changed we should refilter the main view'''

                Thread(target = self.__on_searchentry_threaded, args = ()).start()

        def __on_searchentry_threaded(self):

                gobject.idle_add(self.application_list_filter.refilter)

                gobject.idle_add(self.__enable_disable_selection_menus)

        def __on_edit_paste(self, widget):

                self.w_searchentry_dialog.insert_text(self.main_clipboard_text, \

                    self.w_searchentry_dialog.get_position())

        def __on_clear_paste(self, widget):

                bounds = self.w_searchentry_dialog.get_selection_bounds()

                self.w_searchentry_dialog.delete_text(bounds[0], bounds[1])

                return

        def __on_copy(self, widget):

                bounds = self.w_searchentry_dialog.get_selection_bounds()

                text = self.w_searchentry_dialog.get_chars(bounds[0], bounds[1])

                self.w_main_clipboard.set_text(text)

                return

        def __on_cut(self, widget):

                bounds = self.w_searchentry_dialog.get_selection_bounds()

                text = self.w_searchentry_dialog.get_chars(bounds[0], bounds[1])

                self.w_searchentry_dialog.delete_text(bounds[0], bounds[1])

                self.w_main_clipboard.set_text(text)

                return

        def __on_select_all(self, widget):

                sort_filt_model = \

                    self.w_application_treeview.get_model() #gtk.TreeModelSort

                filt_model = sort_filt_model.get_model() #gtk.TreeModelFilter

                model = filt_model.get_model() #gtk.ListStore

                iter_next = sort_filt_model.get_iter_first()

                list_of_paths = []

                while iter_next != None:

                        sorted_path = sort_filt_model.get_path(iter_next)

                        filtered_path = \

                            sort_filt_model.convert_path_to_child_path(sorted_path)

                        path = filt_model.convert_path_to_child_path(filtered_path)

                        list_of_paths.append(path)

                        iter_next = sort_filt_model.iter_next(iter_next)

                for path in list_of_paths:

                        itr = model.get_iter(path)

                        model.set_value(itr, enumerations.MARK_COLUMN, True)

                self.__enable_disable_selection_menus()

                self.update_statusbar()

                self.__enable_disable_install_update()

                self.__enable_disable_remove()

        def __on_select_updates(self, widget):

                sort_filt_model = \

                    self.w_application_treeview.get_model() #gtk.TreeModelSort

                filt_model = sort_filt_model.get_model() #gtk.TreeModelFilter

                model = filt_model.get_model() #gtk.ListStore

                iter_next = sort_filt_model.get_iter_first()

                list_of_paths = []

                while iter_next != None:

                        sorted_path = sort_filt_model.get_path(iter_next)

                        filtered_iter = sort_filt_model.convert_iter_to_child_iter(None, \

                            iter_next)

                        app_iter = filt_model.convert_iter_to_child_iter(filtered_iter)