// Copyright 2021 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_GRAPHICS_BIN_OPENCL_LOADER_LOADER_H_
#define SRC_GRAPHICS_BIN_OPENCL_LOADER_LOADER_H_
#include <fidl/fuchsia.opencl.loader/cpp/fidl.h>
#include <lib/component/outgoing/cpp/outgoing_directory.h>

#include <list>
#include <string>
#include <vector>

#include "src/graphics/bin/opencl_loader/app.h"

// Implements the opencl loader's Loader service which provides the client
// driver portion to the loader as a VMO.
class LoaderImpl final : public fidl::Server<fuchsia_opencl_loader::Loader>,
                         public LoaderApp::Observer {
 public:
  // Add a handler for this protocol to |outgoing_dir|. Any connections made to the protocol will
  // create a new loader instance, which keeps a reference to |app|. The loader instance will be
  // alive as long as |dispatcher| has tasks with active connections.
  static zx::result<> Add(component::OutgoingDirectory& outgoing_dir, LoaderApp* app,
                          async_dispatcher_t* dispatcher);

  ~LoaderImpl() final;

 private:
  static fidl::ProtocolHandler<fuchsia_opencl_loader::Loader> GetHandler(
      LoaderApp* app, async_dispatcher_t* dispatcher);

  explicit LoaderImpl(LoaderApp* app) : app_(app) {}

  // LoaderApp::Observer implementation.
  void OnIcdListChanged(LoaderApp* app) override;

  // fidl::Server<fuchsia_opencl_loader::Loader> implementation.
  void Get(GetRequest& request, GetCompleter::Sync& completer) override;
  void ConnectToDeviceFs(ConnectToDeviceFsRequest& request,
                         ConnectToDeviceFsCompleter::Sync& completer) override;
  void ConnectToManifestFs(ConnectToManifestFsRequest& request,
                           ConnectToManifestFsCompleter::Sync& completer) override;
  void GetSupportedFeatures(GetSupportedFeaturesCompleter::Sync& completer) override;

  void AddCallback(std::string name, GetCompleter::Async completer);

  bool waiting_for_callbacks() const {
    return !callbacks_.empty() || !connect_manifest_handles_.empty();
  }

  LoaderApp* app_;
  std::list<std::pair<std::string, GetCompleter::Async>> callbacks_;
  std::vector<fidl::ServerEnd<fuchsia_io::Directory>> connect_manifest_handles_;
};

#endif  // SRC_GRAPHICS_BIN_OPENCL_LOADER_LOADER_H_