/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * 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/Portability.h>
#include <folly/executors/InlineExecutor.h>
#include <folly/futures/Promise.h>
#include <folly/lang/Exception.h>

namespace folly {

/*
 * SharedPromise provides the same interface as Promise, but you can extract
 * multiple Futures from it, i.e. you can call getFuture() as many times as
 * you'd like. When the SharedPromise is fulfilled, all of the Futures are
 * completed. Calls to getFuture() after the SharedPromise is fulfilled return
 * a completed Future. If you find yourself constructing collections of Promises
 * and fulfilling them simultaneously with the same value, consider this
 * utility instead. Likewise, if you find yourself in need of setting multiple
 * callbacks on the same Future (which is indefinitely unsupported), consider
 * refactoring to use SharedPromise to "split" the Future.
 *
 * The SharedPromise must be kept alive manually. Consider FutureSplitter for
 * automatic lifetime management.
 */
template <class T>
class SharedPromise {
 public:
  /**
   * Return a Future tied to the shared core state. Unlike Promise::getFuture,
   * this can be called an unlimited number of times per SharedPromise.
   */
  SemiFuture<T> getSemiFuture() const;

  /**
   * Return a Future tied to the shared core state. Unlike Promise::getFuture,
   * this can be called an unlimited number of times per SharedPromise.
   * NOTE: This function is deprecated. Please use getSemiFuture and pass the
   *       appropriate executor to .via on the returned SemiFuture to get a
   *       valid Future where necessary.
   */
  Future<T> getFuture() const;

  /** Return the number of Futures associated with this SharedPromise */
  size_t size() const;

  /** Fulfill the SharedPromise with an exception_wrapper */
  void setException(exception_wrapper ew);

  /** Fulfill the SharedPromise with an exception type E, which can be passed to
    make_exception_wrapper(). Useful for originating exceptions. If you
    caught an exception the exception_wrapper form is more appropriate.
    */
  template <class E>
  typename std::enable_if<std::is_base_of<std::exception, E>::value>::type
  setException(E const&);

  /// Set an interrupt handler to handle interrupts. See the documentation for
  /// Future::raise(). Your handler can do whatever it wants, but if you
  /// bother to set one then you probably will want to fulfill the SharedPromise
  /// with an exception (or special value) indicating how the interrupt was
  /// handled.
  void setInterruptHandler(std::function<void(exception_wrapper const&)>);

  /// Sugar to fulfill this SharedPromise<Unit>
  template <class B = T>
  typename std::enable_if<std::is_same<Unit, B>::value, void>::type setValue() {
    setTry(Try<T>(T()));
  }

  /** Set the value (use perfect forwarding for both move and copy) */
  template <class M>
  void setValue(M&& value);

  void setTry(Try<T>&& t);

  /** Fulfill this SharedPromise with the result of a function that takes no
    arguments and returns something implicitly convertible to T.
    Captures exceptions. e.g.

    p.setWith([] { do something that may throw; return a T; });
  */
  template <class F>
  void setWith(F&& func);

  bool isFulfilled() const;

 private:
  // this allows SharedPromise move-ctor/move-assign to be defaulted
  struct Mutex : std::mutex {
    Mutex() = default;
    Mutex(Mutex&&) noexcept {}
    Mutex& operator=(Mutex&&) noexcept { return *this; }
  };

  template <typename V>
  struct Defaulted {
    using Noexcept = StrictConjunction<
        std::is_nothrow_default_constructible<V>,
        std::is_nothrow_move_constructible<V>,
        std::is_nothrow_move_assignable<V>>;
    V value{V()};
    Defaulted() = default;
    Defaulted(Defaulted&& that) noexcept(Noexcept::value)
        : value(std::exchange(that.value, V())) {}
    Defaulted& operator=(Defaulted&& that) noexcept(Noexcept::value) {
      value = std::exchange(that.value, V());
      return *this;
    }
  };

  bool hasResult() const {
    return try_.value.hasValue() || try_.value.hasException();
  }

  mutable Mutex mutex_;
  mutable Defaulted<size_t> size_;
  Defaulted<Try<T>> try_;
  mutable std::vector<Promise<T>> promises_;
  std::function<void(exception_wrapper const&)> interruptHandler_;
};

} // namespace folly

#include <folly/futures/Future.h>
#include <folly/futures/SharedPromise-inl.h>
