3m/ModList.cpp

355 lines
9.6 KiB
C++

#include "ModList.h"
#include "NetSocket++/NetSocketPP.h"
#include "3mExceptions.h"
#include <vector>
#include <fstream>
#include <iostream>
#include <string>
#include <cstdlib>
using namespace mmm;
ModList::ModList() {
ModListDescription emptymld;
_desc = emptymld;
_modinfos.clear();
_edit = false;
_localPath = "";
_modinfosIterator = -1;
_modinfosAtEnd = true;
}
ModList::ModList(ModListDescription mld) {
_desc = mld;
_modinfos.clear();
_edit = false;
_localPath = "";
_modinfosIterator = -1;
_modinfosAtEnd = false;
NetSocketPP::HTTPReply data;
try {
NetSocketPP::HTTPClientSocket conn(_desc.getServer(), "http", _desc.getPath());
std::cout << "Connection to " << conn.getIP() << " successful!" << std::endl;
data = conn.getReply();
} catch (NetSocketPP::SocketException &exc) {
std::cerr << "SocketException occured while getting remote modlist: " << exc.what() << std::endl;
throw(exc);
} catch (NetSocketPP::NetworkException &exc) {
std::cerr << "NetworkException occured while getting remote modlist: " << exc.what() << std::endl;
throw(exc);
}
if(data.getResponse() != "200 OK") {
throw BadResponseException(data.getResponse());
}
std::string content = data.getContent();
std::vector<std::string> modlist = strtovec(content);
std::string action = "detect";
ModInfoDescription mid;
for(unsigned int i = 0; i < modlist.size(); i++) {
std::string line = "";
line = modlist[i];
if(line[0] != NULL && line[0] != ' ' && line[0] != '\n' && line[0] != '\r') {
if(action == "detect") {
if(line[0] == '{') {
std::string name = "";
for(unsigned int i = 1; line[i] != '}' && i < line.length(); i++) {
name += line[i];
}
mid.setName(name);
action = "parse";
} else {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although { was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
}
} else if(action == "parse") {
if(line[0] == '{') {
if(line[1] == 'e' && line[2] == 'n' && line[3] == 'd' && line[4] == '}') {
if(mid.getName() != "" && mid.getServer() != "" && mid.getPath() != "") {
ModInfo mi;
mi.setModInfoDescription(mid);
_modinfos.push_back(mi);
} else {
throw ParseException(_desc.getServer() + _desc.getPath(), "Data error.");
}
action = "detect";
} else {
std::string msg = "";
msg += "Found ";
msg += line;
msg += " although {end} or action in [] was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
}
} else if(line[0] == '[') {
std::string tmpact = "";
for(unsigned int i = 1; line[i] != ']' && i < line.length(); i++) {
tmpact += line[i];
}
if(tmpact == "server" || tmpact == "modinfo") {
action = tmpact;
} else {
std::string msg = "";
msg += "Found ";
msg += tmpact;
msg += " although server/modinfo was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
}
} else {
std::string msg = "";
msg += "Found ";
msg += line;
msg += " although {end} or action in [] was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
}
} else if(action == "server") {
if(line[0] == '[' || line[0] == '{') {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although string was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
} else {
mid.setServer(line);
action = "parse";
}
} else if(action == "modinfo") {
if(line[0] == '[' || line[0] == '{') {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although string was expected.";
throw ParseException(_desc.getServer() + _desc.getPath(), msg);
} else {
mid.setPath(line);
action = "parse";
}
} else {
throw ParseException(_desc.getServer() + _desc.getPath(), "The program should not reach this place!");
}
}
}
std::cout << "Got all modinfo descriptions from " << _desc.getName() << ", downloading and parsing modinfos..." << std::endl;
for(unsigned int i = 0; i < _modinfos.size(); i++) {
try {
ModInfo mi(_modinfos[i].getModInfoDescription());
std::cout << "Successfully downloaded and parsed: " << mi.getName() << std::endl;
_modinfos[i] = mi;
sleep(2); // This is to prevent DoS-ing rubenwardy' s hosting...
} catch(NetSocketPP::NetworkException &exc) {
std::cerr << "NetworkException occured in NetSocket++: " << exc.what() << std::endl;
} catch(NetSocketPP::SocketException &exc) {
std::cerr << "SocketException occured in NetSocket++: " << exc.what() << std::endl;
} catch(ParseException &exc) {
std::cerr << "ParseException occured: " << exc.what() << std::endl;
} catch(BadResponseException &exc) {
std::cerr << "BadResponseException occured: " << exc.what() << std::endl;
}
}
}
ModList::ModList(std::string path) {
ModListDescription emptymld;
_desc = emptymld;
_modinfos.clear();
_edit = true;
_localPath = path;
_modinfosIterator = -1;
_modinfosAtEnd = false;
std::ifstream modlist(_localPath.c_str());
if(!modlist) {
throw FileException(_localPath, "reading", "Could not open file!");
}
std::string action = "detect";
ModInfoDescription mid;
while(!modlist.eof()) {
std::string line = "";
char c;
do {
modlist.get(c);
if(c != '\n');
line += c;
} while(c != '\n');
line = strip_endl(line);
if(modlist) {
if(line[0] != NULL && line[0] != ' ' && line[0] != '\n' && line[0] != '\r') {
if(action == "detect") {
if(line[0] == '{') {
std::string name = "";
for(unsigned int i = 1; line[i] != '}' && i < line.length(); i++) {
name += line[i];
}
mid.setName(name);
action = "parse";
} else {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although { was expected.";
throw ParseException(_localPath, msg);
}
} else if(action == "parse") {
if(line[0] == '{') {
if(line[1] == 'e' && line[2] == 'n' && line[3] == 'd' && line[4] == '}') {
if(mid.getName() != "" && mid.getServer() != "" && mid.getPath() != "") {
ModInfo mi;
mi.setModInfoDescription(mid);
_modinfos.push_back(mi);
} else {
throw ParseException(_localPath, "Data error.");
}
action = "detect";
} else {
std::string msg = "";
msg += "Found ";
msg += line;
msg += " although {end} or action in [] was expected.";
throw ParseException(_localPath, msg);
}
} else if(line[0] == '[') {
std::string tmpact = "";
for(unsigned int i = 1; line[i] != ']' && i < line.length(); i++) {
tmpact += line[i];
}
if(tmpact == "server" || tmpact == "modinfo") {
action = tmpact;
} else {
std::string msg = "";
msg += "Found ";
msg += tmpact;
msg += " although server/modinfo was expected.";
throw ParseException(_localPath, msg);
}
} else {
std::string msg = "";
msg += "Found ";
msg += line;
msg += " although {end} or action in [] was expected.";
throw ParseException(_localPath, msg);
}
} else if(action == "server") {
if(line[0] == '[' || line[0] == '{') {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although string was expected.";
throw ParseException(_localPath, msg);
} else {
mid.setServer(line);
action = "parse";
}
} else if(action == "modinfo") {
if(line[0] == '[' || line[0] == '{') {
std::string msg = "";
msg += "Found ";
msg += line[0];
msg += " although string was expected.";
throw ParseException(_localPath, msg);
} else {
mid.setPath(line);
action = "parse";
}
} else {
throw ParseException(_localPath, "The program should not reach this place!");
}
}
}
}
modlist.close();
}
ModList::~ModList() {
ModListDescription emptymld;
_desc = emptymld;
_modinfos.clear();
_edit = false;
_localPath = "";
_modinfosIterator = -1;
_modinfosAtEnd = true;
}
void ModList::clear() {
ModListDescription emptymld;
_desc = emptymld;
_modinfos.clear();
_edit = false;
_localPath = "";
_modinfosIterator = -1;
_modinfosAtEnd = true;
}
ModInfo ModList::getNextModInfo() {
if(_modinfosIterator+1 < _modinfos.size()) {
_modinfosIterator++;
return _modinfos[_modinfosIterator];
} else {
static ModInfo emptymi;
_modinfosAtEnd = true;
return emptymi;
}
}
ModInfo ModList::getModInfoByName(std::string name) {
for(unsigned int i = 0; i < _modinfos.size(); i++) {
if(_modinfos[i].getModInfoDescription().getName() == name) {
return _modinfos[i];
}
}
static ModInfo emptymi;
return emptymi;
}
void ModList::insertModInfoDescription(ModInfoDescription mid) {
ModInfo mi;
mi.setModInfoDescription(mid);
_modinfos.push_back(mi);
if(_modinfosAtEnd) {
_modinfosAtEnd = false;
}
}
void ModList::deleteModInfo(std::string name) {
for(std::vector<ModInfo>::iterator i = _modinfos.begin(); i != _modinfos.end(); i++) {
mmm::ModInfoDescription mid = i -> getModInfoDescription();
if(mid.getName() == name) {
_modinfos.erase(i);
return;
}
}
}
void ModList::resetModInfoIterator() {
_modinfosIterator = -1;
if(_modinfosAtEnd) {
_modinfosAtEnd = false;
}
}
bool ModList::modinfosEnd() {
return _modinfosAtEnd;
}
void ModList::write() {
if(!_edit) {
throw NonEditableException("Tried to write back remotely obtained modlist file!");
}
std::ofstream modlist(_localPath.c_str());
if(!modlist) {
throw FileException(_localPath, "writing", "Could not open file!");
}
for(unsigned int i = 0; i < _modinfos.size(); i++) {
ModInfoDescription mid = _modinfos[i].getModInfoDescription();
modlist << "{" << mid.getName() << "}" << std::endl << "[server]" << std::endl << mid.getServer() << std::endl << "[modinfo]" << std::endl << mid.getPath() << std::endl << "{end}" << std::endl;
}
modlist.close();
}
ModListDescription ModList::getModListDescription() {
return _desc;
}
void ModList::setPath(std::string path) {
_localPath = path;
_edit = true;
}