1//===- MlirLspServerMain.cpp - MLIR Language Server main ------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "mlir/Tools/mlir-lsp-server/MlirLspServerMain.h"
10#include "LSPServer.h"
11#include "MLIRServer.h"
12#include "mlir/IR/Dialect.h"
13#include "mlir/Tools/lsp-server-support/Logging.h"
14#include "mlir/Tools/lsp-server-support/Transport.h"
15#include "llvm/Support/CommandLine.h"
16#include "llvm/Support/Program.h"
17
18using namespace mlir;
19using namespace mlir::lsp;
20
21LogicalResult mlir::MlirLspServerMain(int argc, char **argv,
22 DialectRegistryFn registry_fn) {
23 llvm::cl::opt<JSONStreamStyle> inputStyle{
24 "input-style",
25 llvm::cl::desc("Input JSON stream encoding"),
26 llvm::cl::values(clEnumValN(JSONStreamStyle::Standard, "standard",
27 "usual LSP protocol"),
28 clEnumValN(JSONStreamStyle::Delimited, "delimited",
29 "messages delimited by `// -----` lines, "
30 "with // comment support")),
31 llvm::cl::init(Val: JSONStreamStyle::Standard),
32 llvm::cl::Hidden,
33 };
34 llvm::cl::opt<bool> litTest{
35 "lit-test",
36 llvm::cl::desc(
37 "Abbreviation for -input-style=delimited -pretty -log=verbose. "
38 "Intended to simplify lit tests"),
39 llvm::cl::init(Val: false),
40 };
41 llvm::cl::opt<Logger::Level> logLevel{
42 "log",
43 llvm::cl::desc("Verbosity of log messages written to stderr"),
44 llvm::cl::values(
45 clEnumValN(Logger::Level::Error, "error", "Error messages only"),
46 clEnumValN(Logger::Level::Info, "info",
47 "High level execution tracing"),
48 clEnumValN(Logger::Level::Debug, "verbose", "Low level details")),
49 llvm::cl::init(Val: Logger::Level::Info),
50 };
51 llvm::cl::opt<bool> prettyPrint{
52 "pretty",
53 llvm::cl::desc("Pretty-print JSON output"),
54 llvm::cl::init(Val: false),
55 };
56 llvm::cl::ParseCommandLineOptions(argc, argv, Overview: "MLIR LSP Language Server");
57
58 if (litTest) {
59 inputStyle = JSONStreamStyle::Delimited;
60 logLevel = Logger::Level::Debug;
61 prettyPrint = true;
62 }
63
64 // Configure the logger.
65 Logger::setLogLevel(logLevel);
66
67 // Configure the transport used for communication.
68 llvm::sys::ChangeStdinToBinary();
69 JSONTransport transport(stdin, llvm::outs(), inputStyle, prettyPrint);
70
71 // Register the additionally supported URI schemes for the MLIR server.
72 URIForFile::registerSupportedScheme(scheme: "mlir.bytecode-mlir");
73
74 // Configure the servers and start the main language server.
75 MLIRServer server(registry_fn);
76 return runMlirLSPServer(server, transport);
77}
78
79llvm::LogicalResult mlir::MlirLspServerMain(int argc, char **argv,
80 DialectRegistry &registry) {
81 auto registry_fn =
82 [&registry](const lsp::URIForFile &uri) -> DialectRegistry & {
83 return registry;
84 };
85 return MlirLspServerMain(argc, argv, registry_fn);
86}
87

source code of mlir/lib/Tools/mlir-lsp-server/MlirLspServerMain.cpp