//
// Copyright (C) 2016 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

#include "dhcp_client/dbus/dbus_control.h"

#include "dhcp_client/dbus/manager_dbus_adaptor.h"
#include "dhcp_client/dbus/service_dbus_adaptor.h"
#include "dhcp_client/manager.h"
#include "dhcp_client/service.h"


using brillo::dbus_utils::AsyncEventSequencer;
using brillo::dbus_utils::ExportedObjectManager;

namespace dhcp_client {

namespace {
const char kServiceName[] = "org.chromium.dhcp_client";
const char kServicePath[] = "/org/chromium/dhcp_client";
}  // namespace

DBusControl::DBusControl() {}

DBusControl::~DBusControl() {}

void DBusControl::Init() {
  // Setup bus connection.
  dbus::Bus::Options options;
  options.bus_type = dbus::Bus::SYSTEM;
  bus_ = new dbus::Bus(options);
  CHECK(bus_->Connect());

  // Create and register ObjectManager.
  scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer());
  object_manager_.reset(
      new ExportedObjectManager(bus_, dbus::ObjectPath(kServicePath)));
  object_manager_->RegisterAsync(
      sequencer->GetHandler("ObjectManager.RegisterAsync() failed.", true));

  // Create and register Manager.
  manager_.reset(new Manager(this));
  manager_->RegisterAsync(
      sequencer->GetHandler("Manager.RegisterAsync() failed.", true));

  // Take over the service ownership once the objects registration is completed.
  sequencer->OnAllTasksCompletedCall({
    base::Bind(&DBusControl::OnObjectRegistrationCompleted,
               base::Unretained(this))
  });
}

void DBusControl::Shutdown() {
  manager_.reset();
  object_manager_.reset();
  if (bus_) {
    bus_->ShutdownAndBlock();
  }
}

void DBusControl::OnObjectRegistrationCompleted(bool registration_success) {
  // Success should always be true since we've said that failures are fatal.
  CHECK(registration_success) << "Init of one or more objects has failed.";
  CHECK(bus_->RequestOwnershipAndBlock(kServiceName,
                                       dbus::Bus::REQUIRE_PRIMARY))
      << "Unable to take ownership of " << kServiceName;

}

std::unique_ptr<ManagerAdaptorInterface> DBusControl::CreateManagerAdaptor(
    Manager* manager) {
  return std::unique_ptr<ManagerAdaptorInterface>(
      new ManagerDBusAdaptor(bus_, object_manager_.get(), manager));
}

std::unique_ptr<ServiceAdaptorInterface> DBusControl::CreateServiceAdaptor(
    Service* service) {
  return std::unique_ptr<ServiceAdaptorInterface>(
      new ServiceDBusAdaptor(bus_, object_manager_.get(), service));
}


}  // namespace dhcp_client