File indexing completed on 2024-12-08 12:55:20

 // SPDX-FileCopyrightText: 2021 kaniini <https://git.pleroma.social/kaniini>
 // SPDX-FileCopyrightText: 2021 Carl Schwan <carl@carlschwan.eu>
 // SPDX-License-Identifier: GPL-3.0-only
 
 #include "attachmenteditormodel.h"
 #include "account/abstractaccount.h"
 #include "account/account.h"
 #include <QTimer>
 
 AttachmentEditorModel::AttachmentEditorModel(QObject *parent, AbstractAccount *account)
     : QAbstractListModel(parent)
     , m_account(account)
 {
 }
 
 int AttachmentEditorModel::rowCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent);
     return m_attachments.size();
 }
 
 int AttachmentEditorModel::count() const
 {
     return rowCount({});
 }
 
 QVariant AttachmentEditorModel::data(const QModelIndex &index, int role) const
 {
     if (!index.isValid()) {
         return {};
     }
     const int row = index.row();
     const auto &attachment = m_attachments[row];
 
     switch (role) {
     case PreviewRole:
         return attachment->m_preview_url;
     case DescriptionRole:
         return attachment->description();
     }
 
     return {};
 }
 
 QHash<int, QByteArray> AttachmentEditorModel::roleNames() const
 {
     return {
         {PreviewRole, QByteArrayLiteral("preview")},
         {DescriptionRole, QByteArrayLiteral("description")},
     };
 }
 
 QNetworkReply *AttachmentEditorModel::append(const QUrl &filename)
 {
     if (rowCount({}) >= 4) {
         return nullptr;
     }
     return m_account->upload(filename, [=](QNetworkReply *reply) {
         const auto doc = QJsonDocument::fromJson(reply->readAll());
 
         if (!doc.isObject()) {
             return;
         }
 
         beginInsertRows({}, m_attachments.count(), m_attachments.count());
         m_attachments.append(new Attachment{doc.object(), this});
         endInsertRows();
         Q_EMIT countChanged();
     });
 }
 
 void AttachmentEditorModel::appendExisting(Attachment *attachment)
 {
     beginInsertRows({}, m_attachments.count(), m_attachments.count());
     m_attachments.append(attachment);
     endInsertRows();
     Q_EMIT countChanged();
 }
 
 void AttachmentEditorModel::removeAttachment(int row)
 {
     beginRemoveRows({}, row, row);
     m_attachments.removeAt(row);
     endRemoveRows();
     Q_EMIT countChanged();
 }
 
 void AttachmentEditorModel::setDescription(int row, const QString &description)
 {
     auto &attachment = m_attachments[row];
     const auto id = attachment->id();
     attachment->setDescription(description);
 
     const auto attachementUrl = m_account->apiUrl(QStringLiteral("/api/v1/media/%1").arg(id));
     const QJsonObject obj{
         {"description", description},
     };
     const QJsonDocument doc(obj);
     m_account->put(attachementUrl, doc, true, this, nullptr);
     Q_EMIT dataChanged(index(row, 0), index(row, 0), {DescriptionRole});
 }
 
 const QVector<Attachment *> &AttachmentEditorModel::attachments() const
 {
     return m_attachments;
 }