#include "diskspace.h"

namespace bf = boost::filesystem;

#if !defined(_MSC_VER)

#include <sys/statvfs.h>

#include <cerrno>

namespace cpputils {

uint64_t free_disk_space_in_bytes(const bf::path& location) {

struct statvfs stat {};

int result = ::statvfs(location.string().c_str(), &stat);

if (0 != result) {

throw std::runtime_error("Error calling statvfs(). Errno: " + std::to_string(errno));

}

return stat.f_frsize*stat.f_bavail;

}

}

#else

#include <Windows.h>

namespace cpputils {

uint64_t free_disk_space_in_bytes(const bf::path& location) {

ULARGE_INTEGER freeBytes;

if (!GetDiskFreeSpaceEx(location.string().c_str(), &freeBytes, nullptr, nullptr)) {

throw std::runtime_error("Error calling GetDiskFreeSpaceEx(). Error code: " + std::to_string(GetLastError()));

}

return freeBytes.QuadPart;

}

}

#endif