// Copyright 2023 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.

#ifndef SRC_DEVELOPER_DEBUG_DEBUG_AGENT_DEBUG_AGENT_SERVER_H_
#define SRC_DEVELOPER_DEBUG_DEBUG_AGENT_DEBUG_AGENT_SERVER_H_

#include <fidl/fuchsia.debugger/cpp/fidl.h>

#include "src/developer/debug/debug_agent/debug_agent_observer.h"
#include "src/developer/debug/ipc/protocol.h"
#include "src/developer/debug/shared/result.h"
#include "src/developer/debug/shared/status.h"
#include "src/lib/fxl/memory/weak_ptr.h"

namespace debug_agent {

class DebugAgent;
class DebuggedProcess;

class DebugAgentServer : public fidl::Server<fuchsia_debugger::DebugAgent>,
                         public DebugAgentObserver {
 public:
  static DebugAgentServer* BindServer(async_dispatcher_t* dispatcher,
                                      fidl::ServerEnd<fuchsia_debugger::DebugAgent> server_end,
                                      fxl::WeakPtr<DebugAgent> debug_agent);

  DebugAgentServer(fxl::WeakPtr<DebugAgent> agent, async_dispatcher_t* dispatcher);

  // fidl::Server<fuchsia_debugger::DebugAgent> implementation.
  void GetAttachedProcesses(GetAttachedProcessesRequest& request,
                            GetAttachedProcessesCompleter::Sync& completer) override;
  void Connect(ConnectRequest& request, ConnectCompleter::Sync& completer) override;
  void AttachTo(AttachToRequest& request, AttachToCompleter::Sync& completer) override;
  void GetProcessInfo(GetProcessInfoRequest& request,
                      GetProcessInfoCompleter::Sync& completer) override;
  void GetMinidumps(GetMinidumpsRequest& request, GetMinidumpsCompleter::Sync& completer) override;

  void OnUnboundFn(DebugAgentServer* impl, fidl::UnbindInfo info,
                   fidl::ServerEnd<fuchsia_debugger::DebugAgent> server_end);
  void handle_unknown_method(fidl::UnknownMethodMetadata<fuchsia_debugger::DebugAgent> metadata,
                             fidl::UnknownMethodCompleter::Sync& completer) override;

  // DebugAgentObserver implementation. For right now we only care about process starting
  // and exception notifications.
  void OnNotification(const debug_ipc::NotifyProcessStarting& notify) override;
  void OnNotification(const debug_ipc::NotifyException& notify) override;
  void OnNotification(const debug_ipc::NotifyComponentStarting& notify) override;
  void OnNotification(const debug_ipc::NotifyFilterCreated& notify) override;

 private:
  // The test fixture is a friend so that we can access the private methods that are doing all the
  // work directly without having to mock out the FIDL primitives.
  friend class DebugAgentServerTest;

  using AddFilterResult =
      debug::Result<debug_ipc::UpdateFilterReply, fuchsia_debugger::FilterError>;

  // Convert |fidl_filter| to a debug_ipc equivalent, then add that to DebugAgent's set of filters.
  AddFilterResult AddFilter(const fuchsia_debugger::Filter& fidl_filter);
  // Adds |filter| to our list of known filters.
  void AddDebugIpcFilter(const debug_ipc::Filter& filter);
  // Synchronizes our internal list of filters with the DebugAgent.
  debug_ipc::UpdateFilterReply SynchronizeDebugIpcFilters();

  // Attempt to attach to all given FilterMatches, some may fail, which is not reported as an error.
  // If any of the FilterMatch identifiers does not correspond to one of our filters, take no
  // action. The number of successful attaches is returned.
  uint32_t AttachToFilterMatches(const std::vector<debug_ipc::FilterMatch>& filter_matches) const;
  debug::Status AttachWithConfig(zx_koid_t koid, const debug_ipc::AttachConfig& config) const;

  using GetMatchingProcessesResult =
      debug::Result<std::vector<DebuggedProcess*>, fuchsia_debugger::FilterError>;

  // If given, returns all attached processes that match the given filter. This will NOT add the
  // filter to DebugAgent's filters and will NOT issue any attach commands to DebugAgent. Otherwise,
  // returns all currently attached processes.
  GetMatchingProcessesResult GetMatchingProcesses(
      std::optional<fuchsia_debugger::Filter> filter) const;

  std::map<debug_ipc::Filter::Identifier, debug_ipc::Filter> filters_;

  std::optional<fidl::ServerBindingRef<fuchsia_debugger::DebugAgent>> binding_ref_;

  fxl::WeakPtr<DebugAgent> debug_agent_;
  async_dispatcher_t* dispatcher_;
};

}  // namespace debug_agent

#endif  // SRC_DEVELOPER_DEBUG_DEBUG_AGENT_DEBUG_AGENT_SERVER_H_