aboutsummaryrefslogtreecommitdiffstats
path: root/Plugins/JsonExport/jsonexport.h
blob: c9706e000c3106310434caa519a9e7f707c9e1b3 (plain) (blame)
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
#ifndef JSONEXPORT_H
#define JSONEXPORT_H

#include "jsonexport_global.h"
#include "plugins/genericexportplugin.h"
#include "config_builder.h"
#include <QStack>

CFG_CATEGORIES(JsonExportConfig,
     CFG_CATEGORY(JsonExport,
         CFG_ENTRY(QString, Format,       "format")
     )
)

class JSONEXPORTSHARED_EXPORT JsonExport : public GenericExportPlugin
{
        Q_OBJECT
        SQLITESTUDIO_PLUGIN("jsonexport.json")

    public:
        JsonExport();

        QString getFormatName() const;
        ExportManager::StandardConfigFlags standardOptionsToEnable() const;
        QString getExportConfigFormName() const;
        CfgMain* getConfig();
        void validateOptions();
        QString defaultFileExtension() const;
        QString getDefaultEncoding() const;
        bool beforeExportQueryResults(const QString& query, QList<QueryExecutor::ResultColumnPtr>& columns,
                                      const QHash<ExportManager::ExportProviderFlag,QVariant> providedData);
        bool exportQueryResultsRow(SqlResultsRowPtr row);
        bool afterExportQueryResults();
        bool exportTable(const QString& database, const QString& table, const QStringList& columnNames, const QString& ddl, SqliteCreateTablePtr createTable,
                         const QHash<ExportManager::ExportProviderFlag,QVariant> providedData);
        bool exportVirtualTable(const QString& database, const QString& table, const QStringList& columnNames, const QString& ddl, SqliteCreateVirtualTablePtr createTable,
                                const QHash<ExportManager::ExportProviderFlag,QVariant> providedData);
        bool exportTableRow(SqlResultsRowPtr data);
        bool afterExportTable();
        bool beforeExportDatabase(const QString& database);
        bool exportIndex(const QString& database, const QString& name, const QString& ddl, SqliteCreateIndexPtr createIndex);
        bool exportTrigger(const QString& database, const QString& name, const QString& ddl, SqliteCreateTriggerPtr createTrigger);
        bool exportView(const QString& database, const QString& name, const QString& ddl, SqliteCreateViewPtr createView);
        bool afterExportDatabase();
        bool beforeExport();
        bool init();
        void deinit();

    private:
        void setupConfig();
        void incrIndent();
        void decrIndent();
        void updateIndent();
        void incrElementCount();
        void write(const QString& str);
        QString escapeString(const QString& str);
        QString formatValue(const QVariant& val);
        void beginObject();
        void beginObject(const QString& key);
        void endObject();
        void beginArray();
        void beginArray(const QString& key);
        void endArray();
        void writeValue(const QVariant& value);
        void writeValue(const QString& key, const QVariant& value);
        void writePrefixBeforeEnd();
        void writePrefixBeforeNextElement();

        CFG_LOCAL_PERSISTABLE(JsonExportConfig, cfg)
        QStack<int> elementCounter;
        bool indent = false;
        int indentDepth = 0;
        QString indentStr;
        QString newLineStr;
        QString codecName;
};

#endif // JSONEXPORT_H