# coding: utf-8

from argparse import ArgumentParser
import matplotlib.pyplot as plt
from lorem import text as loremtext


def checkArguments():
	args = ArgumentParser(description='Cryptanalyst')
	args.add_argument('-f', '--filename', help='Text to analyze')
	args.add_argument('-s', '--stats', help='Statistics', action='store_true')
	args.add_argument('-d', '--decrypt', help='Decrypt Caesar text', action='store_true')
	return args.parse_args()

def readCipherText(filename):
	file = []
	with open(filename, 'r') as f:
		data = f.readlines()

		for entry in data:
			file.append(entry.strip())

	return file

def main():
	args = checkArguments()
	cipherFile = None

	if args.filename:
		cipherFile = readCipherText(args.filename)

	if args.stats:
		stats(cipherFile)
	if args.decrypt:
		decryptCaesarText(cipherFile)	
	

def stats(file=None):
	if file is None:
		text = loremtext()
	else:
		text = file

	lettersOccurence = {}
	letterUpperToAscii = ord('A') # Get the ASCII code
	for i in range (0, 26):
		lettersOccurence[chr(i + letterUpperToAscii)] = 0

	for line in text:
		for letter in line:
			letter = letter.upper()
			if letter in lettersOccurence:
				lettersOccurence[letter] = lettersOccurence[letter] + 1

	x = [chr(letter + 65) for letter in range(0, 26)]
	y = [lettersOccurence[tmp] for tmp in lettersOccurence]

	largeur = 0.5
	plt.bar(x, y, largeur)
	plt.show()

def decryptCaesarText(cipherFile):
	if cipherFile is None:
		return

	#for lines in cipherFile:
	#	splitLines = lines.split(' ')

	#	for word in splitLines:
	#		if len(word) <= 3:
	#			print(word)
	

if __name__ == "__main__":
	main()