#include <iostream>
#include <string>
#include "ArrayStack.h"

using namespace std;

int getSelection();

int main() {

	ArrayStack<string> *visited = new ArrayStack<string>;
	char answer;
	int selection;
	string curlink;

	do {
	
		selection = getSelection();

		if(selection == 1) {
			visited->push(curlink);			
			cout << "Enter link: ";
			cin >> curlink;
			cout << "Current link is " << curlink << endl;
		} else if(selection == 2) {
			curlink = visited->pop();						
			cout << "Current link is " << curlink << endl;
		} else {
			cout << "Bad Input\n";
			continue;
		}
		cout << "Continue? ";
		cin >> answer;

	} while(answer == 'y');

	return EXIT_SUCCESS;
}

int getSelection() {
	int selection;
	cout << "Enter Choice:\n";
	cout << "\t(1) Follow link\n";
	cout << "\t(2) Go back\n";
	cin >> selection;
	return selection;
}