HECL.hpp

#ifndef HECL_HPP
#define HECL_HPP

#if _WIN32
char* win_realpath(const char* name, char* restrict resolved);
#else
#include <stdlib.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <dirent.h>
#include <fcntl.h>
#endif

#include <time.h>
#include <stdarg.h>
#include <stdio.h>
#include <functional>
#include <stdexcept>
#include <string>
#include <algorithm>
#include <regex>
#include "../extern/blowfish/blowfish.h"

namespace HECL
{

#if _WIN32 && UNICODE
#define HECL_UCS2 1
#endif

std::string WideToUTF8(const std::wstring& src);
std::wstring UTF8ToWide(const std::string& src);

#if HECL_UCS2
typedef wchar_t SystemChar;
typedef std::wstring SystemString;
static inline void ToLower(SystemString& str)
{std::transform(str.begin(), str.end(), str.begin(), towlower);}
static inline void ToUpper(SystemString& str)
{std::transform(str.begin(), str.end(), str.begin(), towupper);}
class SystemUTF8View
{
    std::string m_utf8;
public:
    SystemUTF8View(const SystemString& str)
    : m_utf8(WideToUTF8(str)) {}
    inline const std::string& utf8_str() {return m_utf8;}
};
class SystemStringView
{
    std::wstring m_sys;
public:
    SystemStringView(const std::string& str)
    : m_sys(UTF8ToWide(str)) {}
    inline const std::wstring& sys_str() {return m_sys;}
};
#ifndef _S
#define _S(val) L ## val
#endif
#else
typedef char SystemChar;
typedef std::string SystemString;
static inline void ToLower(SystemString& str)
{std::transform(str.begin(), str.end(), str.begin(), tolower);}
static inline void ToUpper(SystemString& str)
{std::transform(str.begin(), str.end(), str.begin(), toupper);}
class SystemUTF8View
{
    const std::string& m_utf8;
public:
    SystemUTF8View(const SystemString& str)
    : m_utf8(str) {}
    inline const std::string& utf8_str() {return m_utf8;}
};
class SystemStringView
{
    const std::string& m_sys;
public:
    SystemStringView(const std::string& str)
    : m_sys(str) {}
    inline const std::string& sys_str() {return m_sys;}
};
#ifndef _S
#define _S(val) val
#endif
#endif

class Exception : public std::exception
{
    SystemString m_what;
#if HECL_UCS2
    std::string m_utf8what;
#endif
public:
    Exception(const SystemString& what) noexcept
    : m_what(what)
    {
#if HECL_UCS2
        m_utf8what = WideToUTF8(what);
#endif
    }
    const char* what() const noexcept
    {
#if HECL_UCS2
        return m_utf8what.c_str();
#else
        return m_what.c_str();
#endif
    }
    inline const SystemChar* swhat() const noexcept {return m_what.c_str();}
};

static inline void MakeDir(const SystemString& dir)
{
#if _WIN32
    HRESULT err;
    if (!CreateDirectory(dir.c_str(), NULL))
        if ((err = GetLastError()) != ERROR_ALREADY_EXISTS)
            throw std::error_code(err, std::system_category());
#else
    if (mkdir(dir.c_str(), 0755))
        if (errno != EEXIST)
            throw std::error_code(errno, std::system_category());
#endif
}

static inline SystemChar* Getcwd(SystemChar* buf, int maxlen)
{
#if HECL_UCS2
    return wgetcwd(buf, maxlen);
#else
    return getcwd(buf, maxlen);
#endif
}

enum FileLockType
{
    LNONE = 0,
    LREAD,
    LWRITE
};
static inline FILE* Fopen(const SystemChar* path, const SystemChar* mode, FileLockType lock=LNONE)
{
#if HECL_UCS2
    FILE* fp = wfopen(path, mode);
#else
    FILE* fp = fopen(path, mode);
#endif
    if (!fp)
        throw std::error_code(errno, std::system_category());

    if (lock)
    {
#if _WIN32
        HANDLE fhandle = (HANDLE)fileno(fp);
        OVERLAPPED ov = {};
        LockFileEx(fhandle, (lock == LWRITE) ? LOCKFILE_EXCLUSIVE_LOCK : 0, 0, 0, 1, &ov);
#else
        struct flock lk =
        {
            (short)((lock == LREAD) ? F_RDLCK : F_WRLCK),
            SEEK_SET, 0, 0, 0
        };
        fcntl(fileno(fp), F_SETLK, &lk);
#endif
    }

    return fp;
}

static inline int Stat(const SystemChar* path, struct stat* statOut)
{
#if HECL_UCS2
    return wstat(path, statOut);
#else
    return stat(path, statOut);
#endif
}

static inline void Printf(const SystemChar* format, ...)
{
    va_list va;
    va_start(va, format);
#if HECL_UCS2
    vwprintf(format, va);
#else
    vprintf(format, va);
#endif
    va_end(va);
}

static inline void FPrintf(FILE* fp, const SystemChar* format, ...)
{
    va_list va;
    va_start(va, format);
#if HECL_UCS2
    vfwprintf(fp, format, va);
#else
    vfprintf(fp, format, va);
#endif
    va_end(va);
}

typedef std::basic_regex<SystemChar> SystemRegex;
typedef std::regex_token_iterator<SystemString::const_iterator> SystemRegexTokenIterator;
typedef std::match_results<SystemString::const_iterator> SystemRegexMatch;

class ProjectRootPath;

enum LogType
{
    LOG_INFO,
    LOG_WARN,
    LOG_ERROR
};

typedef std::function<void(LogType, std::string&)> FLogger;

class FourCC final
{
    union
    {
        char fcc[4];
        uint32_t num;
    };
public:
    FourCC() /* Sentinel FourCC */
    : num(0) {}
    FourCC(const char* name)
    : num(*(uint32_t*)name) {}
    inline bool operator==(const FourCC& other) const {return num == other.num;}
    inline bool operator!=(const FourCC& other) const {return num != other.num;}
    inline bool operator==(const char* other) const {return num == *(uint32_t*)other;}
    inline bool operator!=(const char* other) const {return num != *(uint32_t*)other;}
    inline std::string toString() const {return std::string(fcc, 4);}
};

class Hash final
{
    int64_t hash;
public:
    Hash(const void* buf, size_t len)
    : hash(Blowfish_hash(buf, len)) {}
    Hash(const std::string& str)
    : hash(Blowfish_hash(str.data(), str.size())) {}
    Hash(int64_t hashin)
    : hash(hashin) {}
    Hash(const Hash& other) {hash = other.hash;}
    inline Hash& operator=(const Hash& other) {hash = other.hash; return *this;}
    inline bool operator==(const Hash& other) const {return hash == other.hash;}
    inline bool operator!=(const Hash& other) const {return hash != other.hash;}
    inline bool operator<(const Hash& other) const {return hash < other.hash;}
    inline bool operator>(const Hash& other) const {return hash > other.hash;}
    inline bool operator<=(const Hash& other) const {return hash <= other.hash;}
    inline bool operator>=(const Hash& other) const {return hash >= other.hash;}
};

class Time final
{
    time_t ts;
public:
    Time() : ts(time(NULL)) {}
    Time(time_t ti) : ts(ti) {}
    Time(const Time& other) {ts = other.ts;}
    inline time_t getTs() const {return ts;}
    inline Time& operator=(const Time& other) {ts = other.ts; return *this;}
    inline bool operator==(const Time& other) const {return ts == other.ts;}
    inline bool operator!=(const Time& other) const {return ts != other.ts;}
    inline bool operator<(const Time& other) const {return ts < other.ts;}
    inline bool operator>(const Time& other) const {return ts > other.ts;}
    inline bool operator<=(const Time& other) const {return ts <= other.ts;}
    inline bool operator>=(const Time& other) const {return ts >= other.ts;}
};

class ProjectPath
{
protected:
    SystemString m_absPath;
    SystemString m_relPath;
    size_t m_hash = 0;
#if HECL_UCS2
    std::string m_utf8AbsPath;
    const char* m_utf8RelPath;
#endif
    ProjectPath() {}
    bool _canonAbsPath(const SystemString& path);
public:
    ProjectPath(const ProjectRootPath& rootPath, const SystemString& path);

    inline bool isRoot() const {return m_relPath.empty();}

    inline const SystemString& getAbsolutePath() const {return m_absPath;}

    inline const SystemString& getRelativePath() const
    {
        if (m_relPath.size())
            return m_relPath;
        static const SystemString dot = _S(".");
        return dot;
    }

    inline const std::string& getAbsolutePathUTF8() const
    {
#if HECL_UCS2
        return m_utf8AbsPath;
#else
        return m_absPath;
#endif
    }

    inline const std::string& getRelativePathUTF8() const
    {
#if HECL_UCS2
        return m_utf8RelPath;
#else
        return m_relPath;
#endif
    }

    enum PathType
    {
        PT_NONE, 
        PT_FILE, 
        PT_DIRECTORY, 
        PT_GLOB 
    };

    PathType getPathType() const;

    Time getModtime() const;

    void getGlobResults(std::vector<SystemString>& outPaths) const;

    inline size_t hash() const {return m_hash;}
    inline bool operator==(const ProjectPath& other) const {return m_hash == other.m_hash;}
    inline bool operator!=(const ProjectPath& other) const {return m_hash != other.m_hash;}

};

class ProjectRootPath : public ProjectPath
{
public:
    ProjectRootPath(const SystemString& path)
    {_canonAbsPath(path);}
};

ProjectRootPath* SearchForProject(const SystemString& path);


/* Type-sensitive byte swappers */
template <typename T>
static inline T bswap16(T val)
{
#if __GNUC__
    return __builtin_bswap16(val);
#elif _WIN32
    return _byteswap_ushort(val);
#else
    return (val = (val << 8) | ((val >> 8) & 0xFF));
#endif
}

template <typename T>
static inline T bswap32(T val)
{
#if __GNUC__
    return __builtin_bswap32(val);
#elif _WIN32
    return _byteswap_ulong(val);
#else
    val = (val & 0x0000FFFF) << 16 | (val & 0xFFFF0000) >> 16;
    val = (val & 0x00FF00FF) << 8 | (val & 0xFF00FF00) >> 8;
    return val;
#endif
}

template <typename T>
static inline T bswap64(T val)
{
#if __GNUC__
    return __builtin_bswap64(val);
#elif _WIN32
    return _byteswap_uint64(val);
#else
    return ((val & 0xFF00000000000000ULL) >> 56) |
           ((val & 0x00FF000000000000ULL) >> 40) |
           ((val & 0x0000FF0000000000ULL) >> 24) |
           ((val & 0x000000FF00000000ULL) >>  8) |
           ((val & 0x00000000FF000000ULL) <<  8) |
           ((val & 0x0000000000FF0000ULL) << 24) |
           ((val & 0x000000000000FF00ULL) << 40) |
           ((val & 0x00000000000000FFULL) << 56);
#endif
}


#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
static inline int16_t ToBig(int16_t val) {return bswap16(val);}
static inline uint16_t ToBig(uint16_t val) {return bswap16(val);}
static inline int32_t ToBig(int32_t val) {return bswap32(val);}
static inline uint32_t ToBig(uint32_t val) {return bswap32(val);}
static inline int64_t ToBig(int64_t val) {return bswap64(val);}
static inline uint64_t ToBig(uint64_t val) {return bswap64(val);}

static inline int16_t ToLittle(int16_t val) {return val;}
static inline uint16_t ToLittle(uint16_t val) {return val;}
static inline int32_t ToLittle(int32_t val) {return val;}
static inline uint32_t ToLittle(uint32_t val) {return val;}
static inline int64_t ToLittle(int64_t val) {return val;}
static inline uint64_t ToLittle(uint64_t val) {return val;}
#else
static inline int16_t ToLittle(int16_t val) {return bswap16(val);}
static inline uint16_t ToLittle(uint16_t val) {return bswap16(val);}
static inline int32_t ToLittle(int32_t val) {return bswap32(val);}
static inline uint32_t ToLittle(uint32_t val) {return bswap32(val);}
static inline int64_t ToLittle(int64_t val) {return bswap64(val);}
static inline uint64_t ToLittle(uint64_t val) {return bswap64(val);}

static inline int16_t ToBig(int16_t val) {return val;}
static inline uint16_t ToBig(uint16_t val) {return val;}
static inline int32_t ToBig(int32_t val) {return val;}
static inline uint32_t ToBig(uint32_t val) {return val;}
static inline int64_t ToBig(int64_t val) {return val;}
static inline uint64_t ToBig(uint64_t val) {return val;}
#endif

}

namespace std
{
template <> struct hash<HECL::ProjectPath>
{
    size_t operator()(const HECL::ProjectPath& val) const noexcept
    {return val.hash();}
};
}

#endif // HECL_HPP