// 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 LIB_DRIVER2_DEVFS_EXPORTER_H_
#define LIB_DRIVER2_DEVFS_EXPORTER_H_

#include <fidl/fuchsia.device.fs/cpp/wire.h>
#include <lib/driver2/namespace.h>
#include <lib/fpromise/promise.h>

namespace driver {

// Allows a driver to export a service to devfs.
class DevfsExporter {
 public:
  // Creates a devfs exporter, where `ns` will be used to connect to
  // fuchsia.device.fs.Exporter, and `svc_dir` will be used to find the services
  // to export.
  static zx::status<DevfsExporter> Create(const Namespace& ns, async_dispatcher_t* dispatcher,
                                          fidl::WireSharedClient<fuchsia_io::Directory> svc_dir);

  DevfsExporter() = default;

  // Exports `service_path` to `devfs_path`, with an optionally associated
  // `protocol_id`.
  fpromise::promise<void, zx_status_t> Export(std::string_view service_path,
                                              std::string_view devfs_path,
                                              uint32_t protocol_id = 0) const;

  // Exports `T` to `devfs_path`, with an optionally associated `protocol_id`.
  template <typename T>
  fpromise::promise<void, zx_status_t> Export(std::string_view devfs_path,
                                              uint32_t protocol_id = 0) const {
    return Export(fidl::DiscoverableProtocolName<T>, devfs_path, protocol_id);
  }

 private:
  DevfsExporter(async_dispatcher_t* dispatcher,
                fidl::WireSharedClient<fuchsia_device_fs::Exporter> exporter,
                fidl::WireSharedClient<fuchsia_io::Directory> svc_dir);

  fpromise::promise<void, zx_status_t> ExportImpl(std::string_view service_path,
                                                  std::string_view devfs_path,
                                                  uint32_t protocol_id = 0) const;

  async_dispatcher_t* dispatcher_;
  fidl::WireSharedClient<fuchsia_device_fs::Exporter> exporter_;
  fidl::WireSharedClient<fuchsia_io::Directory> svc_dir_;
};

}  // namespace driver

#endif  // LIB_DRIVER2_DEVFS_EXPORTER_H_