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

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 = 3   # Index of Status Column in Application TreeView

CLIENT_API_VERSION = 4

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'

import getopt

import os

import sys

import time

import locale

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

from pkg.client import global_settings

# Put _() in the global namespace

import __builtin__

__builtin__._ = gettext.gettext

(

DISPLAY_LINK,

CLICK_LINK,

) = range(2)

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.show_startpage = self.client.get_bool(SHOW_STARTPAGE_PREFERENCES)

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

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

                self.update_all_proceed = False

                self.ua_be_name = None

                self.application_path = None

                self.default_authority = None

                self.first_run = True

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

                self.visible_repository_uptodate = False

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

                # 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

                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)

                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_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_refilter_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 __setup_startpage(self):

                self.opener = urllib.FancyURLopener()

                self.document = gtkhtml2.Document()

                self.document.connect('request_url', self.__request_url)

                self.document.connect('link_clicked', self.__handle_link)

                self.document.clear()

                self.view = gtkhtml2.View()

                self.view.set_document(self.document)

                self.view.connect('request_object', self.__request_object)

                self.view.connect('on_url', self.__on_url)

                try:

                        self.lang, encode = locale.getlocale(locale.LC_CTYPE)

                        if debug:

                                print "Lang: %s: Encode: %s" % (self.lang, encode)

                except locale.Error:

                        self.lang = "C"

                if self.lang == None or self.lang == "":

                        self.lang = "C"

                self.__load_startpage()

                self.w_startpage_frame.add(self.view)

        # Stub handler required by GtkHtml widget

        def __request_object(self, *vargs):

                pass

        def __load_startpage(self):

                self.start_page_url = self.application_dir + \

                    START_PAGE_LANG_BASE % (self.lang, START_PAGE_HOME)

                if not self.__load_uri(self.document, self.start_page_url):

                        self.start_page_url = self.application_dir + \

                            START_PAGE_LANG_BASE % ("C", START_PAGE_HOME)

                        if not self.__load_uri(self.document, self.start_page_url):

                                self.document.open_stream('text/html')

                                self.document.write_stream(_(

                                    "<html><head></head><body><H2>Welcome to \

                                    PackageManager!</H2><br>\

                                    <font color='#0000FF'>Warning: Unable to load \

                                    Start Page:<br>%s</font></body></html>"

                                    % (self.start_page_url)))

                                self.document.close_stream()

                                self.w_main_view_notebook.set_current_page(

                                    NOTEBOOK_PACKAGE_LIST_PAGE)

        def __on_url(self, view, link):

                # Handle mouse over events on links and reset when not on link

                if link == None or link == "":

                        self.update_statusbar()

                else:

                        display_link = self.__handle_link(None, link, DISPLAY_LINK)

                        if display_link != None:

                                self.w_main_statusbar.push(0, display_link)

                        else:

                                self.update_statusbar()

        @staticmethod

        def __is_relative_to_server(url):

                parts = urlparse.urlparse(url)

                if parts[0] or parts[1]:

                        return 0

                return 1

        def __open_url(self, url):

                uri = self.__resolve_uri(url)

                return self.opener.open(uri)

        def __resolve_uri(self, uri):

                if self.__is_relative_to_server(uri) and self.current_url != uri:

                        return urlparse.urljoin(self.current_url, uri)

                return uri

        def __request_url(self, document, url, stream):

                f = self.__open_url(url)

                stream.set_cancel_func(self.__stream_cancel)

                stream.write(f.read())

        # Stub handler required by GtkHtml widget or widget will assert

        def __stream_cancel(self, *vargs):

                pass

        def __load_uri(self, document, link):

                self.w_main_statusbar.push(0, _("Loading... " + link))

                try:

                        f = self.__open_url(link)

                except  (IOError, OSError), err:

                        if debug:

                                print "err: %s" % (err)

                        self.w_main_statusbar.push(0, _("Stopped"))

                        return False

                self.current_url = self.__resolve_uri(link)