/*
 * 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 <react/renderer/graphics/Color.h>
#include <react/renderer/graphics/Float.h>

#include <optional>
#include <string>
#include <string_view>
#include <variant>

namespace facebook::react {

enum class FilterType {
  Blur,
  Brightness,
  Contrast,
  Grayscale,
  HueRotate,
  Invert,
  Opacity,
  Saturate,
  Sepia,
  DropShadow
};

inline std::optional<FilterType> filterTypeFromString(std::string_view filterName)
{
  if (filterName == "blur") {
    return FilterType::Blur;
  } else if (filterName == "brightness") {
    return FilterType::Brightness;
  } else if (filterName == "contrast") {
    return FilterType::Contrast;
  } else if (filterName == "grayscale") {
    return FilterType::Grayscale;
  } else if (filterName == "hueRotate") {
    return FilterType::HueRotate;
  } else if (filterName == "invert") {
    return FilterType::Invert;
  } else if (filterName == "opacity") {
    return FilterType::Opacity;
  } else if (filterName == "saturate") {
    return FilterType::Saturate;
  } else if (filterName == "sepia") {
    return FilterType::Sepia;
  } else if (filterName == "dropShadow") {
    return FilterType::DropShadow;
  } else {
    return std::nullopt;
  }
}

inline std::string toString(const FilterType &filterType)
{
  switch (filterType) {
    case FilterType::Blur:
      return "blur";
    case FilterType::Brightness:
      return "brightness";
    case FilterType::Contrast:
      return "contrast";
    case FilterType::Grayscale:
      return "grayscale";
    case FilterType::HueRotate:
      return "hueRotate";
    case FilterType::Invert:
      return "invert";
    case FilterType::Opacity:
      return "opacity";
    case FilterType::Saturate:
      return "saturate";
    case FilterType::Sepia:
      return "sepia";
    case FilterType::DropShadow:
      return "dropShadow";
    default:
      abort();
  }
}

struct DropShadowParams {
  bool operator==(const DropShadowParams &other) const = default;

  Float offsetX{};
  Float offsetY{};
  Float standardDeviation{};
  SharedColor color{};

#ifdef RN_SERIALIZABLE_STATE
  folly::dynamic toDynamic() const
  {
    folly::dynamic result = folly::dynamic::object();
    result["offsetX"] = offsetX;
    result["offsetY"] = offsetY;
    result["standardDeviation"] = standardDeviation;
    result["color"] = *color;
    return result;
  }
#endif
};

struct FilterFunction {
  bool operator==(const FilterFunction &other) const = default;

  FilterType type{};
  std::variant<Float, DropShadowParams> parameters{};

#ifdef RN_SERIALIZABLE_STATE
  folly::dynamic toDynamic() const
  {
    folly::dynamic result = folly::dynamic::object();
    std::string typeKey = toString(type);
    if (std::holds_alternative<Float>(parameters)) {
      result[typeKey] = std::get<Float>(parameters);
    } else if (std::holds_alternative<DropShadowParams>(parameters)) {
      const auto &dropShadowParams = std::get<DropShadowParams>(parameters);
      result[typeKey] = dropShadowParams.toDynamic();
    }
    return result;
  }
#endif
};

#ifdef RN_SERIALIZABLE_STATE
inline folly::dynamic toDynamic(const FilterFunction &filterFunction)
{
  return filterFunction.toDynamic();
}
#endif

} // namespace facebook::react