Sportstime/SportsTime/Features/Progress/ViewModels/ProgressViewModel.swift

//
//  ProgressViewModel.swift
//  SportsTime
//
//  ViewModel for stadium progress tracking and visualization.
//

import Foundation
import SwiftUI
import SwiftData

@MainActor
@Observable
final class ProgressViewModel {

    // MARK: - State

    var selectedSport: Sport = .mlb
    var isLoading = true
    var error: Error?
    var errorMessage: String?

    // MARK: - Data

    private(set) var visits: [StadiumVisit] = []
    private(set) var stadiums: [Stadium] = []
    private(set) var teams: [Team] = []

    // MARK: - Dependencies

    private var modelContainer: ModelContainer?
    private let dataProvider = AppDataProvider.shared

    // MARK: - Computed Properties

    /// Overall progress for the selected sport
    var leagueProgress: LeagueProgress {
        // Filter stadiums by sport directly (same as sportStadiums)
        let sportStadiums = stadiums.filter { $0.sport == selectedSport }

        let visitedStadiumIds = Set(
            visits
                .filter { $0.sportEnum == selectedSport }
                .compactMap { visit -> String? in
                    // O(1) dictionary lookup via DataProvider
                    dataProvider.stadium(for: visit.stadiumId)?.id
                }
        )

        let visited = sportStadiums.filter { visitedStadiumIds.contains($0.id) }
        let remaining = sportStadiums.filter { !visitedStadiumIds.contains($0.id) }

        return LeagueProgress(
            sport: selectedSport,
            totalStadiums: sportStadiums.count,
            visitedStadiums: visited.count,
            stadiumsVisited: visited,
            stadiumsRemaining: remaining
        )
    }

    /// Stadium visit status indexed by stadium ID
    var stadiumVisitStatus: [String: StadiumVisitStatus] {
        var statusMap: [String: StadiumVisitStatus] = [:]

        // Group visits by stadium
        let visitsByStadium = Dictionary(grouping: visits.filter { $0.sportEnum == selectedSport }) { $0.stadiumId }

        for stadium in stadiums {
            if let stadiumVisits = visitsByStadium[stadium.id], !stadiumVisits.isEmpty {
                let summaries = stadiumVisits.map { visit in
                    VisitSummary(
                        id: visit.id,
                        stadium: stadium,
                        visitDate: visit.visitDate,
                        visitType: visit.visitType,
                        sport: selectedSport,
                        homeTeamName: visit.homeTeamName,
                        awayTeamName: visit.awayTeamName,
                        score: visit.finalScore,
                        photoCount: visit.photoMetadata?.count ?? 0,
                        notes: visit.notes
                    )
                }
                statusMap[stadium.id] = .visited(visits: summaries)
            } else {
                statusMap[stadium.id] = .notVisited
            }
        }

        return statusMap
    }

    /// Stadiums for the selected sport
    var sportStadiums: [Stadium] {
        stadiums.filter { $0.sport == selectedSport }
    }

    /// Visited stadiums for the selected sport
    var visitedStadiums: [Stadium] {
        leagueProgress.stadiumsVisited
    }

    /// Unvisited stadiums for the selected sport
    var unvisitedStadiums: [Stadium] {
        leagueProgress.stadiumsRemaining
    }

    /// Count of trips for the selected sport (stub - can be enhanced)
    var tripCount: Int {
        // TODO: Fetch saved trips count from SwiftData
        0
    }

    /// Recent visits sorted by date
    var recentVisits: [VisitSummary] {
        visits
            .sorted { $0.visitDate > $1.visitDate }
            .prefix(10)
            .compactMap { visit -> VisitSummary? in
                guard let stadium = dataProvider.stadium(for: visit.stadiumId),
                      let sport = visit.sportEnum else {
                    return nil
                }
                return VisitSummary(
                    id: visit.id,
                    stadium: stadium,
                    visitDate: visit.visitDate,
                    visitType: visit.visitType,
                    sport: sport,
                    homeTeamName: visit.homeTeamName,
                    awayTeamName: visit.awayTeamName,
                    score: visit.finalScore,
                    photoCount: visit.photoMetadata?.count ?? 0,
                    notes: visit.notes
                )
            }
    }

    // MARK: - Configuration

    func configure(with container: ModelContainer) {
        self.modelContainer = container
    }

    // MARK: - Actions

    func loadData() async {
        isLoading = true
        error = nil
        errorMessage = nil

        do {
            // Load stadiums and teams from data provider
            if dataProvider.stadiums.isEmpty {
                await dataProvider.loadInitialData()
            }
            stadiums = dataProvider.stadiums
            teams = dataProvider.teams

            // Load visits from SwiftData
            if let container = modelContainer {
                let context = ModelContext(container)
                let descriptor = FetchDescriptor<StadiumVisit>(
                    sortBy: [SortDescriptor(\.visitDate, order: .reverse)]
                )
                visits = try context.fetch(descriptor)
            }
        } catch {
            self.error = error
            self.errorMessage = error.localizedDescription
        }

        isLoading = false
    }

    func selectSport(_ sport: Sport) {
        selectedSport = sport
    }

    func clearError() {
        error = nil
        errorMessage = nil
    }

    // MARK: - Visit Management

    func deleteVisit(_ visit: StadiumVisit) async throws {
        guard let container = modelContainer else { return }

        let context = ModelContext(container)
        context.delete(visit)
        try context.save()

        // Reload data
        await loadData()
    }

    // MARK: - Progress Card Generation

    func progressCardData(includeUsername: Bool = false) -> ProgressCardData {
        ProgressCardData(
            sport: selectedSport,
            progress: leagueProgress,
            username: nil,
            includeMap: true,
            showDetailedStats: false
        )
    }
}