OlaServerServiceImpl.h

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Library General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * OlaServerServiceImpl.h
 * Implementation of the OlaService interface
 * Copyright (C) 2005 Simon Newton
 */

#include <memory>
#include <string>
#include <vector>
#include "common/protocol/Ola.pb.h"
#include "common/protocol/OlaService.pb.h"
#include "ola/Callback.h"
#include "ola/rdm/RDMCommand.h"
#include "ola/rdm/UID.h"
#include "ola/rdm/UIDSet.h"

#ifndef OLAD_OLASERVERSERVICEIMPL_H_
#define OLAD_OLASERVERSERVICEIMPL_H_

namespace ola {

class OlaServerServiceImpl : public ola::proto::OlaServerService {
 public:
  typedef Callback0<void> ReloadPluginsCallback;

  OlaServerServiceImpl(class UniverseStore *universe_store,
                       class DeviceManager *device_manager,
                       class PluginManager *plugin_manager,
                       class PortManager *port_manager,
                       class ClientBroker *broker,
                       const class TimeStamp *wake_up_time,
                       ReloadPluginsCallback *reload_plugins_callback);

  ~OlaServerServiceImpl() {}

  void GetDmx(ola::rpc::RpcController* controller,
              const ola::proto::UniverseRequest* request,
              ola::proto::DmxData* response,
              ola::rpc::RpcService::CompletionCallback* done);


  void RegisterForDmx(ola::rpc::RpcController* controller,
                      const ola::proto::RegisterDmxRequest* request,
                      ola::proto::Ack* response,
                      ola::rpc::RpcService::CompletionCallback* done);

  void UpdateDmxData(ola::rpc::RpcController* controller,
                     const ola::proto::DmxData* request,
                     ola::proto::Ack* response,
                     ola::rpc::RpcService::CompletionCallback* done);
  void StreamDmxData(ola::rpc::RpcController* controller,
                     const ::ola::proto::DmxData* request,
                     ::ola::proto::STREAMING_NO_RESPONSE* response,
                     ola::rpc::RpcService::CompletionCallback* done);


  void SetUniverseName(ola::rpc::RpcController* controller,
                       const ola::proto::UniverseNameRequest* request,
                       ola::proto::Ack* response,
                       ola::rpc::RpcService::CompletionCallback* done);

  void SetMergeMode(ola::rpc::RpcController* controller,
                    const ola::proto::MergeModeRequest* request,
                    ola::proto::Ack* response,
                    ola::rpc::RpcService::CompletionCallback* done);

  void PatchPort(ola::rpc::RpcController* controller,
                 const ola::proto::PatchPortRequest* request,
                 ola::proto::Ack* response,
                 ola::rpc::RpcService::CompletionCallback* done);

  void SetPortPriority(ola::rpc::RpcController* controller,
                       const ola::proto::PortPriorityRequest* request,
                       ola::proto::Ack* response,
                       ola::rpc::RpcService::CompletionCallback* done);

  void GetUniverseInfo(ola::rpc::RpcController* controller,
                       const ola::proto::OptionalUniverseRequest* request,
                       ola::proto::UniverseInfoReply* response,
                       ola::rpc::RpcService::CompletionCallback* done);

  void GetPlugins(ola::rpc::RpcController* controller,
                  const ola::proto::PluginListRequest* request,
                  ola::proto::PluginListReply* response,
                  ola::rpc::RpcService::CompletionCallback* done);

  void ReloadPlugins(ola::rpc::RpcController* controller,
                     const ::ola::proto::PluginReloadRequest* request,
                     ::ola::proto::Ack* response,
                     ola::rpc::RpcService::CompletionCallback* done);

  void GetPluginDescription(
      ola::rpc::RpcController* controller,
      const ola::proto::PluginDescriptionRequest* request,
      ola::proto::PluginDescriptionReply* response,
      ola::rpc::RpcService::CompletionCallback* done);

  void GetPluginState(
      ola::rpc::RpcController* controller,
      const ola::proto::PluginStateRequest* request,
      ola::proto::PluginStateReply* response,
      ola::rpc::RpcService::CompletionCallback* done);

  void GetDeviceInfo(ola::rpc::RpcController* controller,
                     const ola::proto::DeviceInfoRequest* request,
                     ola::proto::DeviceInfoReply* response,
                     ola::rpc::RpcService::CompletionCallback* done);

  void GetCandidatePorts(ola::rpc::RpcController* controller,
                         const ola::proto::OptionalUniverseRequest* request,
                         ola::proto::DeviceInfoReply* response,
                         ola::rpc::RpcService::CompletionCallback* done);

  void ConfigureDevice(ola::rpc::RpcController* controller,
                       const ola::proto::DeviceConfigRequest* request,
                       ola::proto::DeviceConfigReply* response,
                       ola::rpc::RpcService::CompletionCallback* done);

  void GetUIDs(ola::rpc::RpcController* controller,
               const ola::proto::UniverseRequest* request,
               ola::proto::UIDListReply* response,
               ola::rpc::RpcService::CompletionCallback* done);

  void ForceDiscovery(ola::rpc::RpcController* controller,
                      const ola::proto::DiscoveryRequest* request,
                      ola::proto::UIDListReply* response,
                      ola::rpc::RpcService::CompletionCallback* done);

  void RDMCommand(ola::rpc::RpcController* controller,
                  const ::ola::proto::RDMRequest* request,
                  ola::proto::RDMResponse* response,
                  ola::rpc::RpcService::CompletionCallback* done);


  void RDMDiscoveryCommand(ola::rpc::RpcController* controller,
                           const ::ola::proto::RDMDiscoveryRequest* request,
                           ola::proto::RDMResponse* response,
                           ola::rpc::RpcService::CompletionCallback* done);

  void SetSourceUID(ola::rpc::RpcController* controller,
                    const ::ola::proto::UID* request,
                    ola::proto::Ack* response,
                    ola::rpc::RpcService::CompletionCallback* done);

  void SendTimeCode(ola::rpc::RpcController* controller,
                    const ::ola::proto::TimeCode* request,
                    ::ola::proto::Ack* response,
                    ola::rpc::RpcService::CompletionCallback* done);

 private:
  void HandleRDMResponse(ola::proto::RDMResponse* response,
                         ola::rpc::RpcService::CompletionCallback* done,
                         bool include_raw_packets,
                         ola::rdm::rdm_response_code code,
                         const ola::rdm::RDMResponse *rdm_response,
                         const std::vector<std::string> &packets);
  void RDMDiscoveryComplete(unsigned int universe,
                            ola::rpc::RpcService::CompletionCallback* done,
                            ola::proto::UIDListReply *response,
                            const ola::rdm::UIDSet &uids);

  void MissingUniverseError(ola::rpc::RpcController* controller);
  void MissingPluginError(ola::rpc::RpcController* controller);
  void MissingDeviceError(ola::rpc::RpcController* controller);
  void MissingPortError(ola::rpc::RpcController* controller);

  void AddPlugin(class AbstractPlugin *plugin,
                 ola::proto::PluginInfo *plugin_info) const;
  void AddDevice(class AbstractDevice *device,
                 unsigned int alias,
                 ola::proto::DeviceInfoReply* response) const;

  template <class PortClass>
  void PopulatePort(const PortClass &port,
                    ola::proto::PortInfo *port_info) const;

  void SetProtoUID(const ola::rdm::UID &uid, ola::proto::UID *pb_uid);

  class Client* GetClient(ola::rpc::RpcController *controller);

  UniverseStore *m_universe_store;
  DeviceManager *m_device_manager;
  class PluginManager *m_plugin_manager;
  class PortManager *m_port_manager;
  class ClientBroker *m_broker;
  const class TimeStamp *m_wake_up_time;
  std::auto_ptr<ReloadPluginsCallback> m_reload_plugins_callback;
};
}  // namespace ola
#endif  // OLAD_OLASERVERSERVICEIMPL_H_