eqemu-server/common/repositories/task_activities_repository.h
2020-04-04 02:56:14 -05:00

360 lines
10 KiB
C++

/**
* EQEmulator: Everquest Server Emulator
* Copyright (C) 2001-2020 EQEmulator Development Team (https://github.com/EQEmu/Server)
*
* 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; version 2 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY except by those people which sell it, which
* are required to give you total support for your newly bought product;
* 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, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
#ifndef EQEMU_TASK_ACTIVITIES_REPOSITORY_H
#define EQEMU_TASK_ACTIVITIES_REPOSITORY_H
#include "../database.h"
#include "../string_util.h"
class TaskActivitiesRepository {
public:
struct TaskActivities {
int taskid;
int activityid;
int step;
int8 activitytype;
std::string target_name;
std::string item_list;
std::string skill_list;
std::string spell_list;
std::string description_override;
int goalid;
int goalmethod;
int goalcount;
int delivertonpc;
std::string zones;
int8 optional;
};
static std::string PrimaryKey()
{
return std::string("activityid");
}
static std::vector<std::string> Columns()
{
return {
"taskid",
"activityid",
"step",
"activitytype",
"target_name",
"item_list",
"skill_list",
"spell_list",
"description_override",
"goalid",
"goalmethod",
"goalcount",
"delivertonpc",
"zones",
"optional",
};
}
static std::string ColumnsRaw()
{
return std::string(implode(", ", Columns()));
}
static std::string InsertColumnsRaw()
{
std::vector<std::string> insert_columns;
for (auto &column : Columns()) {
if (column == PrimaryKey()) {
continue;
}
insert_columns.push_back(column);
}
return std::string(implode(", ", insert_columns));
}
static std::string TableName()
{
return std::string("task_activities");
}
static std::string BaseSelect()
{
return fmt::format(
"SELECT {} FROM {}",
ColumnsRaw(),
TableName()
);
}
static std::string BaseInsert()
{
return fmt::format(
"INSERT INTO {} ({}) ",
TableName(),
InsertColumnsRaw()
);
}
static TaskActivities NewEntity()
{
TaskActivities entry{};
entry.taskid = 0;
entry.activityid = 0;
entry.step = 0;
entry.activitytype = 0;
entry.target_name = "";
entry.item_list = "";
entry.skill_list = "-1";
entry.spell_list = "0";
entry.description_override = "";
entry.goalid = 0;
entry.goalmethod = 0;
entry.goalcount = 1;
entry.delivertonpc = 0;
entry.zones = "";
entry.optional = 0;
return entry;
}
static TaskActivities GetTaskActivitiesEntry(
const std::vector<TaskActivities> &task_activitiess,
int task_activities_id
)
{
for (auto &task_activities : task_activitiess) {
if (task_activities.activityid == task_activities_id) {
return task_activities;
}
}
return NewEntity();
}
static TaskActivities FindOne(
int task_activities_id
)
{
auto results = content_db.QueryDatabase(
fmt::format(
"{} WHERE id = {} LIMIT 1",
BaseSelect(),
task_activities_id
)
);
auto row = results.begin();
if (results.RowCount() == 1) {
TaskActivities entry{};
entry.taskid = atoi(row[0]);
entry.activityid = atoi(row[1]);
entry.step = atoi(row[2]);
entry.activitytype = atoi(row[3]);
entry.target_name = row[4];
entry.item_list = row[5];
entry.skill_list = row[6];
entry.spell_list = row[7];
entry.description_override = row[8];
entry.goalid = atoi(row[9]);
entry.goalmethod = atoi(row[10]);
entry.goalcount = atoi(row[11]);
entry.delivertonpc = atoi(row[12]);
entry.zones = row[13];
entry.optional = atoi(row[14]);
return entry;
}
return NewEntity();
}
static int DeleteOne(
int task_activities_id
)
{
auto results = content_db.QueryDatabase(
fmt::format(
"DELETE FROM {} WHERE {} = {}",
TableName(),
PrimaryKey(),
task_activities_id
)
);
return (results.Success() ? results.RowsAffected() : 0);
}
static int UpdateOne(
TaskActivities task_activities_entry
)
{
std::vector<std::string> update_values;
auto columns = Columns();
update_values.push_back(columns[2] + " = " + std::to_string(task_activities_entry.step));
update_values.push_back(columns[3] + " = " + std::to_string(task_activities_entry.activitytype));
update_values.push_back(columns[4] + " = '" + EscapeString(task_activities_entry.target_name) + "'");
update_values.push_back(columns[5] + " = '" + EscapeString(task_activities_entry.item_list) + "'");
update_values.push_back(columns[6] + " = '" + EscapeString(task_activities_entry.skill_list) + "'");
update_values.push_back(columns[7] + " = '" + EscapeString(task_activities_entry.spell_list) + "'");
update_values.push_back(columns[8] + " = '" + EscapeString(task_activities_entry.description_override) + "'");
update_values.push_back(columns[9] + " = " + std::to_string(task_activities_entry.goalid));
update_values.push_back(columns[10] + " = " + std::to_string(task_activities_entry.goalmethod));
update_values.push_back(columns[11] + " = " + std::to_string(task_activities_entry.goalcount));
update_values.push_back(columns[12] + " = " + std::to_string(task_activities_entry.delivertonpc));
update_values.push_back(columns[13] + " = '" + EscapeString(task_activities_entry.zones) + "'");
update_values.push_back(columns[14] + " = " + std::to_string(task_activities_entry.optional));
auto results = content_db.QueryDatabase(
fmt::format(
"UPDATE {} SET {} WHERE {} = {}",
TableName(),
implode(", ", update_values),
PrimaryKey(),
task_activities_entry.activityid
)
);
return (results.Success() ? results.RowsAffected() : 0);
}
static TaskActivities InsertOne(
TaskActivities task_activities_entry
)
{
std::vector<std::string> insert_values;
insert_values.push_back(std::to_string(task_activities_entry.step));
insert_values.push_back(std::to_string(task_activities_entry.activitytype));
insert_values.push_back("'" + EscapeString(task_activities_entry.target_name) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.item_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.skill_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.spell_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.description_override) + "'");
insert_values.push_back(std::to_string(task_activities_entry.goalid));
insert_values.push_back(std::to_string(task_activities_entry.goalmethod));
insert_values.push_back(std::to_string(task_activities_entry.goalcount));
insert_values.push_back(std::to_string(task_activities_entry.delivertonpc));
insert_values.push_back("'" + EscapeString(task_activities_entry.zones) + "'");
insert_values.push_back(std::to_string(task_activities_entry.optional));
auto results = content_db.QueryDatabase(
fmt::format(
"{} VALUES ({})",
BaseInsert(),
implode(",", insert_values)
)
);
if (results.Success()) {
task_activities_entry.id = results.LastInsertedID();
return task_activities_entry;
}
task_activities_entry = InstanceListRepository::NewEntity();
return task_activities_entry;
}
static int InsertMany(
std::vector<TaskActivities> task_activities_entries
)
{
std::vector<std::string> insert_chunks;
for (auto &task_activities_entry: task_activities_entries) {
std::vector<std::string> insert_values;
insert_values.push_back(std::to_string(task_activities_entry.step));
insert_values.push_back(std::to_string(task_activities_entry.activitytype));
insert_values.push_back("'" + EscapeString(task_activities_entry.target_name) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.item_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.skill_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.spell_list) + "'");
insert_values.push_back("'" + EscapeString(task_activities_entry.description_override) + "'");
insert_values.push_back(std::to_string(task_activities_entry.goalid));
insert_values.push_back(std::to_string(task_activities_entry.goalmethod));
insert_values.push_back(std::to_string(task_activities_entry.goalcount));
insert_values.push_back(std::to_string(task_activities_entry.delivertonpc));
insert_values.push_back("'" + EscapeString(task_activities_entry.zones) + "'");
insert_values.push_back(std::to_string(task_activities_entry.optional));
insert_chunks.push_back("(" + implode(",", insert_values) + ")");
}
std::vector<std::string> insert_values;
auto results = content_db.QueryDatabase(
fmt::format(
"{} VALUES {}",
BaseInsert(),
implode(",", insert_chunks)
)
);
return (results.Success() ? results.RowsAffected() : 0);
}
static std::vector<TaskActivities> All()
{
std::vector<TaskActivities> all_entries;
auto results = content_db.QueryDatabase(
fmt::format(
"{}",
BaseSelect()
)
);
all_entries.reserve(results.RowCount());
for (auto row = results.begin(); row != results.end(); ++row) {
TaskActivities entry{};
entry.taskid = atoi(row[0]);
entry.activityid = atoi(row[1]);
entry.step = atoi(row[2]);
entry.activitytype = atoi(row[3]);
entry.target_name = row[4];
entry.item_list = row[5];
entry.skill_list = row[6];
entry.spell_list = row[7];
entry.description_override = row[8];
entry.goalid = atoi(row[9]);
entry.goalmethod = atoi(row[10]);
entry.goalcount = atoi(row[11]);
entry.delivertonpc = atoi(row[12]);
entry.zones = row[13];
entry.optional = atoi(row[14]);
all_entries.push_back(entry);
}
return all_entries;
}
};
#endif //EQEMU_TASK_ACTIVITIES_REPOSITORY_H