//Stub.cpp
//Driver for CD database
//CS 2  

#include <fstream.h>
#include "apstring.h"
#include "apvector.h"
#include <stdlib.h>


struct cdType
{
	apstring artist;
	apstring album;
};
		
void menu()
{
	cout<<"A)dd"<<endl;
	cout<<"E)dit"<<endl;
	cout<<"P)rint"<<endl;
	cout<<"S)ave"<<endl;
	cout<<"Q)uit"<<endl << endl; 
}

void print (apvector <cdType> &myCollection)
{
	for (int x=0; x<=myCollection.length()-1; x++)
		cout << x+1 << ") " << myCollection[x].artist << "\t\t-\t" << myCollection[x].album << endl << endl << endl;
}


void add(apvector <cdType> &myCDs)
{
	cdType temp;
	int len = myCDs.length();
	myCDs.resize(len+1);

	cout << "Name of artist ... ";
	getline(cin, temp.artist);
	cout << "Name of album ... ";
	getline(cin, temp.album);

	bool done = false;
	int i = len - 1;
	while (!done && i >= 0)
	{
		if (temp.artist < myCDs[i].artist)
		{
			myCDs[i+1] = myCDs[i];
			i--;
		}
		else 
		{
			myCDs[i+1] = temp;
			done = true;
		}
	}
	if (!done)
		myCDs[0] = temp;
}


void edit (apvector <cdType> &myCollection)
{
	apstring userSearch;
	cout << "What is the name of the album that you would like to edit?" << endl;
	cin >> userSearch;
	bool found;
	for (int x=0; x<myCollection.length(); x++)
	{
		if (myCollection[x].album == userSearch)
		{	
			cout << "Album found.  Album number " << x+1 << "." << endl;
			found = true;
			system ("pause");
			system ("cls");
			cout << "Edit: " << myCollection[x].artist << "\t\t-\t" << myCollection[x].album << endl << endl;
			for (int y=x; y<myCollection.length()-1; y++)
				myCollection[y]=myCollection[y+1];
			add (myCollection);
		}
	}
	if (found != true)
		cout << "Album not found.  Search again or view all albums for correct spelling and" << endl << "capitalization." << endl;
}

void writeFile (apvector <cdType> &myCollection)
{
	ofstream fout;
	apstring line;
	fout.open("C:\\cdCollection\database\txt");
	getline (cin, line);
	fout << line;
	fout << endl;
	fout.close;
}

void readFile (apvector <cdType> &myCollection)
{
	ifstream fin;
	int num;
	apstring garbage;
	fin.open("C:\\cdCollection\database.txt", ios::nocreate);
	if (!fin)
	{
		cout << "File not found.  Please recreate an empty .txt file under the directory" << endl << "C:\\cdCollection\\database.txt" << endl;
		system ("pause");
		system ("cls");
	}
	fin >> num;
	getline(fin, garbage);	
	myCollection.resize(num);
	for (int i=0; i<num; i++)
	{
		getline(fin, myCollection[i].artist);
		getline(fin, myCollection[i].album);
	}
}

int main() 
{
	apvector <cdType> myCollection;
	readFile (myCollection);
	char ch;
	apstring response;
	apstring legal="AaEePpQq";
	do
	{
		do
		{	
			menu();
			getline(cin,response);
			ch = response[0];

		}while(response.find(ch)==-1);

		switch(ch)
		{
			case 'a': 
			case 'A': add (myCollection);break;
			case 'E':
			case 'e': edit (myCollection);break;
			case 'P':
			case 'p': print (myCollection);break;
			case 's':
			case 'S': writeFile (myCollection);break;
			case 'q':
			case 'Q':	break;
		}
		system("pause");
		system("cls");
		
	}while(ch!='q' && ch!= 'Q');
	 
	cout<<endl;
	cout<<"Thank you and have a nice day"<<endl;
	 
	return 0;
}