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 | |
18 | using namespace mlir; |
19 | using namespace mlir::lsp; |
20 | |
21 | LogicalResult mlir::MlirLspServerMain(int argc, char **argv, |
22 | DialectRegistry ®istry) { |
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); |
76 | return runMlirLSPServer(server, transport); |
77 | } |
78 | |