mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Geoff RomerandJon Ross-Perkins
4f10735751
This change splits `NodeKind::IdentifierName` into separate node kinds depending on whether the identifier is followed by parameters, and similarly splits `NameQualifier` based on whether the qualifier has parameters. This enables us to only push a pattern block when it's actually needed, rather than "defensively" pushing one when it might be needed. --------- Co-authored-by: Jon Ross-Perkins <jperkins@google.com>
99 lines
3.7 KiB
C++
99 lines
3.7 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "toolchain/base/shared_value_stores.h"
|
|
#include "toolchain/diagnostics/null_diagnostics.h"
|
|
#include "toolchain/language_server/handle.h"
|
|
#include "toolchain/lex/lex.h"
|
|
#include "toolchain/parse/node_kind.h"
|
|
#include "toolchain/parse/parse.h"
|
|
#include "toolchain/parse/tree_and_subtrees.h"
|
|
#include "toolchain/source/source_buffer.h"
|
|
|
|
namespace Carbon::LanguageServer {
|
|
|
|
// Returns the text of first child of kind IdentifierNameBeforeParams or
|
|
// IdentifierNameNotBeforeParams.
|
|
static auto GetIdentifierName(const SharedValueStores& value_stores,
|
|
const Lex::TokenizedBuffer& tokens,
|
|
const Parse::TreeAndSubtrees& tree_and_subtrees,
|
|
Parse::NodeId node)
|
|
-> std::optional<llvm::StringRef> {
|
|
for (auto child : tree_and_subtrees.children(node)) {
|
|
switch (tree_and_subtrees.tree().node_kind(child)) {
|
|
case Parse::NodeKind::IdentifierNameBeforeParams:
|
|
case Parse::NodeKind::IdentifierNameNotBeforeParams: {
|
|
auto token = tree_and_subtrees.tree().node_token(child);
|
|
if (tokens.GetKind(token) == Lex::TokenKind::Identifier) {
|
|
return value_stores.identifiers().Get(tokens.GetIdentifier(token));
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
auto HandleDocumentSymbol(
|
|
Context& context, const clang::clangd::DocumentSymbolParams& params,
|
|
llvm::function_ref<
|
|
void(llvm::Expected<std::vector<clang::clangd::DocumentSymbol>>)>
|
|
on_done) -> void {
|
|
SharedValueStores value_stores;
|
|
llvm::vfs::InMemoryFileSystem vfs;
|
|
auto lookup = context.files().Lookup(params.textDocument.uri.file());
|
|
CARBON_CHECK(lookup);
|
|
vfs.addFile(lookup.key(), /*mtime=*/0,
|
|
llvm::MemoryBuffer::getMemBufferCopy(lookup.value()));
|
|
|
|
auto source =
|
|
SourceBuffer::MakeFromFile(vfs, lookup.key(), NullDiagnosticConsumer());
|
|
auto tokens = Lex::Lex(value_stores, *source, NullDiagnosticConsumer());
|
|
auto tree = Parse::Parse(tokens, NullDiagnosticConsumer(), nullptr);
|
|
Parse::TreeAndSubtrees tree_and_subtrees(tokens, tree);
|
|
std::vector<clang::clangd::DocumentSymbol> result;
|
|
for (const auto& node : tree.postorder()) {
|
|
clang::clangd::SymbolKind symbol_kind;
|
|
switch (tree.node_kind(node)) {
|
|
case Parse::NodeKind::FunctionDecl:
|
|
case Parse::NodeKind::FunctionDefinitionStart:
|
|
symbol_kind = clang::clangd::SymbolKind::Function;
|
|
break;
|
|
case Parse::NodeKind::Namespace:
|
|
symbol_kind = clang::clangd::SymbolKind::Namespace;
|
|
break;
|
|
case Parse::NodeKind::InterfaceDefinitionStart:
|
|
case Parse::NodeKind::NamedConstraintDefinitionStart:
|
|
symbol_kind = clang::clangd::SymbolKind::Interface;
|
|
break;
|
|
case Parse::NodeKind::ClassDefinitionStart:
|
|
symbol_kind = clang::clangd::SymbolKind::Class;
|
|
break;
|
|
default:
|
|
continue;
|
|
}
|
|
|
|
if (auto name =
|
|
GetIdentifierName(value_stores, tokens, tree_and_subtrees, node)) {
|
|
auto token = tree.node_token(node);
|
|
clang::clangd::Position pos{tokens.GetLineNumber(token) - 1,
|
|
tokens.GetColumnNumber(token) - 1};
|
|
|
|
clang::clangd::DocumentSymbol symbol{
|
|
.name = std::string(*name),
|
|
.kind = symbol_kind,
|
|
.range = {.start = pos, .end = pos},
|
|
.selectionRange = {.start = pos, .end = pos},
|
|
};
|
|
|
|
result.push_back(symbol);
|
|
}
|
|
}
|
|
on_done(result);
|
|
}
|
|
|
|
} // namespace Carbon::LanguageServer
|