#!/bin/env python

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

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

# keyfile_dialog.pyw

# 

# the user should give us a keyfile

#

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

#

# Copyright 2013-2014 X-Net and AIT Austrian Institute of Technology

# 

# 

#     X-Net Services GmbH

#     Elisabethstrasse 1

#     4020 Linz

#     AUSTRIA

#     https://www.x-net.at

# 

#     AIT Austrian Institute of Technology

#     Donau City Strasse 1

#     1220 Wien

#     AUSTRIA

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

# 

# 

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# 

#    http://www.apache.org/licenses/LICENSE-2.0

# 

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

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

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

# imports

import base64

import sys

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

    import win32api

    import win32con

    import win32gui

from PyQt4 import QtCore

from PyQt4 import QtGui

from ui_KeyfileDialog import Ui_KeyfileDialog 

from about_dialog import AboutDialog

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

# code

class KeyfileDialog(QtGui.QDialog):

    """A dialog for letting the user pass on a password/keyfile combo"""

    def __init__(self, user_text = 'Please provide an approbitate password and keyfile to proceed:'):

        QtGui.QDialog.__init__(self)

        # setup the user interface

        self.ui = Ui_KeyfileDialog()

        self.ui.setupUi(self)

        # local members

        self._about_dialog = AboutDialog()

        self._open_file_dialog = QtGui.QFileDialog(self, 'Open Keyfile', QtCore.QDir.homePath(), 'All Files (*.*);;')

        self._open_file_dialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)

        self._open_file_dialog.setFileMode(QtGui.QFileDialog.ExistingFile)

        # connectors

        self.ui.lblText.setText(user_text)

        self.ui.btnAbout.clicked.connect(self.clicked_about)

        self.ui.btnCancel.clicked.connect(self.reject)

        self.ui.btnOk.clicked.connect(self.clicked_ok)

        self.ui.btnBrowse.clicked.connect(self.clicked_browse)

        # positionate ourself central

        screen = QtGui.QDesktopWidget().screenGeometry()

        size = self.geometry()

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

        t = QtCore.QTimer(self)

        t.timeout.connect(self.force_foreground)

        t.setInterval(0)

        t.setSingleShot(True)

        t.start()

    def clicked_about(self):

        """About button has been clicked."""

        self._about_dialog.show()

    def clicked_browse(self):

        """Browse button has been clicked."""

        if self._open_file_dialog.exec_() == QtGui.QDialog.Accepted:

            self.ui.edtKeyfile.setText(self._open_file_dialog.selectedFiles()[0])

    def clicked_ok(self):

        """Ok button has been clicked."""

        # read the content of the keyfile

        keyfile_content = ''

        try:

            keyfile = open(self.ui.edtKeyfile.text(), 'r')

            keyfile_content = keyfile.read()

        except Exception as e:

            sys.stderr.write('failed to read keyfile content:\n' + str(e) + '\n')

            QtGui.QMessageBox.critical(self, 'Failed to read keyfile', str(e))

            return

        # turn into Base64

        keyfile_content_base64 = base64.b64encode(keyfile_content)

        sys.stdout.write('{ ')

        sys.stdout.write('"password": "')

        sys.stdout.write(self.ui.edtPassword.text())

        sys.stdout.write('", ')

        sys.stdout.write('"keyfile": "')

        sys.stdout.write(keyfile_content_base64)

        sys.stdout.write('" ')

        sys.stdout.write('}\n')

        self.accept()

    def force_foreground(self):

        """Force ourselves into foreground"""

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

            w = self

            while not w.nativeParentWidget() is None:

                w = w.nativeParentWidget()

            win32gui.BringWindowToTop(int(w.effectiveWinId()))

            win32gui.SetForegroundWindow(int(w.effectiveWinId()))

    def set_user_text(user_text):

        """Set a text to explain which password we need."""

        self.ui.lblText.setText(user_text)

if __name__ == "__main__":

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

        win32gui.SystemParametersInfo(win32con.SPI_SETFOREGROUNDLOCKTIMEOUT, 0, win32con.SPIF_SENDWININICHANGE | win32con.SPIF_UPDATEINIFILE)

    a = QtGui.QApplication(sys.argv)

    d = KeyfileDialog()

    d.show()

    sys.exit(a.exec_())