sessionstorefile.cpp

/*
 * SPDX-FileCopyrightText: (C) 2015-2022 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "sessionstorefile.h"
 
#include <Cutelyst/Application>
#include <Cutelyst/Context>
 
#include <QCoreApplication>
#include <QDataStream>
#include <QDir>
#include <QFile>
#include <QLockFile>
#include <QLoggingCategory>
 
using namespace Cutelyst;
using namespace Qt::StringLiterals;
 
Q_LOGGING_CATEGORY(C_SESSION_FILE, "cutelyst.plugin.sessionfile", QtWarningMsg)
 
#define SESSION_STORE_FILE_SAVE u"_c_session_store_file_save"_s
#define SESSION_STORE_FILE_DATA u"_c_session_store_file_data"_s
 
namespace {
QVariantHash loadSessionData(Context *c, const QByteArray &sid);
 
inline QString rootPath()
{
    static QString rootPath =
        QDir::tempPath() + u'/' + QCoreApplication::applicationName() + u"/session/data";
    return rootPath;
}
} // namespace
 
SessionStoreFile::SessionStoreFile(QObject *parent)
    : SessionStore(parent)
{
}
 
SessionStoreFile::~SessionStoreFile()
{
}
 
QVariant SessionStoreFile::getSessionData(Context *c,
                                          const QByteArray &sid,
                                          const QString &key,
                                          const QVariant &defaultValue)
{
    const QVariantHash data = loadSessionData(c, sid);
 
    return data.value(key, defaultValue);
}
 
bool SessionStoreFile::storeSessionData(Context *c,
                                        const QByteArray &sid,
                                        const QString &key,
                                        const QVariant &value)
{
    QVariantHash data = loadSessionData(c, sid);
 
    data.insert(key, value);
    c->setStash(SESSION_STORE_FILE_DATA, data);
    c->setStash(SESSION_STORE_FILE_SAVE, true);
 
    return true;
}
 
bool SessionStoreFile::deleteSessionData(Context *c, const QByteArray &sid, const QString &key)
{
    QVariantHash data = loadSessionData(c, sid);
 
    data.remove(key);
    c->setStash(SESSION_STORE_FILE_DATA, data);
    c->setStash(SESSION_STORE_FILE_SAVE, true);
 
    return true;
}
 
bool SessionStoreFile::deleteExpiredSessions(Context *c, quint64 expires)
{
    Q_UNUSED(c)
    if (!expires) {
        QDir dir(rootPath());
        dir.removeRecursively();
    }
    return true;
}
 
namespace {
QVariantHash loadSessionData(Context *c, const QByteArray &sid)
{
    QVariantHash data;
    const QVariant sessionVariant = c->stash(SESSION_STORE_FILE_DATA);
    if (!sessionVariant.isNull()) {
        data = sessionVariant.toHash();
        return data;
    }
 
    const QString root = rootPath();
 
    auto file = new QFile(root + u'/' + QString::fromLatin1(sid), c);
    if (!file->open(QIODevice::ReadWrite)) {
        if (!QDir().mkpath(root)) {
            qCWarning(C_SESSION_FILE) << "Failed to create path for session object" << root;
            return data;
        }
 
        if (!file->open(QIODevice::ReadWrite)) {
            return data;
        }
    }
 
    // Commit data when Context gets deleted
    QObject::connect(c->app(), &Application::afterDispatch, c, [c, file] {
        if (!c->stash(SESSION_STORE_FILE_SAVE).toBool()) {
            return;
        }
 
        const QVariantHash data = c->stash(SESSION_STORE_FILE_DATA).toHash();
 
        if (data.isEmpty()) {
            QFile::remove(file->fileName());
        } else {
            QLockFile lock(file->fileName() + u".lock");
            if (lock.lock()) {
                QDataStream in(file);
 
                if (file->pos()) {
                    file->seek(0);
                }
 
                in << data;
 
                if (file->pos() < file->size()) {
                    file->resize(file->pos());
                }
 
                file->flush();
                lock.unlock();
            }
        }
    });
 
    // Load data
    QLockFile lock(file->fileName() + u".lock");
    if (lock.lock()) {
        QDataStream in(file);
        in >> data;
        lock.unlock();
    }
 
    c->setStash(SESSION_STORE_FILE_DATA, data);
 
    return data;
}
} // namespace
 
#include "moc_sessionstorefile.cpp"