Word.cpp

//
// Created by Matthew Jones on 11/14/16.
//

#include <cstdlib>
#include <ctime>
#include "Word.h"

using json = nlohmann::json;

Word::Word(std::string w) {
	if(w.length() > 1) {
		switch (w.at(0)) {
			case '.':
			case ',':
			case '?':
			case '!':
			case ':':
			case ';':
			case '\n':
				w.erase(w.begin());
				break;
			default:
				break;
		}
	}
	word = w;
	srand(time(NULL));
}



Word* Word::getRandWordFollowing() const{
	int randomIndex = rand() % wordsFollowing.size();
	return wordsFollowing[randomIndex];
}

std::vector<Word*> Word::getWordsFollowing() const{
	return wordsFollowing;
}

void Word::addWordFollowing(Word *wordToAdd) {
	wordsFollowing.push_back(wordToAdd);
 }

std::string Word::toString() const {
//	std::string rval = "{\"Word\": \"" + word + "\"\n\t\"wordsFollowing\": [";
//	for(int i = 0; i < wordsFollowing.size(); i++) {
//		rval.append("\n\t\t\"").append(wordsFollowing[i]->getWord()).append("\",");
//	}
//	rval.pop_back();
//	rval.append("\n\t]\n}");
//	return rval;
	return toJSON().dump();
}

std::string Word::getWord() const {
	return word;
}

json Word::toJSON() const {
	json rval;
	rval["String"] = word;
	json arr;
	for(int i = 0; i < wordsFollowing.size(); i++) {
		arr.push_back(wordsFollowing[i]->getWord());
	}
	rval["Words Following"] = arr;
	return rval;
}