Trio/FreeAPS/Sources/Modules/Home/HomeViewModel.swift

import SwiftDate
import SwiftUI

extension Home {
    class ViewModel<Provider>: BaseViewModel<Provider>, ObservableObject where Provider: HomeProvider {
        @Injected() var apsManager: APSManager!
        @Injected() var history: PumpHistoryStorage!
        @Injected() var temps: TempTargetsStorage!
        @Injected() var glucoseStorage: GlucoseStorage!
        @Injected() var broadcaster: Broadcaster!
        @Injected() var storage: FileStorage!

        @Published var glucose: [BloodGlucose] = []

        @Published var suggestion: Suggestion?

        override func subscribe() {
            glucose = filteredGlucose(glucoseStorage.recent())
            suggestion = try? storage.retrieve(OpenAPS.Enact.suggested, as: Suggestion.self)
            broadcaster.register(GlucoseObserver.self, observer: self)
            broadcaster.register(SuggestionObserver.self, observer: self)
        }

        func addCarbs() {
            showModal(for: .addCarbs)
        }

        func runLoop() {
            apsManager.fetchAndLoop()
        }

        func addTempTarget() {
            showModal(for: .addTempTarget)
        }

        func bolus() {
            showModal(for: .bolus)
        }

        private func filteredGlucose(_ glucose: [BloodGlucose]) -> [BloodGlucose] {
            glucose.filter {
                $0.dateString.addingTimeInterval(3.hours.timeInterval) > Date()
            }
        }
    }
}

extension Home.ViewModel: GlucoseObserver, SuggestionObserver {
    func glucoseDidUpdate(_ glucose: [BloodGlucose]) {
        self.glucose = filteredGlucose(glucose)
    }

    func suggestionDidUpdate(_ suggestion: Suggestion) {
        self.suggestion = suggestion
    }
}