// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2019-2024 Second State INC

#ifdef WASMEDGE_BUILD_FUZZING
#include "driver/fuzzTool.h"
#include "common/configure.h"
#include "loader/loader.h"
#include "validator/validator.h"
#include "llvm/codegen.h"
#include "llvm/compiler.h"

namespace WasmEdge {
namespace Driver {

int FuzzTool(const uint8_t *Data, size_t Size) noexcept {
  using namespace std::literals;
  std::ios::sync_with_stdio(false);
  spdlog::set_level(spdlog::level::critical);

  Configure Conf;
  Conf.getRuntimeConfigure().setForceInterpreter(true);
  Loader::Loader Loader(Conf);

  std::unique_ptr<AST::Module> Module;
  if (auto Res = Loader.parseModule(Span<const uint8_t>(Data, Size))) {
    Module = std::move(*Res);
  } else {
    const auto Err = static_cast<uint32_t>(Res.error());
    spdlog::error("Parse Module failed. Error code: {}"sv, Err);
    return EXIT_FAILURE;
  }

  {
    Validator::Validator ValidatorEngine(Conf);
    if (auto Res = ValidatorEngine.validate(*Module); !Res) {
      const auto Err = static_cast<uint32_t>(Res.error());
      spdlog::error("Validate Module failed. Error code: {}"sv, Err);
      return EXIT_FAILURE;
    }
  }

  LLVM::Compiler Compiler(Conf);
  if (auto Res = Compiler.checkConfigure(); !Res) {
    const auto Err = static_cast<uint32_t>(Res.error());
    spdlog::error("Compiler Configure failed. Error code: {}"sv, Err);
  }

  LLVM::CodeGen CodeGen(Conf);
  if (auto Res = Compiler.compile(*Module); !Res) {
    const auto Err = static_cast<uint32_t>(Res.error());
    spdlog::error("Compilation failed. Error code: {}"sv, Err);
    return EXIT_FAILURE;
  } else if (auto Res2 = CodeGen.codegen(Span<const uint8_t>(Data, Size),
                                         std::move(*Res), "/dev/null"sv);
             !Res2) {
    const auto Err = static_cast<uint32_t>(Res2.error());
    spdlog::error("Code Generation failed. Error code: {}"sv, Err);
    return EXIT_FAILURE;
  }

  return EXIT_SUCCESS;
}

} // namespace Driver
} // namespace WasmEdge
#endif
