carbon-lang/toolchain/driver/driver_main.cpp

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include <cstdlib>

#include "common/bazel_working_dir.h"
#include "common/init_llvm.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/Program.h"
#include "toolchain/driver/driver.h"

auto main(int argc, char** argv) -> int {
  Carbon::InitLLVM init_llvm(argc, argv);

  if (argc < 1) {
    return EXIT_FAILURE;
  }

  // Find the executable without resolving symlinks. Do a PATH lookup if argv[0]
  // isn't a valid path.
  llvm::SmallString<128> exe_path(argv[0]);
  if (!llvm::sys::fs::exists(exe_path)) {
    if (llvm::ErrorOr<std::string> path =
            llvm::sys::findProgramByName(exe_path)) {
      exe_path = *path;
    }
  }

  Carbon::SetWorkingDirForBazel();

  // Printing to stderr should flush stdout. This is most noticeable when stderr
  // is piped to stdout.
  llvm::errs().tie(&llvm::outs());

  llvm::SmallVector<llvm::StringRef> args(argv + 1, argv + argc);
  auto fs = llvm::vfs::getRealFileSystem();

  // Construct the data directory relative to the executable location.
  llvm::SmallString<256> data_dir(llvm::sys::path::parent_path(exe_path));
  llvm::sys::path::append(data_dir, llvm::sys::path::Style::posix,
                          "carbon.runfiles/_main/");

  Carbon::Driver driver(*fs, data_dir, llvm::outs(), llvm::errs());
  bool success = driver.RunCommand(args).success;
  return success ? EXIT_SUCCESS : EXIT_FAILURE;
}