#pragma once
#include <iosfwd>
#include <cstdint>
#include <strings.h> // strncasecmp
#include <panda/string.h>
#include <algorithm>

namespace panda { namespace protocol { namespace websocket {

uint32_t string_hash32_ci (const char *key, size_t len);

struct string_hash_ci {
    size_t operator() (const string& s) const {
        return string_hash32_ci(s.data(), s.length());
    }
};

struct string_equal_ci {
    bool operator() (const string& lhs, const string& rhs) const {
        return lhs.length() == rhs.length() && strncasecmp(lhs.data(), rhs.data(), lhs.length()) == 0;
    }
};

struct char_equal_ci {
    bool operator()(char ch1, char ch2) {
        return std::toupper(ch1) == std::toupper(ch2);
    }
};

inline bool string_contains_ci (const string& haystack, const string& needle) {
    auto iter = std::search(haystack.begin(), haystack.end(), needle.begin(), needle.end(), char_equal_ci());
    return (iter != haystack.end());
}

template <typename T>
inline bool parse_binary_number (T& num, const char*& src, size_t len) {
    if (!num && len >= sizeof(T)) { // common case - have whole number in src
        memcpy(&num, src, sizeof(T));
        src += sizeof(T);
        return true;
    }

    uint8_t bcnt = ((char*)&num)[sizeof(T)-1];
    char* dst = (char*)&num + bcnt;

    if (bcnt + len >= sizeof(T)) { // have everything left in src
        char* end = (char*)&num + sizeof(T);
        while (dst != end) *dst++ = *src++;
        return true;
    }

    // still not enough
    ((char*)&num)[sizeof(T)-1] = bcnt+len;
    char* end = dst + len;
    while (dst != end) *dst++ = *src++;
    return false;
}

void crypt_mask (const char* str, char* dst, size_t len, uint32_t mask, uint64_t bytes_received);

inline void crypt_mask (char* str, size_t len, uint32_t mask, uint64_t bytes_received) {
    crypt_mask(str, str, len, mask, bytes_received);
}

string close_message(uint16_t code);

/// @return true to use in static variable initializators
bool register_close_codes(std::initializer_list<std::pair<uint16_t, string>>);

/// close clode formatter
struct ccfmt {
    uint16_t    code;
    string_view msg;

    ccfmt (uint16_t code, string_view msg) : code(code), msg(msg) {}
};

std::ostream& operator<< (std::ostream&, const ccfmt&); 

}}}