AccountController.cpp

#include "AccountController.h"
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <windows.h>

AccountController::AccountController() {
	fileRead(userData);
}

//Check if the given name exists.
bool AccountController::isId(std::string name) {
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			return true;
		}
	}
	return false;
}
//Check if the given password is matched to the given name.
bool AccountController::isPwMatched(std::string name, std::string pw) {
	std::string temp;
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			temp = it->second.password;
		}
	}
	if (pw == temp) return true;
	else return false;
}
//Check if the given recovery phase is matched to the given name.
bool AccountController::isRpMatched(std::string name, std::string rp) {
	std::string temp;
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			temp = it->second.recoveryPhase;
		}
	}
	if (rp == temp) return true;
	else return false;
}
std::string AccountController::isType(std::string name) {
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			return it->second.accountType;
		}
	}
}

//add ID to id list.
void AccountController::setData(std::string ID, personalData PD) {
	userData.insert(make_pair(ID, PD));
}

//getter of id size
int AccountController::getSize() {
	return userData.size();
}

//getter of password
std::string AccountController::getPW(std::string name) {
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			return it->second.password;
		}
	}
}
std::string AccountController::getRP(std::string name) {
	for (auto it = userData.begin(); it != userData.end(); it++) {
		if (it->first == name) {
			return it->second.recoveryPhase;
		}
	}
}

std::string AccountController::getYourPW(std::string ID, std::string RP) {
	if (isId(ID)) {
		if (isRpMatched(ID, RP)) {
			std::string temp;
			for (auto it = userData.begin(); it != userData.end(); it++) {
				if (it->first == ID) {
					temp = it->second.password;
				}
			}
			return "Your password is [ " + temp + " ]";
		}
		return "Wrong Recovery Phase!";
	}
	return "No account found.";
}

void AccountController::fileRead(std::map<std::string, personalData>& v) {
	std::ifstream fin;
	fin.open("./userData.txt");
	if (fin) {
		userData.clear();
		std::string line;
		while (!fin.eof()) {
			std::getline(fin, line);

			std::string id;
			personalData data;

			std::istringstream iss(line);
			std::getline(iss, id, '|');
			std::getline(iss, data.password, '|');
			std::getline(iss, data.recoveryPhase, '|');
			std::getline(iss, data.accountType, '|');
			v.insert(std::make_pair(id, data));
		}
		fin.close();
	}
	else {
		std::ofstream emptyFile("./userData.txt");
		emptyFile.close();
	}
}
void AccountController::fileWrite(std::map<std::string, personalData>& m) {
	std::ofstream fout;
	fout.open("./userData.txt");

	if (!fout) {
		std::cout << "Error!";
		return;
	}
	for (auto it = ++m.begin(); it != m.end(); it++) {
		fout << it->first << '|';
		fout << it->second.password << '|';
		fout << it->second.recoveryPhase << '|';
		fout << it->second.accountType << '|' << '\n';
	}
	fout.close();
}

void AccountController::printUsers() {
	fileRead(userData);
	int count = 0;
	for (auto it = ++userData.begin(); it != userData.end(); it++) {
		std::cout << "[ " << ++count << " ] " << it->first << std::endl;
	}
}
void AccountController::saveAll() {
	std::ofstream fout;

	fout.open("./userData.txt");
	if (!fout) {
		std::cout << "Data loss error!" << std::endl;
		exit(0);
	}

	for (auto it = userData.begin(); it != userData.end(); it++) {
		fout << it->first << '|';
		fout << it->second.password << '|';
		fout << it->second.recoveryPhase << '|';
		fout << it->second.accountType << '|' << '\n';
	}
	fout.close();
}

void AccountController::deleteAcc(std::string name) {
	userData.erase(name);
	fileWrite(userData);

	system("cls");
	for (int i = 0; i < 3; i++) {
		std::cout << ".";
		Sleep(500);
	}
	std::cout << std::endl;
	std::cout << "Successfully deleted." << std::endl;
	std::cout << std::endl;
	std::cout << "Thank you for using our program." << std::endl;
	Sleep(500);
	std::cout << "We hope to see you again." << std::endl;

}