// Copyright 2014 The Chromium OS 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 BUFFET_DBUS_COMMAND_PROXY_H_
#define BUFFET_DBUS_COMMAND_PROXY_H_

#include <string>

#include <base/macros.h>
#include <base/scoped_observer.h>
#include <brillo/dbus/data_serialization.h>
#include <brillo/dbus/dbus_object.h>
#include <weave/command.h>

#include "buffet/org.chromium.Buffet.Command.h"

namespace brillo {
namespace dbus_utils {
class ExportedObjectManager;
}  // namespace dbus_utils
}  // namespace brillo

namespace buffet {

class DBusCommandProxy : public org::chromium::Buffet::CommandInterface {
 public:
  DBusCommandProxy(brillo::dbus_utils::ExportedObjectManager* object_manager,
                   const scoped_refptr<dbus::Bus>& bus,
                   const std::weak_ptr<weave::Command>& command,
                   std::string object_path);
  ~DBusCommandProxy() override = default;

  void RegisterAsync(
      const brillo::dbus_utils::AsyncEventSequencer::CompletionAction&
          completion_callback);

 private:
  bool SetProgress(brillo::ErrorPtr* error,
                   const brillo::VariantDictionary& progress) override;
  bool Complete(brillo::ErrorPtr* error,
                const brillo::VariantDictionary& results) override;
  bool Abort(brillo::ErrorPtr* error,
             const std::string& code,
             const std::string& message) override;
  bool SetError(brillo::ErrorPtr* error,
                const std::string& code,
                const std::string& message) override;
  bool Cancel(brillo::ErrorPtr* error) override;

  std::weak_ptr<weave::Command> command_;
  org::chromium::Buffet::CommandAdaptor dbus_adaptor_{this};
  brillo::dbus_utils::DBusObject dbus_object_;

  friend class DBusCommandProxyTest;
  friend class DBusCommandDispacherTest;
  DISALLOW_COPY_AND_ASSIGN(DBusCommandProxy);
};

}  // namespace buffet

#endif  // BUFFET_DBUS_COMMAND_PROXY_H_