mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-25 18:20:09 +01:00
Jon Ross-Perkins
96d836f965
I'm trying to make the LSP look more like the rest of the toolchain. I'm trying to separate the handlers from the transport layer, and remove the multiple inheritance aspect. Also fixing some style issues, switching to `Map`, and removing an unnecessary copt. I'm using `Context` for the central object for consistency with other portions of the toolchain. In order to get the `handle_*` files working, I'm using LLVM's registry class. It has a quirk that I can't register two registries in the same cpp file, so there are two one-line cpp files. In order to help show the delta (or lack thereof) for actual implementation, I've copied server.cpp over handle_* and undone that in two commits. See the third and fourth commits on the PR history for that.
93 lines
3.5 KiB
C++
93 lines
3.5 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 Parse::NodeKind::IdentifierName.
|
|
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)) {
|
|
if (tree_and_subtrees.tree().node_kind(child) ==
|
|
Parse::NodeKind::IdentifierName) {
|
|
auto token = tree_and_subtrees.tree().node_token(child);
|
|
if (tokens.GetKind(token) == Lex::TokenKind::Identifier) {
|
|
return value_stores.identifiers().Get(tokens.GetIdentifier(token));
|
|
}
|
|
}
|
|
}
|
|
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
|