// modifications
// Copyright (c) 2022 Marten Richter or other contributers (see commit). All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// original copyright, see LICENSE.chromium
// Copyright (c) 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "src/http3clientsession.h"

#include <utility>

namespace quic
{

  Http3ClientSession::Http3ClientSession(
      const QuicConfig &config, const ParsedQuicVersionVector &supported_versions,
      QuicConnection *connection, const QuicServerId &server_id,
      QuicCryptoClientConfig *crypto_config,
      bool drop_response_body)
      : Http3ClientSession(config, supported_versions, connection, server_id,
                           crypto_config,
                           drop_response_body,
                           /*enable_web_transport=*/false)
  {
  }

  Http3ClientSession::Http3ClientSession(
      const QuicConfig &config, const ParsedQuicVersionVector &supported_versions,
      QuicConnection *connection, const QuicServerId &server_id,
      QuicCryptoClientConfig *crypto_config,
      bool drop_response_body,
      bool enable_web_transport)
      : QuicSpdyClientSession(config, supported_versions, connection, server_id,
                              crypto_config, QuicPriorityType::kWebTransport),
        drop_response_body_(drop_response_body),
        enable_web_transport_(enable_web_transport) {}

  std::unique_ptr<QuicSpdyClientStream>
  Http3ClientSession::CreateClientStream()
  {
    auto stream = std::make_unique<Http3ClientStream>(
        GetNextOutgoingBidirectionalStreamId(), this, BIDIRECTIONAL,
        drop_response_body_);
    stream->set_on_interim_headers(on_interim_headers_);
    return stream;
  }

  WebTransportHttp3VersionSet Http3ClientSession::LocallySupportedWebTransportVersions()
      const
  {
    return enable_web_transport_
               ? kDefaultSupportedWebTransportVersions
               : WebTransportHttp3VersionSet();
  }

  HttpDatagramSupport Http3ClientSession::LocalHttpDatagramSupport()
  {
    return enable_web_transport_ ? HttpDatagramSupport::kRfcAndDraft04
                                 : HttpDatagramSupport::kNone;
  }

  void Http3ClientSession::OnCanCreateNewOutgoingStream(bool unidirectional) {
    if (SupportsWebTransport()) {
      auto itty = svisitors_.begin();
      for (auto itty = svisitors_.begin(); itty != svisitors_.end(); itty++) {
        if (unidirectional) {
          (*itty).second->OnCanCreateNewOutgoingUnidirectionalStream();
        } else {
          (*itty).second->OnCanCreateNewOutgoingBidirectionalStream();
        }
      }
    }
  }
} // namespace quic