import NorgKit
import SwiftUI

struct FlatBlocksView: View {
  let blocks: [NorgBlock]
  let onSetStatus: (Int, TaskStatus) -> Void
  var indent: CGFloat = 0

  @Environment(\.norgTheme) private var theme

  var body: some View {
    let ordinals = orderedOrdinals(blocks)
    VStack(alignment: .leading, spacing: 10) {
      ForEach(Array(blocks.enumerated()), id: \.offset) { index, block in
        row(block, ordinal: ordinals[index])
      }
    }
  }

  @ViewBuilder
  private func row(_ block: NorgBlock, ordinal: Int?) -> some View {
    BlockView(block: block, ordinal: ordinal, onSetStatus: onSetStatus)
      .padding(.leading, indent + levelIndent(block))
      .id(block.line)
      .frame(maxWidth: .infinity, alignment: .leading)

    if let body = block.body, !body.isEmpty {
      FlatBlocksView(blocks: body, onSetStatus: onSetStatus, indent: indent + theme.indentWidth)
    }
  }

  private func levelIndent(_ block: NorgBlock) -> CGFloat {
    switch block {
    case .unorderedListItem(let level, _, _, _),
      .orderedListItem(let level, _, _, _),
      .quote(let level, _, _, _):
      return CGFloat(max(0, level - 1)) * theme.indentWidth
    default:
      return 0
    }
  }
}