#include <iostream>
#include <string>

#include <algorithm>
#include <stdexcept>

#include <cstdlib>

#include "nivector.hpp"

//using Vector = std::vector<std::string>;
using Vector = NIVector;

int main(int argc, char **argv)
{
  if (argc < 3)
    {
      std::cerr << "USAGE: " << argv[0] << " num-to-add-before-getting-refernce  num-to-add-after-getting-referece" << std::endl;
      return 1;
    }
  
  Vector v;

  int x = std::atoi(argv[1]);
  int y = std::atoi(argv[2]);

  for (int i = 0; i < x; i++)
    {
      v.push_back("X");
    }

  std::string& first = v.at(0);

  for (int i = 0; i < y; i++)
    {
      v.push_back("Y");
    }

  // this will fail for std::vector for large enough values of y because
  // the reference is invalidated once the array inside the vector is resized;
  // it should work for NIVector
  first = "Z";

  // std::for_each(v.begin(), v.end(), [](const std::string& s) { std::cout << s; });
  std::cout << std::endl;

  try
    {
      std::string s = *(v.end());
      
      std::cout << "dereferencing end iterator should fail; instead returned"
		<< std::endl
		<< s
		<< std::endl;
    }
  catch (const std::out_of_range& ex)
    {
      std::cout << "can't dereference end iterator, as expected" << std::endl;
    }
}