import ComposableArchitecture
import SwiftUI

// MARK: - List
@Reducer
struct ListTCA {
  @ObservableState
  struct State: Equatable {
    var cells: IdentifiedArrayOf<CellTCA.State> = []
  }
  
  enum Action {
    case addTapped
    case cellAction(IdentifiedActionOf<CellTCA>)
  }
  
  var body: some ReducerOf<Self> {
    Reduce { state, action in
      switch action {

      case .addTapped:
        let newId = state.cells.count
        state.cells.append(.init(id: newId))
        return .none

      case .cellAction:
        return .none
      }
    }
    .forEach(\.cells, action: \.cellAction) {
      CellTCA()
    }
  }
}

struct ContentViewTCA: View {
  let store = Store(initialState: .init()) {
    ListTCA()
  }
  var body: some View {
    List {
      Button("Add row") {
        store.send(.addTapped)
      }

      ForEach(store.scope(state: \.cells, action: \.cellAction)) { cellStore in
        CellView(store: cellStore)
      }
    }
    .listStyle(.plain)
  }
}

// MARK: - Cell
@Reducer
struct CellTCA {
  @ObservableState
  struct State: Equatable, Identifiable {
    let id: Int
    var text: String? = nil

    var cellId: String {
      id.description
    }
  }
  
  enum Action {
    case onAppear
    case contentFetched(String)
  }

  var body: some ReducerOf<Self> {
    Reduce { state, action in
      switch action {
      case .onAppear:
        // asynchronously load content
        return .run { [id = state.id] send in
          await send(.contentFetched("Content of \(id) fetched at \(Date())"))
        }
        
      case .contentFetched(let content):
        state.text = content
        return .none
      }
    }
  }
}

struct CellView: View {
  let store: StoreOf<CellTCA>
  var body: some View {
    let _ = print("rendering cell \(store.cellId)")

    VStack {
      Text(store.text ?? "")
        .onAppear { store.send(.onAppear) }
    }
  }
}

#Preview {
  ContentViewTCA()
}

import SwiftUI

// MARK: - List
@MainActor class ListStoreNonTCA: ObservableObject {
  struct State {
    var cellStores: [CellStoreNonTCA] = []
  }
  @Published var state: State
  init() {
    self.state = State()
  }
  
  func addTapped() {
    let newId = state.cellStores.count
    state.cellStores.append(.init(state: .init(id: newId)))
  }
}

struct ContentViewNonTCA: View {
  @ObservedObject var store = ListStoreNonTCA()
  
  var body: some View {
    List {
      Button("Add row") {
        store.addTapped()
      }

      ForEach(store.state.cellStores, id: \.state.id) { cellStore in
        CellViewNonTCA(store: cellStore)
      }
    }
    .listStyle(.plain)
  }
}

// MARK: - Cell
@MainActor final class CellStoreNonTCA: ObservableObject {
  struct State {
    let id: Int
    var text: String? = nil
  }
  @Published var state: State
  
  init(state: State) {
    self.state = state
  }
  
  func onAppear() {
    Task {
      state.text = "Content fetched at \(Date())"
    }
  }
}

struct CellViewNonTCA: View {
  @ObservedObject var store: CellStoreNonTCA

  var body: some View {
    let _ = print("rendering cell \(store.state.id)")

    VStack {
      Text(store.state.text ?? "")
        .onAppear { store.onAppear() }
    }
  }
}

#Preview {
  ContentViewNonTCA()
}