#! /usr/bin/python

#  Small PyQt applicaion that gives stiata-reader a GUI.  
#
# Pass filename of .emc statement as first argument, for use as "open with" from browser.
# 
#		eg:
#			striata-reader-gui.py ../path/to/statement.emc
#
# Script dependencies:
#     - striata-reader for Linux  (get it from http://www.striata.com/resources/striata-reader/index.php)
#     - PyQt4 (see your distribution's docs about how to install it )

import sys
import os
import subprocess
import string
import shutil
from PyQt4 import QtGui
from PyQt4 import QtCore

global PATH_TO_STRIATA_READER, STRIATA_READER_DEFAULT_OPTIONS, TEMP_FOLDER, STATEMENT_BROWSE_FILE_FILTER_LIST
# NOTE older versions of striata might work with the following path.  Just uncomment to make it work.
# Remember to point it to the *binary* version of striata-reader
#PATH_TO_STRIATA_READER = "/usr/bin/striata-reader-1.0-27-linux"
PATH_TO_STRIATA_READER = "/usr/bin/striata-readerc"
STRIATA_READER_DEFAULT_OPTIONS = ["-openwith=/usr/bin/firefox"]
TEMP_FOLDER = "/tmp/striata-gui/"
STATEMENT_BROWSE_FILE_FILTER_LIST = "Statements (*.emc);;All files (*)"

def createTempDir():
	if not os.path.exists(TEMP_FOLDER):
		os.mkdir(TEMP_FOLDER)
		
def deleteTempDir():
	if os.path.exists(TEMP_FOLDER):
		shutil.rmtree(TEMP_FOLDER)
		
def displayError(message):
	QtGui.QMessageBox.critical(None, "Decoder error", message)
	
def displayWarning(message):
	QtGui.QMessageBox.warning(None, "Decoder warning", message)

class ProcessStarter():
	def __init__(self, filePathGetter, passwordGetter):
		self.filePathGetter = filePathGetter
		self.passwordGetter = passwordGetter

	def startProcess(self):
		argumentList = [PATH_TO_STRIATA_READER] + STRIATA_READER_DEFAULT_OPTIONS + [str("-password=" + self.passwordGetter()), str(self.filePathGetter()), "-outdir="+TEMP_FOLDER]
		try:
			process = subprocess.Popen(argumentList, 0,  stdout=subprocess.PIPE, shell=False)
			output = process.communicate()
		except:
                        errorMessage = "Running " + PATH_TO_STRIATA_READER + " failed for some reason. Does the file exist? Is it the correct path? Did you install striata-reader?"
                        
                        try:
                            # not everyone has the magic lib
                            import magic
                            magicChecker = magic.open(magic.MAGIC_MIME)
                            magicChecker.load()
                            fileTypeAsString = magicChecker.file(PATH_TO_STRIATA_READER)
                            if 'text' in fileTypeAsString:
                                errorMessage = errorMessage + " Please note that the selected executable is not a binary file."
                        except:
                            pass
			displayError(errorMessage)
			return
			
		if (process.returncode != 0):
			decodeError = "Decoding document failed.\n" + PATH_TO_STRIATA_READER + " said:\n"
			for o in output:
				if (o != None):
					decodeError = decodeError + o
			displayError(decodeError)
			 		
class Dialog(QtGui.QDialog):
	def __init__(self, parent =None):
		QtGui.QWidget.__init__(self, parent)
		self.setWindowTitle('Statement decoder')
		mainLayout = QtGui.QVBoxLayout()
		self.setLayout(mainLayout)
		
		openFileLayout = QtGui.QHBoxLayout()
		openFileLabel = QtGui.QLabel("Path to statement")
		mainLayout.addWidget(openFileLabel)
		mainLayout.addLayout(openFileLayout)
		self.fileLineEdit = QtGui.QLineEdit()
		fileOpenButton = QtGui.QPushButton("Browse")
		openFileLayout.addWidget(self.fileLineEdit)
		openFileLayout.addWidget(fileOpenButton)
		
		self.decodeButton = QtGui.QPushButton("Open statement")
		
		passwordLabel = QtGui.QLabel("Enter password")
		self.passwordLineEdit = QtGui.QLineEdit()
		self.passwordLineEdit.setEchoMode(QtGui.QLineEdit.Password)
		self.decodeButton = QtGui.QPushButton("Open statement")
		self.decodeButton.setDefault(True)
		
		mainLayout.addWidget(passwordLabel)
		mainLayout.addWidget(self.passwordLineEdit)
		mainLayout.addStretch()
		mainLayout.addWidget(self.decodeButton)
		self.connect(fileOpenButton, QtCore.SIGNAL('pressed()'), self.askForFile)
		
	def askForFile(self):
		filepath = QtGui.QFileDialog.getOpenFileName(self, "Specify path to statement", "", STATEMENT_BROWSE_FILE_FILTER_LIST)
		self.setFilename(filepath)
		
	def setFilename(self, filepath):
		self.fileLineEdit.setText(filepath)

if __name__ == "__main__":
	app = QtGui.QApplication(sys.argv)
	dialog = Dialog()
	
	processStarter = ProcessStarter(dialog.fileLineEdit.text, dialog.passwordLineEdit.text)
	dialog.connect(dialog.decodeButton, QtCore.SIGNAL('pressed()'), processStarter.startProcess)
	
	createTempDir()
	if len(sys.argv) > 1:
		dialog.setFilename(sys.argv[1])	
	dialog.show()
	
	result = app.exec_()
	try:
		deleteTempDir()
	except:
		displayWarning("Temp dir '" + TEMP_FOLDER +"' could not be deleted.  Your statements were extracted here.")
		
	sys.exit(result)