divkit/client/ios/DivKit/Extensions/DivGridExtensions.swift

import LayoutKit
import VGSL

extension DivGrid: DivBlockModeling {
  public func makeBlock(context: DivBlockModelingContext) throws -> Block {
    let path = context.parentPath + (id ?? DivGrid.type)
    let gridContext = context.modifying(parentPath: path)
    return try modifyError({ DivBlockModelingError($0.message, path: path) }) {
      try applyBaseProperties(
        to: { try makeBaseBlock(context: gridContext) },
        context: gridContext,
        actionsHolder: self
      )
    }
  }

  var nonNilItems: [Div] {
    items ?? []
  }

  private func makeBaseBlock(context: DivBlockModelingContext) throws -> Block {
    let path = context.parentPath
    let itemsContext = context.modifying(errorsStorage: DivErrorsStorage(errors: []))
    let items = nonNilItems.enumerated().compactMap { tuple in
      let itemContext = itemsContext.modifying(parentPath: path + tuple.offset)
      do {
        return try tuple.element.value.makeGridItem(context: itemContext)
      } catch {
        itemContext.addError(error: error)
        return nil
      }
    }

    if nonNilItems.count != items.count {
      throw DivBlockModelingError(
        "Unable to form grid",
        path: path,
        causes: itemsContext.errorsStorage.errors
      )
    }

    if items.isEmpty {
      return EmptyBlock(
        widthTrait: resolveContentWidthTrait(context),
        heightTrait: resolveContentHeightTrait(context)
      )
    }

    let expressionResolver = context.expressionResolver
    return try GridBlock(
      widthTrait: resolveContentWidthTrait(context),
      heightTrait: resolveContentHeightTrait(context),
      contentAlignment: resolveContentAlignment(expressionResolver),
      items: items,
      columnCount: resolveColumnCount(expressionResolver) ?? 0
    )
  }

  private func resolveContentAlignment(
    _ expressionResolver: ExpressionResolver
  ) -> BlockAlignment2D {
    BlockAlignment2D(
      horizontal: resolveContentAlignmentHorizontal(expressionResolver).alignment,
      vertical: resolveContentAlignmentVertical(expressionResolver).alignment
    )
  }
}

extension DivBase {
  fileprivate func makeGridItem(
    context: DivBlockModelingContext
  ) throws -> GridBlock.Item {
    let block = try makeBlock(context: context)
    let expressionResolver = context.expressionResolver
    return .init(
      span: GridBlock.Span(
        rows: resolveRowSpan(expressionResolver) ?? 1,
        columns: resolveColumnSpan(expressionResolver) ?? 1
      ),
      weight: .init(
        column: width.resolveWeight(expressionResolver),
        row: height.resolveWeight(expressionResolver)
      ),
      contents: block,
      alignment: resolveAlignment(context, defaultAlignment: .default)
    )
  }
}

extension DivSize {
  fileprivate func resolveWeight(
    _ expressionResolver: ExpressionResolver
  ) -> LayoutTrait.Weight? {
    switch self {
    case .divFixedSize, .divWrapContentSize:
      nil
    case let .divMatchParentSize(size):
      size.resolveWeight(expressionResolver)
        .flatMap { LayoutTrait.Weight(floatLiteral: $0) }
    }
  }
}