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

from PyQt4 import QtCore

from PyQt4 import QtGui

# local

from environment import Environment

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

# code

class OpenSecurityTrayIcon(QtGui.QSystemTrayIcon):

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

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

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

        # define the tray icon menu

        menu = QtGui.QMenu(parent)

        self.setContextMenu(menu)

        cAcLaunch = menu.addAction(QtGui.QIcon(QtGui.QPixmapCache.find('opensecurity_icon_64')), 'Lauch Application')

        menu.addSeparator()

        cAcExit = menu.addAction("Exit")

        cAcLaunch.triggered.connect(self.clicked_launch_application)

        cAcExit.triggered.connect(self.clicked_exit)

    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], 'launch.py')

        process_command = [sys.executable, dlg_launch_image]

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

        process.communicate()

def main():

    app = QtGui.QApplication(sys.argv)

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

    data_path = Environment("opensecurity").image_path

    for file in os.listdir(data_path):

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

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

    w = QtGui.QWidget()

    trayIcon = OpenSecurityTrayIcon(QtGui.QIcon(QtGui.QPixmapCache.find('opensecurity_icon_64')), w)

    trayIcon.show()

    sys.exit(app.exec_())

# start

if __name__ == "__main__":

    main()