forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			93 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			93 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===- lib/Tooling/Execution.cpp - Standalone clang action execution. -----===//
 | |
| //
 | |
| // 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 "clang/Tooling/StandaloneExecution.h"
 | |
| #include "clang/Tooling/ToolExecutorPluginRegistry.h"
 | |
| 
 | |
| namespace clang {
 | |
| namespace tooling {
 | |
| 
 | |
| static llvm::Error make_string_error(const llvm::Twine &Message) {
 | |
|   return llvm::make_error<llvm::StringError>(Message,
 | |
|                                              llvm::inconvertibleErrorCode());
 | |
| }
 | |
| 
 | |
| const char *StandaloneToolExecutor::ExecutorName = "StandaloneToolExecutor";
 | |
| 
 | |
| static ArgumentsAdjuster getDefaultArgumentsAdjusters() {
 | |
|   return combineAdjusters(
 | |
|       getClangStripOutputAdjuster(),
 | |
|       combineAdjusters(getClangSyntaxOnlyAdjuster(),
 | |
|                        getClangStripDependencyFileAdjuster()));
 | |
| }
 | |
| 
 | |
| StandaloneToolExecutor::StandaloneToolExecutor(
 | |
|     const CompilationDatabase &Compilations,
 | |
|     llvm::ArrayRef<std::string> SourcePaths,
 | |
|     IntrusiveRefCntPtr<llvm::vfs::FileSystem> BaseFS,
 | |
|     std::shared_ptr<PCHContainerOperations> PCHContainerOps)
 | |
|     : Tool(Compilations, SourcePaths, std::move(PCHContainerOps),
 | |
|            std::move(BaseFS)),
 | |
|       Context(&Results), ArgsAdjuster(getDefaultArgumentsAdjusters()) {
 | |
|   // Use self-defined default argument adjusters instead of the default
 | |
|   // adjusters that come with the old `ClangTool`.
 | |
|   Tool.clearArgumentsAdjusters();
 | |
| }
 | |
| 
 | |
| StandaloneToolExecutor::StandaloneToolExecutor(
 | |
|     CommonOptionsParser Options,
 | |
|     std::shared_ptr<PCHContainerOperations> PCHContainerOps)
 | |
|     : OptionsParser(std::move(Options)),
 | |
|       Tool(OptionsParser->getCompilations(), OptionsParser->getSourcePathList(),
 | |
|            std::move(PCHContainerOps)),
 | |
|       Context(&Results), ArgsAdjuster(getDefaultArgumentsAdjusters()) {
 | |
|   Tool.clearArgumentsAdjusters();
 | |
| }
 | |
| 
 | |
| llvm::Error StandaloneToolExecutor::execute(
 | |
|     llvm::ArrayRef<
 | |
|         std::pair<std::unique_ptr<FrontendActionFactory>, ArgumentsAdjuster>>
 | |
|         Actions) {
 | |
|   if (Actions.empty())
 | |
|     return make_string_error("No action to execute.");
 | |
| 
 | |
|   if (Actions.size() != 1)
 | |
|     return make_string_error(
 | |
|         "Only support executing exactly 1 action at this point.");
 | |
| 
 | |
|   auto &Action = Actions.front();
 | |
|   Tool.appendArgumentsAdjuster(Action.second);
 | |
|   Tool.appendArgumentsAdjuster(ArgsAdjuster);
 | |
|   if (Tool.run(Action.first.get()))
 | |
|     return make_string_error("Failed to run action.");
 | |
| 
 | |
|   return llvm::Error::success();
 | |
| }
 | |
| 
 | |
| class StandaloneToolExecutorPlugin : public ToolExecutorPlugin {
 | |
| public:
 | |
|   llvm::Expected<std::unique_ptr<ToolExecutor>>
 | |
|   create(CommonOptionsParser &OptionsParser) override {
 | |
|     if (OptionsParser.getSourcePathList().empty())
 | |
|       return make_string_error(
 | |
|           "[StandaloneToolExecutorPlugin] No positional argument found.");
 | |
|     return std::make_unique<StandaloneToolExecutor>(std::move(OptionsParser));
 | |
|   }
 | |
| };
 | |
| 
 | |
| static ToolExecutorPluginRegistry::Add<StandaloneToolExecutorPlugin>
 | |
|     X("standalone", "Runs FrontendActions on a set of files provided "
 | |
|                     "via positional arguments.");
 | |
| 
 | |
| // This anchor is used to force the linker to link in the generated object file
 | |
| // and thus register the plugin.
 | |
| volatile int StandaloneToolExecutorAnchorSource = 0;
 | |
| 
 | |
| } // end namespace tooling
 | |
| } // end namespace clang
 |