consumer.h

Go to the documentation of this file.

#ifndef REFLECTIVE_RAPIDJSON_CONSUMER_H
#define REFLECTIVE_RAPIDJSON_CONSUMER_H
 
#include "./visitor.h"
 
#include <clang/AST/ASTConsumer.h>
#include <clang/AST/Decl.h>
#include <clang/Basic/DiagnosticIDs.h>
#include <clang/Basic/Version.h>
#include <clang/Frontend/CompilerInstance.h>
#include <clang/Lex/LexDiagnostic.h>
 
namespace ReflectiveRapidJSON {
 
class CodeFactory;

class Consumer : public clang::ASTConsumer {
public:
    Consumer(CodeFactory &factory, clang::CompilerInstance &compilerInstance);
 
    bool HandleTopLevelDecl(clang::DeclGroupRef groupRefDecl) override;
    void HandleTranslationUnit(clang::ASTContext &context) override;
 
    bool shouldParseDecl(clang::Decl *declaration);
 
private:
    void handleNamespaceDefinition(clang::NamespaceDecl *nsDecl);
 
    CodeFactory &m_factory;
    clang::CompilerInstance &m_compilerInstance;
    Visitor m_visitor;
};
 
inline Consumer::Consumer(CodeFactory &factory, clang::CompilerInstance &compilerInstance)
    : m_factory(factory)
    , m_compilerInstance(compilerInstance)
    , m_visitor(factory)
{
}

class DiagConsumer : public clang::DiagnosticConsumer {
public:
    DiagConsumer(std::unique_ptr<DiagnosticConsumer> previous, bool errorResilient);
    unsigned int realErrorCount() const;
 
    void BeginSourceFile(const clang::LangOptions &langOpts, const clang::Preprocessor *pp = nullptr) override;
    void clear() override;
    void EndSourceFile() override;
#if CLANG_VERSION_MAJOR <= 22
    void finish() override;
#endif
    void HandleDiagnostic(clang::DiagnosticsEngine::Level diagLevel, const clang::Diagnostic &info) override;
 
private:
    std::unique_ptr<DiagnosticConsumer> m_proxy;
    unsigned int m_realErrorCount;
    bool m_errorResilient;
};
 
inline DiagConsumer::DiagConsumer(std::unique_ptr<clang::DiagnosticConsumer> previous, bool errorResilient)
    : m_proxy(std::move(previous))
    , m_realErrorCount(0)
    , m_errorResilient(errorResilient)
{
}

inline unsigned int DiagConsumer::realErrorCount() const
{
    return m_realErrorCount;
}
 
} // namespace ReflectiveRapidJSON
 
#endif // REFLECTIVE_RAPIDJSON_CONSUMER_H