redai/src/ArtModel.cpp

// SPDX-FileCopyrightText: 2023 Joshua Goins <josh@redstrate.com>
//
// SPDX-License-Identifier: GPL-3.0-or-later

#include "ArtModel.h"

#include <QDirIterator>
#include <QFile>
#include <QFileInfo>
#include <QIcon>
#include <QJsonDocument>
#include <QJsonObject>
#include <QPixmap>
#include <QtConcurrent>

ArtModel::ArtModel(const QString& definitionDirectory, const QString& assetDirectory) : QAbstractTableModel() {
	piecesFuture = new QFutureWatcher<ArtPiece>(this);
	connect(piecesFuture, &QFutureWatcher<ArtPiece>::resultReadyAt, this, &ArtModel::pieceFinished);
	connect(piecesFuture, &QFutureWatcher<ArtPiece>::finished, this, &ArtModel::finished);

	struct PieceInformation {
		QString definition;
		QString asset;
	};
	QVector<PieceInformation> pieceList;
	QDirIterator it(definitionDirectory);
	while (it.hasNext()) {
		QFileInfo info(it.next());
		if(!info.isFile()) {
			continue;
		}

		pieceList.push_back(PieceInformation{QStringLiteral("%1/%2").arg(definitionDirectory, info.baseName()), QStringLiteral("%1/%2").arg(assetDirectory, info.baseName())});

		beginInsertRows(QModelIndex(), m_artPieces.size(), m_artPieces.size() + 1);

		m_artPieces.push_back(ArtPiece{});

		endInsertRows();
	}

	std::function<ArtPiece(const PieceInformation& info)> loadPiece = [this](const PieceInformation& info) -> ArtPiece {
		ArtPiece p;
		loadData(p, info.definition, info.asset);

		p.image.load(p.filename);
		p.thumbnail = QPixmap::fromImage(p.image).scaled(100, 100, Qt::AspectRatioMode::KeepAspectRatio).toImage();

		return p;
	};

	piecesFuture->setFuture(QtConcurrent::mapped(pieceList, loadPiece));
}

int ArtModel::rowCount(const QModelIndex &parent) const {
	return m_artPieces.size();
}

int ArtModel::columnCount(const QModelIndex &parent) const {
	return 4;
}

QVariant ArtModel::data(const QModelIndex &index, int role) const {
	if (!index.isValid())
		return {};

	if (role == Qt::DisplayRole) {
		switch(index.column()) {
			case 0:
				return m_artPieces[index.row()].filename;
			case 1:
				return {};
			case 2:
				return m_artPieces[index.row()].title;
			case 3:
				return m_artPieces[index.row()].hasAltText;
		}
	} else if (role == Qt::UserRole) {
		return m_artPieces[index.row()].object;
	} else if (role == Qt::DecorationRole) {
		switch(index.column()) {
			case 1:
				return m_artPieces[index.row()].thumbnail;
			case 3:
				return m_artPieces[index.row()].hasAltText  ? QIcon::fromTheme(QStringLiteral("emblem-checked")) : QIcon::fromTheme(QStringLiteral("emblem-error"));
		}
	} else if (role == Qt::UserRole + 1) {
		return m_artPieces[index.row()].jsonFilename;
	}

	return {};
}

QVariant ArtModel::headerData(int section, Qt::Orientation orientation, int role) const {
	if (orientation == Qt::Orientation::Horizontal && role == Qt::DisplayRole) {
		switch(section) {
			case 0:
				return QStringLiteral("Filename");
			case 1:
				return QStringLiteral("Image");
			case 2:
				return QStringLiteral("Title");
			case 3:
				return QStringLiteral("Has Alt Text");
			default:
				Q_UNREACHABLE();
		}
	}

	return QAbstractTableModel::headerData(section, orientation, role);
}


void ArtModel::loadData(ArtPiece& piece, const QString& filename, const QString& assetFilename) {
	piece.jsonFilename = filename + ".json";
	piece.filename = assetFilename;

	QFile artFile(piece.jsonFilename);
	artFile.open(QFile::ReadOnly);
	QJsonDocument artJson = QJsonDocument::fromJson(artFile.readAll());

	if(artJson[QStringLiteral("date")].toString().contains(QStringLiteral("-"))) {
		piece.date = QDate::fromString(artJson[QStringLiteral("date")].toString(), QStringLiteral("yyyy-MM-dd"));
	} else {
		piece.date = QDate::fromString(artJson[QStringLiteral("date")].toString(), QStringLiteral("yyyy"));
	}

	if (artJson.object().contains(QStringLiteral("title"))) {
		piece.title = artJson.object()[QStringLiteral("title")].toString();
	}

	if(artJson.object().contains(QStringLiteral("alt_text"))) {
		piece.hasAltText = true;
	}
}
void ArtModel::pieceFinished(int row) {
	m_artPieces[row] = piecesFuture->resultAt(row);

	Q_EMIT dataChanged(index(row, 0), index(row + 1, 0));
}

void ArtModel::finished() {
	std::sort(m_artPieces.begin(), m_artPieces.end(), [](ArtPiece& a, ArtPiece& b)
			  {
				  return a.date > b.date;
			  });

	Q_EMIT dataChanged(index(0, 0), index(m_artPieces.size(), 0));
}