bin/mdns/util/mdns_impl.h

// Copyright 2017 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 GARNET_BIN_MDNS_UTIL_MDNS_IMPL_H_
#define GARNET_BIN_MDNS_UTIL_MDNS_IMPL_H_

#include <fuchsia/mdns/cpp/fidl.h>
#include <lib/fit/function.h>
#include <lib/zx/channel.h>

#include "garnet/bin/mdns/util/mdns_params.h"
#include "lib/component/cpp/startup_context.h"
#include "lib/fsl/tasks/fd_waiter.h"
#include "lib/fxl/macros.h"
#include "lib/mdns/cpp/service_subscriber.h"

namespace mdns {

class MdnsImpl : public fuchsia::mdns::Responder {
 public:
  MdnsImpl(component::StartupContext* startup_context, MdnsParams* params,
           fit::closure quit_callback);

  ~MdnsImpl() override;

 private:
  void WaitForKeystroke();

  void HandleKeystroke();

  void Resolve(const std::string& host_name, uint32_t timeout_seconds);

  void Subscribe(const std::string& service_name);

  void Publish(const std::string& service_name,
               const std::string& instance_name, uint16_t port,
               const std::vector<std::string>& text);

  void Unpublish(const std::string& service_name,
                 const std::string& instance_name);

  void Respond(const std::string& service_name,
               const std::string& instance_name, uint16_t port,
               const std::vector<std::string>& announce,
               const std::vector<std::string>& text);

  // Responder implementation:
  void UpdateStatus(fuchsia::mdns::Result result) override;

  void GetPublication(bool query, fidl::StringPtr subtype,
                      GetPublicationCallback callback) override;

  fit::closure quit_callback_;
  fuchsia::mdns::ControllerPtr controller_;
  ServiceSubscriber subscriber_;
  fidl::Binding<fuchsia::mdns::Responder> binding_;
  fsl::FDWaiter fd_waiter_;

  uint16_t publication_port_;
  std::vector<std::string> publication_text_;

  FXL_DISALLOW_COPY_AND_ASSIGN(MdnsImpl);
};

}  // namespace mdns

#endif  // GARNET_BIN_MDNS_UTIL_MDNS_IMPL_H_