borrow.cpp

// ------------------------------------------------ borrow.cpp ------------------------------------------------------
// Name: Ihar Simanovich & Jason Kozodoy
// Course: CSS343 A
// Creation Date: 12/01/2016
// Last Modification: 12/14/2016
// ------------------------------------------------------------------------------------------------------------------
#include "borrow.h"
#include <sstream>
#include "movie.h"

using namespace std;

//constructor
Borrow::Borrow() {
	type = 'B';
}

//destructor
Borrow::~Borrow() {

}

//returns if borrow transaction successful
bool Borrow::isSuccess(){
	return success;
}

//does a transaction for borrow
void Borrow::doTransaction(BinTree& tree, char movieType) {

	Movie* result = NULL;

	switch (movieType) {
	case 'F': {

		Comedy* target = new Comedy(title, year);
		if (tree.retrieve(*target, result)) {
			if (result->borrow() == false) {
				cout << "OUT OF STOCK: F, " << target->getTitle() << ", "
						<< target->getYear() << endl;
				cout << endl;
				success = false;
			}

		} else {
			cout << "ERROR - Invalid Movie: F, " << target->getTitle() << ", "
					<< target->getYear() << endl;
			cout << endl;
			success = false;
		}
		delete target;
		break;
	}
	case 'D': {
		Drama* target = new Drama(director, title);

		if (tree.retrieve(*target, result)) {
			if (result->borrow() == false) {
				cout << "OUT OF STOCK: D, " << target->getDirector() << ", "
						<< target->getTitle() << endl;
				cout << endl;
				success = false;
			}
		} else {
			cout << "ERROR - Invalid Movie: D, " << target->getDirector()
					<< ", " << target->getTitle() << endl;
			cout << endl;
			success = false;
		}
		delete target;
		break;
	}
	case 'C': {
		Classic* target = new Classic(month, year, actor);
		if (tree.retrieve(*target, result)) {
			if (result->borrow() == false) {
				cout << "OUT OF STOCK: C, " << target->getActor() << ", "
						<< target->getMonth() << " " << target->getYear()
						<< endl;
				cout << endl;
				success = false;
			}
		} else {
			cout << "ERROR - Invalid Movie: C, " << target->getActor() << ", "
					<< target->getMonth() << " " << target->getYear() << endl;
			cout << endl;
			success = false;
		}
		delete target;
		break;
	}
	default:
		cout << movieType << " - Invalid Movie Type, Can't Borrow" << endl;
		cout << endl;
		success = false;
		break;
	}
	//reset data based on performed transaction
	if (result != NULL) {
		title = result->getTitle();
		year = result->getYear();
	}

}

//sets the data for a movie that is borrowed
void Borrow::setData(string data, char movieType) {

	istringstream dataStream(data);		//opening stream
	type = 'B';
	switch (movieType) {
	case 'F': {
		string temp;
		//reading title
		dataStream >> temp;
		for (;;) {
			//no comma case, adding words to title
			if (temp.find(',') == -1) {
				title += temp;
				title += " ";
				dataStream >> temp;
			}

			//comma case, removing comma and adding to title then break
			else {
				temp = temp.substr(0, temp.size() - 1);
				title += temp;
				break;
			}
		}
		//reading year
		dataStream >> year;

		break;
	}
	case 'D': {

		string temp;
		//reading director
		dataStream >> temp;
		for (;;) {
			//no comma case, adding words to title
			if (temp.find(',') == -1) {
				director += temp;
				director += " ";
				dataStream >> temp;
			}

			//comma case, removing comma and adding to title then break
			else {

				temp = temp.substr(0, temp.size() - 1);
				director += temp;
				break;
			}
		}
		//reading title
		dataStream >> temp;
		for (;;) {
			//no comma case, adding words to title
			if (temp.find(',') == -1) {
				title += temp;
				title += " ";
				dataStream >> temp;
			}

			//comma case, removing comma and adding to title then break
			else {
				temp = temp.substr(0, temp.size() - 1);
				title += temp;
				break;
			}
		}
		break;
	}
	case 'C': {
		string first, last;
		dataStream >> month;
		dataStream >> year;
		dataStream >> first;
		dataStream >> last;
		actor = first + " " + last;
		break;
	}
	default:
		cout << "invalid type of movie, can't set data" << endl;
		break;
	}
}