// Copyright 2015 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 CHROMECAST_BASE_BIND_TO_TASK_RUNNER_H_
#define CHROMECAST_BASE_BIND_TO_TASK_RUNNER_H_

#include <memory>
#include <utility>

#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/callback.h"
#include "base/location.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_vector.h"
#include "base/task_runner.h"
#include "base/threading/thread_task_runner_handle.h"

// This is a helper utility for Bind()ing callbacks to a given TaskRunner.
// The typical use is when |a| (of class |A|) wants to hand a callback such as
// base::BindOnce(&A::AMethod, a) to |b|, but needs to ensure that when |b|
// executes the callback, it does so on a specific TaskRunner (for example,
// |a|'s current MessageLoop).
//
// Typical usage: request to be called back on the current thread:
// other->StartAsyncProcessAndCallMeBack(
//    BindToTaskRunner(my_task_runner_,
//                     base::BindOnce(&MyClass::MyMethod, this)));
//
// Note that the callback is always posted to the target TaskRunner.
//
// As a convenience, you can use BindToCurrentThread() to bind to the
// TaskRunner for the current thread (ie, base::ThreadTaskRunnerHandle::Get()).

namespace chromecast {
namespace bind_helpers {

template <typename Sig>
struct BindToTaskRunnerTrampoline;

template <>
struct BindToTaskRunnerTrampoline<void()> {
  static void RunOnce(base::TaskRunner* task_runner,
                      base::OnceClosure&& callback) {
    task_runner->PostTask(FROM_HERE, std::move(callback));
  }

  static void RunRepeating(base::TaskRunner* task_runner,
                           const base::RepeatingClosure& callback) {
    task_runner->PostTask(FROM_HERE, callback);
  }
};

template <typename... Args>
struct BindToTaskRunnerTrampoline<void(Args...)> {
  static void RunOnce(base::TaskRunner* task_runner,
                      base::OnceCallback<void(Args...)>&& callback,
                      Args... args) {
    task_runner->PostTask(
        FROM_HERE,
        base::BindOnce(std::move(callback), std::forward<Args>(args)...));
  }

  static void RunRepeating(
      base::TaskRunner* task_runner,
      const base::RepeatingCallback<void(Args...)>& callback,
      Args... args) {
    task_runner->PostTask(
        FROM_HERE, base::BindOnce(callback, std::forward<Args>(args)...));
  }
};

}  // namespace bind_helpers

template <typename T>
base::OnceCallback<T> BindToTaskRunner(
    scoped_refptr<base::TaskRunner> task_runner,
    base::OnceCallback<T>&& callback) {
  return base::BindOnce(&bind_helpers::BindToTaskRunnerTrampoline<T>::RunOnce,
                        base::RetainedRef(std::move(task_runner)),
                        std::move(callback));
}

template <typename T>
base::RepeatingCallback<T> BindToTaskRunner(
    scoped_refptr<base::TaskRunner> task_runner,
    const base::RepeatingCallback<T>& callback) {
  return base::BindRepeating(
      &bind_helpers::BindToTaskRunnerTrampoline<T>::RunRepeating,
      base::RetainedRef(std::move(task_runner)), callback);
}

template <typename T>
base::OnceCallback<T> BindToCurrentThread(base::OnceCallback<T>&& callback) {
  return BindToTaskRunner(base::ThreadTaskRunnerHandle::Get(),
                          std::move(callback));
}

template <typename T>
base::RepeatingCallback<T> BindToCurrentThread(
    const base::RepeatingCallback<T>& callback) {
  return BindToTaskRunner(base::ThreadTaskRunnerHandle::Get(), callback);
}

}  // namespace chromecast

#endif  // CHROMECAST_BASE_BIND_TO_TASK_RUNNER_H_