flameshot/src/core/capturerequest.cpp

// SPDX-License-Identifier: GPL-3.0-or-later
// SPDX-FileCopyrightText: 2017-2019 Alejandro Sirgo Rica & Contributors

#include "capturerequest.h"
#include "confighandler.h"
#include "controller.h"
#include "imguruploader.h"
#include "pinwidget.h"
#include "src/utils/screenshotsaver.h"
#include "systemnotification.h"
#include <QApplication>
#include <QClipboard>
#include <QDateTime>
#include <QVector>
#include <stdexcept>

CaptureRequest::CaptureRequest(CaptureRequest::CaptureMode mode,
                               const uint delay,
                               const QVariant& data,
                               CaptureRequest::ExportTask tasks)
  : m_mode(mode)
  , m_delay(delay)
  , m_tasks(tasks)
  , m_data(data)
  , m_forcedID(false)
  , m_id(0)
{}

void CaptureRequest::setStaticID(uint id)
{
    m_forcedID = true;
    m_id = id;
}

uint CaptureRequest::id() const
{
    if (m_forcedID) {
        return m_id;
    }

    uint id = 0;
    QVector<uint> v;
    v << qHash(m_mode) << qHash(m_delay * QDateTime::currentMSecsSinceEpoch())
      << qHash(m_path) << qHash(m_tasks) << m_data.toInt();
    for (uint i : v) {
        id ^= i + 0x9e3779b9 + (id << 6) + (id >> 2);
    }
    return id;
}

QByteArray CaptureRequest::serialize() const
{
    QByteArray data;
    QDataStream stream(&data, QIODevice::WriteOnly);
    // Convert enums to integers
    qint32 tasks = m_tasks, mode = m_mode;
    stream << mode << m_delay << tasks << m_data << m_forcedID << m_id
           << m_path;
    return data;
}

CaptureRequest CaptureRequest::deserialize(const QByteArray& data)
{
    QDataStream stream(data);
    CaptureRequest request;
    qint32 tasks, mode;
    stream >> mode;
    stream >> request.m_delay;
    stream >> tasks;
    stream >> request.m_data;
    stream >> request.m_forcedID;
    stream >> request.m_id;
    stream >> request.m_path;

    // Convert integers to enums
    request.m_tasks = static_cast<ExportTask>(tasks);
    request.m_mode = static_cast<CaptureMode>(mode);
    return request;
}

CaptureRequest::CaptureMode CaptureRequest::captureMode() const
{
    return m_mode;
}

uint CaptureRequest::delay() const
{
    return m_delay;
}

QString CaptureRequest::path() const
{
    return m_path;
}

QVariant CaptureRequest::data() const
{
    return m_data;
}

CaptureRequest::ExportTask CaptureRequest::tasks() const
{
    return m_tasks;
}

void CaptureRequest::addTask(CaptureRequest::ExportTask task)
{
    if (task == SAVE) {
        throw std::logic_error("SAVE task must be added using addSaveTask");
    }
    m_tasks |= task;
}

void CaptureRequest::addSaveTask(const QString& path)
{
    m_tasks |= SAVE;
    m_path = path;
}

void CaptureRequest::addPinTask(const QRect& pinWindowGeometry)
{
    m_tasks |= PIN;
    m_pinWindowGeometry = pinWindowGeometry;
}

void CaptureRequest::exportCapture(const QPixmap& capture)
{
    if (m_tasks & SAVE) {
        if (m_path.isEmpty()) {
            ScreenshotSaver(m_id).saveToFilesystemGUI(capture);
        } else {
            ScreenshotSaver(m_id).saveToFilesystem(capture, m_path);
        }
    }

    if (m_tasks & COPY) {
        ScreenshotSaver().saveToClipboard(capture);
    }

    if (m_tasks & PIN) {
        QWidget* widget = new PinWidget(capture, m_pinWindowGeometry);
        widget->show();
        widget->activateWindow();
        if (m_mode == SCREEN_MODE || m_mode == FULLSCREEN_MODE) {
            SystemNotification().sendMessage(
              QObject::tr("Full screen screenshot pinned to screen"));
        }
    }

    if (m_tasks & UPLOAD) {
        ImgurUploader* widget = new ImgurUploader(capture);
        widget->show();
        widget->activateWindow();
        // NOTE: lambda can't capture 'this' because it might be destroyed later
        ExportTask tasks = m_tasks;
        QObject::connect(
          widget, &ImgurUploader::uploadOk, [widget, tasks](const QUrl& url) {
              if (ConfigHandler().copyAndCloseAfterUpload()) {
                  if (!(tasks & COPY)) {
                      QApplication::clipboard()->setText(url.toString());
                      SystemNotification().sendMessage(
                        QObject::tr("URL copied to clipboard."));
                      widget->close();
                  } else {
                      widget->showPostUploadDialog();
                  }
              } else {
                  widget->showPostUploadDialog();
              }
          });
    }
}