#include<QString>
#include<QWidget>
#include<QHeaderView>
#include<QTableWidgetItem>
#include<iostream>
#include<string>
#include<sstream>
#include<cstdlib>
#include<fstream>
#include<ios>
#include"boost/filesystem.hpp"
#include"KeyManagementWindow.h"
#include"Encryption.h"
#include"GenerateKeys.h"
#include"KeyRetrieval.h"
#include"Conversions.h"
#include"FileNameRetrieval.h"
#include"FolderStructure.h"

KeyManagementWindow::KeyManagementWindow(QWidget* parent) : QDialog(parent)
{
	setupWindow();
	connections();
}


void KeyManagementWindow::setContentsOfComboBox()
{
	selectionBox.get()->clear();
	FileNameRetrieval fnr;
	fnr.retrieveFileNames();
	std::vector<std::string> fn{fnr.fileNameContainer()};

	for (auto fle: fn)
	{
		QString bl{QString::fromStdString(fle)};
		selectionBox.get()->addItem(bl);
	}
}
void KeyManagementWindow::setContentOfKeyView()
{
	QString testQ;
	testQ.append(selectionBox.get()->currentText());
	std::string testS{FolderStructure::keyDirectory+testQ.toStdString()};
	
	KeyRetrieval kr{testS};
	std::vector<int> asciiKeys{kr.codeCharacterStructure()};
	std::vector<std::string> asciiAssignment{kr.keyStructure()};

	auto asciiKeysIter = asciiKeys.begin();
	auto asciiAssignmentIter = asciiAssignment.begin();
	for (auto row=0; row!=rowCount; ++row)
	{
		for (auto column=0; column!=columnCount; ++column)
		{
			if (column==0)
			{
				std::string val{*asciiKeysIter++};
				QString s{val.c_str()};
				elementView.get()->setItem(row, column, new QTableWidgetItem(s));
			}	
			else
			{
				std::string val{*asciiAssignmentIter++};
				QString s{val.c_str()};
				elementView.get()->setItem(row, column, new QTableWidgetItem(s));
			}
		}
	}	
}
void KeyManagementWindow::setupWindow()
{
	windowWidth = 450;
	windowHeight = 450;
	rowCount = 96;
	columnCount = 2;

	elementView = unique_ptr<QTableWidget>{new QTableWidget};

	elementView.get()->setRowCount(rowCount);
	elementView.get()->setColumnCount(columnCount);
	tableHeader<<"character"<<"key";
	elementView.get()->setHorizontalHeaderLabels(tableHeader);
	elementView.get()->verticalHeader()->setVisible(false);

	selectionBox = unique_ptr<QComboBox>{new QComboBox{}};

	generateNewKeys = unique_ptr<QPushButton>{new QPushButton(tr("Generate New Default Key"))};
	closeButton = unique_ptr<QPushButton>{new QPushButton(tr("close"))};
	actionButton = unique_ptr<QPushButton>{new QPushButton};

	mainLayout = unique_ptr<QHBoxLayout>{new QHBoxLayout};
	subLayoutOne = unique_ptr<QVBoxLayout>{new QVBoxLayout};
	subLayoutTwo = unique_ptr<QVBoxLayout>{new QVBoxLayout};

	subLayoutGoonOne = unique_ptr<QVBoxLayout>{new QVBoxLayout};
	subLayoutGoonTwo = unique_ptr<QVBoxLayout>{new QVBoxLayout};

	subLayoutOne.get()->addWidget(elementView.get());

	subLayoutGoonOne.get()->addWidget(selectionBox.get());
	subLayoutGoonOne.get()->addWidget(actionButton.get());
	subLayoutGoonTwo.get()->addWidget(generateNewKeys.get());
	subLayoutGoonTwo.get()->addWidget(closeButton.get());

	subLayoutTwo.get()->addLayout(subLayoutGoonOne.get());
	subLayoutTwo.get()->addLayout(subLayoutGoonTwo.get());


	mainLayout.get()->addLayout(subLayoutOne.get());
	mainLayout.get()->addLayout(subLayoutTwo.get());


	setContentsOfComboBox();

	setLayout(mainLayout.get());

	setFixedWidth(windowWidth);
	setFixedHeight(windowHeight);

	setWindowTitle("Key Management Window");
}
void KeyManagementWindow::connections()
{
	QObject::connect(generateNewKeys.get(), SIGNAL(clicked()), this, SLOT(generation()));
	QObject::connect(closeButton.get(), SIGNAL(clicked()), this, SLOT(exitApplication()));
	QObject::connect(actionButton.get(), SIGNAL(clicked()), this, SLOT(setContentOfKeyView()));
}
void KeyManagementWindow::test()
{
	std::string emp{"d"};
	Encryption ec{emp};	
	std::map<char, std::string> encrypted{ec.encryptedCharactersStructure()};
	QString bo{selectionBox.get()->currentText()};
	std::string k{bo.toStdString()};
	char f{};

	std::stringstream lazy{}; 
	lazy << k;
	lazy >> f;

	std::string value{encrypted[f]};	
	QString v{QString::fromStdString(value)};
}
void KeyManagementWindow::generation()
{
	GenerateKeys gk{};
	gk.keyMove();
	gk.keyDump();
	setContentsOfComboBox();
}
void KeyManagementWindow::exitApplication() { this->hide(); }