// Copyright 2020 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.
#pragma once

#include <array>
#include <span>

#include "pw_assert/light.h"
#include "pw_hdlc_lite/encoder.h"
#include "pw_rpc/channel.h"
#include "pw_stream/stream.h"

namespace pw::hdlc_lite {

// Custom HDLC ChannelOutput class to write and read data through serial using
// the HDLC-Lite protocol.
//
// WARNING: This ChannelOutput is not thread-safe. If thread-safety is required,
// wrap this in a pw::rpc::SynchronizedChannelOutput.
class RpcChannelOutput : public rpc::ChannelOutput {
 public:
  // The RpcChannelOutput class does not own the buffer it uses to store the
  // protobuf bytes. This buffer is specified at the time of creation along with
  // a writer object to which will be used to write and send the bytes.
  constexpr RpcChannelOutput(stream::Writer& writer,
                             std::span<std::byte> buffer,
                             uint8_t address,
                             const char* channel_name)
      : ChannelOutput(channel_name),
        writer_(writer),
        buffer_(buffer),
        address_(address) {}

  std::span<std::byte> AcquireBuffer() override { return buffer_; }

  Status SendAndReleaseBuffer(std::span<const std::byte> buffer) override {
    PW_DASSERT(buffer.data() == buffer_.data());
    if (buffer.empty()) {
      return OkStatus();
    }
    return hdlc_lite::WriteUIFrame(address_, buffer, writer_);
  }

 private:
  stream::Writer& writer_;
  const std::span<std::byte> buffer_;
  const uint8_t address_;
};

// RpcChannelOutput with its own buffer.
//
// WARNING: This ChannelOutput is not thread-safe. If thread-safety is required,
// wrap this in a pw::rpc::SynchronizedChannelOutput.
template <size_t buffer_size>
class RpcChannelOutputBuffer : public rpc::ChannelOutput {
 public:
  constexpr RpcChannelOutputBuffer(stream::Writer& writer,
                                   uint8_t address,
                                   const char* channel_name)
      : ChannelOutput(channel_name), writer_(writer), address_(address) {}

  std::span<std::byte> AcquireBuffer() override { return buffer_; }

  Status SendAndReleaseBuffer(std::span<const std::byte> buffer) override {
    PW_DASSERT(buffer.data() == buffer_.data());
    if (buffer.empty()) {
      return OkStatus();
    }
    return hdlc_lite::WriteUIFrame(address_, buffer, writer_);
  }

 private:
  stream::Writer& writer_;
  std::array<std::byte, buffer_size> buffer_;
  const uint8_t address_;
};

}  // namespace pw::hdlc_lite