#include <iostream> // basic i/o
#include <iomanip>
#include <cstdlib> // rand() and srand()
#include <ctime> // system time
#include <string>
#include <fstream>
#define MAX_HOUSE_SIZE 8
using namespace std;

struct Student{
  string lastname = "";
  string firstname = "";
  string house = "";
  string id = "";
};

struct House{
  string name = "";
  unsigned number = 0;
  Student members[MAX_HOUSE_SIZE];
};

House addStudent(Student s, House h){
  if(h.number < MAX_HOUSE_SIZE){
    h.members[h.number] = s;
    h.number++;
  }else{
    cout << "ERROR: House is full.\n";
  }
  return h;
}

int main() {
  Student students[40];
  House houses[4];
  string houseNames[] = {
    "Gryffindor",
    "Hufflepuff",
    "Ravenclaw",
    "Slytherin"
  };

  cout << "Houses:\n";
  for(int i=0; i < 4; i++){
    houses[i].name = houseNames[i];
    cout << houses[i].name << ": " << houses[i].number << endl;
  }

  string temp;
  ifstream infile;
  ofstream outfile;
  string filename = "students.csv";
  int num = 0;
  int houseNum;
  bool validHouse;
  string tempId = "";
  int padding = 0;
  int studentNum = 1;

  // seed the PRNG
  unsigned seed;
  seed = time(0);
  srand(seed);

  // Get the filename from the user
  // cout << "Please enter the filename: ";
  // cin >> filename;

  // Open the file the user specified
  infile.open(filename);

  // If the file successfully opened, process it.
  if(infile){
    // Read the members from the file and display them.
    while (num < 30){
      // getline(string, token, delimiter);
      getline(infile, temp, ',');
      students[num].lastname = temp;
      getline(infile, temp);
      students[num].firstname = temp;
      // cout << num+1 << "). " << students[num].firstname << " " << students[num].lastname << ": ";

      // add the student ID
      tempId = to_string(studentNum);
      padding = 8-tempId.size();
      // assign to the student
      students[num].id = "H" + tempId.insert(0, padding, '0');
      // advance to the next student number
      studentNum++;

      cout << num+1 << "). " << students[num].firstname << " " << students[num].lastname << ", id: " << students[num].id << ": ";

      do{
        validHouse = false;
        houseNum = rand() % 4;
        // only add the student if there's room
        if(houses[houseNum].number < MAX_HOUSE_SIZE){
          students[num].house = houseNames[houseNum];
          validHouse = true;
          houses[houseNum] = addStudent(students[num], houses[houseNum]);
        }
        // if the house is full, choose another!
      }while(not validHouse);
      cout << students[num].house << endl;
      num++;
    }
    // Close the file.
    infile.close();
  }else{
    // Display an error message.
    cout << "Error opening " << filename << ".\n";
  }
  for(int x = 0; x < 4; x++){
    cout << "\n" << houses[x].name << ": [" << houses[x].number << "]\n";
    filename = houses[x].name + ".csv";
    cout << "Opening " + filename + " for writing.\n";
    outfile.open(filename);
    if(outfile){
      for(int y=0; y < houses[x].number; y++){
        cout << "\n" << setw(2) << right << y+1 << "). " << houses[x].members[y].lastname << ", " << houses[x].members[y].firstname << " | id: " << houses[x].members[y].id;
        outfile << houses[x].members[y].lastname << "," << houses[x].members[y].firstname << "," << houses[x].members[y].id << endl;
      }
    }else{
      cout << "ERROR: unable to open for writing.\n";
    }
    outfile.close();
    cout << endl;
  }
  cout << "\n******************\nThe total is: " << num << endl;
  return 0;
}