67 lines
2.7 KiB
Swift
67 lines
2.7 KiB
Swift
//
|
|
// MoodEntryFunctions.swift
|
|
// Feels (iOS)
|
|
//
|
|
// Created by Trey Tartt on 2/19/22.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
class MoodEntryFunctions {
|
|
static func padMoodEntriesForCalendar(entries grouped: [Int: [Int: [MoodEntry]]]) -> [Int: [Int: [MoodEntry]]] {
|
|
var newGrouped = [Int: [Int: [MoodEntry]]]()
|
|
|
|
let allYears = grouped.keys.sorted(by: > )
|
|
for year in allYears {
|
|
var newMonth = [Int: [MoodEntry]]()
|
|
|
|
let oldMonths = grouped[year]!
|
|
let monthKeys = oldMonths.keys.sorted(by: > )
|
|
for key in monthKeys {
|
|
if let entries = oldMonths[key] {
|
|
newMonth[key] = MoodEntryFunctions.padMoodEntriesMonth(monthEntries: entries)
|
|
}
|
|
newGrouped[year] = newMonth
|
|
}
|
|
}
|
|
return newGrouped
|
|
}
|
|
|
|
static func padMoodEntriesMonth(monthEntries entries: [MoodEntry]) -> [MoodEntry] {
|
|
let sortedEntries = entries.sorted(by: { $0.forDate! < $1.forDate! })
|
|
var mutableEntries = sortedEntries
|
|
|
|
if let firstDate = sortedEntries.first {
|
|
let date = firstDate.forDate!
|
|
|
|
// if the first entry for a month is in the middle of the month we
|
|
// need to add in the missing entries, as placeholders, to the beignning to get
|
|
// the entries on the right day. think user downloads in the middle of the month
|
|
// and entry is on the 13th ... this needs to show on the 13th entry spot
|
|
var startOfMonth = date.startOfMonth
|
|
startOfMonth = Calendar.current.date(byAdding: .hour, value: 9, to: startOfMonth)!
|
|
let lastMissingDate = mutableEntries.first?.forDate ?? date.endOfMonth
|
|
var missingDates = Date.dates(from: startOfMonth, toDate: lastMissingDate, includingToDate: true)
|
|
missingDates = missingDates.dropLast()
|
|
|
|
for date in missingDates {
|
|
mutableEntries.insert(PersistenceController.shared.generateObjectNotInArray(forDate: date, withMood: .placeholder), at: 0)
|
|
}
|
|
|
|
mutableEntries = mutableEntries.sorted(by: {
|
|
$0.forDate! < $1.forDate!
|
|
})
|
|
|
|
// fill in calendar day offset .. if month starts on wed we need to
|
|
// pad the beginning sun, mon, tues
|
|
if let firstDate = mutableEntries.first?.forDate {
|
|
let weekday = Int16(Calendar.current.component(.weekday, from: firstDate))
|
|
for _ in 1..<weekday {
|
|
mutableEntries.insert(PersistenceController.shared.generateObjectNotInArray(withMood: .placeholder), at: 0)
|
|
}
|
|
}
|
|
}
|
|
return mutableEntries
|
|
}
|
|
}
|