IceCubesApp/Packages/Notifications/Sources/Notifications/NotificationsListView.swift

import SwiftUI
import Network
import Models
import Shimmer
import DesignSystem
import Env

public struct NotificationsListView: View {
  @EnvironmentObject private var theme: Theme
  @EnvironmentObject private var watcher: StreamWatcher
  @EnvironmentObject private var client: Client
  @StateObject private var viewModel = NotificationsViewModel()
  
  public init() { }
  
  public var body: some View {
    ScrollView {
      LazyVStack {
        if client.isAuth {
          Picker("", selection: $viewModel.tab) {
            ForEach(NotificationsViewModel.Tab.allCases, id: \.self) { tab in
              Text(tab.rawValue)
            }
          }
          .pickerStyle(.segmented)
          Group {
            notificationsView
          }
          .padding(.top, 16)
        } else {
          Text("Please Sign In to see your notifications")
            .font(.title3)
        }
      }
      .padding(.horizontal, .layoutPadding)
      .padding(.top, .layoutPadding)
    }
    .background(theme.primaryBackgroundColor)
    .task {
      viewModel.client = client
      await viewModel.fetchNotifications()
    }
    .refreshable {
      await viewModel.fetchNotifications()
    }
    .onChange(of: watcher.latestEvent?.id, perform: { _ in
      if let latestEvent = watcher.latestEvent {
        viewModel.handleEvent(event: latestEvent)
      }
    })
    .navigationTitle(Text("Notifications"))
    .navigationBarTitleDisplayMode(.inline)
  }
  
  @ViewBuilder
  private var notificationsView: some View {
    switch viewModel.state {
    case .loading:
      ForEach(Models.Notification.placeholders()) { notification in
        NotificationRowView(notification: notification)
          .redacted(reason: .placeholder)
          .shimmering()
        Divider()
          .padding(.vertical, .dividerPadding)
      }
      
    case let .display(notifications, nextPageState):
      ForEach(notifications) { notification in
        NotificationRowView(notification: notification)
        Divider()
          .padding(.vertical, .dividerPadding)
      }
      
      switch nextPageState {
      case .none:
        EmptyView()
      case .hasNextPage:
        loadingRow
          .onAppear {
            Task {
              await viewModel.fetchNextPage()
            }
          }
      case .loadingNextPage:
        loadingRow
      }
      
    case let .error(error):
      Text(error.localizedDescription)
    }
  }
  
  private var loadingRow: some View {
    HStack {
      Spacer()
      ProgressView()
      Spacer()
    }
  }
}