// Copyright 2018 The Fuchsia 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/developer/debug/debug_agent/local_stream_backend.h"

#include <lib/syslog/cpp/macros.h>

#include "src/developer/debug/ipc/message_reader.h"
#include "src/developer/debug/ipc/protocol.h"
#include "src/developer/debug/shared/logging/logging.h"

namespace debug_agent {

LocalStreamBackend::LocalStreamBackend() { stream().set_writer(this); }

LocalStreamBackend::~LocalStreamBackend() = default;

size_t LocalStreamBackend::ConsumeStreamBufferData(const char* data, size_t len) {
  // We assume we always get a header.
  const debug_ipc::MsgHeader* header = reinterpret_cast<const debug_ipc::MsgHeader*>(data);

  // Buffer for the message and create a reader.
  std::vector<char> msg_buffer;
  msg_buffer.reserve(len);
  msg_buffer.insert(msg_buffer.end(), data, data + len);

  // Dispatch the messages we find interesting.
  // NOTE: Here is where you add more notification handlers as they are sent by
  //       the debug agent.
  DEBUG_LOG(Test) << "Got notification: " << debug_ipc::MsgHeader::TypeToString(header->type);
  switch (header->type) {
    case debug_ipc::MsgHeader::Type::kAttach: {
      debug_ipc::AttachReply attach;
      uint32_t transaction = 0;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &attach, &transaction,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleAttach(std::move(attach));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyComponentDiscovered: {
      debug_ipc::NotifyComponentDiscovered component_discovered;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &component_discovered,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyComponentDiscovered(component_discovered);
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyComponentStarting: {
      debug_ipc::NotifyComponentStarting component_starting;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &component_starting,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyComponentStarting(component_starting);
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyException: {
      debug_ipc::NotifyException exception;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &exception,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyException(std::move(exception));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyFilterCreated: {
      debug_ipc::NotifyFilterCreated filter;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &filter,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyFilterCreated(std::move(filter));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyIO: {
      debug_ipc::NotifyIO io;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &io, debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyIO(std::move(io));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyModules: {
      debug_ipc::NotifyModules modules;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &modules,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyModules(std::move(modules));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyProcessExiting: {
      debug_ipc::NotifyProcessExiting process;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &process,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyProcessExiting(std::move(process));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyProcessStarting: {
      debug_ipc::NotifyProcessStarting process;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &process,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyProcessStarting(std::move(process));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyThreadExiting: {
      debug_ipc::NotifyThreadExiting thread;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &thread,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyThreadExiting(std::move(thread));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyThreadStarting: {
      debug_ipc::NotifyThreadStarting thread;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &thread,
                                  debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyThreadStarting(std::move(thread));
      break;
    }
    case debug_ipc::MsgHeader::Type::kNotifyLog: {
      debug_ipc::NotifyLog log;
      if (!debug_ipc::Deserialize(std::move(msg_buffer), &log, debug_ipc::kCurrentProtocolVersion))
        FX_NOTREACHED();
      HandleNotifyLog(std::move(log));
      break;
    }
    default:
      FX_NOTREACHED() << "Unhandled notification: "
                      << debug_ipc::MsgHeader::TypeToString(header->type);
      break;
  }

  // Say we read the whole message.
  return len;
}

}  // namespace debug_agent