mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-17 16:28:06 +03:00
e636851481
Instead of using a HashMap<ByteString, ByteString, CaseInsensitive...> everywhere, we now encapsulate this in a class. Even better, the new class also allows keeping track of multiple headers with the same name! This will make it possible for HTTP responses to actually retain all their headers on the perilous journey from RequestServer to LibWeb.
66 lines
1.3 KiB
C++
66 lines
1.3 KiB
C++
/*
|
|
* 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;
|
|
}
|
|
|
|
}
|