/*
 * Copyright (c) 2024, Andreas Kling <andreas@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibHTTP/Header.h>

namespace HTTP {

class HeaderMap {
public:
    HeaderMap() = default;
    ~HeaderMap() = default;

    void set(ByteString name, ByteString value)
    {
        m_map.set(name, value);
        m_headers.append({ move(name), move(value) });
    }

    [[nodiscard]] bool contains(ByteString const& name) const
    {
        return m_map.contains(name);
    }

    [[nodiscard]] Optional<ByteString> get(ByteString const& name) const
    {
        return m_map.get(name);
    }

    [[nodiscard]] Vector<Header> const& headers() const
    {
        return m_headers;
    }

private:
    HashMap<ByteString, ByteString, CaseInsensitiveStringTraits> m_map;
    Vector<Header> m_headers;
};

}

namespace IPC {

template<>
inline ErrorOr<void> encode(Encoder& encoder, HTTP::HeaderMap const& header_map)
{
    TRY(encoder.encode(header_map.headers()));
    return {};
}

template<>
inline ErrorOr<HTTP::HeaderMap> decode(Decoder& decoder)
{
    auto headers = TRY(decoder.decode<Vector<HTTP::Header>>());
    HTTP::HeaderMap header_map;
    for (auto& header : headers)
        header_map.set(move(header.name), move(header.value));
    return header_map;
}

}