//===--- CommandLineArgs.cpp ----------------------------------------------===//
//
// 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 "clang/Testing/CommandLineArgs.h"
#include "llvm/MC/TargetRegistry.h"
#include "llvm/Support/ErrorHandling.h"

namespace clang {

std::vector<std::string> getCommandLineArgsForTesting(TestLanguage Lang) {
  std::vector<std::string> Args;
  // Test with basic arguments.
  switch (Lang) {
  case Lang_C89:
    Args = {"-x", "c", "-std=c89"};
    break;
  case Lang_C99:
    Args = {"-x", "c", "-std=c99"};
    break;
  case Lang_CXX03:
    Args = {"-std=c++03", "-frtti"};
    break;
  case Lang_CXX11:
    Args = {"-std=c++11", "-frtti"};
    break;
  case Lang_CXX14:
    Args = {"-std=c++14", "-frtti"};
    break;
  case Lang_CXX17:
    Args = {"-std=c++17", "-frtti"};
    break;
  case Lang_CXX20:
    Args = {"-std=c++20", "-frtti"};
    break;
  case Lang_OBJC:
    Args = {"-x", "objective-c", "-frtti", "-fobjc-nonfragile-abi"};
    break;
  case Lang_OBJCXX:
    Args = {"-x", "objective-c++", "-frtti"};
    break;
  case Lang_OpenCL:
    llvm_unreachable("Not implemented yet!");
  }
  return Args;
}

std::vector<std::string> getCC1ArgsForTesting(TestLanguage Lang) {
  std::vector<std::string> Args;
  switch (Lang) {
  case Lang_C89:
    Args = {"-xc", "-std=c89"};
    break;
  case Lang_C99:
    Args = {"-xc", "-std=c99"};
    break;
  case Lang_CXX03:
    Args = {"-std=c++03"};
    break;
  case Lang_CXX11:
    Args = {"-std=c++11"};
    break;
  case Lang_CXX14:
    Args = {"-std=c++14"};
    break;
  case Lang_CXX17:
    Args = {"-std=c++17"};
    break;
  case Lang_CXX20:
    Args = {"-std=c++20"};
    break;
  case Lang_OBJC:
    Args = {"-xobjective-c"};
    break;
  case Lang_OBJCXX:
    Args = {"-xobjective-c++"};
    break;
  case Lang_OpenCL:
    llvm_unreachable("Not implemented yet!");
  }
  return Args;
}

StringRef getFilenameForTesting(TestLanguage Lang) {
  switch (Lang) {
  case Lang_C89:
  case Lang_C99:
    return "input.c";

  case Lang_CXX03:
  case Lang_CXX11:
  case Lang_CXX14:
  case Lang_CXX17:
  case Lang_CXX20:
    return "input.cc";

  case Lang_OpenCL:
    return "input.cl";

  case Lang_OBJC:
    return "input.m";

  case Lang_OBJCXX:
    return "input.mm";
  }
  llvm_unreachable("Unhandled TestLanguage enum");
}

std::string getAnyTargetForTesting() {
  for (const auto &Target : llvm::TargetRegistry::targets()) {
    std::string Error;
    StringRef TargetName(Target.getName());
    if (TargetName == "x86-64")
      TargetName = "x86_64";
    if (llvm::TargetRegistry::lookupTarget(std::string(TargetName), Error) ==
        &Target) {
      return std::string(TargetName);
    }
  }
  return "";
}

} // end namespace clang
