#!/bin/env python

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

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

 # opensecurity-launcher

 # 

 # launches an application inside a VM

 #

 # 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 urllib

 from PyQt4 import QtCore

 from PyQt4 import QtGui

 # local

 from about import About

 from environment import Environment

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

 # code

 class Chooser(QtGui.QDialog):

     """Ask the user what to launch."""

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

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

         self.setWindowTitle('OpenSecuirty Launch Application')

         self.setup_ui()

         # positionate ourself central

         screen = QtGui.QDesktopWidget().screenGeometry()

         self.resize(self.geometry().width() * 1.25, self.geometry().height())

         size = self.geometry()

         self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2)

     def clicked_about(self):

         """clicked the about button"""

         dlgAbout = About()

         dlgAbout.exec_()

     def clicked_cancel(self):

         """clicked the cancel button"""

         self.reject()

     def clicked_ok(self):

         """clicked the ok button"""

         self.accept()

     def setup_ui(self):

         """Create the widgets."""

         lyMain = QtGui.QVBoxLayout(self)

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

         # content area: left pixmap, right text

         lyContent = QtGui.QHBoxLayout()

         lyMain.addLayout(lyContent)

         # pixmap

         lbPix = QtGui.QLabel()

         lbPix.setPixmap(QtGui.QPixmapCache.find('opensecurity_icon_64'))

         lyContent.addWidget(lbPix, 0, QtCore.Qt.Alignment(QtCore.Qt.AlignTop + QtCore.Qt.AlignHCenter))

         lyContent.addSpacing(16)

         # launch ...

         lyLaunch = QtGui.QGridLayout()

         lyContent.addLayout(lyLaunch)

         lbTitle = QtGui.QLabel('Specify details for application to launch.')

         lyLaunch.addWidget(lbTitle, 0, 0, 1, 2)

         lbVM = QtGui.QLabel('&VM-ID:')

         lyLaunch.addWidget(lbVM, 1, 0)

         self.edVM = QtGui.QLineEdit()

         lyLaunch.addWidget(self.edVM, 1, 1)

         lbVM.setBuddy(self.edVM)

         # TODO: HARD CODED!

         self.edVM.setText('Debian 7')

         lbApplication = QtGui.QLabel('&Application:')

         lyLaunch.addWidget(lbApplication, 2, 0)

         self.cbApplication = QtGui.QComboBox()

         self.cbApplication.setEditable(True)

         lyLaunch.addWidget(self.cbApplication, 2, 1)

         lbApplication.setBuddy(self.cbApplication)

         # TODO: HARD CODED!

         self.cbApplication.addItem('iceweasel')

         self.cbApplication.addItem('vlc')

         self.cbApplication.addItem('xfce4-terminal')

         lyLaunch.addWidget(QtGui.QWidget(), 3, 0, 1, 2)

         lyLaunch.setColumnStretch(1, 1)

         lyLaunch.setRowStretch(3, 1)

         lyMain.addStretch(1)

         # buttons

         lyButton = QtGui.QHBoxLayout()

         lyMain.addLayout(lyButton)

         lyButton.addStretch(1)

         btnOk = QtGui.QPushButton('&Ok', self)

         btnOk.setDefault(True)

         btnOk.setMinimumWidth(100)

         lyButton.addWidget(btnOk)

         btnCancel = QtGui.QPushButton('&Cancel', self)

         btnCancel.setMinimumWidth(100)

         lyButton.addWidget(btnCancel)

         btnAbout = QtGui.QPushButton('&About', self)

         btnAbout.setMinimumWidth(100)

         lyButton.addWidget(btnAbout)

         button_width = max(btnOk.width(), btnCancel.width(), btnAbout.width())

         btnOk.setMinimumWidth(button_width)

         btnCancel.setMinimumWidth(button_width)

         btnAbout.setMinimumWidth(button_width)

         # reduce to the max

         self.resize(lyMain.minimumSize())

         # connectors

         btnOk.clicked.connect(self.clicked_ok)

         btnCancel.clicked.connect(self.clicked_cancel)

         btnAbout.clicked.connect(self.clicked_about)

 def main():

     # parse command line

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

     # we should have now our application icon

     app.setWindowIcon(QtGui.QIcon(QtGui.QPixmapCache.find('opensecurity_icon_64')))

     dlg = Chooser()

     # pop up the dialog

     dlg.show()

     app.exec_()

     if dlg.result() == QtGui.QDialog.Accepted:

         # encode an proper GET request to the opensecurity daemon

         get_vm = urllib.quote(str(dlg.edVM.text()))

         get_app = urllib.quote(str(dlg.cbApplication.currentText()))

         osd_request = 'http://127.0.0.1:8080/application?vm={0}&app={1}'.format(get_vm, get_app)

         urllib.urlopen(osd_request)

         res = 0

     else:

         res = 1

     sys.exit(res)

 # start

 if __name__ == "__main__":

     main()