// Copyright 2017 The Chromium 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 CONTENT_COMMON_POSSIBLY_ASSOCIATED_INTERFACE_PTR_H_
#define CONTENT_COMMON_POSSIBLY_ASSOCIATED_INTERFACE_PTR_H_

#include "base/macros.h"
#include "base/single_thread_task_runner.h"
#include "content/common/possibly_associated_interface_ptr_info.h"
#include "mojo/public/cpp/bindings/associated_interface_ptr.h"
#include "mojo/public/cpp/bindings/interface_ptr.h"

namespace content {

// PossiblyAssociatedInterfacePtr<T> contains mojo::InterfacePtr<T> or
// mojo::AssociatedInterfacePtr<T>, but not both. Mojo-related functions in
// mojo::InterfacePtr<T> and mojo::AssociatedInterfacePtr<T> are not accessible,
// but a user can access the raw pointer to the interface.
template <typename T>
class PossiblyAssociatedInterfacePtr final {
 public:
  using InterfaceType = T;
  using PtrInfoType = PossiblyAssociatedInterfacePtrInfo<T>;

  PossiblyAssociatedInterfacePtr() {}
  PossiblyAssociatedInterfacePtr(std::nullptr_t) {}
  PossiblyAssociatedInterfacePtr(mojo::InterfacePtr<T> independent_ptr)
      : independent_ptr_(std::move(independent_ptr)) {}
  PossiblyAssociatedInterfacePtr(mojo::AssociatedInterfacePtr<T> associated_ptr)
      : associated_ptr_(std::move(associated_ptr)) {}

  PossiblyAssociatedInterfacePtr(PossiblyAssociatedInterfacePtr&& other) =
      default;
  ~PossiblyAssociatedInterfacePtr() {}

  PossiblyAssociatedInterfacePtr& operator=(
      PossiblyAssociatedInterfacePtr&& other) = default;

  void Bind(PossiblyAssociatedInterfacePtrInfo<T> info,
            scoped_refptr<base::SingleThreadTaskRunner> runner = nullptr) {
    independent_ptr_.reset();
    associated_ptr_.reset();

    // A PossiblyAssociatedInterfacePtrInfo is guaranteed to only have either a
    // valid InterfacePtrInfo or valid AssociatedInterfacePtrInfo, never both.
    auto independent_ptr_info = info.TakeIndependentPtrInfo();
    if (independent_ptr_info.is_valid())
      independent_ptr_.Bind(std::move(independent_ptr_info), std::move(runner));

    auto associated_ptr_info = info.TakeAssociatedPtrInfo();
    if (associated_ptr_info.is_valid())
      associated_ptr_.Bind(std::move(associated_ptr_info), std::move(runner));
  }

  PossiblyAssociatedInterfacePtrInfo<T> PassInterface() {
    if (independent_ptr_)
      return independent_ptr_.PassInterface();
    if (associated_ptr_)
      return associated_ptr_.PassInterface();
    return nullptr;
  }

  T* get() const {
    return independent_ptr_ ? independent_ptr_.get() : associated_ptr_.get();
  }
  T* operator->() const { return get(); }
  T& operator*() const { return *get(); }
  explicit operator bool() const { return get(); }

 private:
  mojo::InterfacePtr<T> independent_ptr_;
  mojo::AssociatedInterfacePtr<T> associated_ptr_;

  DISALLOW_COPY_AND_ASSIGN(PossiblyAssociatedInterfacePtr);
};

}  // namespace content

#endif  // CONTENT_COMMON_POSSIBLY_ASSOCIATED_INTERFACE_PTR_H_