# -*- 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 json

import os

import subprocess

import sys

import urllib

import urllib2

import webbrowser

import _winreg

import re

from PyQt4 import QtCore

from PyQt4 import QtGui

# local

import __init__ as opensecurity

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

    from cygwin import Cygwin

import opensecurity_client_restful_server 

from ui import AboutDialog

from ui import ConfigureDialog

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

        self.activated.connect(self.activated_)

    def activated_(self, reason):

        """the system tray icon was activated"""

        self.refresh_format_menu()

    def clicked_about(self):

        """clicked about"""

        d = AboutDialog()

        d.exec_()

    def getProxySettings(self):        

        aReg = _winreg.ConnectRegistry(None,_winreg.HKEY_CURRENT_USER)

        aKey = _winreg.OpenKey(aReg, r"Software\Microsoft\Windows\CurrentVersion\Internet Settings")

        subCount, valueCount, lastModified = _winreg.QueryInfoKey(aKey)

        reg_entries = dict()

        for i in range(valueCount):                                           

            try:

                n,v,t = _winreg.EnumValue(aKey,i)

                reg_entries[n] = v

            except EnvironmentError:                                               

                break

        _winreg.CloseKey(aKey)

        if 'ProxyEnable' in reg_entries.keys() and reg_entries['ProxyEnable'] == 1:

            proxy_search = re.search(r"(?<=http=)(?P<ProxyServer>.*?)(?=;)", reg_entries['ProxyServer'])

            if proxy_search:

                proxies = proxy_search.groupdict()

                if 'ProxyServer' in proxies.keys(): # found http proxy

                    return {'ProxyServer': proxies['ProxyServer']}  

            return {'ProxyServer': reg_entries['ProxyServer']}

        return None

    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

        try:

            # get a proper browsing VM

            Cygwin.start_X11()

            # TODO: HARDCODED ADDRESS OF OPENSECURITYD

            proxy_support = urllib2.ProxyHandler({})

            opener = urllib2.build_opener(proxy_support)

            urllib2.install_opener(opener)

            req_data = ""

            proxy = self.getProxySettings()

            if proxy:

                req_data = '?' + urllib.urlencode(proxy) 

            req = 'http://127.0.0.1:8080/browsing'+ req_data

            browsing_vm = urllib2.urlopen(req).readline()

            print('Called '+ req + ' got: ' + str(browsing_vm))

        except:

            QtGui.QApplication.instance().processEvents()

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

        QtGui.QApplication.instance().processEvents()

    def clicked_configure(self):

        """clicked configure"""

        d = ConfigureDialog()

        d.exec_()

    def clicked_exit(self):

        """clicked exit"""

        opensecurity_client_restful_server.stop()

        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]

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

        try:

            stdout = process.communicate()[0]

            j = json.loads(stdout)

        except:

            return

        try:

            # get a proper browsing VM

            Cygwin.start_X11()

            # TODO: HARDCODED ADDRESS OF OPENSECURITYD

            url = 'http://127.0.0.1:8080/sdvms/' + j['vm'] + '/application' + j['application']

            result = urllib2.urlopen(url).readline()

        except:

            pass 

    def clicked_mail(self):

        """clicked mail"""

        address = 'feedback@opensecurity.at'

        subject = 'Feedback zu OpenSecurity V' + opensecurity.__version__

        if sys.platform == 'linux2':

            subprocess.Popen(['xdg-email', '--subject', subject, address])

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

            mail_url = 'mailto:' + urllib.quote(address, '@') + '?' + urllib.quote('subject=' + subject)

            subprocess.Popen(['cmd', '/C', 'start', mail_url])

    def format_drive(self):

        """format drive clicked (the sender should a QAction created with refresh_format_menu)"""

        try:

            # fiddle the IP of the VM controlling the VM

            s = self.sender()

            ip = str(s.text().split('\\\\')[1])

            # invoke the format drive dialog

            dlg_format_drive = os.path.join(sys.path[0], 'ui', 'format_drive_dialog.py')

            process_command = [sys.executable, dlg_format_drive, ip]

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

            stdout = process.communicate()[0]

        except:

            pass

    def refresh_format_menu(self):

        """create a new list of format 'drives'"""

        self._menu_format.clear()

        a = self._menu_format.addAction('<No Drive given>')

        a.setEnabled(False)

        try:

            # get machines

            machines = json.load(urllib2.urlopen('http://127.0.0.1:8080/sdvms'))

            if len(machines) == 0:

                return

            self._icon_network = QtGui.QIcon()

            self._icon_network.addPixmap(QtGui.QPixmap(":/opensecurity/gfx/network-workgroup.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)

            cleared = False

            for m in machines:

                # do not insert Browsing VM

                if u'SecurityDVM0' in m:

                    continue

                if not cleared:

                    self._menu_format.clear()

                    cleared = True

                a = self._menu_format.addAction(m + '\\\\' + machines[m])

                a.setIcon(self._icon_network)

                a.triggered.connect(self.format_drive)

        except:

            pass

    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_browsing_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_launch_64.png')), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.acLaunch.setIcon(icon)

        menu.addAction(self.acLaunch)

        self.acConfigure = QtGui.QAction('Configuration', self.parent())

        icon = QtGui.QIcon()

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

        self.acConfigure.setIcon(icon)

        menu.addAction(self.acConfigure)

        menu.addSeparator()

        self._menu_format = menu.addMenu('Format')

        menu.addSeparator()

        self.acMail = menu.addAction('Send feedback mail')

        icon = QtGui.QIcon()

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

        self.acMail.setIcon(icon)

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

        # diabled as of TICKET #11

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

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

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

        self.acConfigure.triggered.connect(self.clicked_configure)

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

        # disabled as for TICKET #11

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

        self.acMail.triggered.connect(self.clicked_mail)

def main():

    # parse arguments

    parser = argparse.ArgumentParser(description = 'OpenSecurity Tray Icon.')

    parser.add_argument('-p', '--port', type=int, default=8090, help='port number of the REST API this instance will listen on.')

    args = parser.parse_args()

    # get up Qt

    a = QtGui.QApplication(sys.argv)

    # enforce singelton process

    if opensecurity_client_restful_server.is_already_running(args.port):

        QtGui.QMessageBox.critical(None, 'OpenSecurity Error', 'OpenSecurity Tray Instance already launched.\nClose previous instance first.')

        sys.exit(1)

    # start serving

    opensecurity_client_restful_server.serve(args.port, True)

    # init tray icon widget

    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)

    opensecurity_client_restful_server.tray_icon = trayIcon

    # go!

    trayIcon.show()

    a.setQuitOnLastWindowClosed(False)

    sys.exit(a.exec_())

# start

if __name__ == "__main__":

    main()