@shaykec/app-agent 1.0.5 → 1.0.7

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

@@ -0,0 +1,44 @@
+import CoreData
+import Foundation
+
+/// Core Data managed object for Achievement persistence.
+@objc(AchievementEntity)
+final class AchievementEntity: NSManagedObject {
+    @NSManaged var id: String
+    @NSManaged var title: String
+    @NSManaged var achievementDescription: String
+    @NSManaged var icon: String
+    @NSManaged var requirement: Int32
+    @NSManaged var type: String
+    @NSManaged var isUnlocked: Bool
+    @NSManaged var unlockedAt: Date?
+    @NSManaged var progress: Int32
+
+    // MARK: - Conversion
+
+    func toAchievement() -> Achievement {
+        Achievement(
+            id: id,
+            title: title,
+            description: achievementDescription,
+            icon: icon,
+            requirement: Int(requirement),
+            type: AchievementType(rawValue: type) ?? .totalCompletions,
+            isUnlocked: isUnlocked,
+            unlockedAt: unlockedAt,
+            progress: Int(progress)
+        )
+    }
+
+    func update(from achievement: Achievement) {
+        id = achievement.id
+        title = achievement.title
+        achievementDescription = achievement.description
+        icon = achievement.icon
+        requirement = Int32(achievement.requirement)
+        type = achievement.type.rawValue
+        isUnlocked = achievement.isUnlocked
+        unlockedAt = achievement.unlockedAt
+        progress = Int32(achievement.progress)
+    }
+}

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

@@ -0,0 +1,269 @@
+import CoreData
+import Foundation
+
+/// Programmatic Core Data stack — no .xcdatamodeld file needed.
+/// Defines entities matching TrackTemplate's domain models.
+final class CoreDataStack {
+    static let shared = CoreDataStack()
+
+    let container: NSPersistentContainer
+
+    /// In-memory store for previews/tests; persistent SQLite store for production.
+    private init(inMemory: Bool = false) {
+        let model = Self.buildModel()
+        container = NSPersistentContainer(name: "LocalStorage", managedObjectModel: model)
+        if inMemory {
+            let desc = NSPersistentStoreDescription()
+            desc.type = NSInMemoryStoreType
+            container.persistentStoreDescriptions = [desc]
+        }
+        container.loadPersistentStores { _, error in
+            if let error {
+                fatalError("Core Data failed to load: \(error.localizedDescription)")
+            }
+        }
+        container.viewContext.automaticallyMergesChangesFromParent = true
+        container.viewContext.mergePolicy = NSMergeByPropertyObjectTrumpMergePolicy
+    }
+
+    var viewContext: NSManagedObjectContext { container.viewContext }
+
+    func save() {
+        let ctx = viewContext
+        guard ctx.hasChanges else { return }
+        do {
+            try ctx.save()
+        } catch {
+            print("[CoreDataStack] Save failed: \(error.localizedDescription)")
+        }
+    }
+
+    // MARK: - Programmatic Model
+
+    private static func buildModel() -> NSManagedObjectModel {
+        let model = NSManagedObjectModel()
+
+        // — HabitEntity —
+        let habitEntity = NSEntityDescription()
+        habitEntity.name = "HabitEntity"
+        habitEntity.managedObjectClassName = NSStringFromClass(HabitEntity.self)
+
+        let habitId = NSAttributeDescription()
+        habitId.name = "id"
+        habitId.attributeType = .stringAttributeType
+
+        let habitName = NSAttributeDescription()
+        habitName.name = "name"
+        habitName.attributeType = .stringAttributeType
+        habitName.defaultValue = ""
+
+        let habitDesc = NSAttributeDescription()
+        habitDesc.name = "habitDescription"
+        habitDesc.attributeType = .stringAttributeType
+        habitDesc.isOptional = true
+
+        let habitIcon = NSAttributeDescription()
+        habitIcon.name = "icon"
+        habitIcon.attributeType = .stringAttributeType
+        habitIcon.defaultValue = "star"
+
+        let habitColor = NSAttributeDescription()
+        habitColor.name = "color"
+        habitColor.attributeType = .stringAttributeType
+        habitColor.defaultValue = "#3B82F6"
+
+        let habitCategoryId = NSAttributeDescription()
+        habitCategoryId.name = "categoryId"
+        habitCategoryId.attributeType = .stringAttributeType
+        habitCategoryId.defaultValue = ""
+
+        let habitFrequency = NSAttributeDescription()
+        habitFrequency.name = "frequency"
+        habitFrequency.attributeType = .stringAttributeType
+        habitFrequency.defaultValue = "Daily"
+
+        let habitTargetValue = NSAttributeDescription()
+        habitTargetValue.name = "targetValue"
+        habitTargetValue.attributeType = .integer32AttributeType
+        habitTargetValue.defaultValue = 1
+
+        let habitUnit = NSAttributeDescription()
+        habitUnit.name = "unit"
+        habitUnit.attributeType = .stringAttributeType
+        habitUnit.isOptional = true
+
+        let habitReminderTime = NSAttributeDescription()
+        habitReminderTime.name = "reminderTime"
+        habitReminderTime.attributeType = .stringAttributeType
+        habitReminderTime.isOptional = true
+
+        let habitReminderDaysJSON = NSAttributeDescription()
+        habitReminderDaysJSON.name = "reminderDaysJSON"
+        habitReminderDaysJSON.attributeType = .stringAttributeType
+        habitReminderDaysJSON.isOptional = true
+
+        let habitIsArchived = NSAttributeDescription()
+        habitIsArchived.name = "isArchived"
+        habitIsArchived.attributeType = .booleanAttributeType
+        habitIsArchived.defaultValue = false
+
+        let habitCreatedAt = NSAttributeDescription()
+        habitCreatedAt.name = "createdAt"
+        habitCreatedAt.attributeType = .dateAttributeType
+
+        habitEntity.properties = [
+            habitId, habitName, habitDesc, habitIcon, habitColor, habitCategoryId,
+            habitFrequency, habitTargetValue, habitUnit, habitReminderTime,
+            habitReminderDaysJSON, habitIsArchived, habitCreatedAt
+        ]
+
+        // — HabitEntryEntity —
+        let entryEntity = NSEntityDescription()
+        entryEntity.name = "HabitEntryEntity"
+        entryEntity.managedObjectClassName = NSStringFromClass(HabitEntryEntity.self)
+
+        let entryId = NSAttributeDescription()
+        entryId.name = "id"
+        entryId.attributeType = .stringAttributeType
+
+        let entryHabitId = NSAttributeDescription()
+        entryHabitId.name = "habitId"
+        entryHabitId.attributeType = .stringAttributeType
+
+        let entryDate = NSAttributeDescription()
+        entryDate.name = "date"
+        entryDate.attributeType = .dateAttributeType
+
+        let entryValue = NSAttributeDescription()
+        entryValue.name = "value"
+        entryValue.attributeType = .integer32AttributeType
+        entryValue.defaultValue = 0
+
+        let entryIsCompleted = NSAttributeDescription()
+        entryIsCompleted.name = "isCompleted"
+        entryIsCompleted.attributeType = .booleanAttributeType
+        entryIsCompleted.defaultValue = false
+
+        let entryNote = NSAttributeDescription()
+        entryNote.name = "note"
+        entryNote.attributeType = .stringAttributeType
+        entryNote.isOptional = true
+
+        let entryMood = NSAttributeDescription()
+        entryMood.name = "mood"
+        entryMood.attributeType = .integer32AttributeType
+        entryMood.isOptional = true
+
+        let entryPhotoURL = NSAttributeDescription()
+        entryPhotoURL.name = "photoURL"
+        entryPhotoURL.attributeType = .stringAttributeType
+        entryPhotoURL.isOptional = true
+
+        let entryCompletedAt = NSAttributeDescription()
+        entryCompletedAt.name = "completedAt"
+        entryCompletedAt.attributeType = .dateAttributeType
+        entryCompletedAt.isOptional = true
+
+        entryEntity.properties = [
+            entryId, entryHabitId, entryDate, entryValue, entryIsCompleted,
+            entryNote, entryMood, entryPhotoURL, entryCompletedAt
+        ]
+
+        // — AchievementEntity —
+        let achievementEntity = NSEntityDescription()
+        achievementEntity.name = "AchievementEntity"
+        achievementEntity.managedObjectClassName = NSStringFromClass(AchievementEntity.self)
+
+        let achievementId = NSAttributeDescription()
+        achievementId.name = "id"
+        achievementId.attributeType = .stringAttributeType
+
+        let achievementTitle = NSAttributeDescription()
+        achievementTitle.name = "title"
+        achievementTitle.attributeType = .stringAttributeType
+        achievementTitle.defaultValue = ""
+
+        let achievementDesc = NSAttributeDescription()
+        achievementDesc.name = "achievementDescription"
+        achievementDesc.attributeType = .stringAttributeType
+        achievementDesc.defaultValue = ""
+
+        let achievementIcon = NSAttributeDescription()
+        achievementIcon.name = "icon"
+        achievementIcon.attributeType = .stringAttributeType
+        achievementIcon.defaultValue = "star"
+
+        let achievementRequirement = NSAttributeDescription()
+        achievementRequirement.name = "requirement"
+        achievementRequirement.attributeType = .integer32AttributeType
+        achievementRequirement.defaultValue = 1
+
+        let achievementType = NSAttributeDescription()
+        achievementType.name = "type"
+        achievementType.attributeType = .stringAttributeType
+        achievementType.defaultValue = "total_completions"
+
+        let achievementIsUnlocked = NSAttributeDescription()
+        achievementIsUnlocked.name = "isUnlocked"
+        achievementIsUnlocked.attributeType = .booleanAttributeType
+        achievementIsUnlocked.defaultValue = false
+
+        let achievementUnlockedAt = NSAttributeDescription()
+        achievementUnlockedAt.name = "unlockedAt"
+        achievementUnlockedAt.attributeType = .dateAttributeType
+        achievementUnlockedAt.isOptional = true
+
+        let achievementProgress = NSAttributeDescription()
+        achievementProgress.name = "progress"
+        achievementProgress.attributeType = .integer32AttributeType
+        achievementProgress.defaultValue = 0
+
+        achievementEntity.properties = [
+            achievementId, achievementTitle, achievementDesc, achievementIcon,
+            achievementRequirement, achievementType, achievementIsUnlocked,
+            achievementUnlockedAt, achievementProgress
+        ]
+
+        // — StreakEntity —
+        let streakEntity = NSEntityDescription()
+        streakEntity.name = "StreakEntity"
+        streakEntity.managedObjectClassName = NSStringFromClass(StreakEntity.self)
+
+        let streakHabitId = NSAttributeDescription()
+        streakHabitId.name = "habitId"
+        streakHabitId.attributeType = .stringAttributeType
+
+        let streakCurrent = NSAttributeDescription()
+        streakCurrent.name = "currentStreak"
+        streakCurrent.attributeType = .integer32AttributeType
+        streakCurrent.defaultValue = 0
+
+        let streakLongest = NSAttributeDescription()
+        streakLongest.name = "longestStreak"
+        streakLongest.attributeType = .integer32AttributeType
+        streakLongest.defaultValue = 0
+
+        let streakLastCompleted = NSAttributeDescription()
+        streakLastCompleted.name = "lastCompletedDate"
+        streakLastCompleted.attributeType = .dateAttributeType
+        streakLastCompleted.isOptional = true
+
+        let streakFreezesUsed = NSAttributeDescription()
+        streakFreezesUsed.name = "freezesUsed"
+        streakFreezesUsed.attributeType = .integer32AttributeType
+        streakFreezesUsed.defaultValue = 0
+
+        let streakFreezesAvailable = NSAttributeDescription()
+        streakFreezesAvailable.name = "freezesAvailable"
+        streakFreezesAvailable.attributeType = .integer32AttributeType
+        streakFreezesAvailable.defaultValue = 2
+
+        streakEntity.properties = [
+            streakHabitId, streakCurrent, streakLongest, streakLastCompleted,
+            streakFreezesUsed, streakFreezesAvailable
+        ]
+
+        model.entities = [habitEntity, entryEntity, achievementEntity, streakEntity]
+        return model
+    }
+}

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

@@ -0,0 +1,75 @@
+import CoreData
+import Foundation
+
+/// Core Data managed object for Habit persistence.
+@objc(HabitEntity)
+final class HabitEntity: NSManagedObject {
+    @NSManaged var id: String
+    @NSManaged var name: String
+    @NSManaged var habitDescription: String?
+    @NSManaged var icon: String
+    @NSManaged var color: String
+    @NSManaged var categoryId: String
+    @NSManaged var frequency: String
+    @NSManaged var targetValue: Int32
+    @NSManaged var unit: String?
+    @NSManaged var reminderTime: String?
+    @NSManaged var reminderDaysJSON: String?
+    @NSManaged var isArchived: Bool
+    @NSManaged var createdAt: Date?
+
+    // MARK: - Conversion
+
+    func toHabit() -> Habit {
+        Habit(
+            id: id,
+            name: name,
+            description: habitDescription,
+            icon: icon,
+            color: color,
+            categoryId: categoryId,
+            frequency: HabitFrequency(rawValue: frequency) ?? .daily,
+            targetValue: Int(targetValue),
+            unit: unit,
+            reminderTime: reminderTime,
+            reminderDays: decodeReminderDays(),
+            isArchived: isArchived,
+            createdAt: createdAt ?? Date()
+        )
+    }
+
+    func update(from habit: Habit) {
+        id = habit.id
+        name = habit.name
+        habitDescription = habit.description
+        icon = habit.icon
+        color = habit.color
+        categoryId = habit.categoryId
+        frequency = habit.frequency.rawValue
+        targetValue = Int32(habit.targetValue)
+        unit = habit.unit
+        reminderTime = habit.reminderTime
+        reminderDaysJSON = encodeReminderDays(habit.reminderDays)
+        isArchived = habit.isArchived
+        createdAt = habit.createdAt
+    }
+
+    // MARK: - JSON Helpers
+
+    private func decodeReminderDays() -> [Int]? {
+        guard let json = reminderDaysJSON,
+              let data = json.data(using: .utf8),
+              let days = try? JSONDecoder().decode([Int].self, from: data) else {
+            return nil
+        }
+        return days.isEmpty ? nil : days
+    }
+
+    private func encodeReminderDays(_ days: [Int]?) -> String? {
+        guard let days = days, let data = try? JSONEncoder().encode(days),
+              let json = String(data: data, encoding: .utf8) else {
+            return nil
+        }
+        return json
+    }
+}

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

@@ -0,0 +1,44 @@
+import CoreData
+import Foundation
+
+/// Core Data managed object for HabitEntry persistence.
+@objc(HabitEntryEntity)
+final class HabitEntryEntity: NSManagedObject {
+    @NSManaged var id: String
+    @NSManaged var habitId: String
+    @NSManaged var date: Date?
+    @NSManaged var value: Int32
+    @NSManaged var isCompleted: Bool
+    @NSManaged var note: String?
+    @NSManaged var mood: Int32
+    @NSManaged var photoURL: String?
+    @NSManaged var completedAt: Date?
+
+    // MARK: - Conversion
+
+    func toHabitEntry() -> HabitEntry {
+        HabitEntry(
+            id: id,
+            habitId: habitId,
+            date: date ?? Date(),
+            value: Int(value),
+            isCompleted: isCompleted,
+            note: note,
+            mood: mood > 0 ? Int(mood) : nil,
+            photoURL: photoURL,
+            completedAt: completedAt
+        )
+    }
+
+    func update(from entry: HabitEntry) {
+        id = entry.id
+        habitId = entry.habitId
+        date = entry.date
+        value = Int32(entry.value)
+        isCompleted = entry.isCompleted
+        note = entry.note
+        mood = Int32(entry.mood ?? 0)
+        photoURL = entry.photoURL
+        completedAt = entry.completedAt
+    }
+}

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)
+    }
+}