#include "passenger.h"
#include "flight.h"

int main() {
  
  Flight* flights[3];
  flights[0] = new Flight("Santa Barbara", 150, 3000, DOMESTIC);
  flights[1] = new Flight("Berlin", 230, 6000, INTL);
  flights[2] = new Flight("Miami", 75, 1000, DOMESTIC);

  Passenger p1("Donald Duck", 12345);

  int answer = 1;
  char again = 'n';
  do {
    cout << "Where would " << p1.getName() << " like to travel?\n";
    for (int i = 0; i < 3; i++) {
      cout << "\t(" << (i+1) << ")" << flights[i]->getDestination() << endl;
    }
    cout << "> ";
    cin >> answer;
    while(answer > 3 || answer < 1) {
      cout << "Invalid answer.  Try again. > ";
      cin >> answer;
    }
    
    int index = answer-1;
    if(p1.canFlyFree(flights[index]->getFlightType())) {
      cout << "\tThis passenger can fly free" << endl;
      flights[index]->purchaseTix(1);
      p1.purchaseFreqFlyerTix(flights[index]->getFlightType());
    } else {
      cout << "\tThis passenger must purchase tickets" << endl;
      flights[index]->purchaseTix(1);
      p1.addMiles(flights[index]->getMilesToFly());
    }

    cout << "Miles balance: " << p1.getMiles() << endl;

    cout << "Fly again? ";
    cin >> again;
    while(again != 'y' && again != 'n') {
      cout << "Invalid answer.  Try again. > ";
      cin >> again;
    }

  } while (again != 'n');

  for(int i = 0; i < 3; i++) {
    delete flights[i];
  }		      

  return EXIT_SUCCESS;
}