Add image auto save and load

Author: pavel-cpp

source/interfaces/mainwindow/mainwindow.cpp

@@ -21,7 +21,6 @@
 #include <ui_mainwindow.h>
 
 // Services
-#include <services/image-downloader/image_downloader.h>
 #include <services/image-printer/image_printer.h>
 #include <services/theme-loader/theme_loader.h>
 
@@ -92,28 +91,7 @@ MainWindow::MainWindow(QWidget *parent)
 
     ui->database_table_view->addActions(ui->menubar->actions());
 
-    if (!ImageDownloader::FetchImage(
-        settings_manager_.GetSettings().consts.source_image_path + "." +
-        settings_manager_.GetSettings().image.format,
-        settings_manager_.GetSettings().image.url
-    )) {
-        QMessageBox::warning(this, "Предупреждение!", "Не удалось загрузить изображение!\nНе верно указана ссылка");
-    }
-
-    QImage image;
-
-    if (!image.load(
-        settings_manager_.GetSettings().consts.source_image_path + "." +
-        settings_manager_.GetSettings().image.format
-    )) {
-        QMessageBox::warning(
-            this,
-            "Предупреждение!",
-            "Не удалось открыть изображение!\nНе верно указан формат изображения или ссылка"
-        );
-    }
-
-    text_painter_.SetImage(image);
+    text_painter_.SetImage(settings_manager_.GetSettings().image);
 
     for (int i = 0; i < 3; ++i) {
         items_[i].options = settings_manager_.GetSettings().font_settings[i];
@@ -164,8 +142,7 @@ void MainWindow::on_save_image_triggered() {
         settings_manager_.GetSettings().output_folder
         + "/image_"
         + items_[0].content
-        + "."
-        + settings_manager_.GetSettings().image.format
+        + ".jpeg"
     );
     if (status) {
         ui->statusbar->showMessage(
@@ -313,8 +290,7 @@ void MainWindow::on_save_some_images_triggered() {
             settings_manager_.GetSettings().output_folder
             + "/image_"
             + items_[0].content
-            + "."
-            + settings_manager_.GetSettings().image.format
+            + ".jpeg"
         )) {
             ui->statusbar->showMessage(
                 "image_"

source/interfaces/options/options.cpp

@@ -62,8 +62,6 @@ Options::Options(QWidget *parent) :
     ui->example_3->setStyleSheet(QString("color: %1;\nfont-size: %2px;").arg(
             settings_buffer_.font_settings[2].color.name(QColor::HexRgb)).arg(
             settings_buffer_.font_settings[2].font.pixelSize()));
-    ui->image_url_edit->setText(settings_buffer_.image.url.toString());
-    ui->image_format_edit->setText(settings_buffer_.image.format);
 
     SetAdvancedSettingsVisible(false);
 }
@@ -137,7 +135,7 @@ void Options::on_database_edit_button_clicked() {
 }
 
 void Options::on_save_button_clicked() {
-    if (settings_manager_.GetSettings().image.url != settings_buffer_.image.url ||
+    if (settings_manager_.GetSettings().image != settings_buffer_.image ||
         settings_manager_.GetSettings().database.host != settings_buffer_.database.host ||
         settings_manager_.GetSettings().database.port != settings_buffer_.database.port ||
         settings_manager_.GetSettings().database.schema != settings_buffer_.database.schema) {
@@ -148,21 +146,32 @@ void Options::on_save_button_clicked() {
     close();
 }
 
+void Options::on_upload_button_clicked() {
+    QString path = QFileDialog::getOpenFileName(
+            this,
+            "Загрузить изображение",
+            QDir::homePath(),
+            "All files(*.*);;"
+            "JPEG(*.jpg *.jpeg);;"
+            "PNG(*.png);;"
+            "BITMAP(*.bmp);;"
+            "WEBP(*.webp)"
+        );
+    if (path.isEmpty()) {
+        return;
+    }
+    if (!settings_buffer_.image.load(path)) {
+        QMessageBox::critical(this, "Ошибка", "Файл не загружен."
+                                              "\nНе верно указан путь."
+                                              "\nИли не верное имя файла.");
+    };
+}
+
 void Options::on_path_to_edit_textChanged(const QString &text) {
     settings_buffer_.output_folder = text;
     save_type_ |= SettingsManager::SaveType::SAVE_GENERAL;
 }
 
-void Options::on_image_url_edit_textChanged(const QString &text) {
-    settings_buffer_.image.url = text;
-    save_type_ |= SettingsManager::SaveType::SAVE_SYNCING;
-}
-
-void Options::on_image_format_edit_textChanged(const QString &text) {
-    settings_buffer_.image.format = text;
-    save_type_ |= SettingsManager::SaveType::SAVE_SYNCING;
-}
-
 void Options::SetAdvancedSettingsVisible(bool state) {
     ui->formatting_label->setVisible(state);
     ui->formatting_line->setVisible(state);
@@ -185,10 +194,7 @@ void Options::SetAdvancedSettingsVisible(bool state) {
 
     ui->image_line->setVisible(state);
     ui->image_label->setVisible(state);
-    ui->image_format_label->setVisible(state);
-    ui->image_format_edit->setVisible(state);
-    ui->image_url_label->setVisible(state);
-    ui->image_url_edit->setVisible(state);
+    ui->upload_button->setVisible(state);
 }
 
 void Options::on_advanced_settings_button_clicked() {

source/interfaces/options/options.h

@@ -106,18 +106,9 @@ private slots:
     void on_save_button_clicked();
 
     /**
-     * @brief Slot for handling changes in the image_url_edit field.
-     *
-     * @param text The new text in the image_url_edit field.
-     */
-    void on_image_url_edit_textChanged(const QString &text);
-
-    /**
-     * @brief Slot for handling changes in the image_format_edit field.
-     *
-     * @param text The new text in the image_format_edit field.
+     * @brief Slot for handling the upload_button click event.
      */
-    void on_image_format_edit_textChanged(const QString &text);
+    void on_upload_button_clicked();
 
     /**
      * @brief Slot for handling the advanced_settings_button click event.

source/services/database/models.h

@@ -22,5 +22,4 @@
 // Local
 #include "models/database_options_model.h"
 #include "models/font_settings_model.h"
-#include "models/image_model.h"
 #include "models/passwords_model.h"

source/services/database/models/image_model.h

@@ -19,8 +19,10 @@
 
 #include "settings_manager.h"
 
+#include <QBuffer>
 #include <QDir>
 #include <qfileinfo.h>
+#include <QPixmap>
 #include <QSqlRecord>
 #include <QString>
 #include <QVariant>
@@ -30,6 +32,8 @@ using std::string;
 // Qt
 #include <QSqlQuery>
 
+constexpr char EXTENSION[] = "WEBP";
+
 SettingsManager::SettingsManager(const QString& connection_name) {
     LoadFromIni(connection_name);
     LoadFromDatabase();
@@ -91,12 +95,14 @@ void SettingsManager::Save(int type) {
             query.exec();
         }
 
+        QByteArray image_as_bytes;
+        QBuffer buffer(&image_as_bytes);
+        buffer.open(QIODevice::WriteOnly);
+        settings_.image.save(&buffer, EXTENSION);
         query.prepare(
-            UPDATE_IMAGE
-            .arg(database_.schema)
-            .arg("\'" + settings_.image.url.toString() + "\'")
-            .arg("\'" + settings_.image.format + "\'")
+            UPDATE_IMAGE.arg(settings_.database.schema)
         );
+        query.bindValue(":img", image_as_bytes);
         query.exec();
     }
 
@@ -140,11 +146,10 @@ void SettingsManager::LoadFromDatabase() {
         if (!query.next()) {
             assert(false);
         }
-        {
-            QSqlRecord rec = query.record();
-            settings_.image.url = rec.value("url").toUrl();
-            settings_.image.format = rec.value("format").toString();
-        }
+        settings_.image.loadFromData(
+            query.value("base64").toByteArray(),
+            EXTENSION
+        );
 
         query.prepare(SELECT_PASSWORDS.arg(database_.schema));
         query.exec();

source/services/image-downloader/image_downloader.cpp

@@ -26,6 +26,7 @@
 // Qt
 #include <QColor>
 #include <QFont>
+#include <QImage>
 #include <QPointF>
 #include <QString>
 
@@ -54,7 +55,7 @@ class SettingsManager {
         QString theme;
         std::array<Models::FontSettings, 3> font_settings;
         Models::Database database;
-        Models::Image image;
+        QImage image;
         Models::Passwords passwords;
 
         Settings& operator=(const Settings& other) {
@@ -93,9 +94,9 @@ class SettingsManager {
     const QString UPDATE_FONT_SETTINGS =
         "UPDATE %1.font_settings SET font = %2, color = %3, position_x = %4, position_y = %5, size = %6, bold = %7 WHERE id = %8;";
 
-    const QString SELECT_IMAGE = "SELECT url, format FROM %1.image WHERE id = 1;";
+    const QString SELECT_IMAGE = "SELECT base64 FROM %1.image WHERE id = 1;";
 
-    const QString UPDATE_IMAGE = "UPDATE %1.image SET url = %2, format = %3 WHERE id = 1;";
+    const QString UPDATE_IMAGE = "UPDATE %1.image SET base64 = :img WHERE id = 1;";
 
     const QString SELECT_PASSWORDS = "SELECT * FROM %1.advanced_settings_passwords;";