Fix adding a break

The relationship between TimeEntry and BreakEntry needs to be precisely defined for SwiftData to behave as expected when appending a BreakEntry to an existing TimeEntry.
madbranch · Jan 16, 2025 · e33d2f3 · e33d2f3
1 parent 792a8c5
commit e33d2f3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/Shared/BreakEntry.swift b/Shared/BreakEntry.swift
@@ -3,7 +3,7 @@ import SwiftData
 
 @Model
 class BreakEntry: Codable {
-    init(start: Date, end: Date) {
+    init(start: Date, end: Date, timeEntry: TimeEntry? = nil) {
         assert(BreakEntry.isZeroSecond(start))
         assert(BreakEntry.isZeroSecond(end))
         self.start = start
@@ -16,6 +16,7 @@ class BreakEntry: Codable {
         end = try container.decode(Date.self, forKey: .end)
     }
 
+    var timeEntry: TimeEntry?
     var start: Date
     var end: Date
 

diff --git a/Shared/TimeEntry.swift b/Shared/TimeEntry.swift
@@ -6,6 +6,7 @@ public class TimeEntry: Encodable {
     var start: Date
     var end: Date
     var notes: String = ""
+    @Relationship(deleteRule: .cascade, inverse: \BreakEntry.timeEntry)
     var breaks = [BreakEntry]()
     init(from start: Date, to end: Date, notes: String = "") {
         self.start = start

diff --git a/TimeKeen/Views/PayPeriods/TimeEntryDetails.swift b/TimeKeen/Views/PayPeriods/TimeEntryDetails.swift
@@ -112,7 +112,7 @@ struct TimeEntryDetails: View {
                     IntervalDatePicker(selection: $breakEnd, minuteInterval: minuteInterval, in: timeEntry.start...timeEntry.end, displayedComponents: [.date, .hourAndMinute])
                 }
                 Button(action: {
-                    timeEntry.breaks.append(BreakEntry(start: breakStart, end: breakEnd))
+                    timeEntry.breaks.append(BreakEntry(start: breakStart, end: breakEnd, timeEntry: timeEntry))
                     isAddingBreak = false
                 }) {
                     Text("Add Break")