//------------------------------------------------------------------------------
// File: TestEnv.hh
// Author: Elvin Sindrilaru <esindril@cern.ch>
//------------------------------------------------------------------------------

/************************************************************************
 * EOS - the CERN Disk Storage System                                   *
 * Copyright (C) 2013 CERN/Switzerland                                  *
 *                                                                      *
 * This program is free software: you can redistribute it and/or modify *
 * it under the terms of the GNU General Public License as published by *
 * the Free Software Foundation, either version 3 of the License, or    *
 * (at your option) any later version.                                  *
 *                                                                      *
 * This program is distributed in the hope that it will be useful,      *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
 * GNU General Public License for more details.                         *
 *                                                                      *
 * You should have received a copy of the GNU General Public License    *
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.*
 ************************************************************************/

#include "TestEnv.hh"
#include "XrdCl/XrdClURL.hh"
#include <sstream>
#include <cstdlib>
#include <libgen.h>

#define SSTR(message) static_cast<std::ostringstream&>(std::ostringstream().flush() << message).str()

EOSFSTTEST_NAMESPACE_BEGIN

//------------------------------------------------------------------------------
// Constructor
//
// Notice:
// File file32MB.dat is created as follows:
// dd if=/dev/zero count=32 bs=1M | tr '\000' '\001' > /eos/dev/test/fst/replica/file32MB.dat
//
//------------------------------------------------------------------------------
TestEnv::TestEnv(const std::string& endpoint)
{
  XrdCl::URL url(endpoint);

  if (!url.IsValid()) {
    std::cerr << "error: invalid endpoint - " << endpoint << std::endl;
    exit(1);
  }

  mPathPrefix = url.GetPath();

  if (*mPathPrefix.rbegin() != '/') {
    mPathPrefix += '/';
  }

  mPathPrefix += "fst_unit_tests/dirs/";
  mHostName = url.GetHostName();
  // Note the yes and tr errors are "acceptable"
  system("rm -rf /tmp/file32MB.dat; rm -rf /tmp/file_prefetch.dat");
  system("yes '\\xDE\\xAD\\xBE\\xEF' | tr -d \\\\n | dd of=/tmp/file32MB.dat count=32 bs=1M iflag=fullblock");

  for (int i = 0; i < 4; i++) {
    system("yes '\\xDE\\xAD\\xBE\\xEF' | tr -d \\\\n | dd of=/tmp/file_prefetch.dat count=3 bs=1M iflag=fullblock oflag=append conv=notrunc");
    system("yes '\\xAD\\xAA\\xDA\\xAD' | tr -d \\\\n | dd of=/tmp/file_prefetch.dat count=3 bs=1M iflag=fullblock oflag=append conv=notrunc");
    system("yes '\\xAB\\xCD\\xAB\\xCD' | tr -d \\\\n | dd of=/tmp/file_prefetch.dat count=3 bs=1M iflag=fullblock oflag=append conv=notrunc");
  }

  // Add one last bit to the file so that it has a "random" size
  system("yes '\\xFE\\xDC\\xCB\\xBA' | tr -d \\\\n | dd of=/tmp/file_prefetch.dat count=1 bs=213 iflag=fullblock oflag=append conv=notrunc");
  system(SSTR("eos mkdir -p " << mPathPrefix << "replica ").c_str());
  system(SSTR("eos attr set default=replica " << mPathPrefix <<
              "replica > /dev/null 2>&1").c_str());
  system(SSTR("eos attr rm sys.recycle " << mPathPrefix <<
              "replica > /dev/null 2>&1").c_str());
  system(SSTR("eos mkdir -p " << mPathPrefix << "raiddp").c_str());
  system(SSTR("eos attr set default=raiddp " << mPathPrefix <<
              "raiddp > /dev/null 2>&1").c_str());
  system(SSTR("eos attr rm sys.recycle " << mPathPrefix <<
              "raiddp > /dev/null 2>&1").c_str());
  system(SSTR("eos mkdir -p " << mPathPrefix << "raid6").c_str());
  system(SSTR("eos attr set default=raid6 " << mPathPrefix <<
              "raid6  > /dev/null 2>&1").c_str());
  system(SSTR("eos attr rm sys.recycle " << mPathPrefix <<
              "raid6 > /dev/null 2>&1").c_str());
  system(SSTR("xrdcp -f /tmp/file32MB.dat root://" << mHostName << "/" <<
              mPathPrefix << "replica/ > /dev/null 2>&1").c_str());
  system(SSTR("xrdcp -f /tmp/file32MB.dat root://" << mHostName << "/" <<
              mPathPrefix << "raiddp/ > /dev/null 2>&1").c_str());
  system(SSTR("xrdcp -f /tmp/file32MB.dat root://" << mHostName << "/" <<
              mPathPrefix << "raid6/ > /dev/null 2>&1").c_str());
  system(SSTR("xrdcp -f /tmp/file_prefetch.dat root://" << mHostName << "/" <<
              mPathPrefix << "replica/ > /dev/null 2>&1").c_str());
  mMapParam.insert(std::make_pair("server", mHostName));
  mMapParam.insert(std::make_pair("dummy_file",
                                  mPathPrefix + "replica/dummy.dat"));
  mMapParam.insert(std::make_pair("replica_file",
                                  mPathPrefix + "replica/file32MB.dat"));
  mMapParam.insert(std::make_pair("prefetch_file",
                                  mPathPrefix + "replica/file_prefetch.dat"));
  mMapParam.insert(std::make_pair("raiddp_file",
                                  mPathPrefix + "raiddp/file32MB.dat"));
  mMapParam.insert(std::make_pair("reeds_file",
                                  mPathPrefix + "raid6/file32MB.dat"));
  mMapParam.insert(std::make_pair("file_size", "33554432")); // 32MB
  // ReadV sequences used for testing
  // Test set 1 - 4KB read out of each MB
  mMapParam.insert(std::make_pair("off1",
                                  "0 1048576 2097152 3145728 4194304 5242880 "));
  mMapParam.insert(std::make_pair("len1", "4096 4096 4096 4096 4096 4096"));
  // Correct responses for the set 1
  mMapParam.insert(std::make_pair("off1_stripe0", "0 1048576"));
  mMapParam.insert(std::make_pair("len1_stripe0", "4096 4096"));
  mMapParam.insert(std::make_pair("off1_stripe1", "0 1048576"));
  mMapParam.insert(std::make_pair("len1_stripe1", "4096 4096"));
  mMapParam.insert(std::make_pair("off1_stripe2", "0"));
  mMapParam.insert(std::make_pair("len1_stripe2", "4096"));
  mMapParam.insert(std::make_pair("off1_stripe3", "0"));
  mMapParam.insert(std::make_pair("len1_stripe3", "4096"));
  // Test set 2 - 16KB read around each MB
  mMapParam.insert(std::make_pair("off2",
                                  "1040384 2088960 3137536 4186112 5234688 "
                                  "6283264 7331840 8380416 9428992 10477568"));
  mMapParam.insert(std::make_pair("len2",
                                  "16384 16384 16384 16384 16384 16384 16384 "
                                  "16384 16384 16384"));
  // Correct responses for set 2
  mMapParam.insert(std::make_pair("off2_stripe0",
                                  "1040384 1048576 2088960 2097152 3137536"));
  mMapParam.insert(std::make_pair("len2_stripe0", "8192 8192 8192 8192 8192"));
  mMapParam.insert(std::make_pair("off2_stripe1",
                                  "0 1040384 1048576 2088960 2097152 3137536"));
  mMapParam.insert(std::make_pair("len2_stripe1",
                                  "8192 8192 8192 8192 8192 8192"));
  mMapParam.insert(std::make_pair("off2_stripe2",
                                  "0 1040384 1048576 2088960 2097152"));
  mMapParam.insert(std::make_pair("len2_stripe2", "8192 8192 8192 8192 8192"));
  mMapParam.insert(std::make_pair("off2_stripe3", "0 1040384 1048576 2088960"));
  mMapParam.insert(std::make_pair("len2_stripe3", "8192 8192 8192 8192"));
  // Test set 3
  mMapParam.insert(std::make_pair("off3", "1048576"));
  mMapParam.insert(std::make_pair("len3", "2097169"));
  // Correct responses for set 3
  mMapParam.insert(std::make_pair("off3_stripe0", ""));
  mMapParam.insert(std::make_pair("len3_stripe0", ""));
  mMapParam.insert(std::make_pair("off3_stripe1", "0"));
  mMapParam.insert(std::make_pair("len3_stripe1", "1048576"));
  mMapParam.insert(std::make_pair("off3_stripe2", "0"));
  mMapParam.insert(std::make_pair("len3_stripe2", "1048576"));
  mMapParam.insert(std::make_pair("off3_stripe3", "0"));
  mMapParam.insert(std::make_pair("len3_stripe3", "17"));
}

//------------------------------------------------------------------------------
// Destructor
//------------------------------------------------------------------------------
TestEnv::~TestEnv()
{
  system(SSTR("eos " << "root://" << mHostName << " rm -rF " <<
              mPathPrefix).c_str());
  const std::string dir_to_rm = dirname((char*)std::string(mPathPrefix).c_str());
  system(SSTR("eos " << "root://" << mHostName << " rmdir " <<
              dir_to_rm).c_str());
}

//------------------------------------------------------------------------------
// Set key value mapping
//------------------------------------------------------------------------------
void
TestEnv::SetMapping(const std::string& key, const std::string& value)
{
  auto pair_res = mMapParam.insert(std::make_pair(key, value));

  if (!pair_res.second) {
    std::cerr << "Mapping already exists, key=" << key
              << " value=" << value << std::endl;
  }
}

//------------------------------------------------------------------------------
// Get key value mapping
//------------------------------------------------------------------------------
std::string
TestEnv::GetMapping(const std::string& key) const
{
  auto iter = mMapParam.find(key);

  if (iter != mMapParam.end()) {
    return iter->second;
  } else {
    return std::string("");
  }
}

EOSFSTTEST_NAMESPACE_END