enumdeclarationprinter.cpp

// Copyright (C) 2022 The Qt Company Ltd.

// Copyright (C) 2019 Alexey Edelev <semlanik@gmail.com>

// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0

// Qt-Security score:critical reason:data-parser

#include "enumdeclarationprinter.h"

#include "generatorcommon.h"

using namespace ::QtProtobuf;

using namespace ::qtprotoccommon;

using namespace ::google::protobuf;

using namespace ::google::protobuf::io;

EnumDeclarationPrinter::EnumDeclarationPrinter(const EnumDescriptor *descriptor,

                                               std::shared_ptr<Printer> printer)

    : DescriptorPrinterBase<EnumDescriptor>(descriptor, std::move(printer),

                                            common::produceEnumTypeMap(descriptor, nullptr))

EnumDeclarationPrinter::~EnumDeclarationPrinter() = default;

void EnumDeclarationPrinter::startEnum()

    m_printer->Print(m_typeMap, CommonTemplates::EnumGadgetDeclarationTemplate());

    static const std::string exportMacro = common::buildExportMacro(false);

    if (!exportMacro.empty())

        m_printer->Print({{ "export_macro", exportMacro }},

                         CommonTemplates::QNamespaceDeclarationTemplate());

    else

        m_printer->Print(m_typeMap, CommonTemplates::QNamespaceDeclarationNoExportTemplate());

    if (Options::instance().hasQml())

        m_printer->Print(m_typeMap, CommonTemplates::QmlNamedElement());

    m_printer->Print("\n");

void EnumDeclarationPrinter::printEnum()

    m_printer->Print(m_typeMap, CommonTemplates::EnumClassDefinitionTemplate());

    Indent();

    int numValues = m_descriptor->value_count();

    for (int j = 0; j < numValues; ++j) {

        const EnumValueDescriptor *valueDescr = m_descriptor->value(j);

        m_printer->Print({ { "enumvalue", common::qualifiedCppName(valueDescr->name()) },

                           { "value", std::to_string(valueDescr->number()) } },

                         CommonTemplates::EnumFieldTemplate());

    Outdent();

    m_printer->Print(CommonTemplates::SemicolonBlockEnclosureTemplate());

    m_printer->Print(m_typeMap, CommonTemplates::QEnumNSTemplate());

    if (!m_typeMap["export_macro"].empty())

        m_printer->Print(m_typeMap, CommonTemplates::EnumRegistrationDeclaration());

    else

        m_printer->Print(m_typeMap, CommonTemplates::EnumRegistrationDeclarationNoExport());

void EnumDeclarationPrinter::encloseEnum()

    m_printer->Print({ { "scope_namespaces", m_typeMap["enum_gadget"] } },

                     CommonTemplates::NamespaceClosingTemplate());