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