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

NOTEBOOK_PACKAGE_LIST_PAGE = 0            # Main Package List page index

NOTEBOOK_START_PAGE = 1                   # Main View Start page index

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

SEARCH_STR_FORMAT = "<::description:*%s* OR ::fmri:*%s*>"

SEARCH_LIMIT = 100                        # Maximum number of results shown for

                                          # remote search

INITIAL_SHOW_FILTER_PREFERENCES = "/apps/packagemanager/preferences/initial_show_filter"

INITIAL_SECTION_PREFERENCES = "/apps/packagemanager/preferences/initial_section"

SHOW_STARTPAGE_PREFERENCES = "/apps/packagemanager/preferences/show_startpage"

TYPEAHEAD_SEARCH_PREFERENCES = "/apps/packagemanager/preferences/typeahead_search"

CATEGORIES_STATUS_COLUMN_INDEX = 0   # Index of Status Column in Categories TreeView

STATUS_COLUMN_INDEX = 2   # Index of Status Column in Application TreeView

PKG_CLIENT_NAME = "packagemanager"

# Load Start Page from lang dir if available

START_PAGE_LANG_BASE = "usr/share/package-manager/data/startpagebase/%s/%s"

START_PAGE_HOME = "startpage.html" # Default page

# StartPage Action support for url's on StartPage pages

PM_ACTION = 'pm-action'          # Action field for StartPage url's

# Internal Example: <a href="pm?pm-action=internal&uri=top_picks.html">

ACTION_INTERNAL = 'internal'   # Internal Action value: pm-action=internal

INTERNAL_URI = 'uri'           # Internal field: uri to navigate to in StartPage

                               # without protocol scheme specified

# External Example: <a href="pm?pm-action=external&uri=www.opensolaris.com">

ACTION_EXTERNAL = 'external'   # External Action value: pm-action=external

EXTERNAL_URI = 'uri'           # External field: uri to navigate to in external

                               # default browser without protocol scheme specified

EXTERNAL_PROTOCOL = 'protocol' # External field: optional protocol scheme,

                               # defaults to http

DEFAULT_PROTOCOL = 'http'

PKGINFO_SUFFIX   = '.p5i'      # Mime type file extension

import getopt

import os

import sys

import time

import locale

import itertools

import urllib

import urlparse

import socket

import gettext

import signal

from threading import Thread

from urllib2 import HTTPError, URLError

from cPickle import UnpicklingError

try:

        import gobject

        import gnome

        gobject.threads_init()

        import gconf

        import gtk

        import gtk.glade

        import pygtk

        pygtk.require("2.0")

        import gtkhtml2

except ImportError:

        sys.exit(1)

import pkg.misc as misc

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

import pkg.gui.repository as repository

import pkg.gui.beadmin as beadm

import pkg.gui.cache as cache

import pkg.gui.misc as gui_misc

import pkg.gui.imageinfo as imageinfo

import pkg.gui.installupdate as installupdate

import pkg.gui.enumerations as enumerations

import pkg.gui.parseqs as parseqs

import pkg.gui.webinstall as webinstall

from pkg.client import global_settings

# Put _() in the global namespace

import __builtin__

__builtin__._ = gettext.gettext

(

DISPLAY_LINK,

CLICK_LINK,

) = range(2)

CLIENT_API_VERSION = gui_misc.get_client_api_version()

class PackageManager:

        def __init__(self):

                signal.signal(signal.SIGINT, self.__main_application_quit)

                self.api_o = None

                self.cache_o = None

                self.client = gconf.client_get_default()

                self.initial_show_filter = \

                    self.client.get_int(INITIAL_SHOW_FILTER_PREFERENCES)

                self.initial_section = self.client.get_int(INITIAL_SECTION_PREFERENCES)

                self.show_startpage = self.client.get_bool(SHOW_STARTPAGE_PREFERENCES)

                self.typeahead_search = self.client.get_bool(TYPEAHEAD_SEARCH_PREFERENCES)

                self.set_show_filter = 0

                self.set_section = 0

                self.current_search_option = 0

                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

                main_window_title = _('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

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

                self.main_clipboard_text = None

                self.progress_stop_timer_thread = False

                self.progress_fraction_time_count = 0

                self.progress_canceled = False

                self.catalog_loaded = False

                self.image_dir_arg = None

                self.update_all_proceed = False

                self.ua_be_name = None

                self.application_path = None

                self.default_publisher = None

                self.first_run = True

                self.in_search = False

                self.selected_pkgstem = None

                self.selected_model = None

                self.selected_path = None

                self.info_cache = {}

                self.selected = 0

                self.selected_pkgs = {}

                self.to_install_update = {}

                self.to_remove = {}

                self.in_startpage_startup = self.show_startpage

                self.lang = None

                self.start_page_url = None

                self.visible_status_id = 0

                self.categories_status_id = 0

                self.search_options = [

                    ('ips-search',

                    gui_misc.get_icon_pixbuf(self.application_dir, 'search'),

                    _("_Current Repository"),

                    _("Search Current Repository")),

                    ('ips-search-all',

                    gui_misc.get_icon_pixbuf(self.application_dir, 'search_all'),

                    _("_All Repositories"),

                    _("Search All Repositories"))

                    ]

                self.__register_iconsets(self.search_options)

                self.visible_repository = None

                self.visible_repository_uptodate = False

                self.section_list = None

                self.filter_list = self.__get_new_filter_liststore()

                self.application_list = None

                self.a11y_application_treeview = None

                self.a11y_categories_treeview = None

                self.application_treeview_range = None

                self.application_treeview_initialized = False

                self.categories_treeview_range = None

                self.categories_treeview_initialized = False

                self.category_list = None

                self.repositories_list = None

                self.pr = progress.NullProgressTracker()

                self.pylintstub = None

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

                w_tree_preferences = gtk.glade.XML(self.gladefile, "preferencesdialog")

                self.w_preferencesdialog = \

                    w_tree_preferences.get_widget("preferencesdialog")

                self.w_startpage_checkbutton = \

                    w_tree_preferences .get_widget("startpage_checkbutton")

                self.w_typeaheadsearch_checkbutton = \

                    w_tree_preferences .get_widget("typeaheadsearch_checkbutton")

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

                self.w_generalinfo_textview = \

                    w_tree_main.get_widget("generalinfotextview")

                self.w_generalinfo_textview.set_wrap_mode(gtk.WRAP_WORD)

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

                w_package_hbox = \

                    w_tree_main.get_widget("package_hbox")

                self.w_general_info_label = \

                    w_tree_main.get_widget("general_info_label")

                self.w_startpage_frame = \

                    w_tree_main.get_widget("startpage_frame")

                self.w_startpage_eventbox = \

                    w_tree_main.get_widget("startpage_eventbox")

                self.w_startpage_eventbox.modify_bg(gtk.STATE_NORMAL,

                    gtk.gdk.color_parse("white"))

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

                self.w_main_view_notebook = \

                    w_tree_main.get_widget("main_view_notebook")

                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_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(_("Update All"))

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

                self.w_progressinfo_label.set_text(_(

                    "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.w_clear_search_button = w_tree_main.get_widget("clear_search")

                self.w_clear_search_button.set_sensitive(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.saved_filter_combobox_active = self.initial_show_filter

                self.search_image = w_tree_main.get_widget("search_image")

                self.search_button = w_tree_main.get_widget("set_search")

                self.a11y_search_button = self.search_button.get_accessible()                

                self.is_remote_search = False

                self.need_descriptions = True

                self.searchmenu = gtk.Menu()

                self.search_image.set_from_pixbuf(self.search_options[0][1])

                self.a11y_search_button.set_description(self.search_options[0][3])

                for stock_id, pixbuf, label, description in self.search_options:

                        action = gtk.Action(stock_id, label, None, stock_id)

                        action.connect('activate', 

                            self.__search_menu_item_activate)

                        menu_item = action.create_menu_item()

                        self.searchmenu.append(menu_item)

                        self.pylintstub = description

                        self.pylintstub = pixbuf

                self.changing_search_option = False

                self.saved_repository_combobox_active = -1

                self.saved_sections_combobox_active = 0

                self.saved_application_list = None

                self.saved_category_list = None

                self.saved_section_list = None

                self.saved_selected_application_path = None

                self.search_message_id = 0

                toolbar =  w_tree_main.get_widget("toolbutton2")

                toolbar.set_expand(True)

                self.__init_repository_tree_view()

                self.install_button_tooltip = gtk.Tooltips()

                self.remove_button_tooltip = gtk.Tooltips()

                self.__update_reload_button()

                self.w_main_clipboard.request_text(self.__clipboard_text_received)

                self.w_main_window.set_title(main_window_title)

                # Setup Start Page

                self.document = None

                self.view = None

                self.current_url = None

                self.opener = None

                self.__setup_startpage()

                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_searchentry_key_press_event": \

                                    self.__on_searchentry_key_press_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_set_search_button_press_event":self.__on_set_search,

                                "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,

                                "on_edit_preferences_activate":self.__on_preferences,

                                # 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_help_start_page_activate":self.__on_startpage,

                                "on_details_notebook_switch_page": \

                                    self.__on_notebook_change,

                            }

                        dic_progress = \

                            {

                                "on_cancel_progressdialog_clicked": \

                                    self.__on_cancel_progressdialog_clicked,

                            }

                        dic_preferences = \

                            {

                                "on_startpage_checkbutton_toggled": \

                                    self.__on_startpage_checkbutton_toggled,

                                "on_typeaheadsearch_checkbutton_toggled": \

                                    self.__on_typeaheadsearch_checkbutton_toggled,

                                "on_preferenceshelp_clicked": \

                                    self.__on_preferenceshelp_clicked,

                                "on_preferencesclose_clicked": \

                                    self.__on_preferencesclose_clicked,

                            }

                        w_tree_main.signal_autoconnect(dic_mainwindow)

                        w_tree_progress.signal_autoconnect(dic_progress)

                        w_tree_preferences.signal_autoconnect(dic_preferences)

                except AttributeError, error:

                        print _(

                            "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_list_sort = None

                self.application_refilter_id = 0 

                self.application_refilter_idle_id = 0 

                self.show_info_id = 0 

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

                # Until package icons become available hide Package Icon Panel

                w_package_hbox.hide()

                if self.show_startpage:

                        self.w_main_view_notebook.set_current_page(NOTEBOOK_START_PAGE)

                else:

                        self.w_main_view_notebook.set_current_page(

                            NOTEBOOK_PACKAGE_LIST_PAGE)

        def __register_iconsets(self, icon_info):

                factory = gtk.IconFactory()

                for stock_id, pixbuf, name, description in icon_info:

                        iconset = gtk.IconSet(pixbuf)

                        factory.add(stock_id, iconset)

                        self.pylintstub = name

                        self.pylintstub = description

                factory.add_default()

        def __set_search_option(self, i):

                # The value i is the index in the table search_options

                # of the current choice.

                # Index 0 corresponds to Current Repository.

                # We assume that anything else is remote search.

                # This may need to be revisited if more search options are

                # added.

                if i == self.current_search_option:

                        return

                self.current_search_option = i

                self.changing_search_option = True

                self.__update_repository_combobox_for_search(i != 0)

                self.__update_gui_for_search(i != 0)

                self.changing_search_option = False

        def __update_repository_combobox_for_search(self, is_remote):

                if is_remote:

                        self.saved_repository_combobox_active = \

                            self.w_repository_combobox.get_active()

                self.__disconnect_repository_model()

                if is_remote:

                        self.repositories_list.prepend(

                            [-1, _("All Repositories Search Results"), ])

                else: