Database.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
 
#include <QObject>
#include <QSqlDatabase>
#include <QDateTime>
#include <QJsonObject>
#include "plugin_export.h"
 
class CParameterDatabase;

class PLUGIN_EXPORT CDatabase : public QObject
{
    Q_OBJECT
 
public:
    explicit CDatabase(QObject *parent = nullptr);
    virtual ~CDatabase();


    [[nodiscard]] bool SetDatabase(const CDatabase* db);
    [[nodiscard]] bool SetDatabase(const QSqlDatabase db, const CParameterDatabase* pPara);
 
    [[nodiscard]] QSqlDatabase GetDatabase() const;


    [[nodiscard]] bool OpenDatabase(
        const CParameterDatabase* pPara = nullptr,
        const QString& szConnectName = QString());
    [[nodiscard]] bool OpenMySqlDatabase(
        const CParameterDatabase* pPara,
        const QString& szConnectName = QString());
    [[nodiscard]] bool OpenODBCDatabase(
        const CParameterDatabase* pPara,
        const QString& szConnectName = QString());
    [[nodiscard]] bool OpenSQLiteDatabase(
        const CParameterDatabase* pPara,
        const QString &szConnectionName = QString());
    [[nodiscard]] bool OpenSQLiteDatabase(
        const QString& szFile,
        const QString& szConnectionName = QString());
    
    [[nodiscard]] virtual bool IsOpen() const;
    void CloseDatabase();
 
    [[nodiscard]] const CParameterDatabase* GetParameter() const;
    [[nodiscard]] const QString GetError() const;

    [[nodiscard]] virtual bool ExportToJsonFile(const QString& szFile);
    [[nodiscard]] virtual bool ImportFromJsonFile(const QString& szFile);
 
Q_SIGNALS:
    void sigChanged();
 
protected:

    [[nodiscard]] virtual bool OnInitializeDatabase();
    [[nodiscard]] virtual bool OnInitializeSqliteDatabase();
    [[nodiscard]] virtual bool OnInitializeMySqlDatabase();
 
    void SetError(const QString& szErr = QString());
 
    [[nodiscard]] virtual bool ExportToJson(QJsonObject& obj);
    [[nodiscard]] virtual bool ImportFromJson(const QJsonObject& obj);
 
protected:
    QString m_szConnectName;
    QString m_MinVersion;
    const CParameterDatabase* m_pPara;
 
private:
    QSqlDatabase m_database;
    bool m_bOwner;
    QString m_szError;
};

class PLUGIN_EXPORT CDatabaseIcon : public CDatabase
{
    Q_OBJECT
 
public:
    explicit CDatabaseIcon(const QString& szSuffix = QString(),
                           QObject *parent = nullptr);

    [[nodiscard]] int GetIcon(const QIcon& icon);
    [[nodiscard]] QIcon GetIcon(int id);
 
    [[nodiscard]] static bool ExportIconToJson(/*in*/const QIcon& icon, /*out*/QJsonObject& obj);
    [[nodiscard]] static bool ImportIconFromJson(/*in*/const QJsonObject &obj, /*out*/QIcon& icon);
 
protected:
    bool OnInitializeSqliteDatabase() override;
    bool OnInitializeMySqlDatabase() override;
 
    //[[nodiscard]] virtual bool ExportToJson(QJsonObject& obj) override;
 
private:
    QString m_szTableName;
};

class PLUGIN_EXPORT CDatabaseFile : public CDatabase
{
    Q_OBJECT
    
public:
    explicit CDatabaseFile(const QString& szSuffix = QString(),
                           QObject *parent = nullptr);
 
    bool IsExist(const QString& szFile);

    [[nodiscard]] QByteArray Load(const QString &szFile);
    bool Save(const QString& szFile);
    
    [[nodiscard]] static bool ExportFileToJson(const QString &szFile, QJsonObject &obj);
    [[nodiscard]] static bool ImportFileFromJson(const QJsonObject &obj, QString &szFile);
    [[nodiscard]] bool ImportFileToDatabaseFromJson(const QJsonObject &obj, QString &szFile);

    [[nodiscard]] static QString GetFile(const QString& szFile);
    [[nodiscard]] static QString SetFile(const QString& szFile);
 
protected:
    [[nodiscard]] virtual bool OnInitializeSqliteDatabase() override;
    [[nodiscard]] virtual bool OnInitializeMySqlDatabase() override;
 
private:
    QString m_szTableName;
};