clangd/ClangdUnitStore.cpp

//===--- 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());
}