mirror of
https://github.com/RGBCube/serenity
synced 2025-05-23 13:25:08 +00:00

Previously, to get the globally available declarations in a document (including declarations from headers), we would have to recursively walk the #include tree and get the declarations of each included document. To improve upon this, we now store a HashTable of globally available declaration from included header files in each document, and populate it when we first process the document. Before this, invoking simple autocomplete actions in code documents that had a very large #include tree (e.g when <LibGUI/Widget.h> was included) hang the CppLanguageServer process and used 100% CPU until the process ran out of memory. Now, the autocomplete request in that situation returns immediately :^)
101 lines
4 KiB
C++
101 lines
4 KiB
C++
/*
|
|
* Copyright (c) 2021, Itamar S. <itamar8910@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/Function.h>
|
|
#include <AK/String.h>
|
|
#include <AK/Vector.h>
|
|
#include <DevTools/HackStudio/AutoCompleteResponse.h>
|
|
#include <DevTools/HackStudio/LanguageServers/AutoCompleteEngine.h>
|
|
#include <DevTools/HackStudio/LanguageServers/FileDB.h>
|
|
#include <LibCpp/AST.h>
|
|
#include <LibCpp/Parser.h>
|
|
#include <LibCpp/Preprocessor.h>
|
|
#include <LibGUI/TextPosition.h>
|
|
|
|
namespace LanguageServers::Cpp {
|
|
|
|
using namespace ::Cpp;
|
|
|
|
class ParserAutoComplete : public AutoCompleteEngine {
|
|
public:
|
|
ParserAutoComplete(ClientConnection&, const FileDB& filedb);
|
|
|
|
virtual Vector<GUI::AutocompleteProvider::Entry> get_suggestions(const String& file, const GUI::TextPosition& autocomplete_position) override;
|
|
virtual void on_edit(const String& file) override;
|
|
virtual void file_opened([[maybe_unused]] const String& file) override;
|
|
virtual Optional<GUI::AutocompleteProvider::ProjectLocation> find_declaration_of(const String& filename, const GUI::TextPosition& identifier_position) override;
|
|
|
|
private:
|
|
struct DocumentData {
|
|
const String& filename() const { return m_filename; }
|
|
const String& text() const { return m_text; }
|
|
const Preprocessor& preprocessor() const
|
|
{
|
|
VERIFY(m_preprocessor);
|
|
return *m_preprocessor;
|
|
}
|
|
Preprocessor& preprocessor()
|
|
{
|
|
VERIFY(m_preprocessor);
|
|
return *m_preprocessor;
|
|
}
|
|
const Parser& parser() const
|
|
{
|
|
VERIFY(m_parser);
|
|
return *m_parser;
|
|
}
|
|
Parser& parser()
|
|
{
|
|
VERIFY(m_parser);
|
|
return *m_parser;
|
|
}
|
|
|
|
String m_filename;
|
|
String m_text;
|
|
OwnPtr<Preprocessor> m_preprocessor;
|
|
OwnPtr<Parser> m_parser;
|
|
|
|
// FIXME: This HashTable must be re-computed if a declaration from a header file is modified
|
|
HashTable<NonnullRefPtr<Declaration>> m_declarations_from_headers;
|
|
};
|
|
|
|
Vector<GUI::AutocompleteProvider::Entry> autocomplete_property(const DocumentData&, const MemberExpression&, const String partial_text) const;
|
|
Vector<GUI::AutocompleteProvider::Entry> autocomplete_name(const DocumentData&, const ASTNode&, const String& partial_text) const;
|
|
String type_of(const DocumentData&, const Expression&) const;
|
|
String type_of_property(const DocumentData&, const Identifier&) const;
|
|
String type_of_variable(const Identifier&) const;
|
|
bool is_property(const ASTNode&) const;
|
|
bool is_empty_property(const DocumentData&, const ASTNode&, const Position& autocomplete_position) const;
|
|
RefPtr<Declaration> find_declaration_of(const DocumentData&, const ASTNode&) const;
|
|
NonnullRefPtrVector<Declaration> get_available_declarations(const DocumentData&, const ASTNode&) const;
|
|
|
|
struct PropertyInfo {
|
|
StringView name;
|
|
RefPtr<Type> type;
|
|
};
|
|
Vector<PropertyInfo> properties_of_type(const DocumentData& document, const String& type) const;
|
|
NonnullRefPtrVector<Declaration> get_global_declarations_including_headers(const DocumentData& document) const;
|
|
NonnullRefPtrVector<Declaration> get_global_declarations(const ASTNode& node) const;
|
|
|
|
const DocumentData* get_document_data(const String& file) const;
|
|
const DocumentData* get_or_create_document_data(const String& file);
|
|
void set_document_data(const String& file, OwnPtr<DocumentData>&& data);
|
|
|
|
OwnPtr<DocumentData> create_document_data_for(const String& file);
|
|
String document_path_from_include_path(const StringView& include_path) const;
|
|
void update_declared_symbols(DocumentData&);
|
|
GUI::AutocompleteProvider::DeclarationType type_of_declaration(const Declaration&);
|
|
String scope_of_declaration(const Declaration&);
|
|
Optional<GUI::AutocompleteProvider::ProjectLocation> find_preprocessor_definition(const DocumentData&, const GUI::TextPosition&);
|
|
|
|
OwnPtr<DocumentData> create_document_data(String&& text, const String& filename);
|
|
|
|
HashMap<String, OwnPtr<DocumentData>> m_documents;
|
|
};
|
|
|
|
}
|