//Comments
//Filename
//Purpose

#include <iostream>
#include <fstream>
#include <string>
#include <cctype>

using namespace std;
/////////////////////////////MISC PROGRAM FUNCTIONS//////////////////////////
void instructions ();
void getline (istream& instream, string& str, char sentinel);
/////////////////////////////PLAYLIST FORMAT DEPENDANT FUNCTIONS/////////////
int countEntriesM3U (char file[]);                            //count functions can probably be merged
int countEntriesB4S (char file[]);
void processM3U (ifstream& inputfile, string& title, string& filename, int& gulplength, 
				 string& sfName, int& flag);
void processB4S (ifstream& inputfile, string& title, string& filename, int& gulplength, 
				 string& sfName);
//////////////////////////////PLAYLIST INDEPENDANT FUNCTIONS//////////////////
void parseID3 (ifstream& mp3, int& ID3, string defaultPath, string& songTitle, string& artistName);
void writeEntry (ofstream& outputfile, int gulplength, int ID3, char file[], string songTitle, 
			string artistName, string sfName, string title);
void writeFooter (ofstream& outputfile);
void makeHeader (ofstream& outputfile, int elementcount);


int main (int argc, char* argv[])
{
	if (argc == 1)
	{
	    instructions ();
	}
	else if (argc != 3)
	{
		cout <<endl<<endl<< "Invalid Command Line Argument.  Please note that the proper format is:"
			 <<endl<<"'NRAGen.exe <input_file> <output_file>'"<<endl
			 <<"Example:  'NRAGen.exe c:"<<(char)92<<"Winamp"<<(char)92<<"mylist.m3u c:"<<(char)92<<"coolmusic.nra'"<<endl<<endl;
		string badUser;
		cout << "Press <Enter> to end program.";
		getline(cin, badUser, '\n');
		return 0;
	}
    //////////////////////////////////
  	string defaultPath;             //base directory containing the playlist file
    string title = "";              //non-ID3 based title
	string filename = "";           //MP3 filename without path info
    string sfName;                  //used in case MP3 doesn't conform to standard naming convention
	char empty = 0;                 //used for .get
    char file[500];                 //filenames
	int gulplength = 0;             //necessary part of .nra format
	int elementcount = -1;          //number of songs in current list
	int flag = 0;                   //used to tell various things
	char first = 0;                 //used for .get in ID3 parsing...probably redundant
	string songTitle, artistName;   //used for ID3 info
	string tag = "";                //used to cache ID3 tag...probably unnecessary
	int ID3 = 0;                    //used to tell if ID3 tag is present
	int ID3count = 0;               //used to traverse ID3 tag
	ifstream mp3;                   //self-explanatory
	string devcppsux;               //needed because Dev-CPP doesn't have the correct implementation of the
                                    //.insert function...
	string filetype;                //denotes the filetype of source list
                                    //
    //////////////////////////////////
	
	if (argc == 1)  /////////////////INPUT GATHERING AND CHECKING////////////////////////////////
	{
	    cout << "Enter the full name and path of the input playlist file: ";
        cin.getline(file, 500, '\n');
		if (file[0] == '"')  //drag and drop handling
		{
			while (file[flag] != '\0')
			{
				file[flag] = file[flag + 1];
				flag++;
			}
			file[flag - 2] = '\0';
			flag = 0;
		}

	}
	else 
	{
		strcpy(file, argv[1]);
		cout << "Converting " <<file<<" to .nra format."<<endl<<endl;
	}
	while((file[1] != ':') || (file[2] != 92))
	{
		cout << endl<<"Invalid input file.  Please enter the filename again along with the FULL PATH."
			 <<endl<<"e.g. 'c:"<<(char)92<<"winamp"<<(char)92<<"music"<<(char)92<<"mylist.m3u'"<<endl<<endl;
		cout << "Enter the full name and path of the input playlist file: ";
		cin.getline(file, 500, '\n');
		if (file[0] == '"')  //drag and drop handling
		{
		    while (file[flag] != '\0')
			{
			    file[flag] = file[flag + 1];
			    flag++;
			}
		    file[flag - 2] = '\0';
			flag = 0;
		}
	}
    ifstream inputfile2(file);
	while (!inputfile2.is_open())
	{
		cout <<endl<< "Inputfile Does Not Exist!!!  Please enter the filename again."<<endl<<endl;
		cout << "Enter the full name and path of the input playlist file: ";
		cin.getline(file, 500, '\n');
		if (file[0] == '"')  //drag and drop handling
		{
		    while (file[flag] != '\0')
			{
			    file[flag] = file[flag + 1];
			    flag++;
			}
		    file[flag - 2] = '\0';
			flag = 0;
		}
		while((file[1] != ':') || (file[2] != 92))
		{ 
		    cout << endl<<"Invalid input file.  Please enter the filename again along with the FULL PATH."
			     <<endl<<"e.g. 'c:"<<(char)92<<"winamp"<<(char)92<<"music"<<(char)92<<"mylist.m3u'"<<endl<<endl;
		    cout << "Enter the full name and path of the input playlist file: ";
		    cin.getline(file, 500, '\n');
			if (file[0] == '"')  //drag and drop handling
			{
			    while (file[flag] != '\0')
				{
				    file[flag] = file[flag + 1];
				    flag++;
				}
			    file[flag - 2] = '\0';
				flag = 0;
			}
		}
		inputfile2.close();
	//	cout <<endl <<"Opening: "<< file<<endl;
		inputfile2.open(file);
	}
	inputfile2.clear();
	inputfile2.seekg(0);
	inputfile2.close();
	
	/////////////////END INPUT GATHERING AND CHECKING////////////////////////////////
	
	while (file[flag] != '\0')  //check extension on input
		flag++;
	if (((file[flag - 3] == 'm') || (file[flag - 3] == 'M')) && (file[flag - 2] == '3') && 
		((file[flag -1] == 'u') || (file[flag - 1] == 'U')))
		filetype = "M3U";
	else if (((file[flag - 3] == 'b') || (file[flag - 3] == 'B')) && (file[flag - 2] == '4') && 
		((file[flag -1] == 's') || (file[flag - 1] == 'S')))
		filetype = "B4S";
	else
		filetype = "UNRECOGNIZED";
	flag = 0;

	if (filetype == "UNRECOGNIZED")            //Process options for unrecognized file types
	{
		cout << "Specified input file has an unknown extension...continue anyways (Y/N)? ";
		getline (cin, title, '\n');
		while ((title != "n") && (title != "N") && (title != "y") && (title != "Y"))
		{
			cout << "invalid choice, please re-enter: ";
			getline (cin, title, '\n');
		}
		if ((title == "n") || (title == "N"))
		{
			cout << "program exiting..."<<endl<<endl;
			return 0;
		}
		else
		{
			cout << endl << "You have the following options:"<<endl<<endl;
			cout <<"Press '1' to process the file as a .m3u file."<<endl
				 <<"Press '2' to process the file as a .b4s file."<<endl
				 <<"Press '3' to exit the program without processing the file"<<endl<<endl;
		    cout << "Enter your choice now: ";
			getline (cin, title, '\n');
			while ((title != "1") && (title != "2") && (title != "3"))
			{
				cout << "invalid choice, please re-enter: ";
			    getline (cin, title, '\n');
			}
			if (title == "1")
				filetype = "M3U";
			else if (title == "2")
				filetype = "B4S";
			else
			{
				cout << "program exiting..."<<endl<<endl;
			    return 0;
			}
			title = "";
		}
	}
	ifstream inputfile(file);
	
	if (filetype == "M3U")
	{
		elementcount = countEntriesM3U (file);
		
	    //get base directory because of .m3u's relative file linking
		while(file[ID3count] != '\0')
		{
			if (file[ID3count] == 92)
				ID3 = ID3count;
			ID3count++;
		}
		file[ID3] = '\0';
		defaultPath = file;

		ID3 = 0;
		ID3count = 0;
	}
	else if (filetype == "B4S")
	{
		elementcount = countEntriesB4S (file);  //only preprocessing necessary on a B4S list
	}
    if (argc == 1)  //get outputfile location
	{
        cout << endl<<endl;
		cout << "Enter the name of the desired output file with extension included"<<endl
             << "(include full path to output to a different folder): ";
        cin.getline(file, 500, '\n');
		cout << endl<<endl;
	}
	else
	{
		strcpy(file, argv[2]);
		cout <<"Writing output to: "<< file<<endl<<endl;
	}
	ofstream outputfile(file);

	//////////////////////////////////// NRA Header Info
	makeHeader (outputfile, elementcount);
	///////////////////////////////////////End Header Info

    cout << "Processing.....";

	if (filetype == "M3U") //.m3u formatted playlist handling code
	{
		int count = 0;
		inputfile.clear();
		inputfile.seekg(0);
		while (!inputfile.eof())
		{
			processM3U (inputfile, title, filename, gulplength, sfName, flag);
			//cout << "ProcessM3U returned..."<<endl;                               **DEBUG**
			//cout << endl << "Title: "<<title<<endl<<"Filename: "<<filename<<endl; **DEBUG**
			
			if (flag == 0)  //get the full patch to the MP3 file
			{
			    for(int count6 = 0; count6 < defaultPath.length(); count6++)
				{
					file[count6] = defaultPath.at(count6);
					ID3count = count6;
				}
				ID3count++;
				for (int count7 = 0; count7 < filename.length(); count7++)
				{
					file[ID3count] = filename.at(count7);
					ID3count++;
					file[ID3count] = '\0';
				}
			}
			else if (filename[1] == ':')  //ditto
			{
				int count99 = 0;
				while (count99 < filename.length())
				{
					file[count99] = filename.at(count99);
					count99++;
					file[count99] = '\0';
				}
			}
			else  //ditto (the relative links in a m3u playlist make the 3 differnt methods necessary)
			{
				int count100 = 0;
				devcppsux = (defaultPath + (char)92 + filename);
				while(count100 < devcppsux.length())
				{
					file[count100] = devcppsux.at(count100);
					count100++;
				    file[count100] = '\0';
				}
			}

			ID3count = 0;
			ID3 = 0;
			mp3.open(file, ios::binary);

			parseID3 (mp3, ID3, defaultPath, songTitle, artistName);

			writeEntry (outputfile, gulplength, ID3, file, songTitle, artistName, sfName, title);

			title = "";
			filename = "";
			count++;       //not really used for anythnig...
			gulplength = 0;
			flag = 0;
			songTitle = "";
			artistName = "";

		}
		writeFooter(outputfile);

	}
	else if (filetype == "B4S")
	{
	//	cout << cout << "B4S File..."<<endl<<endl;
	//	string debug;
		inputfile.clear();
		inputfile.seekg(0);
		while (!inputfile.eof())
		{
			processB4S (inputfile, title, filename, gulplength, sfName);
	//		cout << "processB4S Returned"<<endl;
	//		getline (cin, debug, '\n');
	
			ID3count = 0;
			ID3 = 0;
			
			for (int count = 0; count < filename.length(); count++)
			{
				file[count] = filename.at(count);
				file[count + 1] = '\0';
			}

			mp3.open(file, ios::binary);
			
			parseID3 (mp3, ID3, "", songTitle, artistName);
			writeEntry (outputfile, gulplength, ID3, file, songTitle, artistName, sfName, title);

			title = "";
			filename = "";
			gulplength = 0;
			flag = 0;
			songTitle = "";
			artistName = "";
		}
		writeFooter(outputfile);
	}

	inputfile.close();
	outputfile.close();
		
	cout << "Conversion completed successfully." <<endl<<endl;
	cout << "Press <Enter> to end program."<<endl<<endl;
	cin.getline(file, 500, '\n');

	return 0;
}

void instructions()
{
    char any[2];
    cout<<"Welcome to nragen Public Version 1.1 Beta"<<endl<<endl;

	cout<<"This program will convert a Winamp format playlist into"<<endl
		<<"a useable Nero Burning ROM Audio Compilation (.nra) file so that you can "<<endl
		<<"burn all your favorite playlists without having to hunt around for the"<<endl
        <<"individual files.  When entering filenames, ALWAYS include the full extension"<<endl
        <<"of the file you are specifying (normally .m3u for playlist files and .nra for"<<endl
        <<"Nero Audio Compilation files, although if you're running this app then you"<<endl
        <<"probably already know this).  Also, you need to include the FULL PATH of the "<<endl
        <<"input file even if it is in the same directory as this program, no"<<endl<<"exceptions.  "
        <<"When specifying the output file path information is not required "<<endl<<"unless"
        <<" you want to output to a specific folder.  This program may have some"<<endl 
		<<"issues and I strongly recommend"
		<<" that you read the documentation before using it"<<endl<<endl
        <<"This software is provided as is and I assume no liability for anything"<<endl
        <<"that this program may or may not do.  If you have a problem with this"<<endl
        <<"then you must CLOSE this program NOW."
		<<endl<<endl
        <<"                    'nragen' software coded by Some1"<<endl 
		<<"                   email:  Definitive_MP3@hotmail.com"<<endl<<endl;
    cout<<"Press <Enter> to begin.";
    cin.getline(any, 100, '\n');
    cout << endl<<endl;
}
void getline (istream& instream, string& str, char sentinel)
{
	char ch;
	int charcount = 0;
	str = "";
	do
	{
		instream.get(ch);
		if (ch != sentinel)
		{
			str += ch;
			charcount++;
		}

	}while (ch != sentinel);
	if (charcount == 0)
		str = "";
}
int countEntriesM3U (char file[]) 
{
	char empty = 0;
	int elementcount = 0;
	ifstream inputfile2(file);
	while (!(inputfile2.eof()))
	{
		inputfile2.get(empty);
		if (empty == ',')
			elementcount++;
	}
	inputfile2.clear();
	inputfile2.seekg(0);
	inputfile2.close();
	return elementcount;
}
int countEntriesB4S (char file[])
{
	char empty = 0;
	int elementcount = 0;
	ifstream inputfile2(file);
	while (!inputfile2.eof())
	{
		inputfile2.get(empty);
		if (empty == '=')
		{
		    inputfile2.get(empty);
			inputfile2.get(empty);
			if (empty == 'f')
			{
				inputfile2.get(empty);
				if (empty == 'i')
				{
					inputfile2.get(empty);
					if (empty == 'l')
					{
						inputfile2.get(empty);
						if (empty == 'e')
							elementcount++;
					}
				}
			}
			
		}
	}
	inputfile2.clear();
	inputfile2.seekg(0);
	inputfile2.close();

	return elementcount;
}


void makeHeader (ofstream& outputfile, int elementcount)
{
	outputfile.put(13);
	outputfile << "NeroCDAV5" << (char)46 << 0 << (char)46 << 0 << (char)(20 + (4 * elementcount));
	for (int count4 = 0; count4 < 19; count4++)
		outputfile.put(0);
	outputfile.put(1);
	for (int count2 = 0; count2 < 7; count2++)
		outputfile.put(0);
	outputfile.put(9);
	for (int count3 = 0; count3 < 11; count3++)
		outputfile.put(0);
	outputfile << (char)elementcount;
	for (int count5 = 0; count5 < elementcount; count5++)
	{
		outputfile <<(char)0<<(char)0<<(char)0<<(char)1;
	}
	outputfile<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)elementcount<<(char)0<<(char)0<<(char)0;
}
void processM3U (ifstream& inputfile, string& title, string& filename, int& gulplength,
				 string& sfName, int& flag)
{
	char empty = 0;
	while((!(inputfile.eof())) && (empty != ',')) //go to first/next entry
	{
	    inputfile.get(empty);
	}
		
	if (inputfile.eof())
		flag = 1;
	else
	{
		inputfile.get(empty);
		while (empty != 10)
		{
			title += empty;
			//cout << "Current letter: "<<empty<<endl<<"Current title: "<<title<<endl;    **DEBUG**
			inputfile.get(empty);
		}

		//cout << "Title is: "<<title<<endl;     **DEBUG**
		gulplength += (title.length());
		inputfile.get(empty);
		
		if (empty == 92)  //relative link type 1 found (i.e. /MP3s/Albums/whatever.mp3)
		{
               while (empty != 10)
			{
				filename += empty;
				inputfile.get(empty);
			}
			sfName = filename;
			
			for (int  count9 = 0; count9 < sfName.length(); count9++)
			{
				if (((char)sfName.at(count9)) == 92)
				{
					sfName.erase(0, count9 + 1);
					count9 = -1;
				}
			}
			//cout << "sfName is: " << sfName<<endl;       **DEBUG**
			gulplength += sfName.length();
		}
		else   //either an absolute link found (c:\mp3s\albums\whatever.mp3), or a type 2 relative link
		{      //(mp3s\albums\whatever.mp3)
			
			flag = 1;
			filename += empty;
			inputfile.get(empty);
			while (empty != 10)
			{
				filename += empty;
				inputfile.get(empty);
			}
			sfName = filename;
			
			for (int  count10 = 0; count10 < sfName.length(); count10++)
			{
				if (((char)sfName.at(count10)) == 92)
				{
					sfName.erase(0, count10 + 1);
					count10 = -1;
				}
			}
			gulplength += sfName.length();
		}
	}	
}
void processB4S (ifstream& inputfile, string& title, string& filename, int& gulplength, 
				 string& sfName)
{
	char empty = 0;
	int found = 0;

	while (found == 0)
	{
		while (empty != '=')
		{
			inputfile.get(empty);
		}
		inputfile.get(empty);
		inputfile.get(empty);
		if (empty == 'f')
		{
			inputfile.get(empty);
			if (empty == 'i')
			{
				inputfile.get(empty);
				if (empty == 'l')
				{
					inputfile.get(empty);
					if (empty == 'e')
					{
						inputfile.get(empty);
						if (empty == ':')
						{
							getline (inputfile, filename, '>');
							found = 1;
						}
					}
				}
			}
		}
	}
    filename.at(filename.length() - 1) = '\0';
	sfName = filename;

	for (int  count9 = 0; count9 < sfName.length(); count9++)
	{
		if (((char)sfName.at(count9)) == 92)
		{
			sfName.erase(0, count9 + 1);
			count9 = -1;
		}
	}
	sfName.erase(sfName.length() -1, 1);
	inputfile.get(empty);
	inputfile.get(empty);
	inputfile.get(empty);
	
	while (empty != '>')
		inputfile.get(empty);
	getline (inputfile, title, '<');
	inputfile.get(empty);
	for (int count = 0; count < 4; count++)
	{
		while (empty != '<')
		{
			inputfile.get(empty);
		}
		inputfile.get(empty);
	}
	if (empty == '/')
	{
		inputfile.seekg(0, ios::end);
		inputfile.get(empty);
	}
	gulplength += (title.length());
	gulplength += sfName.length();

}

void parseID3 (ifstream& mp3, int& ID3, string defaultPath, string& songTitle, string& artistName)
{
    //defaultPath is an unnecessary parameter
	
    mp3.seekg(0, ios::end);
	int mp3Length = mp3.tellg();
	mp3.seekg(0);
	mp3.seekg((mp3Length - 128));
	char first, temp;
		
	mp3.get(first);
	if (first == 'T')
	{
		mp3.get(first);
					
		//	cout << "T found, next character is:  "<<first<<endl;   **DEBUG**
		if (first == 'A')
		{
			mp3.get(first);
			//		cout << "A found, next character is:  "<<first<<endl;  **DEBUG**
			if (first == 'G')
			{
			     ID3 = 1;
				 mp3.get(first);
                 while ((first != ' ') && (first != 0))
                 {
                     while ((first != ' ') && (first != 0))
                     {
                         songTitle += first;
                         mp3.get(first);
                     }
                     mp3.get(first);
                     if ((first != ' ') && (first != 0))
                         songTitle += ' ';
                 }
                 mp3.get(first);
                 temp = first;
							
				 while ((first == temp) && (!mp3.eof()))
				     mp3.get(first);
							 
				 if ((first != ' ') && (first != 0))
					 artistName = "NULL";
				 else
				 {
                     while ((first != ' ') && (first != 0))
					 {
                         while ((first != ' ') && (first != 0))
						 {
                              artistName += first;
                              mp3.get(first);
						 }
                         mp3.get(first);
                         if ((first != ' ') && (first != 0))
                             artistName += ' ';
					 }
				}
							 
    			if (artistName.length() == 10)
					artistName += " ";
				if (songTitle.length() == 10)
					songTitle += " ";
			}
		}
				
	}
				
	mp3.seekg(0);
	mp3.clear();
	mp3.close();
}

void writeEntry (ofstream& outputfile, int gulplength, int ID3, char file[], string songTitle, 
			string artistName, string sfName, string title)
{
//	cout << "Writing Entry...."<<endl<<endl<<"ID3 is: "<<ID3<<endl<<"File is: " << file<<endl<<"Songtitle is: "<<songTitle<<endl<<"artistName is: "
//		<<artistName<<endl<<"sfName is: "<<sfName<<endl<<"title is: "<<title<<endl;
	int fnameLength = 0;
	while (file[fnameLength] != '\0')
		fnameLength++;
	if (ID3 == 0)
	{
		gulplength += fnameLength;
		gulplength += 81;
	}
	else
	{
		gulplength -= (title.length());
		gulplength += fnameLength;
		gulplength += (artistName.length());
		gulplength += (songTitle.length());
		gulplength += 81;
	}
	outputfile << "GULP";
	outputfile << (char)gulplength<<(char)(gulplength / 255)<<(char)0<<(char)0;
	outputfile.put(fnameLength);
	outputfile <<(char)0<<(char)0<<(char)0;
	outputfile << file;
	outputfile<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0;
	outputfile.put(sfName.length());
	outputfile <<(char)0<<(char)0<<(char)0<< sfName <<(char)0;
	
	if ( ID3 == 0 )
	{
		outputfile<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0;
	    outputfile.put(title.length());
	    outputfile <<(char)0<<(char)0<<(char)0<<title;
	}
	else
	{
		if (artistName.length() != 10)
		    outputfile.put(artistName.length());
		else
		{
			artistName += " ";
			outputfile.put(artistName.length());
		}
		
		outputfile <<(char)0<<(char)0<<(char)0<<artistName<<(char)0;
		if (songTitle.length() != 10)
			outputfile.put(songTitle.length());
		else 
		{
			songTitle += " ";
			outputfile.put(songTitle.length());
		}
              outputfile <<(char)0<<(char)0<<(char)0<<songTitle;
	}
	outputfile     <<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0
		           <<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0
		           <<(char)0<<(char)0<<(char)41<<(char)108<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)150
			       <<(char)0<<(char)0<<(char)0<<(char)190<<(char)108<<(char)0<<(char)0<<(char)150
			       <<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0<<"ENON"
			       <<(char)8<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0;


}
void writeFooter (ofstream& outputfile)
{
	outputfile << "BUSTX"<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0
	           <<(char)1<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0
		       <<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0
			   <<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0
			   <<(char)112<<(char)23<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0
			   <<(char)1<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)1<<(char)0<<(char)0<<(char)0<<(char)0
			   <<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0<<(char)0;
}