base.hpp

// Copyright Take Vos 2023.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "../win32_headers.hpp"
#include "../macros.hpp"
#include <system_error>
#include <utility>
#include <string>
#include <expected>
#include <vector>
#include <algorithm>
#include <bit>
#include <exception>
#include <cstddef>
#include <cstdint>
#include <limits>
 
hi_export_module(hikogui.win32.base);
 
hi_export namespace hi { inline namespace v1 {
 
enum class win32_error : uint32_t {
    success = ERROR_SUCCESS,
    file_not_found = ERROR_FILE_NOT_FOUND,
    more_data = ERROR_MORE_DATA,
    invalid_data = ERROR_INVALID_DATA,
    insufficient_buffer = ERROR_INSUFFICIENT_BUFFER,
    status_pending = STATUS_PENDING,
};
 
}} // namespace hi::v1
 
template<>
struct std::is_error_code_enum<hi::win32_error> : std::true_type {};
 
namespace hi { inline namespace v1 {
 
struct win32_error_category : std::error_category {
    char const *name() const noexcept override
    {
        return "win32";
    }
 
    std::string message(int code) const override;
 
    bool equivalent(int code, std::error_condition const & condition) const noexcept override
    {
        switch (static_cast<hi::win32_error>(code)) {
        case hi::win32_error::file_not_found:
            return condition == std::errc::no_such_file_or_directory;
        case hi::win32_error::more_data:
            return condition == std::errc::message_size;
        case hi::win32_error::invalid_data:
            return condition == std::errc::bad_message;
        case hi::win32_error::status_pending:
            return condition == std::errc::interrupted;
        case hi::win32_error::insufficient_buffer:
            return condition == std::errc::no_buffer_space;
        default:
            return false;
        };
    }
};
 
hi_inline auto global_win32_error_category = win32_error_category{};
 
[[nodiscard]] hi_inline std::error_code make_error_code(win32_error code) noexcept
{
    return {static_cast<int>(code), global_win32_error_category};
}
 
[[nodiscard]] hi_inline win32_error win32_GetLastError() noexcept
{
    return static_cast<win32_error>(::GetLastError());
}

[[nodiscard]] hi_inline std::expected<std::string, win32_error> win32_WideCharToMultiByte(std::wstring_view s, unsigned int code_page = CP_UTF8, uint32_t flags = 0) noexcept
{
    if (s.empty()) {
        // WideCharToMultiByte() does not handle empty strings, if it can not also convert the null-character.
        return std::string{};
    }
 
    auto s_len = static_cast<int>(static_cast<unsigned int>(s.size()));
    auto r_len = ::WideCharToMultiByte(code_page, flags, s.data(), s_len, nullptr, 0, nullptr, nullptr);
    if (r_len == 0) {
        return std::unexpected{win32_GetLastError()};
    }
 
    auto r = std::string(static_cast<size_t>(static_cast<std::make_signed_t<size_t>>(r_len)), '\0');
    r.resize_and_overwrite(r_len, [&](char *p, size_t count) {
        return ::WideCharToMultiByte(code_page, flags, s.data(), s_len, p, static_cast<int>(count), nullptr, nullptr);
    });
 
    if (r.empty()) {
        return std::unexpected{win32_GetLastError()};
    }
 
    return r;
}

[[nodiscard]] hi_inline std::expected<std::wstring, win32_error> win32_MultiByteToWideChar(std::string_view s, unsigned int code_page = CP_UTF8, uint32_t flags = 0) noexcept
{
    if (s.empty()) {
        // MultiByteToWideChar() does not handle empty strings, if it can not also convert the null-character.
        return std::wstring{};
    }
 
    auto s_len = static_cast<int>(static_cast<unsigned int>(s.size()));
    auto r_len = ::MultiByteToWideChar(code_page, flags, s.data(), s_len, nullptr, 0);
    if (r_len == 0) {
        return std::unexpected{win32_GetLastError()};
    }
 
    auto r = std::wstring{};
    r.resize_and_overwrite(r_len, [&](wchar_t *p, size_t count) {
        return ::MultiByteToWideChar(code_page, flags, s.data(), s_len, p, static_cast<int>(count));
    });
 
    if (r.empty()) {
        return std::unexpected{win32_GetLastError()};
    }
 
    return r;
}

[[nodiscard]] hi_inline std::expected<std::vector<std::string>, win32_error> win32_MultiSZToStringVector(wchar_t const *first, wchar_t const *last) noexcept
{
    auto r = std::vector<std::string>{};
 
    while (first != last) {
        auto it_zero = std::find(first, last, wchar_t{0});
        if (it_zero == last) {
            // No termination found.
            return std::unexpected{win32_error::invalid_data};
        }
 
        auto const ws = std::wstring_view{first, static_cast<std::size_t>(it_zero - first)};
        if (ws.empty()) {
            // The list is terminated with an empty string.
            break;
        }
 
        if (auto s = win32_WideCharToMultiByte(ws)) {
            r.push_back(*s);
        } else {
            return std::unexpected{s.error()};
        }
 
        // Continue after the zero terminator.
        first = it_zero + 1;
    }
 
    return r;
}
 
[[nodiscard]] hi_inline std::expected<std::string, win32_error> win32_FormatMessage(win32_error error_code) noexcept
{
    auto const error_code_ = static_cast<DWORD>(std::to_underlying(error_code));
 
    // FormatMessageW() is unable to tell what the buffer size should be.
    // But 64Kbyte is the largest buffer that one should pass.
    LPWSTR buffer = nullptr;
    auto const result = ::FormatMessageW(
        FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
        NULL, // source
        error_code_,
        MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
        reinterpret_cast<LPWSTR>(&buffer),
        0,
        NULL);
 
    if (result == 0) {
        return std::unexpected(win32_GetLastError());
    }
 
    auto r = win32_WideCharToMultiByte(std::wstring_view{buffer, result});
    LocalFree(buffer);
    return r;
}
 
[[nodiscard]] hi_inline std::string win32_error_category::message(int code) const
{
    if (auto msg = win32_FormatMessage(static_cast<win32_error>(code))) {
        return *msg;
 
    } else {
        throw std::system_error(msg.error());
    }
}

[[nodiscard]] hi_inline uint32_t win32_HANDLE_to_int(HANDLE handle) noexcept
{
    auto i = std::bit_cast<uintptr_t>(handle);
    if (std::cmp_greater(i, std::numeric_limits<uint32_t>::max())) {
        std::terminate();
    }
    return static_cast<uint32_t>(i);
}
 
[[nodiscard]] hi_inline HANDLE win32_int_to_HANDLE(uint32_t i) noexcept
{
    return std::bit_cast<HANDLE>(static_cast<uintptr_t>(i));
}
 
}} // namespace hi::v1