#include "rive/text/text_style_background.hpp"
#include "rive/text/text_style_paint.hpp"
#include "rive/text/text.hpp"
#include "rive/shapes/paint/feather.hpp"
#include "rive/shapes/paint/shape_paint.hpp"

using namespace rive;

StatusCode TextStyleBackground::onAddedDirty(CoreContext* context)
{
  StatusCode code = Super::onAddedDirty(context);
  if (code == StatusCode::Ok)
  {
    if (!parent()->is<TextStylePaint>())
    {
      return StatusCode::InvalidObject;
    }
    parent()->as<TextStylePaint>()->background(this);
  }
  return code;
}

TextStylePaint* TextStyleBackground::style() const
{
  return parent()->as<TextStylePaint>();
}

void TextStyleBackground::resetPath()
{
  m_rects.clear();
  m_path.rewind();
}

void TextStyleBackground::addRect(const AABB& rect) { m_rects.push_back(rect); }

void TextStyleBackground::buildDependencies()
{
  Super::buildDependencies();
  // Nothing adds us to the dependency graph otherwise, and a Feather under
  // one of our paints hangs off us -- without this it would never update.
  // Depending on the TextStylePaint (which itself depends on the Text) also
  // guarantees we sort after the text has laid out and filled our rects.
  if (parent() != nullptr)
  {
    parent()->addDependent(this);
  }
}

void TextStyleBackground::updatePath()
{
  m_path.update(m_rects, cornerRadius());
  // Text rebuilds this path from its own update pass; the feathers derive
  // their geometry from it and sort after us, so dirty them to run again.
  for (auto shapePaint : m_ShapePaints)
  {
    auto feather = shapePaint->feather();
    if (feather != nullptr)
    {
      feather->addDirt(ComponentDirt::Path);
    }
  }
}

void TextStyleBackground::draw(Renderer* renderer, const Mat2D& worldTransform)
{
  if (m_rects.empty())
  {
    return;
  }
  for (auto shapePaint : m_ShapePaints)
  {
    if (!shapePaint->shouldDraw())
    {
      continue;
    }
    // A global text style's background lives under the style artboard,
    // not a Text; keep its paint's own blend mode then.
    if (style()->parent()->is<Text>())
    {
      auto text = style()->parent()->as<Text>();
      shapePaint->blendMode(text->blendMode(), text->additiveAmount());
    }
    shapePaint->draw(renderer, &m_path, worldTransform, true);
  }
}

const Mat2D& TextStyleBackground::shapeWorldTransform() const
{
  return style()->shapeWorldTransform();
}

Component* TextStyleBackground::pathBuilder() { return style()->pathBuilder(); }

void TextStyleBackground::cornerRadiusChanged()
{
  if (parent() == nullptr || parent()->parent() == nullptr ||
      !parent()->parent()->is<Text>())
  {
    return;
  }
  parent()->parent()->as<Text>()->markPaintDirty();
}