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

#include "revng/Support/GzipStream.h"
#include "revng/Support/GzipTarFile.h"
#include "revng/Support/Tar.h"

#define BOOST_TEST_MODULE GzipTarFile
bool init_unit_test();
#include "boost/test/unit_test.hpp"

#include "revng/UnitTestHelpers/UnitTestHelpers.h"

static void checkOffset(llvm::SmallVector<char> &Buffer,
                        size_t Start,
                        size_t Size,
                        llvm::StringRef ExpectedValue) {
  llvm::ArrayRef<char> ArrayRef{ Buffer.data() + Start, Size };
  llvm::SmallVector<char> Result = gzipDecompress(ArrayRef);
  BOOST_TEST(llvm::StringRef(Result.begin(), Result.size()) == ExpectedValue);
}

BOOST_AUTO_TEST_CASE(GzipTarFileTest) {
  using revng::OffsetDescriptor;
  using revng::TarReader;

  llvm::SmallVector<char> Buffer;
  llvm::raw_svector_ostream OS(Buffer);

  // Write test data
  revng::GzipTarWriter Writer(OS);

  const char Data1[5] = "foo2";
  OffsetDescriptor Offset1 = Writer.append("foo", { Data1, 4 });

  const char Data2[5] = "bar2";
  OffsetDescriptor Offset2 = Writer.append("bar", { Data2, 4 });

  Writer.close();

  BOOST_TEST(Offset1.Start == 0ULL);
  BOOST_TEST(Offset2.Start == Offset1.End);

  {
    TarReader Reader({ Buffer.data(), Buffer.size() }, revng::TarFormat::Gzip);

    cppcoro::generator<TarReader::Entry> Gen = Reader.entries();
    std::vector<TarReader::Entry> Entries(Gen.begin(), Gen.end());
    BOOST_TEST(Entries.size() == 2ULL);

    llvm::StringRef RefData1(Entries[0].Data.data(), Entries[0].Data.size());
    BOOST_TEST(Entries[0].Filename == "foo");
    BOOST_TEST(RefData1.str() == "foo2");

    llvm::StringRef RefData2(Entries[1].Data.data(), Entries[1].Data.size());
    BOOST_TEST(Entries[1].Filename == "bar");
    BOOST_TEST(RefData2.str() == "bar2");
  }

  checkOffset(Buffer, Offset1.DataStart, Offset1.dataSize(), "foo2");
  checkOffset(Buffer, Offset2.DataStart, Offset2.dataSize(), "bar2");
}