WerkoutIOS/Werkout_ios/Views/AllWorkouts/AllWorkoutsView.swift

//
//  AllWorkoutsView.swift
//  Werkout_ios
//
//  Created by Trey Tartt on 6/15/23.
//

import Foundation
import SwiftUI

struct AllWorkoutsView: View {
    enum MainViews: Int, CaseIterable {
        case AllWorkout = 0
        case MyWorkouts
        
        var title: String {
            switch self {
                
            case .AllWorkout:
                return "All Workouts"
            case .MyWorkouts:
                return "Planned Workouts"
            }
        }
    }
    
    @State var workouts: [Workout]?
    var bridgeModule = BridgeModule.shared
    @State public var needsUpdating: Bool = true
    
    @ObservedObject var dataStore = DataStore.shared
    
    @State private var showWorkoutDetail = false
    @State private var selectedWorkout: Workout? {
        didSet {
            bridgeModule.currentWorkout = selectedWorkout
        }
    }
    
    @State private var selectedPlannedWorkout: Workout? {
        didSet {
            bridgeModule.currentWorkout = selectedPlannedWorkout
        }
    }
    
    @State private var showLoginView = false
    @State private var selectedSegment: MainViews = .AllWorkout
    @State var selectedDate: Date = Date()
    
    let pub = NotificationCenter.default.publisher(for: NSNotification.Name("CreatedNewWorkout"))

    var body: some View {
        ZStack {
            if let workouts = workouts {
                if dataStore.status == .loading {
                    ProgressView()
                        .progressViewStyle(.circular)
                } else {
                    VStack {
                        Picker("", selection: $selectedSegment) {
                            ForEach(MainViews.allCases, id: \.self) { viewType in
                                Text(viewType.title)
                            }
                        }
                        .pickerStyle(.segmented)
                        .padding()
                        
                        switch selectedSegment {
                        case .AllWorkout:
                            allWorkoutView(workouts: workouts)
                        case .MyWorkouts:
                            plannedWorkout(workouts: UserStore.shared.plannedWorkouts)
                        }
                    }
                }
            } else {
                ProgressView()
                    .progressViewStyle(.circular)
            }
        }.onAppear{
//            UserStore.shared.logout()
            maybeUpdateShit()
        }
        .sheet(item: $selectedWorkout) { item in
            let viewModel = WorkoutDetailViewModel(workout: item)
            WorkoutDetailView(viewModel: viewModel)
        }
        .sheet(item: $selectedPlannedWorkout) { item in
            let viewModel = WorkoutDetailViewModel(workout: item)
            WorkoutDetailView(viewModel: viewModel, showAddToCalendar: false)
        }
        .sheet(isPresented: $showLoginView) {
            LoginView(completion: {
                self.needsUpdating = true
                maybeUpdateShit()
            })
        }
        .onReceive(pub) { (output) in
            self.needsUpdating = true
            maybeUpdateShit()
        }
    }
    
    func allWorkoutView(workouts: [Workout]) -> some View {
        List {
            ForEach(workouts, id:\.name) { workout in
                VStack {
                    Text(workout.name)
                        .font(.title2)
                        .frame(maxWidth: .infinity, alignment: .leading)
                    Text(workout.description ?? "")
                        .frame(maxWidth: .infinity, alignment: .leading)
                }
                .contentShape(Rectangle())
                .onTapGesture {
                    selectedWorkout = workout
                }
            }
        }
    }
    
    func plannedWorkout(workouts: [PlannedWorkout]) -> some View {
        List {
            ForEach(workouts, id:\.workout.name) { plannedWorkout in
                HStack {
                    VStack(alignment: .leading) {
                        Text(plannedWorkout.onDate.plannedDate?.weekDay ?? "-")
                            .font(.title)
                        
                        Text(plannedWorkout.onDate.plannedDate?.monthString ?? "-")
                            .font(.title)
                        
                        Text(plannedWorkout.onDate.plannedDate?.dateString ?? "-")
                            .font(.title)
                    }
                    
                    Divider()
                    
                    VStack {
                        Text(plannedWorkout.workout.name)
                            .font(.title)
                            .frame(maxWidth: .infinity, alignment: .leading)
                        
                        Text(plannedWorkout.workout.description ?? "")
                            .font(.body)
                            .frame(maxWidth: .infinity, alignment: .leading)
                        
                        Text(plannedWorkout.onDate)
                            .font(.footnote)
                            .frame(maxWidth: .infinity, alignment: .leading)
                    }
                    .contentShape(Rectangle())
                    .onTapGesture {
                        selectedPlannedWorkout = plannedWorkout.workout
                    }
                }
            }
        }
    }
    
    func maybeUpdateShit() {
        if UserStore.shared.token != nil{
            if UserStore.shared.plannedWorkouts.isEmpty {
                UserStore.shared.fetchPlannedWorkouts()
            }
            
            if needsUpdating {
                dataStore.fetchAllData()
                
                AllWorkoutFetchable().fetch(completion: { result in
                    needsUpdating = false
                    switch result {
                    case .success(let model):
                        DispatchQueue.main.async {
                            self.workouts = model
                        }
                    case .failure(_):
                        fatalError("shit broke")
                    }
                })
            }
        } else {
            showLoginView = true
        }
    }
}

struct AllWorkoutsView_Previews: PreviewProvider {
    static var previews: some View {
        AllWorkoutsView(workouts: PreviewData.allWorkouts())
    }
}