1 //===- TableGenLspServerMain.cpp - TableGen 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/tblgen-lsp-server/TableGenLspServerMain.h"
10 #include "../lsp-server-support/Logging.h"
11 #include "../lsp-server-support/Transport.h"
12 #include "LSPServer.h"
13 #include "TableGenServer.h"
14 #include "llvm/Support/CommandLine.h"
15 #include "llvm/Support/Program.h"
16 
17 using namespace mlir;
18 using namespace mlir::lsp;
19 
20 LogicalResult mlir::TableGenLspServerMain(int argc, char **argv) {
21   llvm::cl::opt<JSONStreamStyle> inputStyle{
22       "input-style",
23       llvm::cl::desc("Input JSON stream encoding"),
24       llvm::cl::values(clEnumValN(JSONStreamStyle::Standard, "standard",
25                                   "usual LSP protocol"),
26                        clEnumValN(JSONStreamStyle::Delimited, "delimited",
27                                   "messages delimited by `// -----` lines, "
28                                   "with // comment support")),
29       llvm::cl::init(JSONStreamStyle::Standard),
30       llvm::cl::Hidden,
31   };
32   llvm::cl::opt<bool> litTest{
33       "lit-test",
34       llvm::cl::desc(
35           "Abbreviation for -input-style=delimited -pretty -log=verbose. "
36           "Intended to simplify lit tests"),
37       llvm::cl::init(false),
38   };
39   llvm::cl::opt<Logger::Level> logLevel{
40       "log",
41       llvm::cl::desc("Verbosity of log messages written to stderr"),
42       llvm::cl::values(
43           clEnumValN(Logger::Level::Error, "error", "Error messages only"),
44           clEnumValN(Logger::Level::Info, "info",
45                      "High level execution tracing"),
46           clEnumValN(Logger::Level::Debug, "verbose", "Low level details")),
47       llvm::cl::init(Logger::Level::Info),
48   };
49   llvm::cl::opt<bool> prettyPrint{
50       "pretty",
51       llvm::cl::desc("Pretty-print JSON output"),
52       llvm::cl::init(false),
53   };
54   llvm::cl::list<std::string> extraIncludeDirs(
55       "tablegen-extra-dir", llvm::cl::desc("Extra directory of include files"),
56       llvm::cl::value_desc("directory"), llvm::cl::Prefix);
57   llvm::cl::list<std::string> compilationDatabases(
58       "tablegen-compilation-database",
59       llvm::cl::desc("Compilation YAML databases containing additional "
60                      "compilation information for .td files"));
61 
62   llvm::cl::ParseCommandLineOptions(argc, argv, "TableGen LSP Language Server");
63 
64   if (litTest) {
65     inputStyle = JSONStreamStyle::Delimited;
66     logLevel = Logger::Level::Debug;
67     prettyPrint = true;
68   }
69 
70   // Configure the logger.
71   Logger::setLogLevel(logLevel);
72 
73   // Configure the transport used for communication.
74   llvm::sys::ChangeStdinToBinary();
75   JSONTransport transport(stdin, llvm::outs(), inputStyle, prettyPrint);
76 
77   // Configure the servers and start the main language server.
78   TableGenServer::Options options(compilationDatabases, extraIncludeDirs);
79   TableGenServer server(options);
80   return runTableGenLSPServer(server, transport);
81 }
82