Worker.c

#include "Worker.h"

Person* initWorker(eShopType dep, char* name, double height, int age){
	Worker* pWorker;
	Person* pBase;
	pBase = initPerson(name, height, age);
	if (!pBase) {

		return NULL;
	}
	pWorker = (Worker*)malloc(sizeof(Worker));
	if (!pWorker) {
		printf("Memory allocation failed\n");
		pBase->freePerson(pBase);
		return NULL;
	}
	//init derived object
	pBase->pDerived = pWorker;
	
	if(!isValidInfoWorker(dep)) {
		pBase->freePerson(pBase);
		free(pWorker);
		return NULL;
	}
	pWorker->WorkerId = generateWorkerID();
	pWorker->department = dep;

	//change base class interfacese
	pBase->freePerson = freeWorker;
	pBase->printPerson = printWorker;
	pBase->loadPersonFromBinFile = loadWorkerFromBinFile;
	pBase->loadPersonFromTextFile = loadWorkerFromTextFile;
	pBase->savePersonToBinFile = saveWorkerToBinFile;
	pBase->savePersonToTextFile = saveWorkerToTextFile;
    return pBase;
}

Person* initWorkerByUser() {
	//make person and then make worker
	Person* tempPerson= initPersonByUser();
	Person* worker = NULL;
	int size = (int)strlen(tempPerson->name);
	int flag = 0;
	int dep = -1;
	do {
		if (flag > 0) {
			//massage for the user for invalid input
			printf("Please try again\n");
		}
		printf("Enter department:\n");
		for (int i = 0; i < eNofShopTypes; i++) {
			printf("%d. %s\n", i+1, ShopTypeStr[i]);
		}
		scanf("%d", &dep);
		flag=1;
		//copy person name again after freeing it because init worker failed
		char* new_name = (char*)malloc(size + 1);
		strcpy(new_name, tempPerson->name);
		//init worker
		worker = initWorker(dep-1,new_name, tempPerson->height, tempPerson->age);
	} while (!worker);
	//free the temp person
	tempPerson->freePerson(tempPerson);
	return worker;
}


int generateWorkerID() {
	int id_temp = randomNum(1, 9);
	for (int i = 0; i < WORK_ID; i++) {
		id_temp = SUM(id_temp * 10, randomNum(1, 9));
	}
	return id_temp;
}

int isValidInfoWorker(eShopType department) {
	if (department < 0 || department >= eNofShopTypes) {
		return 0;
	}
	return 1;
}

int compareWorkerById(Worker* w1, Worker* w2) {
	if (w1->WorkerId == w2->WorkerId) {
		return 0;
	}
	return w1->WorkerId - w2->WorkerId;
}

void printWorker(const Person* pWorker) {
	const Worker* w = pWorker->pDerived;

	printf("--------------------\n");
	printPerson(pWorker);
	printf("Worker ID: %d\n", w->WorkerId);
	printf("Department: %s\n", ShopTypeStr[w->department]);
	printf("--------------------\n");
}

void freeWorker(Person* worker) {
	Worker* w ;
	w = worker->pDerived;
	free(w);
	freePerson(worker);
	
}

int saveWorkerToTextFile(const Person* worker, FILE* fp){
	if(worker == NULL){
		return 0;
	}
	IS_FILE_NULL(fp); //TODO MACRO

	//cast to worker
	Worker* tempWorker = worker->pDerived;

	if(isValidInfoWorker(tempWorker->department) == 0){
		return 0;
	}
	//save person
	savePersonToTextFile(worker, fp);
	//save department
	if (writeIntToTextFile(fp, tempWorker->department) == 0) {
		return 0;
	}
	//save worker id
	if(writeIntToTextFile(fp, tempWorker->WorkerId) == 0){
		return 0;
	}
	return 1;
}
int loadWorkerFromTextFile(Person** worker, FILE* fp){
	IS_FILE_NULL(fp); //TODO MACRO

	Person* tempPerson = NULL;

	//load person
	if(loadPersonFromTextFile(&tempPerson, fp) == 0){
		return 0;
	}
	//load department
	int tempDepartment;
	if (readIntFromTextFile(fp, &tempDepartment) == 0) {
		freePerson(tempPerson);
		return 0;
	}
	//copy person name to worker
	char * personName = (char*)malloc(strlen(tempPerson->name) + 1);
	strcpy(personName, tempPerson->name);

	//init worker
	*worker = initWorker(tempDepartment, personName, tempPerson->height, tempPerson->age);
	freePerson(tempPerson);
	if (*worker == NULL) {
		free(personName);
		return 0;
	}

	//load worker id
	int tempId;
	if (readIntFromTextFile(fp, &tempId) == 0) {
		return 0;
	}
	Worker* temp = (*worker)->pDerived;
	temp->WorkerId = tempId;
	return 1;
}

int saveWorkerToBinFile(const Person* worker, FILE* fp){
	if(worker == NULL){
		return 0;
	}
	IS_FILE_NULL(fp); //TODO MACRO

	//cast to worker
	Worker* tempWorker = worker->pDerived;

	if(isValidInfoWorker(tempWorker->department) == 0){
		return 0;
	}
	//save person
	savePersonToBinFile(worker, fp);
	//save department
	if (writeGeneralToBinFile(fp, &(tempWorker->department), sizeof(eShopType)) == 0) {
		return 0;
	}
	//save worker id
	if (writeGeneralToBinFile(fp, &(tempWorker->WorkerId), sizeof(int)) == 0) {
		return 0;
	}
	return 1;


}
int loadWorkerFromBinFile(Person** worker, FILE* fp){
	IS_FILE_NULL(fp); //TODO MACRO

	Person* tempPerson = NULL;

	//load person
	if(loadPersonFromBinFile(&tempPerson, fp) == 0){
		return 0;
	}
	//load department
	eShopType tempDepartment;
	if (readGeneralFromBinFile(fp, &tempDepartment, sizeof(eShopType)) == 0) {
		freePerson(tempPerson);
		return 0;
	}
	//copy person name to worker
	char * personName = (char*)malloc(strlen(tempPerson->name) + 1);
	strcpy(personName, tempPerson->name);

	//init worker
	*worker = initWorker(tempDepartment, personName, tempPerson->height, tempPerson->age);

	freePerson(tempPerson);
	if(*worker == NULL){
		free(personName);
		return 0;
	}
	//load worker id
	int tempId;
	if (readGeneralFromBinFile(fp, &tempId, sizeof(int)) == 0) {
		return 0;
	}
	Worker* temp = (*worker)->pDerived;
	temp->WorkerId = tempId;
	return 1;
}