#!/bin/env python

# -*- coding: utf-8 -*-

# ------------------------------------------------------------

# opensecurity-dialog

# 

# an opensecurity dialog

#

# Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>

#

# Copyright (C) 2013 AIT Austrian Institute of Technology

# AIT Austrian Institute of Technology GmbH

# Donau-City-Strasse 1 | 1220 Vienna | Austria

# http://www.ait.ac.at

#

# This program is free software; you can redistribute it and/or

# modify it under the terms of the GNU General Public License

# as published by the Free Software Foundation version 2.

# 

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# 

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, 

# Boston, MA  02110-1301, USA.

# ------------------------------------------------------------

# ------------------------------------------------------------

# imports

import argparse

import os

import subprocess

import sys

import urllib2

from PyQt4 import QtCore

from PyQt4 import QtGui

# local

if sys.platform == 'win32' or sys.platform == 'cygwin':

    from cygwin import Cygwin

from ui import AboutDialog

from ui import opensecurity_rc

# ------------------------------------------------------------

# code

class OpenSecurityWait(QtGui.QDialog):

    """OpenSecurity: please wait ..."""

    def __init__(self, parent = None, flags = QtCore.Qt.WindowFlags(0)):

        super(OpenSecurityWait, self).__init__(parent, flags)

        self.setWindowTitle('OpenSecurity')

        self.setup_ui()

    def setup_ui(self):

        """Create the widgets."""

        lyMain = QtGui.QVBoxLayout(self)

        lyMain.setContentsMargins(8, 8, 8, 8)

        # content area: left pixmap, right text

        lbTitle = QtGui.QLabel('Creating secure subsystem. Please stand by ...')

        lyMain.addWidget(lbTitle)

        self.setMinimumSize(400, 50)

        self.resize(lyMain.minimumSize())

class OpenSecurityTrayIcon(QtGui.QSystemTrayIcon):

    """This is the OpenSecuirty Tray Icon"""

    def __init__(self, icon, parent=None):

        super(OpenSecurityTrayIcon, self).__init__(icon, parent)

        self.setup_ui()

    def clicked_about(self):

        """clicked about"""

        d = AboutDialog()

        d.exec_()

    def clicked_browser(self):

        """wish for safe internet browsing"""

        if not (sys.platform == 'win32' or sys.platform == 'cygwin'):

            QtGui.QMessageBox.critical(self.parent(), 'OpenSecurity Error', 'This action is not supported on this platform.\nSorry.')

            return

        # TODO: HARDCODED ADDRESS OF OPENSECURITYD

        # tell the user to wait

        d = OpenSecurityWait()

        d.show()

        QtGui.QApplication.instance().processEvents()

        try:

            # get a proper browsing VM

            Cygwin.start_X11()

            browsing_vm = urllib2.urlopen('http://127.0.0.1:8080/browsing').readline()

            #dlg_launch_image = os.path.join(sys.path[0], 'launch.pyw')

            #process_command = [sys.executable, dlg_launch_image, browsing_vm, '/usr/bin/iceweasel']

            #process_command = [sys.executable, dlg_launch_image, browsing_vm, '/usr/bin/midori']

            #print(process_command)

            #process = subprocess.Popen(process_command, shell = False)

        except:

            d.hide()

            QtGui.QApplication.instance().processEvents()

            QtGui.QMessageBox.critical(None, 'Failed to invoke Safe Internet Browsing', 'OpenSecurity Error')

        d.hide()

        QtGui.QApplication.instance().processEvents()

    def clicked_exit(self):

        """clicked exit"""

        sys.exit(0)

    def clicked_launch_application(self):

        """clicked the launch an application"""

        dlg_launch_image = os.path.join(sys.path[0], 'ui', 'launch_dialog.py')

        process_command = [sys.executable, dlg_launch_image]

        print(process_command)

        process = subprocess.Popen(process_command, shell = False)

    def setup_ui(self):

        """create the user interface

        As for the system tray this is 'just' the context menu.

        """

        # define the tray icon menu

        menu = QtGui.QMenu(self.parent())

        self.setContextMenu(menu)

        # add known apps

        self.acBrowser = QtGui.QAction('Secure Browsing', self.parent())

        icon = QtGui.QIcon()

        icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_icon_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.acBrowser.setIcon(icon)

        menu.addAction(self.acBrowser)

        menu.addSeparator()

        # add standard menu items

        self.acLaunch = QtGui.QAction('Launch Application', self.parent())

        icon = QtGui.QIcon()

        icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(':/opensecurity/gfx/opensecurity_icon_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.acLaunch.setIcon(icon)

        menu.addAction(self.acLaunch)

        menu.addSeparator()

        self.acAbout = menu.addAction('About')

        self.acExit = menu.addAction('Exit')

        self.acBrowser.triggered.connect(self.clicked_browser)

        self.acLaunch.triggered.connect(self.clicked_launch_application)

        self.acAbout.triggered.connect(self.clicked_about)

        self.acExit.triggered.connect(self.clicked_exit)

def main():

    a = QtGui.QApplication(sys.argv)

#    # prebuild the pixmap cache: fetch all jpg, png and jpeg images and load them

#    image_path = os.path.join(Environment("OpenSecurity").data_path, 'gfx')

#    for file in os.listdir(image_path):

#        if file.lower().rpartition('.')[2] in ('jpg', 'png', 'jpeg'):

#            QtGui.QPixmapCache.insert(file.lower().rpartition('.')[0], QtGui.QPixmap(os.path.join(image_path, file)))

    w = QtGui.QWidget()

    icon = QtGui.QIcon()

    icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8(":/opensecurity/gfx/opensecurity_icon_64.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)

    trayIcon = OpenSecurityTrayIcon(icon)

    trayIcon.show()

    a.setQuitOnLastWindowClosed(False)

    sys.exit(a.exec_())

# start

if __name__ == "__main__":

    main()