forked from OSchip/llvm-project
51 lines
1.7 KiB
C++
51 lines
1.7 KiB
C++
//===-- Implementation of IncludeFileCommand ------------------------------===//
|
|
//
|
|
// 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 "IncludeFileCommand.h"
|
|
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
|
|
#include <cstdlib>
|
|
|
|
namespace llvm_libc {
|
|
|
|
const char IncludeFileCommand::Name[] = "include_file";
|
|
|
|
void IncludeFileCommand::run(llvm::raw_ostream &OS, const ArgVector &Args,
|
|
llvm::StringRef StdHeader,
|
|
llvm::RecordKeeper &Records,
|
|
const Command::ErrorReporter &Reporter) const {
|
|
if (Args.size() != 1) {
|
|
Reporter.printFatalError(
|
|
"%%include_file command takes exactly 1 argument.");
|
|
}
|
|
|
|
llvm::StringRef IncludeFile = Args[0];
|
|
auto Buffer = llvm::MemoryBuffer::getFileAsStream(IncludeFile);
|
|
if (!Buffer)
|
|
Reporter.printFatalError(llvm::StringRef("Unable to open ") + IncludeFile);
|
|
|
|
llvm::StringRef Content = Buffer.get()->getBuffer();
|
|
|
|
// If the included file has %%begin() command listed, then we want to write
|
|
// only the content after the begin command.
|
|
// TODO: The way the content is split below does not allow space within the
|
|
// the parentheses and, before and after the command. This probably is too
|
|
// strict and should be relaxed.
|
|
auto P = Content.split("\n%%begin()\n");
|
|
if (P.second.empty()) {
|
|
// There was no %%begin in the content.
|
|
OS << P.first;
|
|
} else {
|
|
OS << P.second;
|
|
}
|
|
}
|
|
|
|
} // namespace llvm_libc
|