import DivKit
import LayoutKit
import VGSL

public final class TextExtensionHandler: DivExtensionHandler {
  public let id: String

  private let text: String?

  public var accessibilityElement: AccessibilityElement? {
    .none(label: text)
  }

  public init(
    id: String,
    text: String?
  ) {
    self.id = id
    self.text = text
  }

  public func applyBeforeBaseProperties(
    to block: Block,
    div _: DivBase,
    context _: DivBlockModelingContext
  ) -> Block {
    guard
      let textBlock = block as? TextBlock,
      let text,
      let newTextBlock = TextBlock(copyingAttributesFrom: textBlock, text: text)
    else {
      return block
    }

    return newTextBlock
  }
}