#include "./codegenerator.h"
#include "./codefactory.h"

#include <c++utilities/application/global.h>

#include <clang/AST/Attr.h>
#include <clang/AST/DeclCXX.h>
#include <clang/Frontend/CompilerInstance.h>
#include <clang/Lex/Lexer.h>

using namespace std;

namespace ReflectiveRapidJSON {

CodeGenerator::~CodeGenerator()
{
}

/*!
 * \brief Adds the specified \a decl to the code generator. The generator might ignore irrelevant declarations.
 */
void CodeGenerator::addDeclaration(clang::Decl *decl)
{
    CPP_UTILITIES_UNUSED(decl)
}

/*!
 * \brief Lazy initializes the source manager.
 * \remarks This method must be called in generate() when subclassing to make use of isOnlyIncluded() and readAnnotation().
 */
void CodeGenerator::lazyInitializeSourceManager() const
{
    if (factory().compilerInstance()) {
        m_sourceManager = &factory().compilerInstance()->getSourceManager();
    }
}

/*!
 * \brief Returns whether the specified \a declaration is only included and not part of the actual file.
 */
bool CodeGenerator::isOnlyIncluded(const clang::Decl *declaration) const
{
    return m_sourceManager
        && m_sourceManager->getFileID(m_sourceManager->getExpansionLoc(declaration->getSourceRange().getBegin())) != m_sourceManager->getMainFileID();
}

/*!
 * \brief Returns the specified \a annotation's text.
 */
std::string_view CodeGenerator::readAnnotation(const clang::Attr *annotation) const
{
    if (!m_sourceManager) {
        return std::string_view();
    }
    auto text = clang::Lexer::getSourceText(sourceManager()->getExpansionRange(annotation->getRange()), *sourceManager(), clang::LangOptions());
    if (text.size() >= 12 && text.startswith("annotate(\"") && text.endswith("\")")) {
        text = text.substr(10, text.size() - 12);
    }
    return text;
}

/*!
 * \brief Returns whether the specified \a record inherits from an instantiation of the specified \a templateClass.
 * \remarks The specified \a record must be defined (not only forward-declared).
 */
bool CodeGenerator::inheritsFromInstantiationOf(clang::CXXRecordDecl *const record, const char *const templateClass)
{
    for (const clang::CXXBaseSpecifier &base : record->bases()) {
        const clang::CXXRecordDecl *const baseDecl = base.getType()->getAsCXXRecordDecl();
        if (baseDecl && baseDecl->getQualifiedNameAsString() == templateClass) {
            return true;
        }
    }
    return false;
}

} // namespace ReflectiveRapidJSON