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

MAX_DESC_LEN = 60                         # Max length of the description

MAX_INFO_CACHE_LIMIT = 100                # Max number of package descriptions to cache

TYPE_AHEAD_DELAY = 600    # The last type in search box after which search is performed

CLIENT_API_VERSION = 4

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.misc as misc

import pkg.client.history as history

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

# Put _() in the global namespace

import __builtin__

__builtin__._ = gettext.gettext

class PackageManager:

        def __init__(self):

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

                global_settings.client_name = PKG_CLIENT_NAME

                try:

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

                except KeyError:

                        self.application_dir = "/"

                misc.setlocale(locale.LC_ALL, "")

                for module in (gettext, gtk.glade):

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

                            "/usr/share/locale")

                        module.textdomain("pkg")

                # XXX Remove and use _() where self._ and self.parent._ are being used

                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.selected_pkgname = None

                self.info_cache = {}

                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)

                toolbar =  w_tree_main.get_widget("toolbutton2")

                toolbar.set_expand(True)

                w_whats_new_button = w_tree_main.get_widget("whats_new_button")

                #The whats new button hidden as this is not yet available

                w_whats_new_button.set_visible_horizontal(False)

                w_whats_new_button.set_visible_vertical(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_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,

                                "on_notebook1_switch_page": \

                                    self.__on_notebook_change,

                            }

                        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.application_refilter_id = 0 

                self.in_setup = True

                self.w_main_window.show_all()

                gdk_win = self.w_main_window.get_window()

                self.gdk_window = gtk.gdk.Window(gdk_win, gtk.gdk.screen_width(), 

                    gtk.gdk.screen_height(), gtk.gdk.WINDOW_CHILD, 0, gtk.gdk.INPUT_ONLY)

                gdk_cursor = gtk.gdk.Cursor(gtk.gdk.WATCH)

                self.gdk_window.set_cursor(gdk_cursor)

        @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.DESCRIPTION_COLUMN

                        gobject.TYPE_INT,         # enumerations.STATUS_COLUMN

                        gobject.TYPE_PYOBJECT,    # enumerations.FMRI_COLUMN

                        gobject.TYPE_STRING,      # enumerations.STEM_COLUMN

                        gobject.TYPE_STRING,      # enumerations.DISPLAY_NAME_COLUMN

                        gobject.TYPE_BOOLEAN,     # enumerations.IS_VISIBLE_COLUMN

                        gobject.TYPE_PYOBJECT     # enumerations.CATEGORY_LIST_COLUMN

                        )

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

                application_list_sort.set_sort_column_id(\

                    enumerations.NAME_COLUMN, gtk.SORT_ASCENDING)

                application_list_sort.set_sort_func(\

                    enumerations.STATUS_ICON_COLUMN, self.__status_sort_func)

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

                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_sort_column_id(enumerations.STATUS_ICON_COLUMN)

                column.set_sort_indicator(True)

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

                self.w_application_treeview.append_column(column)

                description_renderer = gtk.CellRendererText()

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

                    description_renderer, text = enumerations.DESCRIPTION_COLUMN)

                column.set_resizable(True)

                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 __status_sort_func(treemodel, iter1, iter2, user_data=None):

                get_val = treemodel.get_value

                status1 = get_val(iter1, enumerations.STATUS_COLUMN)

                status2 = get_val(iter2, enumerations.STATUS_COLUMN)

                return cmp(status1, status2)

        @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 Categories'), ])

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

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

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

                self.section_list.append([4, self._('Desktop (GNOME)'), ])

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

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

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

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

                self.section_list.append([9, self._('Web Services'), ])