#include <iostream>
#include <fstream>      // for file reading etc.
#include <vector>
#include <algorithm>    // To randomize/sort the vector
#include <cstdlib>
#include "word.h"       // My class for words.
 
using namespace std;
 
void printMain();
void loadWords(vector<Word> &w, string fName);          // If fName == "" then the app will ask for fileName
void addWords(vector<Word> &w, bool clear);             // if clear == true the vector will get empted first.
void displayWords(vector<Word> &w);
void saveWords(vector<Word> &w);
void quiz(vector<Word> &w);
 
int main(int argc, char* argv[])
{
    vector<Word> words; // Create a vector for my class 'word'
    string input;
    if(argv[0] != NULL) loadWords(words, argv[0]);
    while(true)
    {
        printMain();
        getline(cin, input);
 
        if(input=="1") addWords(words, true);
        else if(input=="2") addWords(words, false);
        else if(input=="3") displayWords(words);
        else if(input=="4") quiz(words);
        else if(input=="5") saveWords(words);
        else if(input=="6") loadWords(words, "");
        else if(input=="7") break;
        system("cls");
    }
    return 0;
}
 
void quiz(vector<Word> &w)
{
    random_shuffle(w.begin(), w.end());
    cout << "I'll display the first word, guess the second one." << endl;
    string str;
    int right = 0;
    for(unsigned int i = 0;i < w.size();i++)
    {
        cout << w.at(i).fWord << " : ";
        getline(cin, str);
        if(str==w.at(i).sWord) right++;
    }
    cout << "Your score " << right << " of " << w.size() << ".";
    cin.get();
}
 
void saveWords(vector<Word> &w)
{
    ofstream myFile;
    string fName;
 
    cout << "Filename(no extension): ";
    getline(cin, fName);
 
    myFile.open((fName+".wdb").c_str());
 
    for(unsigned int i = 0; i < w.size();i++)
    {
        myFile << w.at(i).fWord << endl << w.at(i).sWord << endl;
    }
}
 
void displayWords(vector<Word> &w)
{
    for(unsigned int i = 0; i < w.size(); i++)
    {
        cout << w.at(i).fWord << "\t" << w.at(i).sWord << endl;
    }
}
 
void addWords(vector<Word> &w, bool clear)
{
    if(clear==true) w.clear();
    string word[2];
    cout << "End the input by typing # in first word." << endl;
    while(true)
    {
        cout << "First word: ";
        getline(cin, word[0]);
        if(word[0]=="#") break;
        cout << "Second word: ";
        getline(cin, word[1]);
        w.push_back(Word(word[0], word[1]));
    }
 
}
 
void loadWords(vector<Word> &w, string fName)
{
    if(fName == "")
    {
        cout << "Filename(no extension): ";
        getline(cin, fName);
    }
    ifstream myFile;
    string word[2];
 
    myFile.open((fName+".wdb").c_str());
    w.clear();
 
    while(true)
    {
        getline(myFile, word[0]);
        getline(myFile, word[1]);
        if(myFile.fail()) break;
        w.push_back(Word(word[0], word[1]));
    }
}
 
void printMain()
{
    cout << "[1] new wordlist." << endl << "[2] Add words to current wordlist." << endl << "[3] Display current wordlist." << endl << "[4] Take the quiz. (current wordlist)" << endl;
    cout << "[5] Save wordlist to file." << endl << "[6] Load wordlist. (will erease current wordlist)" << endl << "[7] Quit" << endl;
}

#ifndef WORD_H
#define WORD_H
#include <iostream>
 
 
class Word
{
    public:
        Word(std::string _fWord, std::string _sWord); //The constructor.
        std::string fWord; // firstWord
        std::string sWord; // secondWord
 
    protected:
    private:
};
 
#endif // WORD_H

#include "word.h"
 
Word::Word(std::string _fWord, std::string _sWord)
{
    fWord = _fWord;
    sWord = _sWord;
}