WritableStreamDefaultWriter.cpp 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142
  1. /*
  2. * Copyright (c) 2023, Matthew Olsson <mattco@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <LibJS/Runtime/PromiseCapability.h>
  7. #include <LibWeb/Bindings/Intrinsics.h>
  8. #include <LibWeb/Bindings/WritableStreamDefaultWriterPrototype.h>
  9. #include <LibWeb/Streams/AbstractOperations.h>
  10. #include <LibWeb/Streams/WritableStream.h>
  11. #include <LibWeb/Streams/WritableStreamDefaultWriter.h>
  12. #include <LibWeb/WebIDL/ExceptionOr.h>
  13. namespace Web::Streams {
  14. JS_DEFINE_ALLOCATOR(WritableStreamDefaultWriter);
  15. WebIDL::ExceptionOr<JS::NonnullGCPtr<WritableStreamDefaultWriter>> WritableStreamDefaultWriter::construct_impl(JS::Realm& realm, JS::NonnullGCPtr<WritableStream> stream)
  16. {
  17. auto writer = realm.heap().allocate<WritableStreamDefaultWriter>(realm, realm);
  18. // 1. Perform ? SetUpWritableStreamDefaultWriter(this, stream).
  19. TRY(set_up_writable_stream_default_writer(*writer, stream));
  20. return writer;
  21. }
  22. // https://streams.spec.whatwg.org/#default-writer-closed
  23. JS::GCPtr<JS::Object> WritableStreamDefaultWriter::closed()
  24. {
  25. // 1. Return this.[[closedPromise]].
  26. return m_closed_promise->promise();
  27. }
  28. // https://streams.spec.whatwg.org/#default-writer-desired-size
  29. WebIDL::ExceptionOr<Optional<double>> WritableStreamDefaultWriter::desired_size() const
  30. {
  31. // 1. If this.[[stream]] is undefined, throw a TypeError exception.
  32. if (!m_stream)
  33. return WebIDL::SimpleException { WebIDL::SimpleExceptionType::TypeError, "Cannot get desired size of writer that has no locked stream"sv };
  34. // 2. Return ! WritableStreamDefaultWriterGetDesiredSize(this).
  35. return writable_stream_default_writer_get_desired_size(*this);
  36. }
  37. // https://streams.spec.whatwg.org/#default-writer-ready
  38. JS::GCPtr<JS::Object> WritableStreamDefaultWriter::ready()
  39. {
  40. // 1. Return this.[[readyPromise]].
  41. return m_ready_promise->promise();
  42. }
  43. // https://streams.spec.whatwg.org/#default-writer-abort
  44. WebIDL::ExceptionOr<JS::GCPtr<JS::Object>> WritableStreamDefaultWriter::abort(JS::Value reason)
  45. {
  46. auto& realm = this->realm();
  47. // 1. If this.[[stream]] is undefined, return a promise rejected with a TypeError exception.
  48. if (!m_stream) {
  49. auto exception = JS::TypeError::create(realm, "Cannot abort a writer that has no locked stream"sv);
  50. return WebIDL::create_rejected_promise(realm, exception)->promise();
  51. }
  52. // 2. Return ! WritableStreamDefaultWriterAbort(this, reason).
  53. return TRY(writable_stream_default_writer_abort(*this, reason))->promise();
  54. }
  55. // https://streams.spec.whatwg.org/#default-writer-close
  56. WebIDL::ExceptionOr<JS::GCPtr<JS::Object>> WritableStreamDefaultWriter::close()
  57. {
  58. auto& realm = this->realm();
  59. // 1. Let stream be this.[[stream]].
  60. // 2. If stream is undefined, return a promise rejected with a TypeError exception.
  61. if (!m_stream) {
  62. auto exception = JS::TypeError::create(realm, "Cannot close a writer that has no locked stream"sv);
  63. return WebIDL::create_rejected_promise(realm, exception)->promise();
  64. }
  65. // 3. If ! WritableStreamCloseQueuedOrInFlight(stream) is true, return a promise rejected with a TypeError exception.
  66. if (writable_stream_close_queued_or_in_flight(*m_stream)) {
  67. auto exception = JS::TypeError::create(realm, "Cannot close a stream that is already closed or errored"sv);
  68. return WebIDL::create_rejected_promise(realm, exception)->promise();
  69. }
  70. // 4. Return ! WritableStreamDefaultWriterClose(this).
  71. return TRY(writable_stream_default_writer_close(*this))->promise();
  72. }
  73. // https://streams.spec.whatwg.org/#default-writer-release-lock
  74. WebIDL::ExceptionOr<void> WritableStreamDefaultWriter::release_lock()
  75. {
  76. // 1. Let stream be this.[[stream]].
  77. // 2. If stream is undefined, return.
  78. if (!m_stream)
  79. return {};
  80. // 3. Assert: stream.[[writer]] is not undefined.
  81. VERIFY(m_stream->writer());
  82. // 4. Perform ! WritableStreamDefaultWriterRelease(this).
  83. return writable_stream_default_writer_release(*this);
  84. }
  85. // https://streams.spec.whatwg.org/#default-writer-write
  86. WebIDL::ExceptionOr<JS::GCPtr<JS::Object>> WritableStreamDefaultWriter::write(JS::Value chunk)
  87. {
  88. auto& realm = this->realm();
  89. // 1. If this.[[stream]] is undefined, return a promise rejected with a TypeError exception.
  90. if (!m_stream) {
  91. auto exception = JS::TypeError::create(realm, "Cannot write to a writer that has no locked stream"sv);
  92. return WebIDL::create_rejected_promise(realm, exception)->promise();
  93. }
  94. // 2. Return ! WritableStreamDefaultWriterWrite(this, chunk).
  95. return TRY(writable_stream_default_writer_write(*this, chunk))->promise();
  96. }
  97. WritableStreamDefaultWriter::WritableStreamDefaultWriter(JS::Realm& realm)
  98. : Bindings::PlatformObject(realm)
  99. {
  100. }
  101. void WritableStreamDefaultWriter::initialize(JS::Realm& realm)
  102. {
  103. Base::initialize(realm);
  104. WEB_SET_PROTOTYPE_FOR_INTERFACE(WritableStreamDefaultWriter);
  105. }
  106. void WritableStreamDefaultWriter::visit_edges(Cell::Visitor& visitor)
  107. {
  108. Base::visit_edges(visitor);
  109. visitor.visit(m_closed_promise);
  110. visitor.visit(m_ready_promise);
  111. visitor.visit(m_stream);
  112. }
  113. }