#include "Diagnostics.h"
#include "Preamble.h"
#include "CompileDatabase.h"

namespace clice {

class ParsedAST {
private:
    using Decl = clang::Decl*;
    using PathRef = llvm::StringRef;
    using TokenBuffer = clang::syntax::TokenBuffer;
    using ASTConsumer = std::unique_ptr<clang::ASTConsumer>;

    struct FrontendAction : public clang::ASTFrontendAction {
        ASTConsumer CreateASTConsumer(CompilerInstance& instance, PathRef file) override;
    };

private:
    /// path of translation unit
    PathRef path;
    /// llvm version
    std::string version;
    /// headers part of the tu, when a file is loaded, we will build the preamble, the reuse it.
    Preamble preamble;
    /// some extra info
    std::vector<Decl> topLevelDecls;
    std::vector<Diagnostic> diagnostics;
    /// core members for clang frontend
    uninitialized<TokenBuffer> tokens;
    clang::SyntaxOnlyAction action;
    CompilerInstance instance;

    ParsedAST() = default;

public:
    std::unique_ptr<ParsedAST> build(std::string_view path,
                                     const std::shared_ptr<CompilerInvocation>& invocation,
                                     const std::shared_ptr<Preamble>& preamble);

    auto& Tokens() { return tokens.value; }

    auto& Diagnostics() { return diagnostics; }

    auto& ASTContext() { return instance.getASTContext(); }

    auto& TranslationUnit() { return *instance.getASTContext().getTranslationUnitDecl(); }
};

}  // namespace clice