1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
//===--- Protocol.cpp - Language Server Protocol Implementation -----------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file contains the serialization code for the PDLL specific LSP structs.
//
//===----------------------------------------------------------------------===//
#include "Protocol.h"
#include "llvm/ADT/Hashing.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/Format.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/Support/JSON.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/raw_ostream.h"
using namespace mlir;
using namespace mlir::lsp;
// Helper that doesn't treat `null` and absent fields as failures.
template <typename T>
static bool mapOptOrNull(const llvm::json::Value ¶ms,
llvm::StringLiteral prop, T &out,
llvm::json::Path path) {
const llvm::json::Object *o = params.getAsObject();
assert(o);
// Field is missing or null.
auto *v = o->get(prop);
if (!v || v->getAsNull())
return true;
return fromJSON(*v, out, path.field(prop));
}
//===----------------------------------------------------------------------===//
// PDLLViewOutputParams
//===----------------------------------------------------------------------===//
bool mlir::lsp::fromJSON(const llvm::json::Value &value,
PDLLViewOutputKind &result, llvm::json::Path path) {
if (std::optional<StringRef> str = value.getAsString()) {
if (*str == "ast") {
result = PDLLViewOutputKind::AST;
return true;
}
if (*str == "mlir") {
result = PDLLViewOutputKind::MLIR;
return true;
}
if (*str == "cpp") {
result = PDLLViewOutputKind::CPP;
return true;
}
}
return false;
}
bool mlir::lsp::fromJSON(const llvm::json::Value &value,
PDLLViewOutputParams &result, llvm::json::Path path) {
llvm::json::ObjectMapper o(value, path);
return o && o.map("uri", result.uri) && o.map("kind", result.kind);
}
//===----------------------------------------------------------------------===//
// PDLLViewOutputResult
//===----------------------------------------------------------------------===//
llvm::json::Value mlir::lsp::toJSON(const PDLLViewOutputResult &value) {
return llvm::json::Object{{"output", value.output}};
}
|