#include <cstdint>
/**
 * File: fileutils.cpp
 *
 * Author: seichert
 * Created: 1/19/2018
 *
 * Description: Routines for working with files
 *
 * Copyright: Anki, Inc. 2018
 *
 **/

#include "fileutils.h"
#include "log.h"
#include <fstream>
#include <iterator>
#include <sys/stat.h>
#include <sys/types.h>

namespace Anki {

int WriteFileAtomically(const std::string& path,
                        const std::string& data,
                        mode_t mode,
                        uid_t owner,
                        gid_t group)
{
  std::string tmpPath = path + ".tmp";

  {
    std::ofstream ofstrm(tmpPath, std::ios::binary | std::ios::out | std::ios::trunc);
    if (!ofstrm.is_open()) {
      logi("Failed to open '%s'", tmpPath.c_str());
      return -1;
    }

    ofstrm << data;
  }

  int rc = chmod(tmpPath.c_str(), mode);
  if (rc) {
    logi("Error chmoding '%s' errno = %d", tmpPath.c_str(), errno);
    return rc;
  }
  rc = chown(tmpPath.c_str(), owner, group);
  if (rc) {
    logi("Error chowning '%s' errno = %d", tmpPath.c_str(), errno);
    return rc;
  }
  rc = rename(tmpPath.c_str(), path.c_str());
  if (rc) {
    logi("Error renaming '%s' to '%s' errno = %d", tmpPath.c_str(), path.c_str(), errno);
    return rc;
  }
  return 0;
}

int CreateDirectory(const std::string& path,
                    mode_t mode,
                    uid_t owner,
                    gid_t group)
{
  int rc = mkdir(path.c_str(), mode);
  if ((rc == -1) && (errno == EEXIST)) {
    rc = 0;
  }

  if (rc) {
    logi("Error mkdiring '%s' errno = %d", path.c_str(), errno);
    return rc;
  }

  rc = chmod(path.c_str(), mode);
  if (rc) {
    logi("Error chmoding '%s' errno = %d", path.c_str(), errno);
    return rc;
  }
  rc = chown(path.c_str(), owner, group);
  if (rc) {
    logi("Error chowning '%s' errno = %d", path.c_str(), errno);
    return rc;
  }
  return 0;
}

bool ReadFileIntoVector(const std::string& pathToFile,
                        std::vector<uint8_t>& data)
{
  data.clear();
  std::ifstream fileIn;
  fileIn.open(pathToFile, std::ios::in | std::ifstream::binary);
  if (!fileIn.is_open()) {
    return false;
  }
  fileIn.unsetf(std::ios::skipws);

  std::streampos fileSize;

  fileIn.seekg(0, std::ios::end);
  fileSize = fileIn.tellg();
  fileIn.seekg(0, std::ios::beg);

  data.reserve(fileSize);
  data.insert(data.begin(),
              std::istream_iterator<uint8_t>(fileIn),
              std::istream_iterator<uint8_t>());
  fileIn.close();
  return true;
}

} // namespace Anki