src/db/Db2Xml.cpp
browsing at commit = 6ef7244eca99cacc7cde5de361d99bdc566fefcc
/*
===========================================================================
oastat - a program for parsing log files and write the result to a database
Copyright (C) 2012 Poul Sander
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see http://www.gnu.org/licenses/
Source information and contacts persons can be found at
http://code.google.com/p/oastat/
===========================================================================
*/
#include "Db2Xml.hpp"
Db2Xml::Db2Xml()
{
psoudo_playerids.clear();
nextId = 1;
boost::format path("%1%/oastat");
path % getenv("HOME");
p_output_dir = path.str();
boost::format fmkdir("mkdir -p %1%");
fmkdir % p_output_dir;
int fmkdir_ret = system(fmkdir.str().c_str());
if (fmkdir_ret) {
cout << "Attemting to create \"" << path << "\" with \"" << fmkdir.str() << "\", but return code was: " << fmkdir_ret << endl;
}
}
Db2Xml::Db2Xml(string dbargs)
{
boost::format path("%1%/oastat");
path % getenv("HOME");
p_output_dir = path.str();
p_postscript = "";
psoudo_playerids.clear();
nextId = 1;
//mkdir(p_output_dir.c_str());
stringstream stream(stringstream::in | stringstream::out);
stream << dbargs;
string holder;
//stream >> holder;
while (!stream.eof()) {
stream >> holder;
if (!stream.eof()) {
string param;
stream >> param;
if(holder == "outputdir") {
p_output_dir = param;
}
if(holder == "postscript") {
p_postscript = param;
}
}
}
boost::format fmkdir("mkdir -p %1%");
fmkdir % p_output_dir;
int fmkdir_ret = system(fmkdir.str().c_str());
if (fmkdir_ret) {
cout << "Attemting to create \"" << path << "\" with \"" << fmkdir.str() << "\", but return code was: " << fmkdir_ret << endl;
}
}
Db2Xml::Db2Xml(const Db2Xml& orig)
{
}
Db2Xml::~Db2Xml()
{
}
void Db2Xml::startGame(int gametype, const string &mapname, const string &basegame, const string &servername, const OaStatStruct &oss)
{
isOk = true;
p_xmlcontent = "";
p_gametime = oss.getDateTime();
p_servername = servername;
p_xmlcontent += "<OaStatGame>\n";
p_xmlcontent += (boost::format(" <servername>%1%</servername>\n") % getXmlEscaped(p_servername) ).str();
p_xmlcontent += (boost::format(" <gametype>%1%</gametype>\n") % gametype).str();
p_xmlcontent += (boost::format(" <mapname>%1%</mapname>\n") % getXmlEscaped(mapname) ).str();
p_xmlcontent += (boost::format(" <basegame>%1%</basegame>\n") % getXmlEscaped(basegame) ).str();
p_xmlcontent += (boost::format(" <gametime>%04i-%02i-%02iT%02i:%02i:%02i</gametime>\n")
% (p_gametime.tm_year+1900) % (p_gametime.tm_mon+1) % (p_gametime.tm_mday)
% (p_gametime.tm_hour) % (p_gametime.tm_min) % (p_gametime.tm_sec) ).str();
}
void Db2Xml::addGameCvar(const std::string &cvar, const std::string &value)
{
if (!isOk) {
return;
}
p_xmlcontent += " <cvar>\n";
p_xmlcontent += (boost::format(" <key>%1%</key>\n") % getXmlEscaped(cvar) ).str();
p_xmlcontent += (boost::format(" <value>%1%</value>\n") % getXmlEscaped(value) ).str();
p_xmlcontent += " </cvar>\n";
}
void Db2Xml::endGame(int second)
{
if (isOk) {
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += "</OaStatGame>\n";
string filename = (boost::format("%1%/%2%_%3%-%4%-%5%_%6%-%7%-%8%.xml") % p_output_dir % p_servername
% (p_gametime.tm_year+1900) % (p_gametime.tm_mon+1) % (p_gametime.tm_mday)
% (p_gametime.tm_hour) % (p_gametime.tm_min) % (p_gametime.tm_sec) ).str();
ofstream outfile;
outfile.open (filename.c_str());
if (outfile.fail()) {
cout << "could not create " << filename << endl;
}
outfile << p_xmlcontent;
outfile.close();
cout << "End game at " << second << "with size " << p_xmlcontent.length() << " written to " << filename << endl;
}
}
int Db2Xml::getGameNumber()
{
return 1;
}
void Db2Xml::setPlayerInfo(const std::string &guid, const std::string &nickname, bool isBot, int second, int team, const std::string &model, const std::string &headmodel, int skill)
{
if (!isOk) {
return;
}
p_xmlcontent += " <playerinfo>\n";
p_xmlcontent += (boost::format(" <hashedguid>%1%</hashedguid>\n") % guid).str();
p_xmlcontent += (boost::format(" <playerid>%1%</playerid>\n") % getPsoudoId(guid)).str();
p_xmlcontent += (boost::format(" <name>%1%</name>\n") % getXmlEscaped(nickname) ).str();
p_xmlcontent += (boost::format(" <isbot>%1%</isbot>\n") % (isBot? 1:0)).str();
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <model>%1%</model>\n") % getXmlEscaped(model) ).str();
p_xmlcontent += (boost::format(" <headmodel>%1%</headmodel>\n") % getXmlEscaped(headmodel) ).str();
p_xmlcontent += (boost::format(" <skill>%1%</skill>\n") % skill).str();
p_xmlcontent += " </playerinfo>\n";
}
void Db2Xml::addKill(int second, const std::string &attackerID, const std::string &targetID, int type)
{
if (!isOk) {
return;
}
p_xmlcontent += " <kill>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <attacker>%1%</attacker>\n") % attackerID).str();
p_xmlcontent += (boost::format(" <target>%1%</target>\n") % targetID).str();
p_xmlcontent += (boost::format(" <mod_type>%1%</mod_type>\n") % type).str();
p_xmlcontent += " </kill>\n";
}
//void addCapture(int second, string player, int team);
void Db2Xml::addAward(int second, const std::string &player, int award)
{
if (!isOk) {
return;
}
p_xmlcontent += " <award>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <award>%1%</award>\n") % award).str();
p_xmlcontent += " </award>\n";
}
void Db2Xml::addScoreInfo(int second, const std::string &player, int score)
{
if (!isOk) {
return;
}
p_xmlcontent += " <score>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <score>%1%</score>\n") % score).str();
p_xmlcontent += " </score>\n";
}
void Db2Xml::addCtf(int second, const std::string &player, int team, int event)
{
if (!isOk) {
return;
}
p_xmlcontent += " <ctf>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += " </ctf>\n";
}
void Db2Xml::addCtf1f(int second, const std::string &player, int team, int event)
{
if (!isOk) {
return;
}
p_xmlcontent += " <ctf1flag>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += " </ctf1flag>\n";
}
void Db2Xml::addElimination(int second, int roundnumber, int team, int event)
{
if (!isOk) {
return;
}
p_xmlcontent += " <elimination>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <roundnumber>%1%</roundnumber>\n") % roundnumber).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += " </elimination>\n";
}
void Db2Xml::addCtfElimination(int second, int roundnumber, const std::string &player, int team, int event)
{
if (!isOk) {
return;
}
p_xmlcontent += " <ctfelimination>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <roundnumber>%1%</roundnumber>\n") % roundnumber).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += " </ctfelimination>\n";
}
void Db2Xml::addHarvester(int second, const std::string &player1, const std::string &player2, int team, int event, int score)
{
if (!isOk) {
return;
}
p_xmlcontent += " <harvester>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player1>%1%</player1>\n") % player1).str();
p_xmlcontent += (boost::format(" <player2>%1%</player2>\n") % player2).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += (boost::format(" <score>%1%</score>\n") % score).str();
p_xmlcontent += " </harvester>\n";
}
void Db2Xml::addGenericTeamEvent(int second, int team, int amount, const std::string &gametype, const std::string &player1, const std::string &player2, int event, int generic1)
{
if (!isOk) {
return;
}
p_xmlcontent += " <generic_team_event>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <team>%1%</team>\n") % team).str();
p_xmlcontent += (boost::format(" <amount>%1%</amount>\n") % amount).str();
p_xmlcontent += (boost::format(" <gametype>%1%</gametype>\n") % gametype).str();
p_xmlcontent += (boost::format(" <player1>%1%</player1>\n") % player1).str();
p_xmlcontent += (boost::format(" <player2>%1%</player2>\n") % player2).str();
p_xmlcontent += (boost::format(" <event>%1%</event>\n") % event).str();
p_xmlcontent += (boost::format(" <generic1>%1%</generic1>\n") % generic1).str();
p_xmlcontent += " </generic_team_event>\n";
}
void Db2Xml::addChallenge(int second, const std::string &player, int challenge, int amount)
{
if (!isOk) {
return;
}
p_xmlcontent += " <challenge>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <challenge>%1%</challenge>\n") % challenge).str();
p_xmlcontent += (boost::format(" <amount>%1%</amount>\n") % amount).str();
p_xmlcontent += " </challenge>\n";
}
void Db2Xml::addAccuracy(int second, const std::string &player, int type, int shotsFired, int shotsHit)
{
if (!isOk) {
return;
}
p_xmlcontent += " <accuracy>\n";
p_xmlcontent += (boost::format(" <second>%1%</second>\n") % second).str();
p_xmlcontent += (boost::format(" <player>%1%</player>\n") % player).str();
p_xmlcontent += (boost::format(" <mod_type>%1%</mod_type>\n") % type).str();
p_xmlcontent += (boost::format(" <shots_fired>%1%</shots_fired>\n") % shotsFired).str();
p_xmlcontent += (boost::format(" <shots_hit>%1%</shots_hit>\n") % shotsHit).str();
p_xmlcontent += " </accuracy>\n";
}
void Db2Xml::doNotCommit()
{
isOk = false;
}
int Db2Xml::getPsoudoId(const string &guid)
{
int ret = psoudo_playerids[guid];
if (ret > 0) {
return ret;
}
ret = nextId++;
if (ret == 1022) { //world
ret = nextId++;
}
psoudo_playerids[guid] = ret;
return ret;
}
string Db2Xml::getXmlEscaped(const string &org)
{
string result = org;
boost::algorithm::replace_all(result,(string)"&",(string)"&");
boost::algorithm::replace_all(result,(string)"<",(string)"<");
boost::algorithm::replace_all(result,(string)">",(string)">");
boost::algorithm::replace_all(result,(string)"\"",(string)""");
boost::algorithm::replace_all(result,(string)"'",(string)"'");
return result;
}
void Db2Xml::createTables()
{
}