#!/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    # 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

PKG_CLIENT_NAME = "packagemanager"

import getopt

import os

import sys

import time

import locale

import urlparse

import socket

import gettext

from threading import Thread

from urllib2 import HTTPError, URLError

try:

        import gobject

        import gnome

        gobject.threads_init()

        import gtk

        import gtk.glade

        import pygtk

        pygtk.require("2.0")

except ImportError:

        sys.exit(1)

import pkg.client.history as history

import pkg.client.image as image

import pkg.client.progress as progress

import pkg.client.api_errors as api_errors

import pkg.client.api as api

import pkg.client.retrieve as retrieve

import pkg.portable as portable

import pkg.gui.repository as repository

import pkg.gui.beadmin as beadm

import pkg.gui.imageinfo as imageinfo

import pkg.gui.installupdate as installupdate

import pkg.gui.enumerations as enumerations

from pkg.client import global_settings

class PackageManager:

        def __init__(self):

                self.image_o = None

                self.api_o = None

                socket.setdefaulttimeout(

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

                # Override default PKG_TIMEOUT_MAX if a value has been specified

                # in the environment.

                global_settings.PKG_TIMEOUT_MAX = int(os.environ.get("PKG_TIMEOUT_MAX",

                    global_settings.PKG_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')

                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 = -1                  # 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.first_run = True

                self.provided_image_dir = True

                self.section_list = self.__get_new_section_liststore()

                self.filter_list = self.__get_new_filter_liststore()

                self.application_list = None

                self.category_list = None

                self.repositories_list = None

                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_info_notebook = w_tree_main.get_widget("notebook1")

                self.w_generalinfo_textview = \

                    w_tree_main.get_widget("generalinfotextview")

                self.w_installedfiles_textview = \

                    w_tree_main.get_widget("installedfilestextview")

                self.w_license_textview = \

                    w_tree_main.get_widget("licensetextview")

                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

                clear_search_image = w_tree_main.get_widget("clear_image")

                clear_search_image.set_from_stock(gtk.STOCK_CLEAR, gtk.ICON_SIZE_MENU)

                self.__update_reload_button()

                self.w_main_clipboard.request_text(self.__clipboard_text_received)

                self.w_main_window.set_title(main_window_title)

                self.w_license_page = self.w_info_notebook.get_nth_page(3)

                self.w_info_notebook.remove_page(3)

                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_file_be_activate":self.__on_file_be_activate,

                                "on_package_install_update_activate": \

                                    self.__on_install_update,

                                "on_settings_edit_repositories_activate": \

                                    self.__on_edit_repositories_activate,

                                "on_package_remove_activate":self.__on_remove,

                                "on_help_about_activate":self.__on_help_about,

                                "on_help_help_activate":self.__on_help_help,

                                "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_clear_search_clicked":self.__on_clear_search,

                                "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.application_list_filter = None 

                self.in_setup = True

                self.w_main_window.show_all()

        @staticmethod

        def __get_new_application_liststore():

                return 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_BOOLEAN,     # enumerations.IS_VISIBLE_COLUMN

                        gobject.TYPE_PYOBJECT     # enumerations.CATEGORY_LIST_OBJECT

                        )

        @staticmethod

        def __get_new_category_liststore():

                return 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

                        )

        @staticmethod

        def __get_new_section_liststore():

                return gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.SECTION_ID

                        gobject.TYPE_STRING,      # enumerations.SECTION_NAME

                        )

        @staticmethod                    

        def __get_new_filter_liststore():

                return gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.FILTER_ID

                        gobject.TYPE_STRING,      # enumerations.FILTER_NAME

                        )

        @staticmethod

        def __get_new_repositories_liststore():

                return gtk.ListStore(

                        gobject.TYPE_INT,         # enumerations.REPOSITORY_ID

                        gobject.TYPE_STRING,      # enumerations.REPOSITORY_NAME

                        )

        def __init_tree_views(self, application_list, category_list, repositories_list):

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

                filters'''

                self.__disconnect_models()

                self.__remove_treeview_columns(self.w_application_treeview)

                self.__remove_treeview_columns(self.w_categories_treeview)

                ##APPLICATION MAIN TREEVIEW

                application_list_filter = application_list.filter_new()

                application_list_sort = \

                    gtk.TreeModelSort(application_list_filter)

                toggle_renderer = gtk.CellRendererToggle()

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

                column = gtk.TreeViewColumn()

                column.set_title(self._("Status"))

                #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 = True)

                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)

                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)

                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)

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

                ##CATEGORIES TREEVIEW

                #enumerations.CATEGORY_NAME

                category_list_filter = category_list.filter_new()

                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)

                if self.first_run:

                        ##SECTION COMBOBOX

                        #enumerations.SECTION_NAME

                        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

                        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

                        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.application_list = application_list

                self.category_list = category_list

                self.repositories_list = repositories_list

                self.category_list_filter = category_list_filter

                self.application_list_filter = application_list_filter

                self.w_sections_combobox.set_model(self.section_list)

                self.w_sections_combobox.set_active(0)

                self.w_filter_combobox.set_model(self.filter_list)

                self.w_filter_combobox.set_active(0)

                self.w_repository_combobox.set_model(repositories_list)

                self.w_categories_treeview.set_model(category_list_filter)

                self.w_application_treeview.set_model(application_list_sort)

                application_list_filter.set_visible_func(self.__application_filter)

                category_list_filter.set_visible_func(self.category_filter)

                toggle_renderer.connect('toggled', self.__active_pane_toggle, \

                    application_list_sort)

                category_selection.connect("changed", \

                    self.__on_category_selection_changed, None)

                self.package_selection.set_mode(gtk.SELECTION_SINGLE)

                self.package_selection.connect("changed", \

                    self.__on_package_selection_changed, None)

                self.first_run = False

        def __disconnect_models(self):

                self.w_application_treeview.set_model(None)

                self.w_categories_treeview.set_model(None)

                self.w_repository_combobox.set_model(None)

                self.w_sections_combobox.set_model(None)

                self.w_filter_combobox.set_model(None)

        @staticmethod

        def __remove_treeview_columns(treeview):

                columns = treeview.get_columns()

                if columns:

                        for column in columns:

                                treeview.remove_column(column)

        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: