// Copyright 2021 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.

#include "pw_rpc/integration_testing.h"

#include <limits>

#include "gtest/gtest.h"
#include "pw_log/log.h"
#include "pw_rpc/integration_test_socket_client.h"
#include "pw_unit_test/logging_event_handler.h"

namespace pw::rpc::integration_test {
namespace {

// Hard-coded to 1055 bytes, which is enough to fit 512-byte payloads when using
// HDLC framing.
SocketClientContext<1055> context;
unit_test::LoggingEventHandler log_test_events;

}  // namespace

Client& client() { return context.client(); }

int GetClientSocketFd() { return context.GetSocketFd(); }

void SetEgressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
  context.SetEgressChannelManipulator(new_channel_manipulator);
}

void SetIngressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
  context.SetIngressChannelManipulator(new_channel_manipulator);
}

Status InitializeClient(int port) {
  unit_test::RegisterEventHandler(&log_test_events);
  if (port <= 0 || port > std::numeric_limits<uint16_t>::max()) {
    PW_LOG_CRITICAL("Port numbers must be between 1 and 65535; %d is invalid",
                    port);
    return Status::InvalidArgument();
  }

  PW_LOG_INFO("Connecting to pw_rpc client at localhost:%d", port);
  return context.Start(port);
}

void TerminateClient() { context.Terminate(); }

Status InitializeClient(int argc, char* argv[], const char* usage_args) {
  if (argc < 2) {
    PW_LOG_INFO("Usage: %s %s", argv[0], usage_args);
    return Status::InvalidArgument();
  }

  const int port = std::atoi(argv[1]);
  return InitializeClient(port);
}

}  // namespace pw::rpc::integration_test