mirror of https://github.com/Nheko-Reborn/nheko
Nicolas Werner
3 years ago
parent
09c041c8ac
commit
ef068ac2b3
@ -0,0 +1,164 @@ |
||||
// SPDX-FileCopyrightText: 2021 Nheko Contributors
|
||||
//
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
|
||||
#include "MxcAnimatedImage.h" |
||||
|
||||
#include <QDir> |
||||
#include <QFileInfo> |
||||
#include <QMimeDatabase> |
||||
#include <QQuickWindow> |
||||
#include <QSGImageNode> |
||||
#include <QStandardPaths> |
||||
|
||||
#include "EventAccessors.h" |
||||
#include "Logging.h" |
||||
#include "MatrixClient.h" |
||||
#include "timeline/TimelineModel.h" |
||||
|
||||
void |
||||
MxcAnimatedImage::startDownload() |
||||
{ |
||||
if (!room_) |
||||
return; |
||||
if (eventId_.isEmpty()) |
||||
return; |
||||
|
||||
auto event = room_->eventById(eventId_); |
||||
if (!event) { |
||||
nhlog::ui()->error("Failed to load media for event {}, event not found.", |
||||
eventId_.toStdString()); |
||||
return; |
||||
} |
||||
|
||||
QByteArray mimeType = QString::fromStdString(mtx::accessors::mimetype(*event)).toUtf8(); |
||||
|
||||
animatable_ = QMovie::supportedFormats().contains(mimeType.split('/').back()); |
||||
animatableChanged(); |
||||
|
||||
if (!animatable_) |
||||
return; |
||||
|
||||
QString mxcUrl = QString::fromStdString(mtx::accessors::url(*event)); |
||||
QString originalFilename = QString::fromStdString(mtx::accessors::filename(*event)); |
||||
|
||||
auto encryptionInfo = mtx::accessors::file(*event); |
||||
|
||||
// If the message is a link to a non mxcUrl, don't download it
|
||||
if (!mxcUrl.startsWith("mxc://")) { |
||||
return; |
||||
} |
||||
|
||||
QString suffix = QMimeDatabase().mimeTypeForName(mimeType).preferredSuffix(); |
||||
|
||||
const auto url = mxcUrl.toStdString(); |
||||
const auto name = QString(mxcUrl).remove("mxc://"); |
||||
QFileInfo filename(QString("%1/media_cache/media/%2.%3") |
||||
.arg(QStandardPaths::writableLocation(QStandardPaths::CacheLocation)) |
||||
.arg(name) |
||||
.arg(suffix)); |
||||
if (QDir::cleanPath(name) != name) { |
||||
nhlog::net()->warn("mxcUrl '{}' is not safe, not downloading file", url); |
||||
return; |
||||
} |
||||
|
||||
QDir().mkpath(filename.path()); |
||||
|
||||
QPointer<MxcAnimatedImage> self = this; |
||||
|
||||
auto processBuffer = [this, mimeType, encryptionInfo, self](QIODevice &device) { |
||||
if (!self) |
||||
return; |
||||
|
||||
if (buffer.isOpen()) { |
||||
movie.stop(); |
||||
movie.setDevice(nullptr); |
||||
buffer.close(); |
||||
} |
||||
|
||||
if (encryptionInfo) { |
||||
QByteArray ba = device.readAll(); |
||||
std::string temp(ba.constData(), ba.size()); |
||||
temp = mtx::crypto::to_string( |
||||
mtx::crypto::decrypt_file(temp, encryptionInfo.value())); |
||||
buffer.setData(temp.data(), temp.size()); |
||||
} else { |
||||
buffer.setData(device.readAll()); |
||||
} |
||||
buffer.open(QIODevice::ReadOnly); |
||||
buffer.reset(); |
||||
|
||||
QTimer::singleShot(0, this, [this, mimeType] { |
||||
nhlog::ui()->info("Playing movie with size: {}, {}", |
||||
buffer.bytesAvailable(), |
||||
buffer.isOpen()); |
||||
movie.setFormat(mimeType); |
||||
movie.setDevice(&buffer); |
||||
movie.start(); |
||||
emit loadedChanged(); |
||||
}); |
||||
}; |
||||
|
||||
if (filename.isReadable()) { |
||||
QFile f(filename.filePath()); |
||||
if (f.open(QIODevice::ReadOnly)) { |
||||
processBuffer(f); |
||||
return; |
||||
} |
||||
} |
||||
|
||||
http::client()->download( |
||||
url, |
||||
[filename, url, processBuffer](const std::string &data, |
||||
const std::string &, |
||||
const std::string &, |
||||
mtx::http::RequestErr err) { |
||||
if (err) { |
||||
nhlog::net()->warn("failed to retrieve media {}: {} {}", |
||||
url, |
||||
err->matrix_error.error, |
||||
static_cast<int>(err->status_code)); |
||||
return; |
||||
} |
||||
|
||||
try { |
||||
QFile file(filename.filePath()); |
||||
|
||||
if (!file.open(QIODevice::WriteOnly)) |
||||
return; |
||||
|
||||
QByteArray ba(data.data(), (int)data.size()); |
||||
file.write(ba); |
||||
file.close(); |
||||
|
||||
QBuffer buf(&ba); |
||||
buf.open(QBuffer::ReadOnly); |
||||
processBuffer(buf); |
||||
} catch (const std::exception &e) { |
||||
nhlog::ui()->warn("Error while saving file to: {}", e.what()); |
||||
} |
||||
}); |
||||
} |
||||
|
||||
QSGNode * |
||||
MxcAnimatedImage::updatePaintNode(QSGNode *oldNode, QQuickItem::UpdatePaintNodeData *) |
||||
{ |
||||
imageDirty = false; |
||||
QSGImageNode *n = static_cast<QSGImageNode *>(oldNode); |
||||
if (!n) |
||||
n = window()->createImageNode(); |
||||
|
||||
// n->setTexture(nullptr);
|
||||
auto img = movie.currentImage(); |
||||
if (!img.isNull()) |
||||
n->setTexture(window()->createTextureFromImage(img)); |
||||
else |
||||
return nullptr; |
||||
|
||||
n->setSourceRect(img.rect()); |
||||
n->setRect(QRect(0, 0, width(), height())); |
||||
n->setFiltering(QSGTexture::Linear); |
||||
n->setMipmapFiltering(QSGTexture::Linear); |
||||
|
||||
return n; |
||||
} |
||||
@ -0,0 +1,79 @@ |
||||
// SPDX-FileCopyrightText: 2021 Nheko Contributors
|
||||
//
|
||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||
|
||||
#pragma once |
||||
|
||||
#include <QBuffer> |
||||
#include <QMovie> |
||||
#include <QObject> |
||||
#include <QQuickItem> |
||||
|
||||
class TimelineModel; |
||||
|
||||
// This is an AnimatedImage, that can draw encrypted images
|
||||
class MxcAnimatedImage : public QQuickItem |
||||
{ |
||||
Q_OBJECT |
||||
Q_PROPERTY(TimelineModel *roomm READ room WRITE setRoom NOTIFY roomChanged REQUIRED) |
||||
Q_PROPERTY(QString eventId READ eventId WRITE setEventId NOTIFY eventIdChanged) |
||||
Q_PROPERTY(bool animatable READ animatable NOTIFY animatableChanged) |
||||
Q_PROPERTY(bool loaded READ loaded NOTIFY loadedChanged) |
||||
public: |
||||
MxcAnimatedImage(QQuickItem *parent = nullptr) |
||||
: QQuickItem(parent) |
||||
{ |
||||
connect(this, &MxcAnimatedImage::eventIdChanged, &MxcAnimatedImage::startDownload); |
||||
connect(this, &MxcAnimatedImage::roomChanged, &MxcAnimatedImage::startDownload); |
||||
connect(&movie, &QMovie::frameChanged, this, &MxcAnimatedImage::newFrame); |
||||
setFlag(QQuickItem::ItemHasContents); |
||||
// setAcceptHoverEvents(true);
|
||||
} |
||||
|
||||
bool animatable() const { return animatable_; } |
||||
bool loaded() const { return buffer.size() > 0; } |
||||
QString eventId() const { return eventId_; } |
||||
TimelineModel *room() const { return room_; } |
||||
void setEventId(QString newEventId) |
||||
{ |
||||
if (eventId_ != newEventId) { |
||||
eventId_ = newEventId; |
||||
emit eventIdChanged(); |
||||
} |
||||
} |
||||
void setRoom(TimelineModel *room) |
||||
{ |
||||
if (room_ != room) { |
||||
room_ = room; |
||||
emit roomChanged(); |
||||
} |
||||
} |
||||
|
||||
QSGNode *updatePaintNode(QSGNode *oldNode, |
||||
QQuickItem::UpdatePaintNodeData *updatePaintNodeData) override; |
||||
|
||||
signals: |
||||
void roomChanged(); |
||||
void eventIdChanged(); |
||||
void animatableChanged(); |
||||
void loadedChanged(); |
||||
|
||||
private slots: |
||||
void startDownload(); |
||||
void newFrame(int frame) |
||||
{ |
||||
currentFrame = frame; |
||||
imageDirty = true; |
||||
update(); |
||||
} |
||||
|
||||
private: |
||||
TimelineModel *room_ = nullptr; |
||||
QString eventId_; |
||||
QString filename_; |
||||
bool animatable_ = false; |
||||
QBuffer buffer; |
||||
QMovie movie; |
||||
int currentFrame = 0; |
||||
bool imageDirty = true; |
||||
}; |
||||
Loading…
Reference in new issue