ladybird/Libraries/LibWeb/ContentSecurityPolicy/Directives/DirectiveFactory.cpp

/*
 * Copyright (c) 2025, Luke Wilde <luke@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <LibJS/Runtime/Realm.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ChildSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ConnectSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/DefaultSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/Directive.h>
#include <LibWeb/ContentSecurityPolicy/Directives/DirectiveFactory.h>
#include <LibWeb/ContentSecurityPolicy/Directives/FontSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/FrameSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ImageSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ManifestSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/MediaSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/Names.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ObjectSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ScriptSourceAttributeDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ScriptSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/ScriptSourceElementDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/StyleSourceAttributeDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/StyleSourceDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/StyleSourceElementDirective.h>
#include <LibWeb/ContentSecurityPolicy/Directives/WorkerSourceDirective.h>

namespace Web::ContentSecurityPolicy::Directives {

GC::Ref<Directive> create_directive(GC::Heap& heap, String name, Vector<String> value)
{
    if (name == Names::ChildSrc)
        return heap.allocate<ChildSourceDirective>(move(name), move(value));

    if (name == Names::ConnectSrc)
        return heap.allocate<ConnectSourceDirective>(move(name), move(value));

    if (name == Names::DefaultSrc)
        return heap.allocate<DefaultSourceDirective>(move(name), move(value));

    if (name == Names::FontSrc)
        return heap.allocate<FontSourceDirective>(move(name), move(value));

    if (name == Names::FrameSrc)
        return heap.allocate<FrameSourceDirective>(move(name), move(value));

    if (name == Names::ImgSrc)
        return heap.allocate<ImageSourceDirective>(move(name), move(value));

    if (name == Names::ManifestSrc)
        return heap.allocate<ManifestSourceDirective>(move(name), move(value));

    if (name == Names::MediaSrc)
        return heap.allocate<MediaSourceDirective>(move(name), move(value));

    if (name == Names::ObjectSrc)
        return heap.allocate<ObjectSourceDirective>(move(name), move(value));

    if (name == Names::ScriptSrcAttr)
        return heap.allocate<ScriptSourceAttributeDirective>(move(name), move(value));

    if (name == Names::ScriptSrc)
        return heap.allocate<ScriptSourceDirective>(move(name), move(value));

    if (name == Names::ScriptSrcElem)
        return heap.allocate<ScriptSourceElementDirective>(move(name), move(value));

    if (name == Names::StyleSrcAttr)
        return heap.allocate<StyleSourceAttributeDirective>(move(name), move(value));

    if (name == Names::StyleSrc)
        return heap.allocate<StyleSourceDirective>(move(name), move(value));

    if (name == Names::StyleSrcElem)
        return heap.allocate<StyleSourceElementDirective>(move(name), move(value));

    if (name == Names::WorkerSrc)
        return heap.allocate<WorkerSourceDirective>(move(name), move(value));

    return heap.allocate<Directive>(move(name), move(value));
}

}