// Unit 5: File I/O
// - Determine hash size based on the number of records in the file
// - Read data from the input file and insert them into the hash table and BST
//
// Written by: Kevin Cremin
// Modified by: Iurii Tatishchev

#include <sstream>
#include "fio.h"

bool isInteger(const string &str);

bool isDouble(const string &str);

using namespace std;

int findHashSize(const string &filename) {
    ifstream inputFile(filename);

    if (!inputFile.good()) {
        // cout << "Error opening the input file: \"" << filename << "\"" << endl;
        return -1;
    }
    // cout << "Reading data from \"" << filename << "\"" << endl;

    int count = 0;
    string line;
    while (getline(inputFile, line)) {
        if (line.size() != 0) {
            count++;
        }
    }

    count *= 2;
    return findNextPrime(count);
}

void insertFile(const string &filename, BinarySearchTree<string> &bst, HashTable<CPU> &hash) {
    ifstream inputFile(filename);
    // cout << "Reading data from \"" << filename << "\"" << endl;

    if (!inputFile.good()) {
        cout << "Error opening the input file: \"" << filename << "\". Skipping...\n";
        return;
    }

    string line;
    while (getline(inputFile, line)) {
        int releaseYear, coreCount;
        string name, architecture, strToNum;
        double baseClock;

        stringstream temp(line);

        getline(temp, name, ';');

        // check if the CPU is already in the hash table
        CPU key(name, -1, -1, "", -1);
        if (hash.search(key, key, key_to_index) != -1) {
            cout << "Duplicate CPU \"" << name << "\" found in file. Skipping...\n";
            continue;
        }
        temp.ignore();
        getline(temp, strToNum, ';');
        releaseYear = stoi(strToNum);

        temp.ignore();
        getline(temp, strToNum, ';');
        coreCount = stoi(strToNum);

        temp.ignore();
        getline(temp, architecture, ';');

        temp.ignore();
        getline(temp, strToNum, ';');
        baseClock = stod(strToNum);


        // create a CPU object and initialize it with data from file
        CPU aCPU(name, releaseYear, coreCount, architecture, baseClock);
        bst.insert(name);

        hash.insert(aCPU, key_to_index);
    }

    inputFile.close();
    cout << "Data from file \"" << filename << "\" added.\n";
}


void insertCPU(BinarySearchTree<string> &bst, HashTable<CPU> &hash) {
    string tester, name, architecture;
    int releaseYear, coreCount;
    double baseClock;

    cout << "Please enter the name of the CPU: ";
    cin.ignore();
    getline(cin, name);

    CPU key(name, -1, -1, "", -1);
    CPU checker;

    if (hash.search(checker, key, key_to_index) != -1) {
        cout << "Duplicate CPUs are not allowed!\n";
        return;
    }


    cout << "Please enter the year the CPU was released: ";
    cin >> tester;

    while (!(isInteger(tester))) {
        cout << "Please enter an integer: ";
        cin >> tester;
    }

    releaseYear = stoi(tester);


    cout << "Please enter the number of cores in the CPU: ";
    cin >> tester;

    while (!(isInteger(tester))) {
        cout << "Please enter an integer: ";
        cin >> tester;
    }

    coreCount = stoi(tester);


    // Strings don't need to be tested
    cout << "Please enter the architecture of the CPU: ";
    cin.ignore();
    getline(cin, architecture);

    cout << "Please enter the base clock of the CPU: ";
    cin >> tester;

    while (!(isDouble(tester))) {
        cout << "Please enter a decimal: ";
        cin >> tester;
    }

    baseClock = stod(tester);


    CPU aCPU(name, releaseYear, coreCount, architecture, baseClock);
    bst.insert(name);

    hash.insert(aCPU, key_to_index);
}

/*
 * Name: isInteger
 * Written By: Kevin Cremin
 * Modified By:
 * Purpose: Determine if input is an integer
 * Input: String to test
 * Output: Bool of whether input was an integer
 * Procedure: Checks that all characters are digits.
 */
bool isInteger(const string &str) {
    for (int i = 0; i < str.length(); i++) {
        if (!(isdigit(str[i]))) {
            return false;
        }
    }

    return true;
}

/*
 * Name: isDouble
 * Written By: Kevin Cremin
 * Modified By:
 * Purpose: Determine if input is a double
 * Input: String to test
 * Output: Bool of whether input was an double
 * Procedure: Checks that all characters are digits and that there is at most one period.
 */
bool isDouble(const string &str) {
    int chance = 0;

    for (int i = 0; i < str.length(); i++) {
        if (!(isdigit(str[i]))) {
            if (str[i] == '.') {
                if (chance >= 1) {
                    return false;
                } else {
                    chance++;
                }
            } else {
                return false;
            }
        }
    }
    return true;
}