revng-revng/tools/model/apply/Main.cpp

/// \file Main.cpp
/// \brief

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include <string>

#include "llvm/Support/CommandLine.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/ToolOutputFile.h"
#include "llvm/Support/raw_ostream.h"

#include "revng/Model/ToolHelpers.h"
#include "revng/Support/MetaAddress/YAMLTraits.h"
#include "revng/Support/YAMLTraits.h"
#include "revng/TupleTree/TupleTreeDiff.h"

using namespace llvm;

static cl::OptionCategory ThisToolCategory("Tool options", "");

static cl::opt<std::string> PathModel(cl::Positional,
                                      cl::cat(ThisToolCategory),
                                      cl::desc("<input model>"),
                                      cl::init("-"),
                                      cl::value_desc("model"));

static cl::opt<std::string> DiffPath(cl::Positional,
                                     cl::cat(ThisToolCategory),
                                     cl::desc("<model diff>"),
                                     cl::value_desc("model"));

static ModelOutputOptions<false> Options(ThisToolCategory);

int main(int Argc, char *Argv[]) {
  cl::HideUnrelatedOptions({ &ThisToolCategory });
  cl::ParseCommandLineOptions(Argc, Argv);

  ExitOnError ExitOnError;

  auto Model = ModelInModule::load(PathModel);
  if (not Model)
    ExitOnError(Model.takeError());

  auto Diff = deserializeFile<TupleTreeDiff<model::Binary>>(DiffPath);
  if (not Diff)
    ExitOnError(Diff.takeError());

  Diff->apply(Model->getModel());

  auto DesiredOutput = Options.getDesiredOutput(Model->hasModule());
  ExitOnError(Model->save(Options.getPath(), DesiredOutput));

  return EXIT_SUCCESS;
}