mirror of
https://github.com/MaskRay/ccls.git
synced 2024-11-28 18:41:57 +00:00
69 lines
2.3 KiB
C++
69 lines
2.3 KiB
C++
#include "message_handler.h"
|
|
#include "query_utils.h"
|
|
#include "queue_manager.h"
|
|
#include "symbol.h"
|
|
|
|
namespace {
|
|
MethodType kMethodType = "textDocument/documentHighlight";
|
|
|
|
struct In_TextDocumentDocumentHighlight : public RequestInMessage {
|
|
MethodType GetMethodType() const override { return kMethodType; }
|
|
lsTextDocumentPositionParams params;
|
|
};
|
|
MAKE_REFLECT_STRUCT(In_TextDocumentDocumentHighlight, id, params);
|
|
REGISTER_IN_MESSAGE(In_TextDocumentDocumentHighlight);
|
|
|
|
struct Out_TextDocumentDocumentHighlight
|
|
: public lsOutMessage<Out_TextDocumentDocumentHighlight> {
|
|
lsRequestId id;
|
|
std::vector<lsDocumentHighlight> result;
|
|
};
|
|
MAKE_REFLECT_STRUCT(Out_TextDocumentDocumentHighlight, jsonrpc, id, result);
|
|
|
|
struct Handler_TextDocumentDocumentHighlight
|
|
: BaseMessageHandler<In_TextDocumentDocumentHighlight> {
|
|
MethodType GetMethodType() const override { return kMethodType; }
|
|
void Run(In_TextDocumentDocumentHighlight* request) override {
|
|
int file_id;
|
|
QueryFile* file;
|
|
if (!FindFileOrFail(db, project, request->id,
|
|
request->params.textDocument.uri.GetPath(), &file,
|
|
&file_id)) {
|
|
return;
|
|
}
|
|
|
|
WorkingFile* working_file =
|
|
working_files->GetFileByFilename(file->def->path);
|
|
|
|
Out_TextDocumentDocumentHighlight out;
|
|
out.id = request->id;
|
|
|
|
for (SymbolRef sym :
|
|
FindSymbolsAtLocation(working_file, file, request->params.position)) {
|
|
// Found symbol. Return references to highlight.
|
|
EachOccurrence(db, sym, true, [&](Use use) {
|
|
if (use.file_id != file_id)
|
|
return;
|
|
if (std::optional<lsLocation> ls_loc =
|
|
GetLsLocation(db, working_files, use)) {
|
|
lsDocumentHighlight highlight;
|
|
highlight.range = ls_loc->range;
|
|
if (use.role & Role::Write)
|
|
highlight.kind = lsDocumentHighlightKind::Write;
|
|
else if (use.role & Role::Read)
|
|
highlight.kind = lsDocumentHighlightKind::Read;
|
|
else
|
|
highlight.kind = lsDocumentHighlightKind::Text;
|
|
highlight.role = use.role;
|
|
out.result.push_back(highlight);
|
|
}
|
|
});
|
|
break;
|
|
}
|
|
|
|
QueueManager::WriteStdout(kMethodType, out);
|
|
}
|
|
};
|
|
REGISTER_MESSAGE_HANDLER(Handler_TextDocumentDocumentHighlight);
|
|
} // namespace
|