@shaykec/app-agent 1.0.4 → 1.0.6

package/templates/ios/TrackTemplate/TrackTemplate/Data/LocalStorage/LocalStorageProvider.swift

@@ -0,0 +1,180 @@
+import CoreData
+import Foundation
+
+/// Core Data–backed implementation of DataRepository.
+/// Stores habit tracking data in a local SQLite database.
+/// Pre-seeds default data on first launch (empty database).
+///
+/// Usage: LocalStorageProvider.shared (singleton, like MockDataProvider)
+final class LocalStorageProvider: ObservableObject, DataRepository {
+    static let shared = LocalStorageProvider()
+
+    @Published var habits: [Habit] = []
+    @Published var achievements: [Achievement] = []
+
+    private let stack: CoreDataStack
+    private var context: NSManagedObjectContext { stack.viewContext }
+
+    private init(stack: CoreDataStack = .shared) {
+        self.stack = stack
+        loadData()
+    }
+
+    // MARK: - Data Loading
+
+    func loadData() {
+        fetchHabits()
+        fetchAchievements()
+
+        // Pre-seed from MockDataProvider on first launch (empty DB)
+        if habits.isEmpty {
+            seedDefaults()
+            fetchHabits()
+            fetchAchievements()
+        }
+    }
+
+    // MARK: - CRUD Operations - Habits
+
+    func createHabit(_ habit: Habit) {
+        let entity = HabitEntity(context: context)
+        entity.update(from: habit)
+        stack.save()
+        fetchHabits()
+    }
+
+    func updateHabit(_ habit: Habit) {
+        guard let entity = fetchHabitEntity(byId: habit.id) else { return }
+        entity.update(from: habit)
+        stack.save()
+        fetchHabits()
+    }
+
+    func deleteHabit(_ habit: Habit) {
+        guard let entity = fetchHabitEntity(byId: habit.id) else { return }
+        // Delete related entries and streaks
+        deleteHabitEntries(for: habit.id)
+        deleteStreak(for: habit.id)
+        context.delete(entity)
+        stack.save()
+        fetchHabits()
+    }
+
+    func archiveHabit(_ habit: Habit) {
+        var updated = habit
+        updated.isArchived = true
+        updateHabit(updated)
+    }
+
+    // MARK: - CRUD Operations - Achievements
+
+    func unlockAchievement(_ achievement: Achievement) {
+        guard let entity = fetchAchievementEntity(byId: achievement.id) else { return }
+        var updated = achievement
+        updated.isUnlocked = true
+        updated.unlockedAt = Date()
+        entity.update(from: updated)
+        stack.save()
+        fetchAchievements()
+    }
+
+    func updateAchievementProgress(_ achievement: Achievement, progress: Int) {
+        guard let entity = fetchAchievementEntity(byId: achievement.id) else { return }
+        var updated = achievement
+        updated.progress = progress
+        entity.update(from: updated)
+        stack.save()
+        fetchAchievements()
+    }
+
+    // MARK: - Reset
+
+    func reset() {
+        // Delete all entities
+        let entities = ["HabitEntity", "HabitEntryEntity", "AchievementEntity", "StreakEntity"]
+        for entityName in entities {
+            let request = NSFetchRequest<NSFetchRequestResult>(entityName: entityName)
+            let delete = NSBatchDeleteRequest(fetchRequest: request)
+            _ = try? context.execute(delete)
+        }
+        stack.save()
+
+        // Reset to defaults
+        seedDefaults()
+        loadData()
+    }
+
+    // MARK: - Private Fetch Helpers
+
+    private func fetchHabits() {
+        let request = NSFetchRequest<HabitEntity>(entityName: "HabitEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "createdAt", ascending: false)]
+        do {
+            let entities = try context.fetch(request)
+            habits = entities.map { $0.toHabit() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch habits: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchAchievements() {
+        let request = NSFetchRequest<AchievementEntity>(entityName: "AchievementEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "requirement", ascending: true)]
+        do {
+            let entities = try context.fetch(request)
+            achievements = entities.map { $0.toAchievement() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch achievements: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchHabitEntity(byId id: String) -> HabitEntity? {
+        let request = NSFetchRequest<HabitEntity>(entityName: "HabitEntity")
+        request.predicate = NSPredicate(format: "id == %@", id)
+        request.fetchLimit = 1
+        return try? context.fetch(request).first
+    }
+
+    private func fetchAchievementEntity(byId id: String) -> AchievementEntity? {
+        let request = NSFetchRequest<AchievementEntity>(entityName: "AchievementEntity")
+        request.predicate = NSPredicate(format: "id == %@", id)
+        request.fetchLimit = 1
+        return try? context.fetch(request).first
+    }
+
+    private func deleteHabitEntries(for habitId: String) {
+        let request = NSFetchRequest<HabitEntryEntity>(entityName: "HabitEntryEntity")
+        request.predicate = NSPredicate(format: "habitId == %@", habitId)
+        if let entries = try? context.fetch(request) {
+            entries.forEach { context.delete($0) }
+        }
+    }
+
+    private func deleteStreak(for habitId: String) {
+        let request = NSFetchRequest<StreakEntity>(entityName: "StreakEntity")
+        request.predicate = NSPredicate(format: "habitId == %@", habitId)
+        if let streak = try? context.fetch(request).first {
+            context.delete(streak)
+        }
+    }
+
+    // MARK: - Seeding
+
+    private func seedDefaults() {
+        let mockProvider = MockDataProvider.shared
+
+        // Seed habits
+        for habit in mockProvider.habits {
+            let entity = HabitEntity(context: context)
+            entity.update(from: habit)
+        }
+        stack.save()
+
+        // Seed achievements
+        for achievement in mockProvider.achievements {
+            let entity = AchievementEntity(context: context)
+            entity.update(from: achievement)
+        }
+        stack.save()
+    }
+}

package/templates/ios/TrackTemplate/TrackTemplate/Data/LocalStorage/StreakEntity.swift

@@ -0,0 +1,35 @@
+import CoreData
+import Foundation
+
+/// Core Data managed object for Streak persistence.
+@objc(StreakEntity)
+final class StreakEntity: NSManagedObject {
+    @NSManaged var habitId: String
+    @NSManaged var currentStreak: Int32
+    @NSManaged var longestStreak: Int32
+    @NSManaged var lastCompletedDate: Date?
+    @NSManaged var freezesUsed: Int32
+    @NSManaged var freezesAvailable: Int32
+
+    // MARK: - Conversion
+
+    func toStreak() -> Streak {
+        Streak(
+            habitId: habitId,
+            currentStreak: Int(currentStreak),
+            longestStreak: Int(longestStreak),
+            lastCompletedDate: lastCompletedDate,
+            freezesUsed: Int(freezesUsed),
+            freezesAvailable: Int(freezesAvailable)
+        )
+    }
+
+    func update(from streak: Streak) {
+        habitId = streak.habitId
+        currentStreak = Int32(streak.currentStreak)
+        longestStreak = Int32(streak.longestStreak)
+        lastCompletedDate = streak.lastCompletedDate
+        freezesUsed = Int32(streak.freezesUsed)
+        freezesAvailable = Int32(streak.freezesAvailable)
+    }
+}

package/templates/ios/TrackTemplate/TrackTemplate/Features/Dashboard/TodayView.swift

@@ -27,7 +27,12 @@ struct TodayView: View {
                             .padding(.horizontal)
                         
                         if dailySummary.habits.isEmpty {
-                            EmptyHabitsView(onAddHabit: { showAddHabit = true })
+                            EmptyStateView(
+                                config: .noItems(
+                                    message: "Start building better habits today",
+                                    onAdd: { showAddHabit = true }
+                                )
+                            )
                         } else {
                             ForEach(Array(dailySummary.habits.enumerated()), id: \.element.habit.id) { index, habitWithEntry in
                                 HabitRowView(
@@ -255,38 +260,6 @@ struct HabitRowView: View {
     }
 }
 
-// MARK: - Empty Habits View
-struct EmptyHabitsView: View {
-    let onAddHabit: () -> Void
-    
-    var body: some View {
-        VStack(spacing: AppConfig.Theme.Spacing.md) {
-            Image(systemName: "leaf.fill")
-                .font(.system(size: 50))
-                .foregroundColor(AppConfig.Theme.primaryColor.opacity(0.5))
-            
-            Text("No habits yet")
-                .font(AppConfig.Theme.Typography.title3)
-            
-            Text("Start building better habits today")
-                .font(AppConfig.Theme.Typography.body)
-                .foregroundColor(.secondary)
-            
-            Button(action: onAddHabit) {
-                Text("Add Your First Habit")
-                    .font(AppConfig.Theme.Typography.headline)
-                    .foregroundColor(.white)
-                    .padding(.horizontal, AppConfig.Theme.Spacing.lg)
-                    .padding(.vertical, AppConfig.Theme.Spacing.sm)
-                    .background(AppConfig.Theme.primaryColor)
-                    .cornerRadius(AppConfig.Theme.CornerRadius.medium)
-            }
-            .accessibilityIdentifier("today_add_habit_button")
-        }
-        .padding(AppConfig.Theme.Spacing.xl)
-    }
-}
-
 // MARK: - Completion Celebration View
 struct CompletionCelebrationView: View {
     var body: some View {

package/templates/ios/TrackTemplate/TrackTemplate/Features/Habits/HabitsListView.swift

@@ -49,9 +49,11 @@ struct HabitsListView: View {
                 
                 // Habits List
                 if filteredHabits.isEmpty {
-                    EmptyHabitsListView(
-                        isArchived: showArchived,
-                        onAddHabit: { showAddHabit = true }
+                    EmptyStateView(
+                        config: .noItems(
+                            message: showArchived ? "Archived habits will appear here" : "Create your first habit to get started",
+                            onAdd: showArchived ? nil : { showAddHabit = true }
+                        )
                     )
                 } else {
                     List {
@@ -177,46 +179,6 @@ struct HabitListRowView: View {
     }
 }
 
-// MARK: - Empty Habits List View
-struct EmptyHabitsListView: View {
-    let isArchived: Bool
-    let onAddHabit: () -> Void
-    
-    var body: some View {
-        VStack(spacing: AppConfig.Theme.Spacing.lg) {
-            Spacer()
-            
-            Image(systemName: isArchived ? "archivebox" : "list.bullet.clipboard")
-                .font(.system(size: 60))
-                .foregroundColor(.secondary)
-            
-            Text(isArchived ? "No Archived Habits" : "No Habits Yet")
-                .font(AppConfig.Theme.Typography.title2)
-            
-            Text(isArchived ? "Archived habits will appear here" : "Create your first habit to get started")
-                .font(AppConfig.Theme.Typography.body)
-                .foregroundColor(.secondary)
-                .multilineTextAlignment(.center)
-            
-            if !isArchived {
-                Button(action: onAddHabit) {
-                    Text("Create Habit")
-                        .font(AppConfig.Theme.Typography.headline)
-                        .foregroundColor(.white)
-                        .padding(.horizontal, AppConfig.Theme.Spacing.lg)
-                        .padding(.vertical, AppConfig.Theme.Spacing.sm)
-                        .background(AppConfig.Theme.primaryColor)
-                        .cornerRadius(AppConfig.Theme.CornerRadius.medium)
-                }
-                .accessibilityIdentifier("habits_add_button")
-            }
-            
-            Spacer()
-        }
-        .padding()
-    }
-}
-
 // MARK: - Habit Detail View
 struct HabitDetailView: View {
     let habit: Habit

package/templates/ios/TrackTemplate/TrackTemplate/MockData/MockDataProvider.swift

@@ -4,7 +4,7 @@ import Combine
 
 // MARK: - Mock Data Provider
 // CUSTOMIZE:API - Replace mock data with real data source
-class MockDataProvider: ObservableObject {
+class MockDataProvider: ObservableObject, DataRepository {
     static let shared = MockDataProvider()
     
     // MARK: - Published Properties

package/templates/ios/TrackTemplate/TrackTemplate/Services/HabitManager.swift

@@ -260,7 +260,7 @@ class HabitManager: ObservableObject {
         }
         
         // Load achievements
-        achievements = MockDataProvider.shared.achievements
+        achievements = DataSourceResolver.repository.achievements
     }
     
     private func saveHabits() {

package/templates/ios/TrackTemplate/TrackTemplate.xcodeproj/project.pbxproj

@@ -7,24 +7,35 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
+		02F29184D468849A9A819D72 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 8ED6A1D3EDCDB73F6FC871AB /* Assets.xcassets */; };
+		09F0EB35872444CBCA95B104 /* AchievementEntity.swift in Sources */ = {isa = PBXBuildFile; fileRef = DF98B15B7BAA55DD5F78F264 /* AchievementEntity.swift */; };
 		1218E606CF66A0D4A18708B3 /* TodayView.swift in Sources */ = {isa = PBXBuildFile; fileRef = D476F4337FD1FEB067699E04 /* TodayView.swift */; };
 		174A0960906E74C2CBDDD87A /* NetworkMonitor.swift in Sources */ = {isa = PBXBuildFile; fileRef = 916488B857AA1227C3C3F75F /* NetworkMonitor.swift */; };
 		1815B71DBE91C3CEBB2557AF /* MockDataProviderTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = 21C020A57D7DC0FFE0EE804F /* MockDataProviderTests.swift */; };
 		32905A6C3620A57E260A6C28 /* OfflineBanner.swift in Sources */ = {isa = PBXBuildFile; fileRef = AAD5DAE138A58423B6A54BB4 /* OfflineBanner.swift */; };
 		3E3927E3E76E54AD821CB6AE /* MockDataProvider.swift in Sources */ = {isa = PBXBuildFile; fileRef = F1BD78CD036AB49AF9F3465C /* MockDataProvider.swift */; };
+		4354E33CEC6CE73C60A29945 /* CoreDataStack.swift in Sources */ = {isa = PBXBuildFile; fileRef = 40CA2A3A0A608491C1ABABC1 /* CoreDataStack.swift */; };
 		44E138C734742D539C10E138 /* SyncManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = 58F6D94727655DD21CE775BB /* SyncManager.swift */; };
+		4659DA932F7ABCB49DD2D59C /* EmptyStateView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 49EF2A1842F6A12F66A3D844 /* EmptyStateView.swift */; };
 		4D69A25B53C888EC06F81E6D /* SettingsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 537B40D07584475F7533438F /* SettingsView.swift */; };
+		612B16AC311689F90C4FA00C /* StreakEntity.swift in Sources */ = {isa = PBXBuildFile; fileRef = EBBD676ADA1E6765785B5CD2 /* StreakEntity.swift */; };
 		641D3A5E4470B7411CA4FAE1 /* Models.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4BBECF5056929F118A7E25F7 /* Models.swift */; };
 		706FFDD3A658E862F89A080C /* Colors.swift in Sources */ = {isa = PBXBuildFile; fileRef = C17D71982F494C65E44D6EAF /* Colors.swift */; };
+		72AEA463326017C6DE116F46 /* DataRepository.swift in Sources */ = {isa = PBXBuildFile; fileRef = 6E33AF1B700D2FE931720948 /* DataRepository.swift */; };
 		7F35CBCE4AEA63FF5A850B7B /* HabitsListView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 140B3031202FE5621EE6DC05 /* HabitsListView.swift */; };
 		80A3B6FA4B26DC90A77B5A44 /* AppConfig.swift in Sources */ = {isa = PBXBuildFile; fileRef = F307278DC21A4EA1617A9A1F /* AppConfig.swift */; };
 		824BEA38DA708AFDEB077CF0 /* ModelsTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = 74D3B6A0318540E54ED54B44 /* ModelsTests.swift */; };
+		82A6422D651EFFB5CDC57F15 /* LocalStorageProvider.swift in Sources */ = {isa = PBXBuildFile; fileRef = 1FBB2866168F55A19E4BAF48 /* LocalStorageProvider.swift */; };
 		8A98725AE1F4241D749DF6C7 /* HabitManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = E0F008AF07733F22B39D1906 /* HabitManager.swift */; };
+		96A9C7E362A913881D1B2F59 /* HabitEntryEntity.swift in Sources */ = {isa = PBXBuildFile; fileRef = 5EF90E41F7D9E82936DF0480 /* HabitEntryEntity.swift */; };
 		BAA2A80C82CD2499B76D275A /* AchievementsView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 930C3798AD395C075882C679 /* AchievementsView.swift */; };
 		C9EE29659550DBC2FEFD2FC2 /* LocalPersistence.swift in Sources */ = {isa = PBXBuildFile; fileRef = 764297F97E31AD41EFAF50C3 /* LocalPersistence.swift */; };
+		CE9D1E74EF5F3EE6417ED041 /* DataRepositoryTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = CA34DF418F91B3BD48B5BCA7 /* DataRepositoryTests.swift */; };
 		D41EA3BF151A201AAE4F565A /* TrackTemplateApp.swift in Sources */ = {isa = PBXBuildFile; fileRef = E5AA67B3E855FDCC14039B9D /* TrackTemplateApp.swift */; };
 		D81FE6BAD1B8E6664AB5605D /* HabitManagerTests.swift in Sources */ = {isa = PBXBuildFile; fileRef = F3AB1929449D19B2EBF13012 /* HabitManagerTests.swift */; };
 		DA4D5E59C395F915A2534444 /* DashboardView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 17E093C7FB3C44904ADE7D30 /* DashboardView.swift */; };
+		E74C405F55B279DB44AD487D /* DataSourceResolver.swift in Sources */ = {isa = PBXBuildFile; fileRef = 6B193217FC577BBD6BE3AC27 /* DataSourceResolver.swift */; };
+		F80302863F51FD58DCE0C3B5 /* HabitEntity.swift in Sources */ = {isa = PBXBuildFile; fileRef = 4E9DCF6CE0B4270D309731A0 /* HabitEntity.swift */; };
 /* End PBXBuildFile section */
 
 /* Begin PBXContainerItemProxy section */
@@ -41,21 +52,32 @@
 		11B590A337F70B003C7F32C7 /* TrackTemplate.app */ = {isa = PBXFileReference; includeInIndex = 0; lastKnownFileType = wrapper.application; path = TrackTemplate.app; sourceTree = BUILT_PRODUCTS_DIR; };
 		140B3031202FE5621EE6DC05 /* HabitsListView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HabitsListView.swift; sourceTree = "<group>"; };
 		17E093C7FB3C44904ADE7D30 /* DashboardView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = DashboardView.swift; sourceTree = "<group>"; };
+		1FBB2866168F55A19E4BAF48 /* LocalStorageProvider.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = LocalStorageProvider.swift; sourceTree = "<group>"; };
 		21C020A57D7DC0FFE0EE804F /* MockDataProviderTests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = MockDataProviderTests.swift; sourceTree = "<group>"; };
 		37563DEFD9B6436651D42B62 /* TrackTemplateTests.xctest */ = {isa = PBXFileReference; includeInIndex = 0; lastKnownFileType = wrapper.cfbundle; path = TrackTemplateTests.xctest; sourceTree = BUILT_PRODUCTS_DIR; };
+		40CA2A3A0A608491C1ABABC1 /* CoreDataStack.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = CoreDataStack.swift; sourceTree = "<group>"; };
+		49EF2A1842F6A12F66A3D844 /* EmptyStateView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = EmptyStateView.swift; sourceTree = "<group>"; };
 		4BBECF5056929F118A7E25F7 /* Models.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Models.swift; sourceTree = "<group>"; };
+		4E9DCF6CE0B4270D309731A0 /* HabitEntity.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HabitEntity.swift; sourceTree = "<group>"; };
 		537B40D07584475F7533438F /* SettingsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SettingsView.swift; sourceTree = "<group>"; };
 		58F6D94727655DD21CE775BB /* SyncManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SyncManager.swift; sourceTree = "<group>"; };
+		5EF90E41F7D9E82936DF0480 /* HabitEntryEntity.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HabitEntryEntity.swift; sourceTree = "<group>"; };
+		6B193217FC577BBD6BE3AC27 /* DataSourceResolver.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = DataSourceResolver.swift; sourceTree = "<group>"; };
+		6E33AF1B700D2FE931720948 /* DataRepository.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = DataRepository.swift; sourceTree = "<group>"; };
 		74D3B6A0318540E54ED54B44 /* ModelsTests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ModelsTests.swift; sourceTree = "<group>"; };
 		764297F97E31AD41EFAF50C3 /* LocalPersistence.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = LocalPersistence.swift; sourceTree = "<group>"; };
+		8ED6A1D3EDCDB73F6FC871AB /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
 		916488B857AA1227C3C3F75F /* NetworkMonitor.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = NetworkMonitor.swift; sourceTree = "<group>"; };
 		930C3798AD395C075882C679 /* AchievementsView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AchievementsView.swift; sourceTree = "<group>"; };
 		A0A6BA7C62E6967468E72E56 /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist; path = Info.plist; sourceTree = "<group>"; };
 		AAD5DAE138A58423B6A54BB4 /* OfflineBanner.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = OfflineBanner.swift; sourceTree = "<group>"; };
 		C17D71982F494C65E44D6EAF /* Colors.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Colors.swift; sourceTree = "<group>"; };
+		CA34DF418F91B3BD48B5BCA7 /* DataRepositoryTests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = DataRepositoryTests.swift; sourceTree = "<group>"; };
 		D476F4337FD1FEB067699E04 /* TodayView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = TodayView.swift; sourceTree = "<group>"; };
+		DF98B15B7BAA55DD5F78F264 /* AchievementEntity.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AchievementEntity.swift; sourceTree = "<group>"; };
 		E0F008AF07733F22B39D1906 /* HabitManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HabitManager.swift; sourceTree = "<group>"; };
 		E5AA67B3E855FDCC14039B9D /* TrackTemplateApp.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = TrackTemplateApp.swift; sourceTree = "<group>"; };
+		EBBD676ADA1E6765785B5CD2 /* StreakEntity.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = StreakEntity.swift; sourceTree = "<group>"; };
 		F1BD78CD036AB49AF9F3465C /* MockDataProvider.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = MockDataProvider.swift; sourceTree = "<group>"; };
 		F307278DC21A4EA1617A9A1F /* AppConfig.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AppConfig.swift; sourceTree = "<group>"; };
 		F3AB1929449D19B2EBF13012 /* HabitManagerTests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HabitManagerTests.swift; sourceTree = "<group>"; };
@@ -91,9 +113,12 @@
 		501177E904BC0E0C437D5D84 /* Data */ = {
 			isa = PBXGroup;
 			children = (
+				6E33AF1B700D2FE931720948 /* DataRepository.swift */,
+				6B193217FC577BBD6BE3AC27 /* DataSourceResolver.swift */,
 				764297F97E31AD41EFAF50C3 /* LocalPersistence.swift */,
 				916488B857AA1227C3C3F75F /* NetworkMonitor.swift */,
 				58F6D94727655DD21CE775BB /* SyncManager.swift */,
+				67AD7BB9594DB55096C04EBD /* LocalStorage */,
 			);
 			path = Data;
 			sourceTree = "<group>";
@@ -106,6 +131,19 @@
 			path = Achievements;
 			sourceTree = "<group>";
 		};
+		67AD7BB9594DB55096C04EBD /* LocalStorage */ = {
+			isa = PBXGroup;
+			children = (
+				DF98B15B7BAA55DD5F78F264 /* AchievementEntity.swift */,
+				40CA2A3A0A608491C1ABABC1 /* CoreDataStack.swift */,
+				4E9DCF6CE0B4270D309731A0 /* HabitEntity.swift */,
+				5EF90E41F7D9E82936DF0480 /* HabitEntryEntity.swift */,
+				1FBB2866168F55A19E4BAF48 /* LocalStorageProvider.swift */,
+				EBBD676ADA1E6765785B5CD2 /* StreakEntity.swift */,
+			);
+			path = LocalStorage;
+			sourceTree = "<group>";
+		};
 		712260C4171E2E455DFB955B /* Habits */ = {
 			isa = PBXGroup;
 			children = (
@@ -117,6 +155,7 @@
 		7B580897CA530FD7F9621F5A /* Core */ = {
 			isa = PBXGroup;
 			children = (
+				49EF2A1842F6A12F66A3D844 /* EmptyStateView.swift */,
 				496E799D113D629DDBBA1F1A /* Theme */,
 			);
 			path = Core;
@@ -178,6 +217,7 @@
 		DD2FA9BE601E04CBC9C1326C /* Resources */ = {
 			isa = PBXGroup;
 			children = (
+				8ED6A1D3EDCDB73F6FC871AB /* Assets.xcassets */,
 				A0A6BA7C62E6967468E72E56 /* Info.plist */,
 			);
 			path = Resources;
@@ -194,6 +234,7 @@
 		E90EB891C9AA7A825074CC36 /* TrackTemplateTests */ = {
 			isa = PBXGroup;
 			children = (
+				CA34DF418F91B3BD48B5BCA7 /* DataRepositoryTests.swift */,
 				F3AB1929449D19B2EBF13012 /* HabitManagerTests.swift */,
 				21C020A57D7DC0FFE0EE804F /* MockDataProviderTests.swift */,
 				74D3B6A0318540E54ED54B44 /* ModelsTests.swift */,
@@ -233,6 +274,7 @@
 			buildConfigurationList = CD46C91699D88DE77A90F117 /* Build configuration list for PBXNativeTarget "TrackTemplate" */;
 			buildPhases = (
 				937775264687FA0966D38548 /* Sources */,
+				B26124E8AA9DA5B81E58D371 /* Resources */,
 			);
 			buildRules = (
 			);
@@ -292,11 +334,23 @@
 		};
 /* End PBXProject section */
 
+/* Begin PBXResourcesBuildPhase section */
+		B26124E8AA9DA5B81E58D371 /* Resources */ = {
+			isa = PBXResourcesBuildPhase;
+			buildActionMask = 2147483647;
+			files = (
+				02F29184D468849A9A819D72 /* Assets.xcassets in Resources */,
+			);
+			runOnlyForDeploymentPostprocessing = 0;
+		};
+/* End PBXResourcesBuildPhase section */
+
 /* Begin PBXSourcesBuildPhase section */
 		37B2A2D0D3B672444A75D75C /* Sources */ = {
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				CE9D1E74EF5F3EE6417ED041 /* DataRepositoryTests.swift in Sources */,
 				D81FE6BAD1B8E6664AB5605D /* HabitManagerTests.swift in Sources */,
 				1815B71DBE91C3CEBB2557AF /* MockDataProviderTests.swift in Sources */,
 				824BEA38DA708AFDEB077CF0 /* ModelsTests.swift in Sources */,
@@ -307,18 +361,27 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				09F0EB35872444CBCA95B104 /* AchievementEntity.swift in Sources */,
 				BAA2A80C82CD2499B76D275A /* AchievementsView.swift in Sources */,
 				80A3B6FA4B26DC90A77B5A44 /* AppConfig.swift in Sources */,
 				706FFDD3A658E862F89A080C /* Colors.swift in Sources */,
+				4354E33CEC6CE73C60A29945 /* CoreDataStack.swift in Sources */,
 				DA4D5E59C395F915A2534444 /* DashboardView.swift in Sources */,
+				72AEA463326017C6DE116F46 /* DataRepository.swift in Sources */,
+				E74C405F55B279DB44AD487D /* DataSourceResolver.swift in Sources */,
+				4659DA932F7ABCB49DD2D59C /* EmptyStateView.swift in Sources */,
+				F80302863F51FD58DCE0C3B5 /* HabitEntity.swift in Sources */,
+				96A9C7E362A913881D1B2F59 /* HabitEntryEntity.swift in Sources */,
 				8A98725AE1F4241D749DF6C7 /* HabitManager.swift in Sources */,
 				7F35CBCE4AEA63FF5A850B7B /* HabitsListView.swift in Sources */,
 				C9EE29659550DBC2FEFD2FC2 /* LocalPersistence.swift in Sources */,
+				82A6422D651EFFB5CDC57F15 /* LocalStorageProvider.swift in Sources */,
 				3E3927E3E76E54AD821CB6AE /* MockDataProvider.swift in Sources */,
 				641D3A5E4470B7411CA4FAE1 /* Models.swift in Sources */,
 				174A0960906E74C2CBDDD87A /* NetworkMonitor.swift in Sources */,
 				32905A6C3620A57E260A6C28 /* OfflineBanner.swift in Sources */,
 				4D69A25B53C888EC06F81E6D /* SettingsView.swift in Sources */,
+				612B16AC311689F90C4FA00C /* StreakEntity.swift in Sources */,
 				44E138C734742D539C10E138 /* SyncManager.swift in Sources */,
 				1218E606CF66A0D4A18708B3 /* TodayView.swift in Sources */,
 				D41EA3BF151A201AAE4F565A /* TrackTemplateApp.swift in Sources */,

package/templates/ios/TrackTemplate/TrackTemplateTests/DataRepositoryTests.swift

@@ -0,0 +1,26 @@
+import XCTest
+@testable import TrackTemplate
+
+/// Tests that MockDataProvider conforms to DataRepository and that
+/// DataSourceResolver returns the correct implementation.
+final class DataRepositoryTests: XCTestCase {
+
+    // MARK: - DataRepository Protocol Conformance
+
+    func testMockDataProviderConformsToDataRepository() {
+        let repo: any DataRepository = MockDataProvider.shared
+        XCTAssertNotNil(repo, "MockDataProvider should conform to DataRepository")
+    }
+
+    func testDataSourceResolverReturnsMock() {
+        let repo = DataSourceResolver.repository
+        XCTAssertNotNil(repo, "DataSourceResolver should return a non-nil repository")
+        XCTAssertTrue(repo is MockDataProvider, "When active is .mock, should return MockDataProvider")
+    }
+
+    func testRepositoryLoadData() {
+        let repo: any DataRepository = MockDataProvider.shared
+        XCTAssertFalse(repo.habits.isEmpty, "habits should not be empty")
+        XCTAssertFalse(repo.achievements.isEmpty, "achievements should not be empty")
+    }
+}

package/templates/ios/TrackTemplate/tests/01_launch_today.yaml

@@ -1,14 +1,53 @@
 appId: ${APP_BUNDLE_ID}
 ---
+- clearState
 - launchApp
+- extendedWaitUntil:
+    visible:
+      - "Today"
+      - "Habits"
+      - "Dashboard"
+      - "Profile"
+    timeout: 10000
+# Skip onboarding if present
 - tapOn:
     text: "Skip"
     optional: true
 - tapOn:
     text: "Get Started"
     optional: true
-- assertVisible: "Today"
-- assertVisible: ".*Morning Exercise.*"
-- assertVisible: ".*Read.*"
-- assertVisible: ".*Drink Water.*"
-- assertVisible: "Today's Habits"
+- tapOn:
+    text: "Continue"
+    optional: true
+# Verify we're on Today tab (default)
+- assertVisible:
+    text: "Today"
+# Verify today screen content
+- assertVisible:
+    id: "today_add_habit_button"
+- assertVisible:
+    id: "today_progress_ring"
+# Check for habits or empty state
+- assertVisible:
+    id: "today_empty_state"
+    optional: true
+# If habits exist, verify at least one habit card
+- assertVisible:
+    id: "today_habit_0_card"
+    optional: true
+- assertVisible:
+    id: "today_habit_0_toggle"
+    optional: true
+# Verify date selector buttons
+- assertVisible:
+    id: "today_date_0_button"
+    optional: true
+# Verify tab bar is visible
+- assertVisible:
+    text: "Today"
+- assertVisible:
+    text: "Habits"
+- assertVisible:
+    text: "Dashboard"
+- assertVisible:
+    text: "Profile"