#include <iostream>
#include <string>
#include <fstream>
#include <conio.h>
using namespace std;
 
string lineInput ();
string createPassword ();
string encryptLine (string line, string password);
 
 
int main()
{
	bool loopHold = false;
	string menuChoice, tempLine, tempPassword;
 
	do  {
		cout << "Welcome, do you wish to:\n"
			"1 - Encrypt\n"
			"2 - Decrypt\n"
			"3 - Exit\n" << endl;
		getline(cin, menuChoice);
 
		if (menuChoice == "1")  {
			system("cls");
 
			do  {
				cout << "Options:\n"
					"1 - Encrypt a text string\n"
					"2 - Encrypt a text string and save it to a file\n"
					"3 - Encrypt a target file and save it to another\n"
					"4 - Back to previous menu\n" << endl;
				getline(cin, menuChoice);
 
				if (menuChoice == "1")  {
					system("cls");
					cout << "Enter the string to be encrypted, finish with #" << endl;
					tempLine = lineInput();
					tempPassword = createPassword();
					cout << encryptLine(tempLine, tempPassword) << endl;
					_getch();
				}
 
				else if (menuChoice == "4")  {
					system("cls");
					break;
				}
				system("cls");
			} while (loopHold);
		}
 
		else if (menuChoice == "3")  {
			return 0;
		}
 
		system("cls");
	} while (loopHold);
 
	return 0;
}
 
string lineInput ()
{
	string tempLine;
	char in;
	while (cin.get(in) && in != '#')  {
		tempLine += in;
	}
	return tempLine;
}
 
string createPassword ()
{
	cin.ignore(1000, '\n');
	string tempPassword, confirmPassword;
	bool loopHold;
 
	do  {
 
		do  {
			cout << "Enter desired password (min. 6 characters): ";
			getline(cin, tempPassword);
 
			if (tempPassword.size() < 6)  {
				cout << "You must enter a longer password!" << endl;
				loopHold = true;
			}
			else  {
				loopHold = false;
			}
		} while (loopHold);
 
		cout << "Confirm the password: ";
		getline(cin, confirmPassword);
 
		if (confirmPassword != tempPassword)  {
			cout << "The passwords do not match!" << endl;
			loopHold = true;
		}
		else  {
			loopHold = false;
		}
	} while (loopHold);
 
	return confirmPassword;
}
 
string encryptLine (string line, string password)
{
	string encryptedLine;
 
	for (unsigned pos = 0, pos2 = 0; pos < line.size(); pos ++, pos2 ++)  {
		encryptedLine += (line.at(pos) + password.at(pos2));
 
		if (pos2 > password.size())  {
			pos2 = 0;
		}
	}
	return encryptedLine;
}