usercard.cpp

#include "usercard.h"

#include <QBuffer>
#include <QDebug>
#include <QFileDialog>
#include <QHBoxLayout>
#include <QLabel>
#include <QPushButton>
#include <QSqlDatabase>
#include <QSqlError>
#include <QSqlQuery>
#include <QStackedWidget>
#include <QStandardPaths>
#include <QTextEdit>
#include <QVBoxLayout>

class UserCard::Implementation
{
public:
    explicit Implementation(const QString& _user_name);
    QString getUsercardText();
    void saveUsercardText(const QString& _text);
    void updateUserPhoto(const QByteArray& _photo, const QString& _user_name);
    void deleteUserPhoto(const QString& _user_name);
    const QByteArray getUserPhoto(const QString& _user_name);

    const QString user_name;
    QTextEdit* textedit;
    QLabel* photo_lbl;
    QPushButton* pb_add_photo;
};

UserCard::Implementation::Implementation(const QString& _user_name)
    : user_name(_user_name)
{
}

QString UserCard::Implementation::getUsercardText()
{
    QString user_card_text;
    {
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE", "connection_name");
        db.setDatabaseName("ImagineTalkDB.db");
        db.open();

        QSqlQuery query(db);
        query.prepare("SELECT user_card FROM Users WHERE user=?");
        query.addBindValue(user_name);
        if (!query.exec()) {
            qDebug() << "Error in UserCard::Implementation::getUsercardText: "
                     << query.lastError().text();
        }
        query.first();
        user_card_text.append(query.value("user_card").toString());
        query.finish();
        db.close();
    }
    QSqlDatabase::removeDatabase("connection_name");
    return user_card_text;
}

void UserCard::Implementation::saveUsercardText(const QString& _text)
{
    QByteArray in_byte_array;
    QBuffer in_buffer(&in_byte_array);
    in_buffer.open(QIODevice::WriteOnly);
    in_buffer.write(_text.toUtf8());

    {
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE", "connection_name");
        db.setDatabaseName("ImagineTalkDB.db");
        db.open();

        QSqlQuery query(db);
        query.prepare("UPDATE Users SET user_card=? WHERE user=?");
        query.addBindValue(in_byte_array);
        query.addBindValue(user_name);

        if (!query.exec()) {
            qDebug() << "Error in UserCard::Implementation::saveUsercardText: "
                     << query.lastError().text();
        }
        query.finish();
        db.close();
    }
    QSqlDatabase::removeDatabase("connection_name");
}

void UserCard::Implementation::updateUserPhoto(const QByteArray& _photo, const QString& _user_name)
{
    {
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE", "connection_name");
        db.setDatabaseName("ImagineTalkDB.db");
        db.open();

        QSqlQuery query(db);
        query.prepare("UPDATE Users SET user_photo=? WHERE user=?");
        query.addBindValue(_photo);
        query.addBindValue(_user_name);

        if (!query.exec()) {
            qDebug() << "Error in UserCard::Implementation::saveUsercardText: "
                     << query.lastError().text();
        }
        query.finish();
        db.close();
    }
    QSqlDatabase::removeDatabase("connection_name");
}

void UserCard::Implementation::deleteUserPhoto(const QString& _user_name)
{
    {
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE", "connection_name");
        db.setDatabaseName("ImagineTalkDB.db");
        db.open();

        QSqlQuery query(db);
        query.prepare("UPDATE Users SET user_photo=NULL WHERE user=?");
        query.addBindValue(_user_name);

        if (!query.exec()) {
            qDebug() << "Error in UserCard::Implementation::saveUsercardText: "
                     << query.lastError().text();
        }
        query.finish();
        db.close();
    }
    QSqlDatabase::removeDatabase("connection_name");
}

const QByteArray UserCard::Implementation::getUserPhoto(const QString& _user_name)
{
    QByteArray user_photo;
    {
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE", "connection_name");
        db.setDatabaseName("ImagineTalkDB.db");
        db.open();

        QSqlQuery query(db);
        query.prepare("SELECT user_photo FROM Users WHERE user=?");
        query.addBindValue(_user_name);
        if (!query.exec()) {
            qDebug() << "Error in UserCard::Implementation::getUsercardText: "
                     << query.lastError().text();
        }
        query.first();
        user_photo = query.value("user_photo").toByteArray();
        query.finish();
        db.close();
    }
    QSqlDatabase::removeDatabase("connection_name");
    return user_photo;
}

//
// ***
//

UserCard::UserCard(QStackedWidget* _stack, QWidget* _main_window, const QString& _user_name)
    : d(new Implementation(_user_name))
{
    //
    // Определяем размер кнопок
    //
    const QFontMetrics fm(font());
    const int height = fm.height() + 20;
    const QSize size_pb(320, height);
    const int spacing = 1;

    //
    // Текстовый редактор
    //
    d->textedit = new QTextEdit(this);
    d->textedit->setText(d->getUsercardText());

    //
    // Кнопки
    //
    QPushButton* pb_home = new QPushButton("Назад", this);
    pb_home->setFixedSize(size_pb);
    connect(pb_home, &QPushButton::clicked, this, [this, _stack, _main_window]() {
        _stack->removeWidget(this);
        _stack->setCurrentWidget(_main_window);
        delete this;
    });

    d->pb_add_photo = new QPushButton("Добавить фото", this);
    d->pb_add_photo->setFixedSize(size_pb);
    connect(d->pb_add_photo, &QPushButton::clicked, this, [this, size_pb]() {
        QStringList desktop_path
            = QStandardPaths::standardLocations(QStandardPaths::DesktopLocation);
        QString image_path
            = QFileDialog::getOpenFileName(this, "Открыть изображение", desktop_path.first(),
                                           "Image Files (*.jpg *.png *.bmp *.jpeg)");
        if (image_path.isEmpty())
            return;

        QPixmap pix(image_path);
        QByteArray in_byte_array;
        QBuffer in_buffer(&in_byte_array);
        in_buffer.open(QIODevice::WriteOnly);
        if (pix.save(&in_buffer, "PNG")) {
            //
            // Сохраняем фото в БД
            //
            d->updateUserPhoto(in_byte_array, d->user_name);
        }
        pix = pix.scaledToWidth(size_pb.width(), Qt::SmoothTransformation);
        d->photo_lbl->setPixmap(pix);
        d->pb_add_photo->setText("Изменить фото");
    });

    QPushButton* pb_delete_photo = new QPushButton("Удалить фото", this);
    pb_delete_photo->setFixedSize(size_pb);
    connect(pb_delete_photo, &QPushButton::clicked, this, [this]() {
        d->deleteUserPhoto(d->user_name);
        d->pb_add_photo->setText("Добавить фото");
        d->photo_lbl->setText("Фото");
    });

    //
    // Фото
    //
    d->photo_lbl = new QLabel(this);
    d->photo_lbl->setFixedWidth(size_pb.width());
    d->photo_lbl->setAlignment(Qt::AlignCenter);
    QByteArray photo_bytearray = d->getUserPhoto(d->user_name);
    if (photo_bytearray.isEmpty()) {
        d->photo_lbl->setText("Фото");
    } else {
        QPixmap photo;
        photo.loadFromData(photo_bytearray);
        photo = photo.scaledToWidth(size_pb.width(), Qt::SmoothTransformation);
        d->photo_lbl->setPixmap(photo);
        d->pb_add_photo->setText("Изменить фото");
    }

    QVBoxLayout* vbx_layout = new QVBoxLayout;
    vbx_layout->addStretch();
    vbx_layout->addWidget(d->photo_lbl);
    vbx_layout->addStretch();
    vbx_layout->addWidget(d->pb_add_photo);
    vbx_layout->addWidget(pb_delete_photo);
    vbx_layout->addWidget(pb_home);
    vbx_layout->setSpacing(spacing);
    vbx_layout->setAlignment(Qt::AlignBottom);

    QHBoxLayout* hbx_layout = new QHBoxLayout;
    hbx_layout->addWidget(d->textedit);
    hbx_layout->addLayout(vbx_layout);
    setLayout(hbx_layout);
}

UserCard::~UserCard()
{
    qDebug() << "UserCard destructor";
    d->saveUsercardText(d->textedit->toPlainText());
}