import CommonCorePublic
import LayoutKit

extension [Div] {
  func makeBlocks<T>(
    context: DivBlockModelingContext,
    sizeModifier: DivSizeModifier? = nil,
    mappedBy modificator: (Div, Block, DivBlockModelingContext) throws -> T
  ) rethrows -> [T] {
    try iterativeFlatMap { div, index in
      let itemContext = context.modifying(
        parentPath: context.parentPath + index,
        sizeModifier: sizeModifier
      )
      let block: Block
      do {
        block = try modifyError({
          DivBlockModelingError($0.message, path: itemContext.parentPath)
        }) {
          try div.value.makeBlock(context: itemContext)
        }
      } catch {
        context.addError(error: error)
        return nil
      }
      return try modificator(div, block, itemContext)
    }
  }

  func makeBlocks(context: DivBlockModelingContext) -> [Block] {
    makeBlocks(context: context, mappedBy: { _, block, _ in block })
  }
}