Reflect/Shared/Persisence/DataControllerDELETE.swift

//
//  DataControllerDELETE.swift
//  Feels
//
//  SwiftData DELETE operations.
//

import SwiftData
import Foundation

extension DataController {
    func clearDB() {
        do {
            try modelContext.delete(model: MoodEntryModel.self)
            saveAndRunDataListeners()
        } catch {
            fatalError("Failed to clear database: \(error)")
        }
    }

    func deleteLast(numberOfEntries: Int) {
        let startDate = Calendar.current.date(byAdding: .day, value: -numberOfEntries, to: Date())!
        let entries = getData(startDate: startDate, endDate: Date(), includedDays: [])

        for entry in entries {
            modelContext.delete(entry)
        }
        save()
    }

    func deleteRandomFromLast(numberOfEntries: Int) {
        let startDate = Calendar.current.date(byAdding: .day, value: -numberOfEntries, to: Date())!
        let entries = getData(startDate: startDate, endDate: Date(), includedDays: [])

        for entry in entries where Bool.random() {
            modelContext.delete(entry)
        }
        save()
    }
}