/*
1. Към класа CAddress от упражнение 3 да се добави глобална приятелска функция (оператор) за извеждане на данните за адреса по указан като параметър изходен поток,например:
friend ostream& operator<<(ostream& toStream, const CAddress& addr)
2. Към класа CPerson от упражнение 3 да се добави частна функция за получаване на година, месец и ден от подадения като параметър единен граждански номер,например:
void getYMD(const string& strEGN,int &iYear,int &iMonth, int &iDay)const;
3. Към класа CPerson от упражнение 3 да се добави глобална приятелска функция (оператор) за извеждане на данните за личността по указан като параметър изходен поток,например:
friend ostream& operator<<(ostream& toStream, const CPerson& pers)
4. Към класа CStudent от упражнение 3 да се добави глобална приятелска функция за конзолен вход към единен граждански номер на подадения като параметър обект от класа CStudent,например:
friend void ConsoleInputEGN(CStudent& stud)
5. Към класа CStudent от упражнение 3 да се добави глобална приятелска функция (оператор) за извеждане на данните за студента по указан като параметър изходен поток,например:
friend ostream& operator<<(ostream& toStream, const CStudent& stud)
6. Към класа CStudent от упражнение 3 да се добавят следните глобални приятелски функции (оператори) за сравнение на двама студента (параметри):
? За равенство по възраст ==;
? За неравенство по възраст !=;
? За по-малко по възраст <;
? За по-голямо по възраст >;
7. Да се състави подходяща главна функция за проверка на създадените функции и оператори.*/


#include <iostream>
#include <string>
#include <ctime>
using namespace std;

class CAddress {
string street;
string post;
string city;
public:

friend ostream& operator<<(ostream& toStream, const CAddress& addr);

CAddress() {
street = "";
post = "";
city = "";
}

explicit CAddress(string s, string p, string c) {
street = s;
post = p;
city = c;
}

CAddress(const CAddress& a) //copy constructor
{
street = a.street;
post = a.post;
city = a.city;
}

CAddress(const CAddress* a) //copy constructor
{
street = a->street;
post = a->post;
city = a->city;
}

string GetStreet() {
return street;
}

void SetStreet(string s) {
street = s;
}

string GetPost() {
return post;
}

void SetPost(string p) {
post = p;
}

string GetCity() {
return city;
}

void SetCity(string c) {
city = c;
}

ostream& Output(ostream& toStream) {
toStream << street << ";" << post << ";" << city;
return toStream;
}

};

class CPerson {
public:
enum Gender { male, female, undef };
private:
string name;
string egn;
Gender gender;
public:
friend ostream& operator<<(ostream& toStream, const CPerson& pers);
CPerson() {
name = "";
gender = undef;
egn = "";
}

explicit CPerson(string szName, Gender szGender) {
name = szName;
gender = szGender;
egn = "";
}

explicit CPerson(string name, string egn, Gender gender) {
this->name = name;
if (checkEGN(egn))
this->egn = egn;
else
this->egn = "1111111111";
this->gender = gender;
}

void SetName(string szName) {
name = szName;
}

string GetName() {
return name;
}

void SetGender(Gender szGender) {
gender = szGender;
}

Gender GetGender() {
return gender;
}

bool SetEGN(string e) {
if (checkEGN(e)) {
egn = e;
return true;
}
else if (e == "1111111111") {
egn = e;
return true;
}
else
return false;
}

string GetEGN() {
return egn;
}
private:
bool isCorrectDateOfEGN(std::string yymmdd) {
if (yymmdd.length() != 6)
return false;

int dayspermon[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };

int month, day, year;

year = atoi(yymmdd.substr(0, 2).c_str());
month = atoi(yymmdd.substr(2, 2).c_str());
day = atoi(yymmdd.substr(4, 2).c_str());

if (!(year != 0 && month != 0 && day != 0))
return false;
if (year < 0 || month < 1 || day < 1) {
return false;
}

if (month >= 13 && month <= 20) {
return false;
}
else if (month >= 21 && month <= 32) {
month = month - 20;
}
else if (month >= 33 && month <= 40) {
return false;
}
else if (month >= 41 && month <= 52) {
month = month - 40;
}
else if (month > 52) {
return false;
}

if (month == 2)
if ((year % 4) == 0)
dayspermon[2] = 29;

if (day > dayspermon[month - 1]) {
return false;
}

return true;
}

bool checkEGN(std::string egn) {
int ctrSum = 0;
int ctrField = 0;
int teglo[] = { 2, 4, 8, 5, 10, 9, 7, 3, 6 };

if (egn.length() != 10)
return false;

long long tmpEgn = atoi(egn.c_str());
if (tmpEgn == 0)
return false;

if (!isCorrectDateOfEGN(egn.substr(0, 6))) {
return false;
}

for (int i = 0; i < egn.length() - 1; ++i) {
string t = egn.substr(i, 1);
ctrSum += teglo[i] * atoi(t.c_str());
}

ctrField = (ctrSum % 11) % 10;

if (ctrField != atoi(egn.substr(9, 1).c_str()))
return false;
else return true;
}

void getYMD(const string& strEGN, int& iYear, int& iMonth, int& iDay) const {
iYear = atoi(strEGN.substr(0, 2).c_str());
iMonth = atoi(strEGN.substr(2, 2).c_str());
iDay = atoi(strEGN.substr(4, 2).c_str());

if (iMonth >= 21 && iMonth <= 32) {
iMonth = iMonth - 20;
iYear = iYear + 1800;
}
else if (iMonth >= 41 && iMonth <= 52) {
iMonth = iMonth - 40;
iYear = iYear + 2000;
}
else iYear = iYear + 1900;
}

public:

void getYMD(int& iYear, int& iMonth, int& iDay) const {
iYear = atoi(egn.substr(0, 2).c_str());
iMonth = atoi(egn.substr(2, 2).c_str());
iDay = atoi(egn.substr(4, 2).c_str());

if (iMonth >= 21 && iMonth <= 32) {
iMonth = iMonth - 20;
iYear = iYear + 1800;
}
else if (iMonth >= 41 && iMonth <= 52) {
iMonth = iMonth - 40;
iYear = iYear + 2000;
}
else iYear = iYear + 1900;
}

Gender stringToGender(const string& strGender) const {
if (strGender == "male")
return Gender::male;
else if (strGender == "female")
return Gender::female;
else return Gender::undef;
}

string genderToString() const {
switch (gender) {
case male:
return "male";
case female:
return "female";
case undef:
return "undef";
}
}
};

class CStudent : public CPerson {
string fnumber;
CAddress address;
public:
friend ostream& operator<<(ostream& toStream, CStudent& stud);
friend void ConsoleInputEGN(CStudent& stud);
friend bool operator==(CStudent& s1, CStudent& s2);
friend bool operator!=(CStudent& s1, CStudent& s2);
friend bool operator>(CStudent& s1, CStudent& s2);
friend bool operator<(CStudent& s1, CStudent& s2);
CStudent() {
fnumber = "";
address = new CAddress();
}

explicit CStudent(string n, string egn, Gender g, string s, string c, string p) : CPerson(n, egn, g) {
address = new CAddress();
address.SetStreet(s);
address.SetCity(c);
address.SetPost(p);
}

explicit CStudent(string n, string egn, Gender g, string s, string c, string p, string fn) : CPerson(n, egn, g) {
address = new CAddress();
address.SetStreet(s);
address.SetCity(c);
address.SetPost(p);
fnumber = fn;
}

CStudent(CStudent& a) {
address = a.address;
fnumber = a.fnumber;
this->SetName(a.GetName());
this->SetGender(a.GetGender());
this->SetEGN(a.GetEGN());
}

void SetAddress(CAddress& a) {
address = a;
}

CAddress& GetAddress() {
return address;
}

void SetFNum(string fn) {
this->fnumber = fn;
}

string GetFNum() {
return fnumber;
}

ostream& Output(ostream& out) {
out << GetName() << ";" << GetGender() << ";" << GetEGN() << ";";
address.Output(out);
out << ";" << fnumber << endl;
return out;
}
};

ostream& operator<<(ostream& toStream, const CAddress& addr) {
toStream << addr.street << ";" << addr.post << ";" << addr.city;
return toStream;
}

ostream& operator<<(ostream& toStream, const CPerson& pers) {
toStream << pers.name << ";" << pers.gender << ";" << pers.egn;
return toStream;
}

ostream& operator<<(ostream& toStream, CStudent& stud) {
toStream << stud.GetName() << ";" << stud.GetGender() << ";" << stud.GetEGN() << ";";
toStream << stud.GetAddress();
toStream << ";" << stud.fnumber;
return toStream;
}

void ConsoleInputEGN(CStudent& stud) {
string input;
while (true) {
cout << "Enter valid EGN:";
cin >> input;
if (stud.SetEGN(input))
break;
else cout << "Invalid EGN." << endl;
}
}

bool operator==(CStudent& s1, CStudent& s2) {
int s1year, s1month, s1day;
int s2year, s2month, s2day;
s1.getYMD(s1year, s1month, s1day);
s2.getYMD(s2year, s2month, s2day);

if ((s1year == s2year) && (s1month == s2month) && (s1day == s2day))
return true;
else return false;
}

bool operator!=(CStudent& s1, CStudent& s2) {
int s1year, s1month, s1day;
int s2year, s2month, s2day;
s1.getYMD(s1year, s1month, s1day);
s2.getYMD(s2year, s2month, s2day);
if ((s1year != s2year) || (s1month != s2month) || (s1day != s2day))
return true;
else return false;
}

bool operator>(CStudent& s1, CStudent& s2) {
int s1year, s1month, s1day;
int s2year, s2month, s2day;
s1.getYMD(s1year, s1month, s1day);
s2.getYMD(s2year, s2month, s2day);

if (s1year > s2year)
return false;
else if (s1year == s2year) {
if (s1month > s2month)
return false;
else if (s1month == s2month) {
if (s1day > s2day)
return false;
else return true;
}
else return true;
}
else return true;
}

bool operator<(CStudent& s1, CStudent& s2) {
int s1year, s1month, s1day;
int s2year, s2month, s2day;
s1.getYMD(s1year, s1month, s1day);
s2.getYMD(s2year, s2month, s2day);

if (s1year < s2year)
return false;
else if (s1year == s2year) {
if (s1month < s2month)
return false;
else if (s1month == s2month) {
if (s1day < s2day)
return false;
else return true;
}
else return true;
}
else return true;
}


int main() {
// https://georgi.unixsol.org/programs/egn.php

CStudent s1("Gosho", "5412065927", CPerson::male, "Roza", "Varna", "9000", "24621449");
CStudent s2("Mariya", "3108208010", CPerson::female, "Kon", "Dobrich", "8000", "24621507");

if (s1 > s2) {
"S1 e po-vuzrasten ot S2";
}
else if (s1 < s2) {
"S2 e po-vuzrasten ot S1";
}
else {
cout << "Dvata studenta sa astralni bliznaci";
}
return 0;
}