mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
26fd10c9e8
* clang-tidy * fix * fix script * test clang compiler * fix clang-tidy rules * Fix windows and other issues. * Fix * Improve information when running check-git-clang-tidy-output.sh on different OS
121 lines
3.9 KiB
C++
121 lines
3.9 KiB
C++
/*
|
|
* Copyright 2016 The Kythe Authors. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* Adapted from
|
|
* https://github.com/xulongwu4/bazel-compilation-database/blob/master/kythe/generate_compile_commands/extract_compile_command.cc
|
|
*/
|
|
#include <fcntl.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <cstdio>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "google/protobuf/io/coded_stream.h"
|
|
#include "google/protobuf/io/zero_copy_stream.h"
|
|
#include "google/protobuf/io/zero_copy_stream_impl.h"
|
|
#include "google/protobuf/stubs/common.h"
|
|
#include "rapidjson/stringbuffer.h"
|
|
#include "rapidjson/writer.h"
|
|
#include "thirdparty/protobuf/extra_actions_base.pb.h"
|
|
|
|
namespace {
|
|
using ::google::protobuf::io::CodedInputStream;
|
|
using ::google::protobuf::io::FileInputStream;
|
|
|
|
bool ReadExtraAction(const std::string &path, blaze::ExtraActionInfo *info,
|
|
blaze::CppCompileInfo *cpp_info) {
|
|
int fd = ::open(path.c_str(), O_RDONLY, S_IREAD | S_IWRITE);
|
|
if (fd < 0) {
|
|
perror("Failed to open input: ");
|
|
return false;
|
|
}
|
|
FileInputStream file_input(fd);
|
|
file_input.SetCloseOnDelete(true);
|
|
|
|
CodedInputStream input(&file_input);
|
|
if (!info->ParseFromCodedStream(&input)) return false;
|
|
if (!info->HasExtension(blaze::CppCompileInfo::cpp_compile_info)) return false;
|
|
*cpp_info = info->GetExtension(blaze::CppCompileInfo::cpp_compile_info);
|
|
return true;
|
|
}
|
|
|
|
std::string JoinCommand(const std::vector<std::string> &command) {
|
|
std::string output;
|
|
if (command.empty()) return output;
|
|
|
|
// TODO(shahms): Deal with embedded spaces and quotes.
|
|
auto iter = command.begin();
|
|
output = *iter++;
|
|
for (; iter != command.end(); ++iter) {
|
|
output += " " + *iter;
|
|
}
|
|
return output;
|
|
}
|
|
|
|
std::string FormatCompilationCommand(const std::string &source_file,
|
|
const std::vector<std::string> &command) {
|
|
rapidjson::StringBuffer buffer;
|
|
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
|
|
writer.StartObject();
|
|
writer.Key("file");
|
|
writer.String(source_file.c_str());
|
|
writer.Key("directory");
|
|
writer.String("@BAZEL_ROOT@");
|
|
writer.Key("command");
|
|
writer.String(JoinCommand(command).c_str());
|
|
writer.EndObject();
|
|
return buffer.GetString();
|
|
}
|
|
} // namespace
|
|
|
|
int main(int argc, char **argv) {
|
|
GOOGLE_PROTOBUF_VERIFY_VERSION;
|
|
if (argc != 3) {
|
|
std::cerr << "usage: " << argv[0] << " extra-action-file output-file" << std::endl;
|
|
return 1;
|
|
}
|
|
std::string extra_action_file = argv[1];
|
|
std::string output_file = argv[2];
|
|
blaze::ExtraActionInfo info;
|
|
blaze::CppCompileInfo cpp_info;
|
|
if (!ReadExtraAction(extra_action_file, &info, &cpp_info)) return 1;
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back(cpp_info.tool());
|
|
args.insert(args.end(), cpp_info.compiler_option().begin(),
|
|
cpp_info.compiler_option().end());
|
|
if (std::find(args.begin(), args.end(), "-c") == args.end()) {
|
|
args.push_back("-c");
|
|
args.push_back(cpp_info.source_file());
|
|
}
|
|
if (std::find(args.begin(), args.end(), "-o") == args.end()) {
|
|
args.push_back("-o");
|
|
args.push_back(cpp_info.output_file());
|
|
}
|
|
|
|
FILE *output = ::fopen(output_file.c_str(), "w");
|
|
if (output == nullptr) {
|
|
perror("Unable to open file for writing: ");
|
|
return 1;
|
|
}
|
|
::fputs(FormatCompilationCommand(cpp_info.source_file(), args).c_str(), output);
|
|
::fclose(output);
|
|
|
|
google::protobuf::ShutdownProtobufLibrary();
|
|
return 0;
|
|
}
|