54 lines
1.5 KiB
C++
54 lines
1.5 KiB
C++
#include "document_symbol.h"
|
|
|
|
#include "third-party/json.hpp"
|
|
|
|
using json = nlohmann::json;
|
|
|
|
std::optional<json> ir_symbols(Workspace& workspace, LSPSpec::DocumentSymbolParams params) {
|
|
json symbols = json::array();
|
|
auto maybe_tracked_file = workspace.get_tracked_ir_file(params.m_textDocument.m_uri);
|
|
if (!maybe_tracked_file) {
|
|
return symbols;
|
|
}
|
|
|
|
const auto& tracked_file = maybe_tracked_file.value().get();
|
|
for (const auto& symbol : tracked_file.m_symbols) {
|
|
symbols.push_back(symbol);
|
|
}
|
|
|
|
return symbols;
|
|
}
|
|
|
|
std::optional<json> og_symbols(Workspace& workspace, LSPSpec::DocumentSymbolParams params) {
|
|
json symbols = json::array();
|
|
auto maybe_tracked_file = workspace.get_tracked_og_file(params.m_textDocument.m_uri);
|
|
if (!maybe_tracked_file) {
|
|
return symbols;
|
|
}
|
|
|
|
const auto& tracked_file = maybe_tracked_file.value().get();
|
|
for (const auto& symbol : tracked_file.m_symbols) {
|
|
symbols.push_back(symbol);
|
|
}
|
|
|
|
return symbols;
|
|
}
|
|
|
|
namespace lsp_handlers {
|
|
|
|
std::optional<json> document_symbols(Workspace& workspace, int /*id*/, json params) {
|
|
auto converted_params = params.get<LSPSpec::DocumentSymbolParams>();
|
|
const auto file_type =
|
|
workspace.determine_filetype_from_uri(converted_params.m_textDocument.m_uri);
|
|
|
|
if (file_type == Workspace::FileType::OpenGOALIR) {
|
|
return ir_symbols(workspace, converted_params);
|
|
} else if (file_type == Workspace::FileType::OpenGOAL) {
|
|
return og_symbols(workspace, converted_params);
|
|
}
|
|
|
|
return json::array();
|
|
}
|
|
|
|
} // namespace lsp_handlers
|