forked from OSchip/llvm-project
79 lines
2.8 KiB
C++
79 lines
2.8 KiB
C++
//===--- ClangdUnitStore.cpp - A ClangdUnits container -----------*-C++-*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "ClangdUnitStore.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include <algorithm>
|
|
|
|
using namespace clang::clangd;
|
|
using namespace clang;
|
|
|
|
std::shared_ptr<CppFile> CppFileCollection::removeIfPresent(PathRef File) {
|
|
std::lock_guard<std::mutex> Lock(Mutex);
|
|
|
|
auto It = OpenedFiles.find(File);
|
|
if (It == OpenedFiles.end())
|
|
return nullptr;
|
|
|
|
std::shared_ptr<CppFile> Result = It->second;
|
|
OpenedFiles.erase(It);
|
|
return Result;
|
|
}
|
|
|
|
CppFileCollection::RecreateResult
|
|
CppFileCollection::recreateFileIfCompileCommandChanged(
|
|
PathRef File, PathRef ResourceDir, GlobalCompilationDatabase &CDB,
|
|
std::shared_ptr<PCHContainerOperations> PCHs,
|
|
IntrusiveRefCntPtr<vfs::FileSystem> VFS, clangd::Logger &Logger) {
|
|
auto NewCommand = getCompileCommand(CDB, File, ResourceDir);
|
|
|
|
std::lock_guard<std::mutex> Lock(Mutex);
|
|
|
|
RecreateResult Result;
|
|
|
|
auto It = OpenedFiles.find(File);
|
|
if (It == OpenedFiles.end()) {
|
|
It = OpenedFiles
|
|
.try_emplace(File, CppFile::Create(File, std::move(NewCommand),
|
|
std::move(PCHs), Logger))
|
|
.first;
|
|
} else if (!compileCommandsAreEqual(It->second->getCompileCommand(),
|
|
NewCommand)) {
|
|
Result.RemovedFile = std::move(It->second);
|
|
It->second =
|
|
CppFile::Create(File, std::move(NewCommand), std::move(PCHs), Logger);
|
|
}
|
|
Result.FileInCollection = It->second;
|
|
return Result;
|
|
}
|
|
|
|
tooling::CompileCommand
|
|
CppFileCollection::getCompileCommand(GlobalCompilationDatabase &CDB,
|
|
PathRef File, PathRef ResourceDir) {
|
|
std::vector<tooling::CompileCommand> Commands = CDB.getCompileCommands(File);
|
|
if (Commands.empty())
|
|
// Add a fake command line if we know nothing.
|
|
Commands.push_back(getDefaultCompileCommand(File));
|
|
|
|
// Inject the resource dir.
|
|
// FIXME: Don't overwrite it if it's already there.
|
|
Commands.front().CommandLine.push_back("-resource-dir=" +
|
|
std::string(ResourceDir));
|
|
return std::move(Commands.front());
|
|
}
|
|
|
|
bool CppFileCollection::compileCommandsAreEqual(
|
|
tooling::CompileCommand const &LHS, tooling::CompileCommand const &RHS) {
|
|
// tooling::CompileCommand.Output is ignored, it's not relevant for clangd.
|
|
return LHS.Directory == RHS.Directory &&
|
|
LHS.CommandLine.size() == RHS.CommandLine.size() &&
|
|
std::equal(LHS.CommandLine.begin(), LHS.CommandLine.end(),
|
|
RHS.CommandLine.begin());
|
|
}
|