dc9fb65c4f
This provides a format for externally specifying the include directories for a source file. The format of the tablegen database is exactly the same as that for PDLL, namely it includes the absolute source file name and the set of include directories. The database format is shared to simplify the infra, and also because the format itself is general enough to share. Even if we desire to expand in the future to contain the actual compilation command, nothing there is specific enough that we would need two different formats. As with PDLL, support for generating the database is added to our mlir_tablegen cmake command. Differential Revision: https://reviews.llvm.org/D125441
82 lines
3.1 KiB
C++
82 lines
3.1 KiB
C++
//===- TableGenLspServerMain.cpp - TableGen Language Server main ----------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Tools/tblgen-lsp-server/TableGenLspServerMain.h"
|
|
#include "../lsp-server-support/Logging.h"
|
|
#include "../lsp-server-support/Transport.h"
|
|
#include "LSPServer.h"
|
|
#include "TableGenServer.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Program.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::lsp;
|
|
|
|
LogicalResult mlir::TableGenLspServerMain(int argc, char **argv) {
|
|
llvm::cl::opt<JSONStreamStyle> inputStyle{
|
|
"input-style",
|
|
llvm::cl::desc("Input JSON stream encoding"),
|
|
llvm::cl::values(clEnumValN(JSONStreamStyle::Standard, "standard",
|
|
"usual LSP protocol"),
|
|
clEnumValN(JSONStreamStyle::Delimited, "delimited",
|
|
"messages delimited by `// -----` lines, "
|
|
"with // comment support")),
|
|
llvm::cl::init(JSONStreamStyle::Standard),
|
|
llvm::cl::Hidden,
|
|
};
|
|
llvm::cl::opt<bool> litTest{
|
|
"lit-test",
|
|
llvm::cl::desc(
|
|
"Abbreviation for -input-style=delimited -pretty -log=verbose. "
|
|
"Intended to simplify lit tests"),
|
|
llvm::cl::init(false),
|
|
};
|
|
llvm::cl::opt<Logger::Level> logLevel{
|
|
"log",
|
|
llvm::cl::desc("Verbosity of log messages written to stderr"),
|
|
llvm::cl::values(
|
|
clEnumValN(Logger::Level::Error, "error", "Error messages only"),
|
|
clEnumValN(Logger::Level::Info, "info",
|
|
"High level execution tracing"),
|
|
clEnumValN(Logger::Level::Debug, "verbose", "Low level details")),
|
|
llvm::cl::init(Logger::Level::Info),
|
|
};
|
|
llvm::cl::opt<bool> prettyPrint{
|
|
"pretty",
|
|
llvm::cl::desc("Pretty-print JSON output"),
|
|
llvm::cl::init(false),
|
|
};
|
|
llvm::cl::list<std::string> extraIncludeDirs(
|
|
"tablegen-extra-dir", llvm::cl::desc("Extra directory of include files"),
|
|
llvm::cl::value_desc("directory"), llvm::cl::Prefix);
|
|
llvm::cl::list<std::string> compilationDatabases(
|
|
"tablegen-compilation-database",
|
|
llvm::cl::desc("Compilation YAML databases containing additional "
|
|
"compilation information for .td files"));
|
|
|
|
llvm::cl::ParseCommandLineOptions(argc, argv, "TableGen LSP Language Server");
|
|
|
|
if (litTest) {
|
|
inputStyle = JSONStreamStyle::Delimited;
|
|
logLevel = Logger::Level::Debug;
|
|
prettyPrint = true;
|
|
}
|
|
|
|
// Configure the logger.
|
|
Logger::setLogLevel(logLevel);
|
|
|
|
// Configure the transport used for communication.
|
|
llvm::sys::ChangeStdinToBinary();
|
|
JSONTransport transport(stdin, llvm::outs(), inputStyle, prettyPrint);
|
|
|
|
// Configure the servers and start the main language server.
|
|
TableGenServer::Options options(compilationDatabases, extraIncludeDirs);
|
|
TableGenServer server(options);
|
|
return runTableGenLSPServer(server, transport);
|
|
}
|