DataManager.cpp

#pragma once

#include "DataManager.h"
#include "json/json.h"

using namespace std;

DataManager::DataManager() {}

DataManager::DataManager(string usrDataName, string pkgDataName) {
	setUsrFile(usrDataName);
	setPkgFile(pkgDataName);
	setAdminFile("./data/admin.dat");

}

DataManager::~DataManager() {
	UsrFile.close();
	PkgFile.close();
	AdminFile.close();
}


int DataManager::setUsrFile(string name) {
	UsrFile.open(name, ios::in | ios::out | ios::binary);
	if (UsrFile) {
		User tmp;
		int n = 0;
		UsrFile.seekg(0, ios::beg);
		while (!UsrFile.eof() && UsrFile.peek() != EOF) {
			UsrFile.read((char*)&tmp, sizeof(User));
			users.push_back(tmp);
			int n = users.size() - 1;//vector下标从0开始
			UsrMap[tmp.userName] = n;
			UsrMap[tmp.UUID] = n;
		}
	}
	else {
		cout << "usrFileNotFound,CreatNewOne.\n" << endl;
		UsrFile.open(name, ios::out | ios::binary);
		UsrFile.close(); UsrFile.clear();
		UsrFile.open(name, ios::in | ios::out | ios::binary);
	}

	UsrFile.seekg(0);
	if (!UsrFile) { return -1; }
	return 0;
}

int DataManager::setPkgFile(string name) {
	PkgFile.open(name, ios::in | ios::out | ios::binary);
	if (PkgFile) {
		Package tmp;
		PkgFile.seekg(0);
		while (!PkgFile.eof() && PkgFile.peek() != EOF) {
			PkgFile.read((char*)&tmp, sizeof(Package));
			packages.push_back(tmp);
			int n = packages.size() - 1;//vector下标从0开始
			PkgMap[tmp.UUID] = n;
			PkgFromMap[tmp.from].push_back(n);
			PkgToMap[tmp.to].push_back(n);
		}
	}
	else {
		cout << "pkgFileNotFound,CreatNewOne.\n" << endl;
		PkgFile.open(name, ios::out | ios::binary);
		PkgFile.close(); PkgFile.clear();
		PkgFile.open(name, ios::in | ios::out | ios::binary);
	}

	PkgFile.seekg(0);
	if (!PkgFile) { return -1; }
	return 0;
}

int DataManager::setAdminFile(string name) {
	AdminFile.open(name, ios::in | ios::out | ios::binary);
	if (AdminFile) {
		AdminFile.seekg(0);
		while (!AdminFile.eof() && AdminFile.peek() != EOF) {
			AdminFile.read((char*)&admin, sizeof(Admin));
		}
	}
	else {
		cout << "adminFileNotFound,CreatNewOne.\n" << endl;
		AdminFile.open(name, ios::out | ios::binary);
		strcpy_s(admin.pwd, 17, "12345678");
		cout << "admin defaut pwd:12345678\n" << endl;
		admin.ballance = 0;
		AdminFile.write((char*)&admin, sizeof(Admin));

		AdminFile.close(); AdminFile.clear();
		AdminFile.open(name, ios::in | ios::out | ios::binary);
	}

	AdminFile.seekg(0);
	if (!AdminFile) { return -1; }
	return 0;
}

int DataManager::checkExistUsr(string usrName) {
	if (!UsrFile) { return -1; }
	map<string, int>::iterator iter;
	iter = UsrMap.find(usrName);
	if (iter == UsrMap.end()) { return -404; }
	else { return UsrMap[usrName]; }
}

int DataManager::verifyUsr(string usrName, string pwd) {
	if (!UsrFile) { return -1; }
	int id = checkExistUsr(usrName);
	if (id < 0) { return -404; }
	else {
		if (strcmp(users[id].password, pwd.c_str()) == 0) {
			return 0;
		}
		else { return -403; }
	}
}

int DataManager::changePwd(string usrName, string newpwd) {
	if (strcmp(usrName.c_str(), "admin") == 0) {
		if (checkPwd(newpwd, 0) == 0) {
			strcpy_s(admin.pwd, 17, newpwd.c_str());
			AdminFile.seekp(0);
			AdminFile.write((char*)&admin, sizeof(Admin));
			return 0;
		}
		else {
			return -400;
		}
	}
	int seq = checkExistUsr(usrName);
	int res = users[seq].changePwd(newpwd);
	if (res == 0) {
		UsrFile.seekp(seq * sizeof(User), ios::beg);
		UsrFile.write((char*)&users[seq], sizeof(User));
		return 0;
	}
	else { return -400; }

}

int DataManager::addUsr(User& tmp) {
	if (!UsrFile) { return -1; }
	int id = checkExistUsr(tmp.userName);
	if (id != -404) {
		return -2;
	}
	users.push_back(tmp);
	int n = users.size() - 1;//vector下标从0开始
	UsrMap[tmp.userName] = n;
	UsrMap[tmp.UUID] = n;

	UsrFile.seekp(0, ios::end);
	UsrFile.write((char*)&tmp, sizeof(User));

	return 0;
}

int DataManager::charge(string usr, double amount) {
	if (!UsrFile) { return -1; }
	int seq = UsrMap[usr];
	if (seq < 0) { return -404; }
	else {
		int res = users[seq].charge(amount);
		UsrFile.seekp(seq * sizeof(User), ios::beg);
		UsrFile.write((char*)&users[seq], sizeof(User));
		return res;
	}
}

Json::Value DataManager::getUsrData(string usr) {
	Json::Value j;
	if (strcmp(usr.c_str(), "admin") == 0) {
		for (int seq = 0; seq < users.size(); ++seq) {
			Json::Value res;
			res["usrname"] = users[seq].userName;
			res["pwd"] = users[seq].password;
			res["realName"] = users[seq].realName;
			res["address"] = users[seq].address;
			res["tel"] = users[seq].tel;
			res["ballance"] = users[seq].AccBalance;
			res["uuid"] = users[seq].UUID;
			j.append(res);
		}
	}
	else {
		int seq = UsrMap[usr];
		if (seq >= 0) {
			Json::Value res;
			res["ballance"] = users[seq].AccBalance;
			res["usrname"] = users[seq].userName;
			res["pwd"] = users[seq].password;
			res["realName"] = users[seq].realName;
			res["address"] = users[seq].address;
			res["tel"] = users[seq].tel;
			res["uuid"] = users[seq].UUID;
			j.append(res);
		}
	}
	return j;
}

//包裹
int DataManager::checkExistPkg(string uuid) {
	if (!PkgFile) { return -1; }
	map<string, int>::iterator iter;
	iter = PkgMap.find(uuid);
	if (iter == PkgMap.end()) { return -404; }
	else { return PkgMap[uuid]; }
}

int DataManager::_addPkg(Package& tmp) {
	if (!PkgFile || !UsrFile) { return -1; }
	int id = checkExistPkg(tmp.UUID);
	if (id != -404) {
		return -500;
	}
	packages.push_back(tmp);
	int n = packages.size() - 1;//vector下标从0开始
	PkgMap[tmp.UUID] = n;
	PkgFromMap[tmp.from].push_back(n);
	PkgToMap[tmp.to].push_back(n);
	PkgFile.seekp(0, ios::end);
	PkgFile.write((char*)&tmp, sizeof(Package));
	return 0;
}

int DataManager::addPkg(User& f, User& t, double amount, int type, string d) {
	if (!PkgFile || !UsrFile) { return -1; }
	if (type != 2 && type != 5 && type != 8) {
		return -400;
	}
	if (f.AccBalance < amount * type) { return -402; }//没钱
	else {
		f.AccBalance -= amount * type;
		admin.ballance += amount * type;

		return _addPkg(Package(f, t, type, d));
	}
}

Json::Value DataManager::getPkgData(string usr, char w) {
	Json::Value res;
	if ((w == 'a' || w == 'A') && (strcmp(usr.c_str(), "admin") == 0)) {
		//admin、all，返回全部包裹信息
		for (int seq = 0; seq < packages.size(); ++seq) {
			Json::Value j;

			j["from"] = packages[seq].from;
			j["fAddr"] = packages[seq].fAddress;
			j["to"] = packages[seq].to;
			j["tAddr"] = packages[seq].tAddress;

			j["sendTime"] = packages[seq].sendTime;
			j["rcvTime"] = packages[seq].rcvTime;

			j["type"] = packages[seq].type;
			j["description"] = packages[seq].description;
			j["UUID"] = packages[seq].UUID;
			j["status"] = packages[seq].status;

			res.append(j);
		}
	}
	if (w == 's' || w == 'S') {//send
		for (int i = 0; i < PkgFromMap[usr].size(); ++i) {
			int seq = PkgFromMap[usr][i];
			Json::Value j;

			j["from"] = packages[seq].from;
			j["fAddr"] = packages[seq].fAddress;
			j["to"] = packages[seq].to;
			j["tAddr"] = packages[seq].tAddress;

			j["sendTime"] = packages[seq].sendTime;
			j["rcvTime"] = packages[seq].rcvTime;

			j["type"] = packages[seq].type;
			j["description"] = packages[seq].description;
			j["UUID"] = packages[seq].UUID;
			j["status"] = packages[seq].status;

			res.append(j);
		}
	}
	if (w == 'r' || w == 'R') {//收件
		for (int i = 0; i < PkgToMap[usr].size(); ++i) {
			int seq = PkgToMap[usr][i];
			Json::Value j;

			j["from"] = packages[seq].from;
			j["fAddr"] = packages[seq].fAddress;
			j["to"] = packages[seq].to;
			j["tAddr"] = packages[seq].tAddress;

			j["sendTime"] = packages[seq].sendTime;
			j["rcvTime"] = packages[seq].rcvTime;

			j["type"] = packages[seq].type;
			j["description"] = packages[seq].description;
			j["UUID"] = packages[seq].UUID;
			j["status"] = packages[seq].status;

			res.append(j);
		}
	}
	if (w == 'u' || w == 'U') {//uuid
		int seq = PkgMap[usr];
		Json::Value j;

		j["from"] = packages[seq].from;
		j["fAddr"] = packages[seq].fAddress;
		j["to"] = packages[seq].to;
		j["tAddr"] = packages[seq].tAddress;

		j["sendTime"] = packages[seq].sendTime;
		j["rcvTime"] = packages[seq].rcvTime;

		j["type"] = packages[seq].type;
		j["description"] = packages[seq].description;
		j["UUID"] = packages[seq].UUID;
		j["status"] = packages[seq].status;

		res.append(j);
	}
	return res;
}

int DataManager::receiptPkg(string usr, string pkgUid) {
	int seq = PkgMap[pkgUid];
	if (seq < 0) { return -404; }
	if (strcmp(packages[seq].to, usr.c_str()) != 0) {
		return -400;
	}
	int res = packages[seq].receiption();
	PkgFile.seekp(seq * sizeof(Package), ios::beg);
	PkgFile.write((char*)&packages[seq], sizeof(Package));
	return res;
}