/*
 * Copyright 2015 Facebook, Inc.
 *
 * 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.
 */

#pragma once
#include <folly/futures/Future.h>

namespace folly { namespace wangle {

template <typename ExecutorImpl>
class FutureExecutor : public ExecutorImpl {
 public:
  template <typename... Args>
  explicit FutureExecutor(Args&&... args)
    : ExecutorImpl(std::forward<Args>(args)...) {}

  /*
   * Given a function func that returns a Future<T>, adds that function to the
   * contained Executor and returns a Future<T> which will be fulfilled with
   * func's result once it has been executed.
   *
   * For example: auto f = futureExecutor.addFuture([](){
   *                return doAsyncWorkAndReturnAFuture();
   *              });
   */
  template <typename F>
  typename std::enable_if<isFuture<typename std::result_of<F()>::type>::value,
                          typename std::result_of<F()>::type>::type
  addFuture(F func) {
    typedef typename std::result_of<F()>::type::value_type T;
    Promise<T> promise;
    auto future = promise.getFuture();
    auto movePromise = folly::makeMoveWrapper(std::move(promise));
    auto moveFunc = folly::makeMoveWrapper(std::move(func));
    ExecutorImpl::add([movePromise, moveFunc] () mutable {
      (*moveFunc)().then([movePromise] (Try<T>&& t) mutable {
        movePromise->setTry(std::move(t));
      });
    });
    return future;
  }

  /*
   * Similar to addFuture above, but takes a func that returns some non-Future
   * type T.
   *
   * For example: auto f = futureExecutor.addFuture([]() {
   *                return 42;
   *              });
   */
  template <typename F>
  typename std::enable_if<!isFuture<typename std::result_of<F()>::type>::value,
                          Future<typename Unit::Lift<typename std::result_of<F()>::type>::type>>::type
  addFuture(F func) {
    using T = typename Unit::Lift<typename std::result_of<F()>::type>::type;
    Promise<T> promise;
    auto future = promise.getFuture();
    auto movePromise = folly::makeMoveWrapper(std::move(promise));
    auto moveFunc = folly::makeMoveWrapper(std::move(func));
    ExecutorImpl::add([movePromise, moveFunc] () mutable {
      movePromise->setWith(std::move(*moveFunc));
    });
    return future;
  }
};

}}
