WerkoutIOS/iphone/Werkout_ios/Views/AllWorkouts/AllWorkoutsView.swift

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

import Foundation
import SwiftUI
import HealthKit

enum MainViewTypes: Int, CaseIterable {
    case AllWorkout = 0
    case MyWorkouts
    
    var title: String {
        switch self {
            
        case .AllWorkout:
            return "All Workouts"
        case .MyWorkouts:
            return "Planned Workouts"
        }
    }
}

struct AllWorkoutsView: View {
    @State var isUpdating = false
    @State var workouts: [Workout]?
    @State var uniqueWorkoutUsers: [RegisteredUser]?
    let healthStore = HKHealthStore()
    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.currentExerciseInfo.workout = selectedWorkout
        }
    }
    
    @State private var selectedPlannedWorkout: Workout? {
        didSet {
            bridgeModule.currentExerciseInfo.workout = selectedPlannedWorkout
        }
    }
    
    @State private var showLoginView = false
    @State private var selectedSegment: MainViewTypes = .AllWorkout
    @State var selectedDate: Date = Date()
    
    let pub = NotificationCenter.default.publisher(for: NSNotification.Name("CreatedNewWorkout"))

    var body: some View {
        ZStack {
            if let workouts = workouts {
                VStack {
                    AllWorkoutPickerView(mainViews: MainViewTypes.allCases,
                                         selectedSegment: $selectedSegment,
                                         showCurrentWorkout: {
                        selectedWorkout = bridgeModule.currentExerciseInfo.workout
                    })
                    
                    
                    switch selectedSegment {
                    case .AllWorkout:
                        if isUpdating {
                            ProgressView()
                                .progressViewStyle(.circular)
                        }
                        
                        AllWorkoutsListView(uniqueWorkoutUsers: $uniqueWorkoutUsers,
                                            workouts: workouts,
                                            selectedWorkout: { workout in
                            selectedWorkout = workout
                        }, refresh: {
                            self.needsUpdating = true
                            maybeUpdateShit()
                        })
                        Divider()
                    case .MyWorkouts:
                        PlannedWorkoutView(workouts: UserStore.shared.plannedWorkouts,
                                           selectedPlannedWorkout: $selectedPlannedWorkout)
                    }
                }
            } else {
                ProgressView("Updating")
            }
        }.onAppear{
//            UserStore.shared.logout()
            authorizeHealthKit()
            maybeUpdateShit()
        }
        .sheet(item: $selectedWorkout) { item in
            let isPreview = item.id == bridgeModule.currentExerciseInfo.workout?.id
            let viewModel = WorkoutDetailViewModel(workout: item, isPreview: isPreview)
            WorkoutDetailView(viewModel: viewModel)
        }
        .sheet(item: $selectedPlannedWorkout) { item in
            let viewModel = WorkoutDetailViewModel(workout: item, isPreview: true)
            WorkoutDetailView(viewModel: viewModel)
        }
        .sheet(isPresented: $showLoginView) {
            LoginView(completion: {
                self.needsUpdating = true
                maybeUpdateShit()
            })
            .interactiveDismissDisabled()
        }
        .onReceive(pub) { (output) in
            self.needsUpdating = true
            maybeUpdateShit()
        }
    }

    func maybeUpdateShit() {
        if UserStore.shared.token != nil{
            if UserStore.shared.plannedWorkouts.isEmpty {
                UserStore.shared.fetchPlannedWorkouts()
            }
            
            if needsUpdating {
                self.isUpdating = true
                dataStore.fetchAllData(completion: {
                    DispatchQueue.main.async {
                        guard let allWorkouts = dataStore.allWorkouts else {
                            return
                        }
                        self.workouts = allWorkouts.sorted(by: {
                            $0.createdAt ?? Date() < $1.createdAt ?? Date()
                        })
                        self.isUpdating = false
                        self.uniqueWorkoutUsers = dataStore.workoutsUniqueUsers
                    }
                    
                    self.isUpdating = false
                })
            }
        } else {
            showLoginView = true
        }
    }
    
    func authorizeHealthKit() {
        let healthKitTypes: Set = [
            HKObjectType.quantityType(forIdentifier: .heartRate)!,
            HKObjectType.quantityType(forIdentifier: .activeEnergyBurned)!,
            HKObjectType.quantityType(forIdentifier: .oxygenSaturation)!,
            HKQuantityType.workoutType()
        ]
        
        healthStore.requestAuthorization(toShare: healthKitTypes, read: healthKitTypes) { (succ, error) in
            if !succ {
                fatalError("Error requesting authorization from health store: \(String(describing: error)))")
            }
        }
    }
}

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