mirror of
https://github.com/FunKey-Project/RetroFE.git
synced 2025-12-12 17:58:53 +01:00
133 lines
4.8 KiB
C++
133 lines
4.8 KiB
C++
/* This file is part of RetroFE.
|
|
*
|
|
* RetroFE 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* RetroFE 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 RetroFE. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "MamelistMetadata.h"
|
|
#include "DB.h"
|
|
#include "../Utility/Log.h"
|
|
#include "Metadata.h"
|
|
#include <rapidxml.hpp>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include <sqlite3.h>
|
|
|
|
|
|
MamelistMetadata::MamelistMetadata(DB &dbInstance)
|
|
: DBInstance(dbInstance)
|
|
{
|
|
}
|
|
|
|
MamelistMetadata::~MamelistMetadata()
|
|
{
|
|
}
|
|
|
|
bool MamelistMetadata::Import(std::string filename, std::string collection)
|
|
{
|
|
bool retVal = true;
|
|
rapidxml::xml_document<> doc;
|
|
rapidxml::xml_node<> * rootNode;
|
|
char *error = NULL;
|
|
sqlite3 *handle = DBInstance.GetHandle();
|
|
|
|
std::ifstream f(filename.c_str());
|
|
|
|
if (!f.good())
|
|
{
|
|
Logger::Write(Logger::ZONE_ERROR, "Mamelist", "Could not find mamelist metadata file at \"" + filename + "\"");
|
|
|
|
retVal = false;
|
|
}
|
|
|
|
f.close();
|
|
|
|
if(retVal)
|
|
{
|
|
Logger::Write(Logger::ZONE_INFO, "Mamelist", "Importing mamelist file \"" + filename + "\"");
|
|
std::ifstream file(filename.c_str());
|
|
std::vector<char> buffer((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
|
|
|
|
buffer.push_back('\0');
|
|
|
|
doc.parse<0>(&buffer[0]);
|
|
|
|
rootNode = doc.first_node("mame");
|
|
|
|
sqlite3_exec(handle, "BEGIN IMMEDIATE TRANSACTION;", NULL, NULL, &error);
|
|
for (rapidxml::xml_node<> * game = rootNode->first_node("game"); game; game = game->next_sibling())
|
|
{
|
|
rapidxml::xml_attribute<> *nameNode = game->first_attribute("name");
|
|
rapidxml::xml_attribute<> *cloneOfXml = game->first_attribute("cloneof");
|
|
|
|
if(nameNode != NULL)
|
|
{
|
|
std::string name = nameNode->value();
|
|
rapidxml::xml_node<> *descriptionNode = game->first_node("description");
|
|
rapidxml::xml_node<> *yearNode = game->first_node("year");
|
|
rapidxml::xml_node<> *manufacturerNode = game->first_node("manufacturer");
|
|
rapidxml::xml_node<> *inputNode = game->first_node("input");
|
|
|
|
std::string description = (descriptionNode == NULL) ? nameNode->value() : descriptionNode->value();
|
|
std::string year = (yearNode == NULL) ? "" : yearNode->value();
|
|
std::string manufacturer = (manufacturerNode == NULL) ? "" : manufacturerNode->value();
|
|
std::string cloneOf = (cloneOfXml == NULL) ? "" : cloneOfXml->value();
|
|
std::string players;
|
|
std::string buttons;
|
|
|
|
if(inputNode != NULL)
|
|
{
|
|
rapidxml::xml_attribute<> *playersAttribute = inputNode->first_attribute("players");
|
|
rapidxml::xml_attribute<> *buttonsAttribute = inputNode->first_attribute("buttons");
|
|
|
|
if(playersAttribute)
|
|
{
|
|
players = playersAttribute->value();
|
|
}
|
|
|
|
if(buttonsAttribute)
|
|
{
|
|
buttons = buttonsAttribute->value();
|
|
}
|
|
|
|
}
|
|
|
|
sqlite3_stmt *stmt;
|
|
sqlite3_prepare_v2(handle,
|
|
"UPDATE OR REPLACE Meta SET title=?, year=?, manufacturer=?, players=?, buttons=?, cloneOf=? WHERE name=? AND collectionName=?;",
|
|
-1, &stmt, 0);
|
|
|
|
sqlite3_bind_text(stmt, 1, description.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 2, year.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 3, manufacturer.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 4, players.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 5, buttons.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 6, cloneOf.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 7, name.c_str(), -1, SQLITE_TRANSIENT);
|
|
sqlite3_bind_text(stmt, 8, collection.c_str(), -1, SQLITE_TRANSIENT);
|
|
|
|
sqlite3_step(stmt);
|
|
sqlite3_finalize(stmt);
|
|
|
|
}
|
|
}
|
|
sqlite3_exec(handle, "COMMIT TRANSACTION;", NULL, NULL, &error);
|
|
|
|
}
|
|
|
|
|
|
|
|
return retVal;
|
|
}
|