#include <iostream>
#include <fstream>

#include "apstring.h"

apstring parseOut(ifstream &, apstring);

int main() {
	ifstream stdFile, headFile;
	ofstream output;
	char ch;
	apstring str1 = "", head = "", mid = "", tail = "", str2 = "", column = "", rootName, rootPage, rootPage2, temp1, temp2, head_torpg = "", mid_torpg = "";
	int num;
	bool invisFlag;

	headFile.open("head.dat");
	while (headFile.get(ch))
		head = head + ch;
	headFile.close();

        headFile.open("head_torpg.dat");
        while (headFile.get(ch))
                head_torpg = head_torpg + ch;
        headFile.close();

	headFile.open("mid.dat");
	while (headFile.get(ch))
		mid = mid + ch;
	headFile.close();

	headFile.open("mid_torpg.dat");
	while (headFile.get(ch))
                mid_torpg = mid_torpg + ch;
        headFile.close();
	
	headFile.open("pages.dat");
	output.open("pages2.dat");
	while (headFile.get(ch))
		output << ch;
	output.close();
	headFile.close();

	headFile.open("tail.dat");
	while (headFile.get(ch))
		tail = tail + ch;
	headFile.close();


	stdFile.open("pages.dat");
	while (stdFile >> str1) {

		column = "";
		rootPage2 = "";
		if (str1 == "<ROOT>") {
			invisFlag = true;
			stdFile >> str1;
			rootPage = str1;
			headFile.open("pages2.dat");
			invisFlag = false;
			while (headFile >> temp1) {
				if (temp1 == "<ROOT>" || temp1 == "<INVIS>") {
					if (temp1 == "<ROOT>")
						invisFlag = false;
					else if (temp1 == "<INVIS>")
						invisFlag = true;
					headFile >> rootPage2;
					getline(headFile, temp1);
					if (!invisFlag)
						column = column + "<img src = \"bullet.gif\"><a href = \"" + rootPage2 + ".html\">" + temp1 + "</a><br>\n";
				}
				else if (rootPage2 == rootPage && !invisFlag) {
					getline(headFile, temp2);
					column = column + "<img src = \"arrow.gif\"><a href = \"";
					column = column + rootPage2 + '_' + temp1 + ".html\">" + temp2 + "</a><br>\n";
				}
			}
			headFile.close();

			str2 = str1 + ".html";
			output.open(str2.c_str());
			getline(stdFile, str2);
			str2 = str1 + ".dat";
			headFile.open(str2.c_str());
                        if (rootPage == "torpg")
                                output << head_torpg << column << mid_torpg;
                        else
                                output << head << column << mid;
                        output << parseOut(headFile, rootPage) << tail;
			headFile.close();
			output.close();
		}
		else {
			headFile.open("pages2.dat");
			invisFlag = false;
			while (headFile >> temp1) {
				if (temp1 == "<ROOT>" || temp1 == "<INVIS>") {
					if (temp1 == "<ROOT>")
						invisFlag = false;
					else if (temp1 == "<INVIS>")
						invisFlag = true;
					headFile >> rootPage2;
					getline(headFile, temp1);
					if (!invisFlag)
						column = column + "<img src = \"bullet.gif\"><a href = \"" + rootPage2 + ".html\">" + temp1 + "</a><br>\n";
				}
				else if (rootPage2 == rootPage && !invisFlag) {
					getline(headFile, temp2);
					column = column + "<img src = \"arrow.gif\"><a href = \"";
					column = column + rootPage2 + '_' + temp1 + ".html\">" + temp2 + "</a><br>\n";
				}

			}
			headFile.close();
                        str2 = rootPage + '_' + str1 + ".html";
                        output.open(str2.c_str());
                        getline(stdFile, str2);
                        str2 = rootPage + '_' + str1 + ".dat";
                        headFile.open(str2.c_str());
                        if (rootPage == "torpg")
                                output << head_torpg << column << mid_torpg;
                        else
                                output << head << column << mid;
                        output << parseOut(headFile, rootPage) << tail;
			headFile.close();
                        output.close();
		}
	}
	stdFile.close();

	cout << "Website build completed.\n";

	return 0;
}

apstring parseOut(ifstream & inFile, apstring pageType) {
	apstring toReturn, temp, temp2;
	int count = 4, entries = 0;
	char ch;
	bool torpg = false;
	if (pageType == "torpg")
		torpg = true;

	while (getline(inFile, temp) && count <= 4) {
		if (temp == "<ENTRY>") {
			count = 4;
			getline(inFile, temp2);
			getline(inFile, temp);
			toReturn += "<table border=0 cellspacing=0 cellpadding=0 bgcolor=\"#FFFFFF\">\n";
			toReturn += "<tr height = 20><td width = 20>\n";
			if (torpg) {
				toReturn += "<img src = \"bar_top_left_torpg.gif\" valign = \"bottom\" halign = \"right\">";
				toReturn += "</td><td background = \"bar_mid_torpg.gif\">";
				toReturn = toReturn + "<div align = \"center\"><font color = \"#FFFF00\">" + temp;
				toReturn = toReturn + "</font></div></td><td><img src = \"bar_right_torpg.gif\" valign = \"bottom\" halign = \"left\"></td></tr>\n";
				toReturn += "<tr><td width = 20 background = \"bar_left_down_torpg.gif\" valign = \"top\">";
				toReturn += "<img src = \"bar_left_down_top_torpg.gif\"> &nbsp </td><td>";
			}
			else {
				toReturn += "<img src = \"bar_top_left.gif\" valign = \"bottom\" halign = \"right\">";			
				toReturn += "</td><td background = \"bar_mid.gif\">";
				toReturn = toReturn + "<div align = \"center\"><font color = \"#FFFFFF\">" + temp;
				toReturn = toReturn + "</font></div></td><td><img src = \"bar_right.gif\" valign = \"bottom\" halign = \"left\"></td></tr>\n";
				toReturn += "<tr><td width = 20 background = \"bar_left_down.gif\" valign = \"top\">";
				toReturn += "<img src = \"bar_left_down_top.gif\"> &nbsp </td><td>";
			}
			getline(inFile, temp);
			toReturn = toReturn + "<b>Contributed by:" + temp2 + "</b><br>";
		}
		else if (temp == "</ENTRY>") {
			count = 0;
			toReturn += "</td><td> &nbsp ";
			if (torpg)
				toReturn += "</td></tr><tr><td><img src = \"bar_left_bottom_torpg.gif\" halign = \"left\"></td></tr></table>\n<br>";
			else
				toReturn += "</td></tr><tr><td><img src = \"bar_left_bottom.gif\" halign = \"left\"></td></tr></table>\n<br>";				
			entries++;
		}
		else if (temp == "<TEXT>") {
			count = 5;
		}
		else if (temp == "<HTML>") {
			count = 6;
		}
		if (entries == 5) {
			toReturn += "<br><br><br><hr><div align = \"center\"><b>ARCHIVED</b></div><br><br><br>\n";
			entries++;
		}
		if (count == 4)
			toReturn = toReturn + temp;
	}

	if (count == 5) {
		while (inFile.get(ch)) {
			if (ch == '\n')
				toReturn += "<br>";
			else
				toReturn += ch;
		}
	}
	else if (count == 6) {
		while (inFile.get(ch)) {
			toReturn += ch;
		}
	}
	toReturn += "</tr></td></table><br><br>";

	return toReturn;
}