// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/media_router/common/providers/cast/channel/keep_alive_delegate.h"

#include <string>
#include <utility>

#include "base/functional/bind.h"
#include "components/media_router/common/providers/cast/channel/cast_channel_enum.h"
#include "components/media_router/common/providers/cast/channel/cast_socket.h"
#include "components/media_router/common/providers/cast/channel/logger.h"
#include "net/base/net_errors.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
#include "third_party/openscreen/src/cast/common/channel/proto/cast_channel.pb.h"

namespace cast_channel {

KeepAliveDelegate::KeepAliveDelegate(
    CastSocket* socket,
    scoped_refptr<Logger> logger,
    std::unique_ptr<CastTransport::Delegate> inner_delegate,
    base::TimeDelta ping_interval,
    base::TimeDelta liveness_timeout)
    : inner_delegate_(std::move(inner_delegate)),
      handler_(socket,
               std::move(logger),
               ping_interval,
               liveness_timeout,
               base::BindRepeating(&KeepAliveDelegate::OnError,
                                   base::Unretained(this))) {
  DCHECK(inner_delegate_);
}

KeepAliveDelegate::~KeepAliveDelegate() = default;

void KeepAliveDelegate::SetTimersForTest(
    std::unique_ptr<base::RetainingOneShotTimer> injected_ping_timer,
    std::unique_ptr<base::RetainingOneShotTimer> injected_liveness_timer) {
  handler_.SetTimersForTest(std::move(injected_ping_timer),
                            std::move(injected_liveness_timer));
}

// CastTransport::Delegate interface.
void KeepAliveDelegate::Start() {
  handler_.Start();
  inner_delegate_->Start();
}

void KeepAliveDelegate::OnError(ChannelError error_state) {
  DVLOG(1) << "KeepAlive::OnError: "
           << ::cast_channel::ChannelErrorToString(error_state);
  inner_delegate_->OnError(error_state);
  handler_.Stop();
}

void KeepAliveDelegate::OnMessage(const CastMessage& message) {
  DVLOG(2) << "KeepAlive::OnMessage : " << message.payload_utf8();
  if (!handler_.HandleMessage(message)) {
    inner_delegate_->OnMessage(message);
  }
}

}  // namespace cast_channel