Differential D12068 Diff 33147 src/screenshotdialog.cpp

Changeset View

Standalone View

src/screenshotdialog.cpp

This file was added.

1		/*
2		* Copyright © 2018 Red Hat, Inc
3		*
4		* This program is free software; you can redistribute it and/or
5		* modify it under the terms of the GNU Lesser General Public
6		* License as published by the Free Software Foundation; either
7		* version 2 of the License, or (at your option) any later version.
8		*
9		* This library is distributed in the hope that it will be useful,
10		* but WITHOUT ANY WARRANTY; without even the implied warranty of
11		* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12		* Lesser General Public License for more details.
13		*
14		* You should have received a copy of the GNU Lesser General Public
15		* License along with this library. If not, see <http://www.gnu.org/licenses/>.
16		*
17		* Authors:
18		* Jan Grulich <jgrulich@redhat.com>
19		*/
20
21		#include "screenshotdialog.h"
22		#include "ui_screenshotdialog.h"
23
24		#include <QLoggingCategory>
25		#include <QPushButton>
26
27		#include <QDBusInterface>
28		#include <QDBusPendingCall>
29		#include <QDBusPendingReply>
30		#include <QDBusUnixFileDescriptor>
31		#include <QTimer>
32		#include <QFutureWatcher>
33		#include <QtConcurrent/QtConcurrentRun>
34		#include <qplatformdefs.h>
35
36		#include <fcntl.h>
37		#include <unistd.h>
38
39		Q_LOGGING_CATEGORY(XdgDesktopPortalKdeScreenshotDialog, "xdp-kde-screenshot-dialog")
40
41		static int readData(int fd, QByteArray &data)
42		{
43		// implementation based on QtWayland file qwaylanddataoffer.cpp
44		char buf[4096];
45		int retryCount = 0;
46		int n;
47		while (true) {
48		n = QT_READ(fd, buf, sizeof buf);
49		// give user 30 sec to click a window, afterwards considered as error
50		if (n == -1 && (errno == EAGAIN) && ++retryCount < 30000) {
51		usleep(1000);
52		} else {
53		break;
54		}
55		}
56		if (n > 0) {
57		data.append(buf, n);
58		n = readData(fd, data);
59		}
60		return n;
61		}
62
63		static QImage readImage(int pipeFd)
64		{
65		QByteArray content;
66		if (readData(pipeFd, content) != 0) {
67		close(pipeFd);
68		return QImage();
69		}
70		close(pipeFd);
71		QDataStream ds(content);
72		QImage image;
73		ds >> image;
74		return image;
75		};
76
77		ScreenshotDialog::ScreenshotDialog(QDialog *parent, Qt::WindowFlags flags)
78		: QDialog(parent, flags)
79		, m_dialog(new Ui::ScreenshotDialog)
80		{
81		m_dialog->setupUi(this);
82
83		connect(m_dialog->buttonBox, &QDialogButtonBox::accepted, this, &ScreenshotDialog::accept);
84		connect(m_dialog->buttonBox, &QDialogButtonBox::rejected, this, &ScreenshotDialog::reject);
85		connect(m_dialog->takeScreenshotButton, &QPushButton::clicked, this, [this] () {
86		QTimer::singleShot(1000 * m_dialog->delaySpinBox->value(), this, &ScreenshotDialog::takeScreenshot);
87		});
88
89		connect(m_dialog->areaComboBox, static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), this, [this] (int index) {
90		m_dialog->includeBordersCheckbox->setEnabled(index == 2);
91		});
92
93		m_dialog->buttonBox->button(QDialogButtonBox::Save)->setEnabled(false);
94
95		setWindowTitle(i18n("Request screenshot"));
96		}
97
98		ScreenshotDialog::~ScreenshotDialog()
99		{
100		delete m_dialog;
101		}
102		void ScreenshotDialog::takeScreenshot()
103		{
104		int pipeFds[2];
105		if (pipe2(pipeFds, O_CLOEXEC\|O_NONBLOCK) != 0) {
106		Q_EMIT failed();
107		return;
108		}
109
110		QDBusInterface interface(QLatin1String("org.kde.KWin"), QLatin1String("/Screenshot"), QLatin1String("org.kde.kwin.Screenshot"));
111		if (m_dialog->areaComboBox->currentIndex() < 2) {
112		interface.asyncCall(m_dialog->areaComboBox->currentIndex() ? QLatin1String("screenshotScreen") : QLatin1String("screenshotFullscreen"), QVariant::fromValue(QDBusUnixFileDescriptor(pipeFds[1])), m_dialog->includeCursorCheckbox->isChecked());
113		} else {
114		interface.asyncCall(QLatin1String("interactive"), QVariant::fromValue(QDBusUnixFileDescriptor(pipeFds[1])), mask());
115		}
116
117		QFutureWatcher<QImage> *watcher = new QFutureWatcher<QImage>(this);
118		QObject::connect(watcher, &QFutureWatcher<QImage>::finished, this,
119		[watcher, this] {
120		watcher->deleteLater();
121		m_image = watcher->result();
122		m_dialog->image->setPixmap(QPixmap::fromImage(m_image).scaled(400, 320, Qt::KeepAspectRatio));
123		m_dialog->buttonBox->button(QDialogButtonBox::Save)->setEnabled(true);
124		}
125		);
126
127		watcher->setFuture(QtConcurrent::run(readImage, pipeFds[0]));
128
129		::close(pipeFds[1]);
	KanediasUnsubmitted Not Done Are we safe closing the fd 1 here? Are we sure it already got through asyncCall? Kanedias: Are we safe closing the fd 1 here? Are we sure it already got through asyncCall?
130		}
131
132		QImage ScreenshotDialog::image() const
133		{
134		return m_image;
135		}
136
137		int ScreenshotDialog::mask()
138		{
139		int mask = 0;
140		if (m_dialog->includeBordersCheckbox->isChecked()) {
141		mask = 1;
142		}
143		if (m_dialog->includeCursorCheckbox->isChecked()) {
144		mask \|= 1 << 1;
145		}
146		return mask;
147		}