HeadersIterator.cpp 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. /*
  2. * Copyright (c) 2022, Linus Groh <linusg@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <LibJS/Runtime/Array.h>
  7. #include <LibJS/Runtime/Iterator.h>
  8. #include <LibWeb/Bindings/HeadersIteratorPrototype.h>
  9. #include <LibWeb/Bindings/Intrinsics.h>
  10. #include <LibWeb/Fetch/HeadersIterator.h>
  11. #include <LibWeb/Infra/Strings.h>
  12. namespace Web::Bindings {
  13. template<>
  14. void Intrinsics::create_web_prototype_and_constructor<HeadersIteratorPrototype>(JS::Realm& realm)
  15. {
  16. auto prototype = realm.create<HeadersIteratorPrototype>(realm);
  17. m_prototypes.set("HeadersIterator"_fly_string, prototype);
  18. }
  19. }
  20. namespace Web::Fetch {
  21. GC_DEFINE_ALLOCATOR(HeadersIterator);
  22. GC::Ref<HeadersIterator> HeadersIterator::create(Headers const& headers, JS::Object::PropertyKind iteration_kind)
  23. {
  24. return headers.realm().create<HeadersIterator>(headers, iteration_kind);
  25. }
  26. HeadersIterator::HeadersIterator(Headers const& headers, JS::Object::PropertyKind iteration_kind)
  27. : PlatformObject(headers.realm())
  28. , m_headers(headers)
  29. , m_iteration_kind(iteration_kind)
  30. {
  31. }
  32. HeadersIterator::~HeadersIterator() = default;
  33. void HeadersIterator::initialize(JS::Realm& realm)
  34. {
  35. Base::initialize(realm);
  36. WEB_SET_PROTOTYPE_FOR_INTERFACE(HeadersIterator);
  37. }
  38. void HeadersIterator::visit_edges(JS::Cell::Visitor& visitor)
  39. {
  40. Base::visit_edges(visitor);
  41. visitor.visit(m_headers);
  42. }
  43. // https://webidl.spec.whatwg.org/#es-iterable, Step 2
  44. GC::Ref<JS::Object> HeadersIterator::next()
  45. {
  46. // The value pairs to iterate over are the return value of running sort and combine with this’s header list.
  47. auto value_pairs_to_iterate_over = [&]() {
  48. return m_headers->m_header_list->sort_and_combine();
  49. };
  50. auto pairs = value_pairs_to_iterate_over();
  51. if (m_index >= pairs.size())
  52. return create_iterator_result_object(vm(), JS::js_undefined(), true);
  53. auto const& pair = pairs[m_index++];
  54. auto pair_name = Infra::isomorphic_decode(pair.name);
  55. auto pair_value = Infra::isomorphic_decode(pair.value);
  56. switch (m_iteration_kind) {
  57. case JS::Object::PropertyKind::Key:
  58. return create_iterator_result_object(vm(), JS::PrimitiveString::create(vm(), pair_name), false);
  59. case JS::Object::PropertyKind::Value:
  60. return create_iterator_result_object(vm(), JS::PrimitiveString::create(vm(), pair_value), false);
  61. case JS::Object::PropertyKind::KeyAndValue: {
  62. auto array = JS::Array::create_from(realm(), { JS::PrimitiveString::create(vm(), pair_name), JS::PrimitiveString::create(vm(), pair_value) });
  63. return create_iterator_result_object(vm(), array, false);
  64. }
  65. default:
  66. VERIFY_NOT_REACHED();
  67. }
  68. }
  69. }