108 lines
3.3 KiB
C++
108 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include "Resolver.h"
|
|
#include "Directive.h"
|
|
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
|
|
namespace clice {
|
|
|
|
/// All AST related information needed for language server.
|
|
class ASTInfo {
|
|
public:
|
|
ASTInfo(clang::FileID interested,
|
|
std::unique_ptr<clang::FrontendAction> action,
|
|
std::unique_ptr<clang::CompilerInstance> instance,
|
|
std::optional<TemplateResolver> resolver,
|
|
std::optional<clang::syntax::TokenBuffer> buffer,
|
|
llvm::DenseMap<clang::FileID, Directive> directives) :
|
|
interested(interested), action(std::move(action)), instance(std::move(instance)),
|
|
m_resolver(std::move(resolver)), buffer(std::move(buffer)),
|
|
m_directives(std::move(directives)) {}
|
|
|
|
ASTInfo(const ASTInfo&) = delete;
|
|
|
|
ASTInfo(ASTInfo&&) = default;
|
|
|
|
~ASTInfo() {
|
|
if(action) {
|
|
action->EndSourceFile();
|
|
}
|
|
}
|
|
|
|
public:
|
|
auto& srcMgr() {
|
|
return instance->getSourceManager();
|
|
}
|
|
|
|
auto& pp() {
|
|
return instance->getPreprocessor();
|
|
}
|
|
|
|
auto& context() {
|
|
return instance->getASTContext();
|
|
}
|
|
|
|
auto& sema() {
|
|
return instance->getSema();
|
|
}
|
|
|
|
auto& tokBuf() {
|
|
assert(buffer && "Token buffer is not available");
|
|
return *buffer;
|
|
}
|
|
|
|
auto& resolver() {
|
|
assert(m_resolver && "Template resolver is not available");
|
|
return *m_resolver;
|
|
}
|
|
|
|
auto& directives() {
|
|
return m_directives;
|
|
}
|
|
|
|
auto tu() {
|
|
return instance->getASTContext().getTranslationUnitDecl();
|
|
}
|
|
|
|
/// ============================================================================
|
|
/// Utility Functions
|
|
/// ============================================================================
|
|
|
|
/// @brief Get the length of the token at the given location.
|
|
/// All SourceLocation instances in the Clang AST originate from the start position of tokens,
|
|
/// which helps reduce memory usage. When token length information is needed, a simple lexing
|
|
/// operation based on the start position can be performed.
|
|
auto getTokenLength(clang::SourceLocation loc) {
|
|
return clang::Lexer::MeasureTokenLength(loc, srcMgr(), instance->getLangOpts());
|
|
}
|
|
|
|
/// @brief Get the spelling of the token at the given location.
|
|
llvm::StringRef getTokenSpelling(clang::SourceLocation loc) {
|
|
return llvm::StringRef(srcMgr().getCharacterData(loc), getTokenLength(loc));
|
|
}
|
|
|
|
private:
|
|
/// The interested file ID. For file without header context, it is the main file ID.
|
|
/// For file with header context, it is the file ID of header file.
|
|
clang::FileID interested;
|
|
|
|
/// The frontend action used to build the AST.
|
|
std::unique_ptr<clang::FrontendAction> action;
|
|
|
|
/// Compiler instance, responsible for performing the actual compilation and managing the
|
|
/// lifecycle of all objects during the compilation process.
|
|
std::unique_ptr<clang::CompilerInstance> instance;
|
|
|
|
/// The template resolver used to resolve dependent name.
|
|
std::optional<TemplateResolver> m_resolver;
|
|
|
|
/// Token information collected during the preprocessing.
|
|
std::optional<clang::syntax::TokenBuffer> buffer;
|
|
|
|
/// All diretive information collected during the preprocessing.
|
|
llvm::DenseMap<clang::FileID, Directive> m_directives;
|
|
};
|
|
|
|
} // namespace clice
|