//===-- DependencyInfo.h --------------------------------------------------===//
//
// 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 "llvm/ADT/StringRef.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/WithColor.h"
#include "llvm/Support/raw_ostream.h"

#include <set>

class DependencyInfo {
public:
  explicit DependencyInfo(std::string DependencyInfoPath)
      : DependencyInfoPath(DependencyInfoPath) {}

  virtual ~DependencyInfo(){};

  virtual void addMissingInput(llvm::StringRef Path) {
    NotFounds.insert(Path.str());
  }

  // Writes the dependencies to specified path. The content is first sorted by
  // OpCode and then by the filename (in alphabetical order).
  virtual void write(llvm::Twine Version,
                     const std::vector<std::string> &Inputs,
                     std::string Output) {
    std::error_code EC;
    llvm::raw_fd_ostream OS(DependencyInfoPath, EC, llvm::sys::fs::OF_None);
    if (EC) {
      llvm::WithColor::defaultErrorHandler(llvm::createStringError(
          EC,
          "failed to write to " + DependencyInfoPath + ": " + EC.message()));
      return;
    }

    auto AddDep = [&OS](DependencyInfoOpcode Opcode,
                        const llvm::StringRef &Path) {
      OS << static_cast<uint8_t>(Opcode);
      OS << Path;
      OS << '\0';
    };

    AddDep(DependencyInfoOpcode::Tool, Version.str());

    // Sort the input by its names.
    std::vector<llvm::StringRef> InputNames;
    InputNames.reserve(Inputs.size());
    for (const auto &F : Inputs)
      InputNames.push_back(F);
    llvm::sort(InputNames);

    for (const auto &In : InputNames)
      AddDep(DependencyInfoOpcode::InputFound, In);

    for (const std::string &F : NotFounds)
      AddDep(DependencyInfoOpcode::InputMissing, F);

    AddDep(DependencyInfoOpcode::Output, Output);
  }

private:
  enum DependencyInfoOpcode : uint8_t {
    Tool = 0x00,
    InputFound = 0x10,
    InputMissing = 0x11,
    Output = 0x40,
  };

  const std::string DependencyInfoPath;
  std::set<std::string> NotFounds;
};

// Subclass to avoid any overhead when not using this feature
class DummyDependencyInfo : public DependencyInfo {
public:
  DummyDependencyInfo() : DependencyInfo("") {}
  void addMissingInput(llvm::StringRef Path) override {}
  void write(llvm::Twine Version, const std::vector<std::string> &Inputs,
             std::string Output) override {}
};