main ajeitada 2

#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <fstream>
#include <locale.h>
#include "CifraDeCesar.h"
#include "criptografia.h"
#include "Polialfabetica.h"
#include "CifraDeUsoUnico.h"
#include "CifraDeTransposicaoDeColunas.h"
#include "enigma.h"

using namespace std;

int main()
{
	ofstream arquivoS;
	ifstream arquivoE;

	CifraDeCesar c(0, "", "", "");
	Polialfabetica p("", "", "");
	CifraDeUsoUnico u("", "", "");
	CifraDeTransposicaoDeColunas t("", "", "");
	enigma e("","","",0);
    int i=1,k =0, j=0, menu=0;
    char c1,c2;
	string s;
	setlocale(LC_ALL,"");

	while(i)
	{
		system("cls");
		cout<<"Programa de Criptografia, Digite:"<<endl<<"1: para usar a cifra de César"<<endl<<"2: para usar a cifra polialfabética"<<endl<<"3: para usar a cifra de uso único"<<endl<<"4: para usar a cifra de transposição de colunas"<<endl<<"5: para usar a máquina enigma"<<endl<<"6: para acessar o histórico:"<<endl<<"0: para sair"<<endl;
    	cin>>menu;
    	if(menu==1)
        {
			system("cls");
			arquivoS.open("cezar.txt", ios::app);
			if(arquivoS.is_open())
			{
				cout<<"Cifra de césar:"<<endl<<"Digite 1 para encriptar:"<<endl<<"Digite 2 para decriptar:"<<endl;
				cin>>i;
				if(i==1)
				{
			   	   	system("cls");
 			   	   	cout<<endl<<"Digite a frase que será encriptada: (letras maiúsculas e minúsculas)"<<endl;
 			  	    	cin>>s;
 			  	    	c.setEntrada(s);
				   	cout<<"Digite o quanto quer deslocar os caracteres: (número inteiro de tamanho infinito)"<<endl;
			  	    	cin>>i;
 			  	    	c.setdesloca(i);
 			  	    	c.encripta();
 			  	    	arquivoS<<"criptografia:  "<<"  Entrada:"<<c.getEntrada()<<"  Deslocamento:"<<c.getdesloca()<<"  Saida:"<<c.getSaida()<<endl;
				   	arquivoS.close();
 			  	    	cout<<endl<<"A mensagem encriptada é: "<<c.getSaida();
		   		}
				if(i==2)
				{
			   		system("cls");
 			   		cout<<endl<<"Digite a frase encriptada: (letras maiúsculas e minúsculas)"<<endl;
 			   		cin>>s;
 			   		c.setSaida(s);
			   	   	cout<<"Digite o quanto os caracteres foram deslocados: (número inteiro de tamanho infinito)"<<endl;
			   		cin>>i;
 			   		c.setdesloca(i);
 			   	   	c.desencripta();
	   			   	arquivoS<<"decriptografia:   "<<"Entrada:"<<c.getSaida()<<"  Deslocamento:"<<c.getdesloca()<<"  Saida:"<<c.getEntrada()<<endl;
				   	arquivoS.close();
 			   	   	cout<<endl<<"A mensagem decriptada é:"<<c.getEntrada();
	  	  	    }
			}
			else
			{
				system("cls");
				cout<<"Não foi possível abrir o arquivo";
			}

	   	}
	   	if(menu==2)
		{
		   	system("cls");
		   	arquivoS.open("poli.txt", ios::app);
		   	if(arquivoS.is_open())
			{
				cout<<"Cifra polialfabética"<<endl<<"Digite 1 para encriptar:"<<endl<<"Digite 2 para decriptar:"<<endl;
		   		cin>>i;
		   		if(i==1)
				{
					system("cls");
					cout<<"Digite frase que seré encriptada: (letras maiúsculas e minúsculas)"<<endl;
					cin>>s;
				   	p.setEntrada(s);
				   	cout<<endl<<"Digite a palavra chave: (uma palavra de qualquer tamanho contendo apenas letras minúsculas)"<<endl;
				   	cin>>s;
				   	p.setChave(s);
				   	p.encripta();
				   	arquivoS<<"criptografia:   "<<"Entrada:"<<p.getEntrada()<<"  Chave:"<<p.getChave()<<"  Saida:"<<p.getSaida()<<endl;
				   	arquivoS.close();
				   	cout<<endl<<"A mensagem encriptada é:"<<p.getSaida();
			   	}

				if(i==2)
		   		 {
					system("cls");
				   	cout<<"Digite a frase que será decriptada: (letras maiúsculas e minúsculas)"<<endl;
				   	cin>>s;
				   	p.setSaida(s);
				   	cout<<endl<<"Digite a palavra chave:  (uma palavra de qualquer tamanho contendo apenas letras minúsculas)"<<endl;
				   	cin>>s;
				   	p.setChave(s);
				   	p.desencripta();
				   	cout<<endl<<"A mensagem decriptada é:"<<p.getEntrada();
				   	arquivoS<<"descriptografia:   "<<"Entrada:"<<p.getSaida()<<"  Chave:"<<p.getChave()<<"  Saida:"<<p.getEntrada()<<endl;
				   	arquivoS.close();
				}
			}
			else
			{
				system("cls");
				cout<<"Não foi possivel abrir o arquivo";
			}

	    }
 	   	if(menu==3)
		{
			system("cls");
			arquivoS.open("unico.txt", ios::app);
			if(arquivoS.is_open())
			{
				cout<<"Cifra de uso único:"<<endl<<"Digite 1 para encriptar:"<<endl<<"Digite 2 para decriptar:"<<endl;
			   	cin>>i;
				if(i==1)
				{
				   	system("cls");
 			   		cout<<endl<<"Digite a frase que será encriptada: (letras maiúsculas e minúsculas)"<<endl;
 			   		cin>>s;
 			   		u.setEntrada(s);
 			   		cout<<"Gerando chave aleatória:";
 			   		u.tamanhoChaveEntrada();
 			   		u.geraChave();
					   	for(i=0; i<u.getEntrada().size(); i++)
					   	{
						  		cout<<"  "<<u.getChaves(i);
				   	}
					u.encripta();
					arquivoS<<"criptografia:   "<<"Entrada:"<<u.getEntrada()<<"  Chave:";
					for(i=0; i<u.getEntrada().size(); i++)
					   	{
						  		arquivoS<<" "<<u.getChaves(i);
				   	}
				   	arquivoS<<"  Saida:"<<u.getSaida()<<endl;
				   	arquivoS.close();
				   	cout<<endl<<endl<<"A mensagem encriptada é:"<<u.getSaida();
				}
				if(i==2)
				   	{
					system("cls");
 			   		cout<<endl<<"Digite a frase encriptada: (letras maiúsculas e minúsculas)"<<endl;
 			   	   	cin>>s;
 			   	   	u.setSaida(s);
 			   	   	u.tamanhoChaveSaida();
 			   	   	cout<<endl<<"Digite as chaves:"<<endl;
					   	for(i=0; i<u.getSaida().size(); i++)
					   	{
					   	cout<<i +1<<"º"<<" chave:";
						  		cin>>k;
						  		u.setChaves(i,k);
				   	}
				   	u.descriptografa();
				   	arquivoS<<"descriptografia:   "<<"Entrada:"<<u.getSaida()<<"  Chave:";
					   	for(i=0; i<u.getSaida().size(); i++)
					   	{
						  		arquivoS<<" "<<u.getChaves(i);
				   	}
				   	arquivoS<<"  Saida:"<<u.getEntrada()<<endl;
				   	arquivoS.close();
				   	cout<<endl<<endl<<"A frase decriptada é:"<<u.getEntrada();
				  }
			}
			else
			{
				system("cls");
				cout<<"Não foi possivel abrir o arquivo";
			}
		}
		if(menu==4)
		{

		   	system("cls");
		   	arquivoS.open("coluna.txt", ios::app);
			if(arquivoS.is_open())
			{
				cout<<"Cifra de transposição de colunas:"<<endl<<"Digite 1 para encriptar:"<<endl<<"Digite 2 para decriptar:"<<endl;
		   		cin>>i;
		   		if(i==1)
				{
				    try{
                        system("cls");
                        cout<<endl<<"Digite a palavra chave: (Palavra de qualquer tamanho sem caracteres repetidos)"<<endl;
                        cin>>s;
                        t.setChave(s);
                        cout<<"Digite a frase que será encriptada: (Texto qualquer sem espaços)"<<endl;
                        cin>>s;
                        t.setEntrada(s);
                        t.encripta();
                        arquivoS<<"criptografia:  "<<"Entrada:"<<t.getEntrada()<<"  Chave:"<<t.getChave()<<"  Saida:"<<t.getSaida()<<endl;
                        arquivoS.close();
                        cout<<endl<<"A mensagem encriptada é:"<<t.getSaida();
				    }
                        catch(DivideByZeroException &divideByZeroException){
                            cout<<"A chave deve ter, no mínimo, um caractere."<<endl;
                        }

				}

				if(i==2)
		    	{
				   	try{
                        system("cls");
                        cout<<endl<<"Digite a palavra chave:";
                        cin>>s;
                        t.setChave(s);
                        cout<<"Digite a mensagem que será decriptada:";
                        cin>>s;
                        t.setEntrada(s);
                        t.descriptografa();
                        arquivoS<<"descriptografia:  "<<"Entrada:"<<t.getEntrada()<<"  Chave:"<<t.getChave()<<"  Saida:"<<t.getSaida()<<endl;
                        arquivoS.close();
                        cout<<endl<<"A mensagem decriptada é:"<<t.getSaida();
                        }
                        catch(DivideByZeroException &divideByZeroException){
                            cout<<"A chave deve ter, no mínimo, um caractere"<<endl;
                        }

		    		 }
			}
			else
			{
				system("cls");
				cout<<"Não foi possível abrir o arquivo.";
			}
		}
		if(menu==5)
		{
			system("cls");
			arquivoS.open("enigma.txt", ios::app);
			if(arquivoS.is_open())
			{
				cout<<"Máquina enigma"<<endl;
				cout<<endl<<"Defina quais rotores irá usar(3 números diferentes de 1 a 5)"<<endl;
				for(k=0;k<3;k++)
				{
				   	cout<<"rotor "<<k+1<<":";
					cin>>i;
					e.setescolha(k,i);
				}

				cout<<endl<<"Defina os anéis dos rotores (3 números  de 1 a 26)"<<endl;
				for(k=0;k<3;k++)
				{
					cout<<"anel "<<k+1<<":";
					cin>>i;
					e.setanel(k,i);
				}

				cout<<endl<<"Defina a rotação dos rotores(3 números de 1 a 26)"<<endl;
				for(k=0;k<3;k++)
				{
					cout<<"rotação "<<k+1<<":";
					cin>>i;
					e.setrotacao(k,i);
				}

				arquivoS<<"Rotores escolhidos: "<<e.getescolha(0)<<","<<e.getescolha(1)<<","<<e.getescolha(2);
				arquivoS<<"  Posicão dos anéis: "<<e.getanel(0)<<","<<e.getanel(1)<<","<<e.getanel(2);
				arquivoS<<"  Rotacão dos rotores: "<<e.getrotacao(0)<<","<<e.getrotacao(1)<<","<<e.getrotacao(2);
				cout<<endl<<"Digite a frase de entrada: (sem espacos)"<<endl;
				cin>>s;
				e.setEntrada(s);
				arquivoS<<"  Letras trocadas: ";
				while(i)
				{
					cout<<endl<<"Digite 1 para conectar letras, 0 para continuar:";
					cin>>i;
					if(i==1)
					{
						cout<<"Digite a 1 letra:";
						cin>>c1;
						cout<<"Digite a 2 letra:";
						cin>>c2;
						e.trocaletras(c1,c2);
						arquivoS<<c1<<"&"<<c2<<" , ";
					}

				}
				e.encripta();
				cout<<endl<<"A mensagem encriptada é: "<<e.getSaida();
				arquivoS<<" Entrada:"<<s<<"  Saída:"<<e.getSaida()<<endl;
				arquivoS.close();
			}
			else
			{
				system("cls");
				cout<<"Não foi possível abrir o arquivo.";
			}

		}
		if(menu==6)
		{
			system("cls");
			cout<<"Digite qual método quer acessar o historico:"<<endl<<"1:Cifra de César:"<<endl<<"2:Cifra polialfabética:"<<endl<<"3:Cifra de uso único:"<<endl<<"4:Cifra de transposição de colunas:"<<endl<<"5:Máquina enigma"<<endl;
			cin>>i;
			if(i==1)
			{
				system("cls");
				arquivoE.open("cezar.txt");
				if(arquivoE.is_open())
				{
					while(getline(arquivoE,s))
						cout<<s<<endl<<endl;
					arquivoE.close();
				}
				else
				{
				   	system("cls");
				   	cout<<"Não foi possível abrir o arquivo";
				}
			}

			if(i==2)
			{
				arquivoE.open("poli.t");
				if(arquivoE.is_open())
				{
					while(getline(arquivoE,s))
					   	cout<<s<<endl<<endl;
				   	arquivoE.close();
				}
				else
				{
				   	system("cls");
				   	cout<<"Não foi possível abrir o arquivo";
				}

			}
			if(i==3)
			{
				system("cls");
				arquivoE.open("unico.txt");
				if(arquivoE.is_open())
				{
					while(getline(arquivoE,s))
						cout<<s<<endl<<endl;
					arquivoE.close();
				}
				else
				{
				   	system("cls");
				   	cout<<"Não foi possível abrir o arquivo";
				}

			}
			if(i==4)
			{
				system("cls");
				arquivoE.open("coluna.txt");
				if(arquivoE.is_open())
				{
					while(getline(arquivoE,s))
						cout<<s<<endl<<endl;
					arquivoE.close();
				}
				else
				{
				   	system("cls");
				   	cout<<"Não foi possível abrir o arquivo";
				}

			}
			if(i==5)
			{
				system("cls");
				arquivoE.open("enigma.txt");
				if(arquivoE.is_open())
				{
					while(getline(arquivoE,s))
						cout<<s<<endl<<endl;
					arquivoE.close();
				}
				else
				{
				   	system("cls");
				   	cout<<"Não foi possível abrir o arquivo";
				}
			}
		}
		if (menu){
            cout<<endl<<"Digite 1 para continuar usando o programa ou 0 para sair";
            cin>>i;
		}
		else
            i = 0;

	}

	return 0;
}