Files
workouts/Shared/Persistence/WorkoutsModelContainer.swift
T
rzen 2c1e4759ae Add machine comfort settings and tighten the document schemas
Machine-based exercises now carry ordered name/value comfort settings
(seat height, back-rest position, ...) on both ExerciseDocument and, as a
plan-time snapshot, WorkoutLogDocument — the optional array doubles as the
machine flag. Editable from the exercise editor's new Machine section and
from the workout row's settings sheet, whose mid-workout edits write back
to the originating split (workout saved first, so seed clone-on-edit
repointing can't clobber the log edit).

Schema tightening rides the same rev: splits bump to v2 (weight-reminder
fields and the unused exercise category removed), workouts to v3 (the
derived `completed` flag removed; status is the single source). Starter
seeds regenerated at v2 with unchanged ULIDs; SwiftData cache schema
bumped to rebuild. SCHEMA.md documents the shapes.

Claude-Session: https://claude.ai/code/session_01LEoff8bXGBS83tK1c55Mf7
2026-07-06 16:29:44 -04:00

88 lines
3.9 KiB
Swift

import Foundation
import SwiftData
/// Builds the SwiftData cache container. The iCloud Drive JSON documents are the
/// source of truth, so this store is a rebuildable cache — wiping it is always
/// safe; `SyncEngine` repopulates it from the container on next launch.
enum WorkoutsModelContainer {
/// Bump whenever the cache schema changes — wipes and rebuilds from files.
/// 2: added health-metric columns to `Workout` and `activityTypeRaw` to `Split`.
/// 3: added `categoryRaw` to `Exercise`.
/// 4: removed the weight-reminder columns and `categoryRaw` from `Exercise` and
/// the `completed` column from `WorkoutLog`; added `machineSettings` to both.
static let currentSchemaVersion = 4
private static let schemaVersionKey = "Workouts.persistenceSchemaVersion"
private static let identityTokenKey = "Workouts.iCloudIdentityToken"
private static var storeURL: URL {
URL.applicationSupportDirectory.appending(path: "Workouts.store")
}
static func make() -> ModelContainer {
let schema = Schema([Split.self, Exercise.self, Workout.self, WorkoutLog.self])
ensureStoreDirectoryExists()
wipeIfNeeded()
wipeIfAccountChanged()
do {
// `.none` is load-bearing: the default `.automatic` would silently
// enable CloudKit mirroring on top of our iCloud Drive file sync.
let config = ModelConfiguration(schema: schema, url: storeURL, cloudKitDatabase: .none)
let container = try ModelContainer(for: schema, configurations: [config])
UserDefaults.standard.set(currentSchemaVersion, forKey: schemaVersionKey)
return container
} catch {
print("Workouts: ModelContainer creation failed at \(storeURL.path): \(error). Falling back to in-memory.")
}
do {
let config = ModelConfiguration(schema: schema, isStoredInMemoryOnly: true)
return try ModelContainer(for: schema, configurations: [config])
} catch {
fatalError("Workouts: could not create even an in-memory ModelContainer: \(error)")
}
}
/// Records the current iCloud identity token as the cache's owner. Call after
/// rebuilding the cache for a newly-signed-in account.
static func persistCurrentIdentityToken() {
UserDefaults.standard.set(currentIdentityTokenData(), forKey: identityTokenKey)
}
// MARK: - Wipe helpers
private static func ensureStoreDirectoryExists() {
let parent = storeURL.deletingLastPathComponent()
try? FileManager.default.createDirectory(at: parent, withIntermediateDirectories: true)
}
private static func wipeIfNeeded() {
let stored = UserDefaults.standard.integer(forKey: schemaVersionKey)
guard stored < currentSchemaVersion else { return }
wipeStore()
}
/// Wipes the cache when the signed-in iCloud account differs from the one the
/// cache was built for. A nil token is "not ready yet", not "no account" — so
/// only compare once a token resolves; mid-session changes are handled live.
private static func wipeIfAccountChanged() {
guard let current = currentIdentityTokenData() else { return }
let stored = UserDefaults.standard.data(forKey: identityTokenKey)
guard current != stored else { return }
wipeStore()
UserDefaults.standard.set(current, forKey: identityTokenKey)
}
private static func currentIdentityTokenData() -> Data? {
guard let token = FileManager.default.ubiquityIdentityToken else { return nil }
return try? NSKeyedArchiver.archivedData(withRootObject: token, requiringSecureCoding: false)
}
private static func wipeStore() {
let base = storeURL
for url in [base, URL(fileURLWithPath: base.path + "-wal"), URL(fileURLWithPath: base.path + "-shm")] {
try? FileManager.default.removeItem(at: url)
}
}
}