93 lines
3.8 KiB
Swift
93 lines
3.8 KiB
Swift
//
|
|
// PersistenceGET.swift
|
|
// Feels
|
|
//
|
|
// Created by Trey Tartt on 2/17/22.
|
|
//
|
|
|
|
import CoreData
|
|
|
|
extension PersistenceController {
|
|
public func getEntry(byDate date: Date) -> MoodEntry? {
|
|
let startDate = Calendar.current.startOfDay(for: date)
|
|
let endDate = Calendar.current.date(byAdding: .day, value: 1, to: startDate)!
|
|
|
|
let predicate = NSPredicate(format: "forDate >= %@ && forDate <= %@ ",
|
|
startDate as NSDate,
|
|
endDate as NSDate)
|
|
|
|
let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
|
|
fetchRequest.predicate = predicate
|
|
fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: true)]
|
|
let data = try! viewContext.fetch(fetchRequest)
|
|
return data.first
|
|
}
|
|
|
|
public func getData(startDate: Date, endDate: Date, includedDays: [Int]) -> [MoodEntry] {
|
|
try! viewContext.setQueryGenerationFrom(.current)
|
|
// viewContext.refreshAllObjects()
|
|
|
|
var includedDays16 = [Int16]()
|
|
|
|
if includedDays.isEmpty {
|
|
includedDays16 = [Int16(1), Int16(2), Int16(3), Int16(4), Int16(5), Int16(6), Int16(7)]
|
|
} else {
|
|
includedDays16 = includedDays.map({
|
|
Int16($0)
|
|
})
|
|
}
|
|
let predicate = NSPredicate(format: "forDate >= %@ && forDate <= %@ && weekDay IN %@",
|
|
startDate as NSDate,
|
|
endDate as NSDate,
|
|
includedDays16)
|
|
|
|
let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
|
|
fetchRequest.predicate = predicate
|
|
fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: true)]
|
|
let data = try! viewContext.fetch(fetchRequest)
|
|
return data
|
|
}
|
|
|
|
public func splitIntoYearMonth(includedDays: [Int]) -> [Int: [Int: [MoodEntry]]] {
|
|
let data = PersistenceController.shared.getData(startDate: Date(timeIntervalSince1970: 0),
|
|
endDate: Date(),
|
|
includedDays: includedDays).sorted(by: {
|
|
$0.forDate! < $1.forDate!
|
|
})
|
|
var returnData = [Int: [Int: [MoodEntry]]]()
|
|
|
|
if let earliestEntry = data.first,
|
|
let lastEntry = data.last {
|
|
|
|
let calendar = Calendar.current
|
|
let components = calendar.dateComponents([.year], from: earliestEntry.forDate!)
|
|
let earliestYear = components.year!
|
|
|
|
let latestComponents = calendar.dateComponents([.year], from: lastEntry.forDate!)
|
|
let latestYear = latestComponents.year!
|
|
|
|
for year in earliestYear...latestYear {
|
|
var allMonths = [Int: [MoodEntry]]()
|
|
|
|
for month in (1...12) {
|
|
var components = DateComponents()
|
|
components.month = month
|
|
components.year = year
|
|
components.day = 1
|
|
let startDateOfMonth = Calendar.current.date(from: components)!
|
|
|
|
let items = PersistenceController.shared.getData(startDate: startDateOfMonth,
|
|
endDate: startDateOfMonth.endOfMonth,
|
|
includedDays: [1,2,3,4,5,6,7])
|
|
|
|
if !items.isEmpty {
|
|
allMonths[month] = items
|
|
}
|
|
}
|
|
returnData[year] = allMonths
|
|
}
|
|
}
|
|
return returnData
|
|
}
|
|
}
|