-
Notifications
You must be signed in to change notification settings - Fork 0
/
Database.cpp
175 lines (154 loc) · 3.8 KB
/
Database.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
#include "../include/FILEIO/Database.h"
#include <fstream>
#include <sstream>
#include <iostream>
Database::Database() = default;
Database::~Database() = default;
/**
* @brief Load a CSV file into the database
* @param fileName The name of the file to load
* @return True if the file was loaded successfully, false otherwise
*/
bool Database::load(const std::string& fileName)
{
std::ifstream file(fileName);
if (!file.is_open())
{
std::cerr << "Error: Could not open file " << fileName << std::endl;
return false;
}
std::string line;
if (std::getline(file, line))
{
headers = split(line, ',');
}
while (std::getline(file, line))
{
Row row;
auto values = split(line, ',');
for (size_t i = 0; i < headers.size() && i < values.size(); ++i)
{
row[headers[i]] = values[i];
}
rows.push_back(row);
}
file.close();
return true;
}
/**
* @brief Save the database to a CSV file
* @param fileName The name of the file to save
* @return True if the file was saved successfully, false otherwise
*/
bool Database::save(const std::string& fileName) const
{
std::ofstream file(fileName);
if (!file.is_open())
{
std::cerr << "Error: Could not open file " << fileName << std::endl;
return false;
}
file << join(headers, ',') << std::endl;
for (const auto& row : rows)
{
std::vector<std::string> values;
for (const auto& header : headers)
{
auto it = row.find(header);
if (it != row.end())
{
values.push_back(it->second);
}
else
{
values.emplace_back("");
}
}
file << join(values, ',') << std::endl;
}
file.close();
return true;
}
/**
* @brief Add a row to the database
* @param row The row to add
*/
void Database::addRow(const Row& row)
{
rows.push_back(row);
}
/**
* @brief Remove a row from the database
* @param index The index of the row to remove
*/
const std::vector<Database::Row>& Database::getRows() const
{
return rows;
}
/**
* @brief The path to the database file
* @param path The path to the database file
*/
void Database::databasePath(const std::filesystem::path& path)
{
filePath = path;
}
/**
* @brief The path to the database file
* @return The path to the database file
*/
const std::filesystem::path& Database::databasePath() const
{
return filePath;
}
/**
* @brief Split a string into tokens
* @param str The string to split
* @param delimiter The delimiter to split on
* @return A vector of tokens
*/
std::vector<std::string> Database::split(const std::string& str, char delimiter)
{
std::vector<std::string> tokens;
std::string token;
std::istringstream tokenStream(str);
while (std::getline(tokenStream, token, delimiter))
{
tokens.push_back(token);
}
return tokens;
}
/**
* @brief Join a vector of strings into a single string
* @param elements The elements to join
* @param delimiter The delimiter to join on
* @return The joined string
*/
std::string Database::join(const std::vector<std::string>& elements, char delimiter)
{
std::ostringstream os;
for (auto it = elements.begin(); it != elements.end(); ++it)
{
if (it != elements.begin())
{
os << delimiter;
}
os << *it;
}
return os.str();
}
/**
* @brief Trim whitespace from a string
* @param str The string to trim
* @return The trimmed string
*/
std::string Database::trim(const std::string& str)
{
size_t first = str.find_first_not_of(' ');
if (std::string::npos == first)
{
return str;
}
size_t last = str.find_last_not_of(' ');
return str.substr(first, (last - first + 1));
}