roleacl.cpp

/*
 * Copyright (C) 2014-2017 Daniel Nicoletti <dantti12@gmail.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
#include "roleacl_p.h"

#include "common.h"

#include <Cutelyst/Plugins/Authentication/authentication.h>
#include <Cutelyst/Controller>
#include <Cutelyst/Dispatcher>

using namespace Cutelyst;

RoleACL::RoleACL(QObject *parent) : Component(new RoleACLPrivate, parent)
{
}

Component::Modifiers RoleACL::modifiers() const
{
    return AroundExecute;
}

bool RoleACL::init(Cutelyst::Application *application, const QVariantHash &args)
{
    Q_D(RoleACL);
    Q_UNUSED(application)

    const auto attributes = args.value(QLatin1String("attributes")).value<ParamsMultiMap>();
    d->actionReverse = args.value(QLatin1String("reverse")).toString();

    if (!attributes.contains(QLatin1String("RequiresRole")) && !attributes.contains(QLatin1String("AllowedRole"))) {
        qFatal("RoleACL: Action %s requires at least one RequiresRole or AllowedRole attribute", qPrintable(d->actionReverse));
    } else {
        const QStringList required = attributes.values(QLatin1String("RequiresRole"));
        for (const QString &role : required) {
            d->requiresRole.append(role);
        }

        const QStringList allowed = attributes.values(QLatin1String("AllowedRole"));
        for (const QString &role : allowed) {
            d->allowedRole.append(role);
        }
    }

    auto it = attributes.constFind(QLatin1String("ACLDetachTo"));
    if (it == attributes.constEnd() || it.value().isEmpty()) {
        qFatal("RoleACL: Action %s requires the ACLDetachTo(<action>) attribute", qPrintable(d->actionReverse));
    }
    d->aclDetachTo = it.value();

    return true;
}

bool RoleACL::aroundExecute(Context *c, QStack<Cutelyst::Component *> stack)
{
    Q_D(const RoleACL);

    if (canVisit(c)) {
        return Component::aroundExecute(c, stack);
    }

    c->detach(d->detachTo);

    return false;
}

bool RoleACL::canVisit(Context *c) const
{
    Q_D(const RoleACL);

    const QStringList user_has = Authentication::user(c).value(QStringLiteral("roles")).toStringList();

    const QStringList required = d->requiresRole;
    const QStringList allowed = d->allowedRole;

    if (!required.isEmpty() && !allowed.isEmpty()) {
        for (const QString &role : required) {
            if (!user_has.contains(role)) {
                return false;
            }
        }

        for (const QString &role : allowed) {
            if (user_has.contains(role)) {
                return true;
            }
        }
    }  else if (!required.isEmpty()) {
        for (const QString &role : required) {
            if (!user_has.contains(role)) {
                return false;
            }
        }
        return true;
    } else if (!allowed.isEmpty()) {
        for (const QString &role : allowed) {
            if (user_has.contains(role)) {
                return true;
            }
        }
    }

    return false;
}

bool RoleACL::dispatcherReady(const Dispatcher *dispatcher, Cutelyst::Controller *controller)
{
    Q_D(RoleACL);
    Q_UNUSED(dispatcher)

    d->detachTo = controller->actionFor(d->aclDetachTo);
    if (!d->detachTo) {
        d->detachTo = dispatcher->getActionByPath(d->aclDetachTo);
        if (!d->detachTo) {
            qFatal("RoleACL: Action '%s' requires a valid action set on the ACLDetachTo(%s) attribute",
                   qPrintable(d->actionReverse), qPrintable(d->aclDetachTo));
        }
    }

    return true;
}

#include "moc_roleacl.cpp"