import SwiftUI
import WaterfallGrid

struct ThumbnailGridView<Item: Hashable & Identifiable, Content: View>: View {
  let items: [Item]
  let columnCount: Int
  let useAlternativeGrid: Bool
  let columnsData: [[Item]]
  let content: (Item) -> Content

  var body: some View {
    if useAlternativeGrid {
      HStack(alignment: .top) {
        ForEach(0..<columnCount, id: \.self) { columnIndex in
          LazyVStack {
            ForEach(columnsData[columnIndex], id: \.id) { item in
              content(item)
                .id(item.id)
            }
          }
          .transaction { $0.animation = nil }
        }
      }
      #if os(macOS)
        .padding(8)
      #else
        .padding(.horizontal)
      #endif
      .transition(.opacity)
    } else {
      WaterfallGrid(items, id: \.id) { item in
        content(item)
          .id(item.id)
      }
      .gridStyle(columns: columnCount)
      .transaction { $0.animation = nil }
      #if os(macOS)
        .padding(8)
      #else
        .padding(.horizontal)
      #endif
      .transition(.opacity)
    }
  }
}