forked from keepassxreboot/keepassxc
/
CsvExporter.cpp
113 lines (95 loc) · 3.02 KB
/
CsvExporter.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
/*
* Copyright (C) 2015 Florian Geyer <blueice@fobos.de>
* Copyright (C) 2015 Felix Geyer <debfx@fobos.de>
*
* 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 or (at your option)
* version 3 of the License.
*
* 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/>.
*/
#include "CsvExporter.h"
#include <QFile>
#include "core/Database.h"
#include "core/Group.h"
bool CsvExporter::exportDatabase(const QString& filename, const QSharedPointer<const Database>& db)
{
QFile file(filename);
if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
m_error = file.errorString();
return false;
}
return exportDatabase(&file, db);
}
bool CsvExporter::exportDatabase(QIODevice* device, const QSharedPointer<const Database>& db)
{
if (device->write(exportHeader().toUtf8()) == -1) {
m_error = device->errorString();
return false;
}
if (device->write(exportGroup(db->rootGroup()).toUtf8()) == -1) {
m_error = device->errorString();
return false;
}
return true;
}
QString CsvExporter::exportDatabase(const QSharedPointer<const Database>& db)
{
return exportHeader() + exportGroup(db->rootGroup());
}
QString CsvExporter::errorString() const
{
return m_error;
}
QString CsvExporter::exportHeader()
{
QString header;
addColumn(header, "Group");
addColumn(header, "Title");
addColumn(header, "Username");
addColumn(header, "Password");
addColumn(header, "URL");
addColumn(header, "Notes");
return header + QString("\n");
}
QString CsvExporter::exportGroup(const Group* group, QString groupPath)
{
QString response;
if (!groupPath.isEmpty()) {
groupPath.append("/");
}
groupPath.append(group->name());
const QList<Entry*>& entryList = group->entries();
for (const Entry* entry : entryList) {
QString line;
addColumn(line, groupPath);
addColumn(line, entry->title());
addColumn(line, entry->username());
addColumn(line, entry->password());
addColumn(line, entry->url());
addColumn(line, entry->notes());
line.append("\n");
response.append(line);
}
const QList<Group*>& children = group->children();
for (const Group* child : children) {
response.append(exportGroup(child, groupPath));
}
return response;
}
void CsvExporter::addColumn(QString& str, const QString& column)
{
if (!str.isEmpty()) {
str.append(",");
}
str.append("\"");
str.append(QString(column).replace("\"", "\"\""));
str.append("\"");
}