@shaykec/app-agent 1.0.4 → 1.0.6

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

@@ -0,0 +1,365 @@
+import CoreData
+import Foundation
+
+/// Programmatic Core Data stack — no .xcdatamodeld file needed.
+/// Defines CategoryEntity, ProductEntity, OrderEntity, ReviewEntity, and BannerEntity models in code.
+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()
+
+        // — CategoryEntity —
+        let categoryEntity = NSEntityDescription()
+        categoryEntity.name = "CategoryEntity"
+        categoryEntity.managedObjectClassName = NSStringFromClass(CategoryEntity.self)
+
+        let catId = NSAttributeDescription()
+        catId.name = "id"
+        catId.attributeType = .stringAttributeType
+
+        let catName = NSAttributeDescription()
+        catName.name = "name"
+        catName.attributeType = .stringAttributeType
+        catName.defaultValue = ""
+
+        let catIcon = NSAttributeDescription()
+        catIcon.name = "icon"
+        catIcon.attributeType = .stringAttributeType
+        catIcon.defaultValue = ""
+
+        let catImage = NSAttributeDescription()
+        catImage.name = "image"
+        catImage.attributeType = .stringAttributeType
+        catImage.isOptional = true
+
+        let catParentId = NSAttributeDescription()
+        catParentId.name = "parentId"
+        catParentId.attributeType = .stringAttributeType
+        catParentId.isOptional = true
+
+        let catProductCount = NSAttributeDescription()
+        catProductCount.name = "productCount"
+        catProductCount.attributeType = .integer32AttributeType
+        catProductCount.isOptional = false
+        catProductCount.defaultValue = 0
+
+        categoryEntity.properties = [catId, catName, catIcon, catImage, catParentId, catProductCount]
+
+        // — ProductEntity —
+        let productEntity = NSEntityDescription()
+        productEntity.name = "ProductEntity"
+        productEntity.managedObjectClassName = NSStringFromClass(ProductEntity.self)
+
+        let prodId = NSAttributeDescription()
+        prodId.name = "id"
+        prodId.attributeType = .stringAttributeType
+
+        let prodName = NSAttributeDescription()
+        prodName.name = "name"
+        prodName.attributeType = .stringAttributeType
+        prodName.defaultValue = ""
+
+        let prodDesc = NSAttributeDescription()
+        prodDesc.name = "productDescription"
+        prodDesc.attributeType = .stringAttributeType
+        prodDesc.defaultValue = ""
+
+        let prodPrice = NSAttributeDescription()
+        prodPrice.name = "price"
+        prodPrice.attributeType = .doubleAttributeType
+
+        let prodOriginalPrice = NSAttributeDescription()
+        prodOriginalPrice.name = "originalPrice"
+        prodOriginalPrice.attributeType = .doubleAttributeType
+        prodOriginalPrice.isOptional = true
+
+        let prodImagesJSON = NSAttributeDescription()
+        prodImagesJSON.name = "imagesJSON"
+        prodImagesJSON.attributeType = .stringAttributeType
+        prodImagesJSON.isOptional = true
+
+        let prodCategoryId = NSAttributeDescription()
+        prodCategoryId.name = "categoryId"
+        prodCategoryId.attributeType = .stringAttributeType
+
+        let prodBrand = NSAttributeDescription()
+        prodBrand.name = "brand"
+        prodBrand.attributeType = .stringAttributeType
+        prodBrand.isOptional = true
+
+        let prodRating = NSAttributeDescription()
+        prodRating.name = "rating"
+        prodRating.attributeType = .doubleAttributeType
+
+        let prodReviewCount = NSAttributeDescription()
+        prodReviewCount.name = "reviewCount"
+        prodReviewCount.attributeType = .integer32AttributeType
+
+        let prodInStock = NSAttributeDescription()
+        prodInStock.name = "inStock"
+        prodInStock.attributeType = .booleanAttributeType
+
+        let prodStockCount = NSAttributeDescription()
+        prodStockCount.name = "stockCount"
+        prodStockCount.attributeType = .integer32AttributeType
+        prodStockCount.isOptional = true
+
+        let prodVariantsJSON = NSAttributeDescription()
+        prodVariantsJSON.name = "variantsJSON"
+        prodVariantsJSON.attributeType = .stringAttributeType
+        prodVariantsJSON.isOptional = true
+
+        let prodSpecsJSON = NSAttributeDescription()
+        prodSpecsJSON.name = "specificationsJSON"
+        prodSpecsJSON.attributeType = .stringAttributeType
+        prodSpecsJSON.isOptional = true
+
+        let prodTagsJSON = NSAttributeDescription()
+        prodTagsJSON.name = "tagsJSON"
+        prodTagsJSON.attributeType = .stringAttributeType
+        prodTagsJSON.isOptional = true
+
+        let prodCreatedAt = NSAttributeDescription()
+        prodCreatedAt.name = "createdAt"
+        prodCreatedAt.attributeType = .dateAttributeType
+        prodCreatedAt.isOptional = true
+
+        productEntity.properties = [
+            prodId, prodName, prodDesc, prodPrice, prodOriginalPrice, prodImagesJSON,
+            prodCategoryId, prodBrand, prodRating, prodReviewCount, prodInStock,
+            prodStockCount, prodVariantsJSON, prodSpecsJSON, prodTagsJSON, prodCreatedAt
+        ]
+
+        // — OrderEntity —
+        let orderEntity = NSEntityDescription()
+        orderEntity.name = "OrderEntity"
+        orderEntity.managedObjectClassName = NSStringFromClass(OrderEntity.self)
+
+        let orderId = NSAttributeDescription()
+        orderId.name = "id"
+        orderId.attributeType = .stringAttributeType
+
+        let orderNumber = NSAttributeDescription()
+        orderNumber.name = "orderNumber"
+        orderNumber.attributeType = .stringAttributeType
+
+        let orderItemsJSON = NSAttributeDescription()
+        orderItemsJSON.name = "itemsJSON"
+        orderItemsJSON.attributeType = .stringAttributeType
+
+        let orderStatus = NSAttributeDescription()
+        orderStatus.name = "status"
+        orderStatus.attributeType = .stringAttributeType
+
+        let orderShippingAddressJSON = NSAttributeDescription()
+        orderShippingAddressJSON.name = "shippingAddressJSON"
+        orderShippingAddressJSON.attributeType = .stringAttributeType
+
+        let orderBillingAddressJSON = NSAttributeDescription()
+        orderBillingAddressJSON.name = "billingAddressJSON"
+        orderBillingAddressJSON.attributeType = .stringAttributeType
+        orderBillingAddressJSON.isOptional = true
+
+        let orderPaymentMethodJSON = NSAttributeDescription()
+        orderPaymentMethodJSON.name = "paymentMethodJSON"
+        orderPaymentMethodJSON.attributeType = .stringAttributeType
+
+        let orderSubtotal = NSAttributeDescription()
+        orderSubtotal.name = "subtotal"
+        orderSubtotal.attributeType = .doubleAttributeType
+
+        let orderDiscount = NSAttributeDescription()
+        orderDiscount.name = "discount"
+        orderDiscount.attributeType = .doubleAttributeType
+
+        let orderShipping = NSAttributeDescription()
+        orderShipping.name = "shipping"
+        orderShipping.attributeType = .doubleAttributeType
+
+        let orderTax = NSAttributeDescription()
+        orderTax.name = "tax"
+        orderTax.attributeType = .doubleAttributeType
+
+        let orderTotal = NSAttributeDescription()
+        orderTotal.name = "total"
+        orderTotal.attributeType = .doubleAttributeType
+
+        let orderDiscountCode = NSAttributeDescription()
+        orderDiscountCode.name = "discountCode"
+        orderDiscountCode.attributeType = .stringAttributeType
+        orderDiscountCode.isOptional = true
+
+        let orderTrackingNumber = NSAttributeDescription()
+        orderTrackingNumber.name = "trackingNumber"
+        orderTrackingNumber.attributeType = .stringAttributeType
+        orderTrackingNumber.isOptional = true
+
+        let orderTrackingURL = NSAttributeDescription()
+        orderTrackingURL.name = "trackingURL"
+        orderTrackingURL.attributeType = .stringAttributeType
+        orderTrackingURL.isOptional = true
+
+        let orderEstimatedDelivery = NSAttributeDescription()
+        orderEstimatedDelivery.name = "estimatedDelivery"
+        orderEstimatedDelivery.attributeType = .dateAttributeType
+        orderEstimatedDelivery.isOptional = true
+
+        let orderCreatedAt = NSAttributeDescription()
+        orderCreatedAt.name = "createdAt"
+        orderCreatedAt.attributeType = .dateAttributeType
+
+        let orderUpdatedAt = NSAttributeDescription()
+        orderUpdatedAt.name = "updatedAt"
+        orderUpdatedAt.attributeType = .dateAttributeType
+
+        let orderTimelineJSON = NSAttributeDescription()
+        orderTimelineJSON.name = "timelineJSON"
+        orderTimelineJSON.attributeType = .stringAttributeType
+        orderTimelineJSON.isOptional = true
+
+        orderEntity.properties = [
+            orderId, orderNumber, orderItemsJSON, orderStatus, orderShippingAddressJSON,
+            orderBillingAddressJSON, orderPaymentMethodJSON, orderSubtotal, orderDiscount,
+            orderShipping, orderTax, orderTotal, orderDiscountCode, orderTrackingNumber,
+            orderTrackingURL, orderEstimatedDelivery, orderCreatedAt, orderUpdatedAt, orderTimelineJSON
+        ]
+
+        // — ReviewEntity —
+        let reviewEntity = NSEntityDescription()
+        reviewEntity.name = "ReviewEntity"
+        reviewEntity.managedObjectClassName = NSStringFromClass(ReviewEntity.self)
+
+        let revId = NSAttributeDescription()
+        revId.name = "id"
+        revId.attributeType = .stringAttributeType
+
+        let revProductId = NSAttributeDescription()
+        revProductId.name = "productId"
+        revProductId.attributeType = .stringAttributeType
+
+        let revUserId = NSAttributeDescription()
+        revUserId.name = "userId"
+        revUserId.attributeType = .stringAttributeType
+
+        let revUserName = NSAttributeDescription()
+        revUserName.name = "userName"
+        revUserName.attributeType = .stringAttributeType
+
+        let revUserAvatar = NSAttributeDescription()
+        revUserAvatar.name = "userAvatar"
+        revUserAvatar.attributeType = .stringAttributeType
+        revUserAvatar.isOptional = true
+
+        let revRating = NSAttributeDescription()
+        revRating.name = "rating"
+        revRating.attributeType = .integer32AttributeType
+
+        let revTitle = NSAttributeDescription()
+        revTitle.name = "title"
+        revTitle.attributeType = .stringAttributeType
+        revTitle.isOptional = true
+
+        let revComment = NSAttributeDescription()
+        revComment.name = "comment"
+        revComment.attributeType = .stringAttributeType
+
+        let revImagesJSON = NSAttributeDescription()
+        revImagesJSON.name = "imagesJSON"
+        revImagesJSON.attributeType = .stringAttributeType
+        revImagesJSON.isOptional = true
+
+        let revVerifiedPurchase = NSAttributeDescription()
+        revVerifiedPurchase.name = "isVerifiedPurchase"
+        revVerifiedPurchase.attributeType = .booleanAttributeType
+
+        let revHelpfulCount = NSAttributeDescription()
+        revHelpfulCount.name = "helpfulCount"
+        revHelpfulCount.attributeType = .integer32AttributeType
+
+        let revCreatedAt = NSAttributeDescription()
+        revCreatedAt.name = "createdAt"
+        revCreatedAt.attributeType = .dateAttributeType
+
+        reviewEntity.properties = [
+            revId, revProductId, revUserId, revUserName, revUserAvatar, revRating,
+            revTitle, revComment, revImagesJSON, revVerifiedPurchase, revHelpfulCount, revCreatedAt
+        ]
+
+        // — BannerEntity —
+        let bannerEntity = NSEntityDescription()
+        bannerEntity.name = "BannerEntity"
+        bannerEntity.managedObjectClassName = NSStringFromClass(BannerEntity.self)
+
+        let banId = NSAttributeDescription()
+        banId.name = "id"
+        banId.attributeType = .stringAttributeType
+
+        let banTitle = NSAttributeDescription()
+        banTitle.name = "title"
+        banTitle.attributeType = .stringAttributeType
+
+        let banSubtitle = NSAttributeDescription()
+        banSubtitle.name = "subtitle"
+        banSubtitle.attributeType = .stringAttributeType
+        banSubtitle.isOptional = true
+
+        let banImageURL = NSAttributeDescription()
+        banImageURL.name = "imageURL"
+        banImageURL.attributeType = .stringAttributeType
+
+        let banActionType = NSAttributeDescription()
+        banActionType.name = "actionType"
+        banActionType.attributeType = .stringAttributeType
+
+        let banActionValue = NSAttributeDescription()
+        banActionValue.name = "actionValue"
+        banActionValue.attributeType = .stringAttributeType
+
+        let banBackgroundColor = NSAttributeDescription()
+        banBackgroundColor.name = "backgroundColor"
+        banBackgroundColor.attributeType = .stringAttributeType
+        banBackgroundColor.isOptional = true
+
+        bannerEntity.properties = [
+            banId, banTitle, banSubtitle, banImageURL, banActionType, banActionValue, banBackgroundColor
+        ]
+
+        model.entities = [categoryEntity, productEntity, orderEntity, reviewEntity, bannerEntity]
+        return model
+    }
+}

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

@@ -0,0 +1,304 @@
+import CoreData
+import Foundation
+
+/// Core Data–backed implementation of DataRepository.
+/// Stores categories, products, orders, reviews, and banners in a local SQLite database.
+/// Pre-seeds default data from MockDataProvider on first launch.
+///
+/// Usage: LocalStorageProvider.shared (singleton, like MockDataProvider)
+final class LocalStorageProvider: ObservableObject, DataRepository {
+    static let shared = LocalStorageProvider()
+
+    @Published var categories: [Category] = []
+    @Published var products: [Product] = []
+    @Published var banners: [Banner] = []
+    @Published var orders: [Order] = []
+    @Published var userReviews: [Review] = []
+    @Published var favorites: Set<String> = []
+
+    private let stack: CoreDataStack
+    private var context: NSManagedObjectContext { stack.viewContext }
+
+    private init(stack: CoreDataStack = .shared) {
+        self.stack = stack
+        loadData()
+    }
+
+    // MARK: - Data Loading
+
+    func loadData() {
+        fetchCategories()
+        fetchProducts()
+        fetchOrders()
+        fetchReviews()
+        fetchBanners()
+        loadFavorites()
+
+        // Pre-seed from MockDataProvider on first launch (empty DB)
+        if categories.isEmpty && products.isEmpty {
+            seedDefaultData()
+            fetchCategories()
+            fetchProducts()
+            fetchOrders()
+            fetchReviews()
+            fetchBanners()
+        }
+    }
+
+    // MARK: - CRUD Operations - Orders
+
+    func addOrder(_ order: Order) {
+        let entity = OrderEntity(context: context)
+        entity.update(from: order)
+        stack.save()
+        fetchOrders()
+    }
+
+    func updateOrder(_ order: Order) {
+        guard let entity = fetchOrderEntity(byId: order.id) else { return }
+        entity.update(from: order)
+        stack.save()
+        fetchOrders()
+    }
+
+    // MARK: - CRUD Operations - Reviews
+
+    func addReview(_ review: Review) {
+        let entity = ReviewEntity(context: context)
+        entity.update(from: review)
+        stack.save()
+        fetchReviews()
+    }
+
+    // MARK: - CRUD Operations - Favorites
+
+    func toggleFavorite(_ productId: String) {
+        if favorites.contains(productId) {
+            favorites.remove(productId)
+        } else {
+            favorites.insert(productId)
+        }
+        saveFavorites()
+    }
+
+    func isFavorite(_ productId: String) -> Bool {
+        favorites.contains(productId)
+    }
+
+    // MARK: - CRUD Operations - Products
+
+    func addProduct(_ product: Product) {
+        let entity = ProductEntity(context: context)
+        entity.update(from: product)
+        stack.save()
+        fetchProducts()
+        updateCategoryCounts()
+    }
+
+    func updateProduct(_ product: Product) {
+        guard let entity = fetchProductEntity(byId: product.id) else { return }
+        entity.update(from: product)
+        stack.save()
+        fetchProducts()
+    }
+
+    func deleteProduct(_ id: String) {
+        guard let entity = fetchProductEntity(byId: id) else { return }
+        context.delete(entity)
+        stack.save()
+        fetchProducts()
+        updateCategoryCounts()
+    }
+
+    // MARK: - Query Methods
+
+    func product(by id: String) -> Product? {
+        products.first { $0.id == id }
+    }
+
+    func featuredProducts() -> [Product] {
+        Array(products.filter { $0.rating >= 4.5 }.prefix(10))
+    }
+
+    func saleProducts() -> [Product] {
+        products.filter { $0.originalPrice != nil }
+    }
+
+    func reviews(for productId: String) -> [Review] {
+        userReviews.filter { $0.productId == productId }
+    }
+
+    // MARK: - Reset
+
+    func reset() {
+        // Delete all entities
+        fetchAllCategories().forEach { context.delete($0) }
+        fetchAllProducts().forEach { context.delete($0) }
+        fetchAllOrders().forEach { context.delete($0) }
+        fetchAllReviews().forEach { context.delete($0) }
+        fetchAllBanners().forEach { context.delete($0) }
+
+        favorites = []
+        saveFavorites()
+
+        seedDefaultData()
+        fetchCategories()
+        fetchProducts()
+        fetchOrders()
+        fetchReviews()
+        fetchBanners()
+    }
+
+    // MARK: - Private Fetch Helpers
+
+    private func fetchCategories() {
+        let request = NSFetchRequest<CategoryEntity>(entityName: "CategoryEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "name", ascending: true)]
+        do {
+            let entities = try context.fetch(request)
+            categories = entities.map { $0.toCategory() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch categories: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchProducts() {
+        let request = NSFetchRequest<ProductEntity>(entityName: "ProductEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "name", ascending: true)]
+        do {
+            let entities = try context.fetch(request)
+            let categoriesMap = Dictionary(uniqueKeysWithValues: categories.map { ($0.id, $0) })
+            products = entities.compactMap { entity in
+                categoriesMap[entity.categoryId].map { entity.toProduct(category: $0) }
+            }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch products: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchOrders() {
+        let request = NSFetchRequest<OrderEntity>(entityName: "OrderEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "createdAt", ascending: false)]
+        do {
+            let entities = try context.fetch(request)
+            orders = entities.compactMap { $0.toOrder() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch orders: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchReviews() {
+        let request = NSFetchRequest<ReviewEntity>(entityName: "ReviewEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "createdAt", ascending: false)]
+        do {
+            let entities = try context.fetch(request)
+            userReviews = entities.map { $0.toReview() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch reviews: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchBanners() {
+        let request = NSFetchRequest<BannerEntity>(entityName: "BannerEntity")
+        request.sortDescriptors = [NSSortDescriptor(key: "title", ascending: true)]
+        do {
+            let entities = try context.fetch(request)
+            banners = entities.map { $0.toBanner() }
+        } catch {
+            print("[LocalStorageProvider] Failed to fetch banners: \(error.localizedDescription)")
+        }
+    }
+
+    private func fetchAllCategories() -> [CategoryEntity] {
+        let request = NSFetchRequest<CategoryEntity>(entityName: "CategoryEntity")
+        return (try? context.fetch(request)) ?? []
+    }
+
+    private func fetchAllProducts() -> [ProductEntity] {
+        let request = NSFetchRequest<ProductEntity>(entityName: "ProductEntity")
+        return (try? context.fetch(request)) ?? []
+    }
+
+    private func fetchAllOrders() -> [OrderEntity] {
+        let request = NSFetchRequest<OrderEntity>(entityName: "OrderEntity")
+        return (try? context.fetch(request)) ?? []
+    }
+
+    private func fetchAllReviews() -> [ReviewEntity] {
+        let request = NSFetchRequest<ReviewEntity>(entityName: "ReviewEntity")
+        return (try? context.fetch(request)) ?? []
+    }
+
+    private func fetchAllBanners() -> [BannerEntity] {
+        let request = NSFetchRequest<BannerEntity>(entityName: "BannerEntity")
+        return (try? context.fetch(request)) ?? []
+    }
+
+    private func fetchProductEntity(byId id: String) -> ProductEntity? {
+        let request = NSFetchRequest<ProductEntity>(entityName: "ProductEntity")
+        request.predicate = NSPredicate(format: "id == %@", id)
+        request.fetchLimit = 1
+        return try? context.fetch(request).first
+    }
+
+    private func fetchOrderEntity(byId id: String) -> OrderEntity? {
+        let request = NSFetchRequest<OrderEntity>(entityName: "OrderEntity")
+        request.predicate = NSPredicate(format: "id == %@", id)
+        request.fetchLimit = 1
+        return try? context.fetch(request).first
+    }
+
+    // MARK: - Seeding
+
+    /// Seeds default data from MockDataProvider on first launch.
+    private func seedDefaultData() {
+        let mockProvider = MockDataProvider.shared
+
+        // Seed categories
+        for category in mockProvider.categories {
+            let entity = CategoryEntity(context: context)
+            entity.update(from: category)
+        }
+
+        // Seed products
+        for product in mockProvider.products {
+            let entity = ProductEntity(context: context)
+            entity.update(from: product)
+        }
+
+        // Seed banners
+        for banner in mockProvider.banners {
+            let entity = BannerEntity(context: context)
+            entity.update(from: banner)
+        }
+
+        stack.save()
+    }
+
+    // MARK: - Favorites Persistence
+
+    private func loadFavorites() {
+        if let saved = LocalPersistence.load(Set<String>.self, filename: "favorites") {
+            favorites = saved
+        }
+    }
+
+    private func saveFavorites() {
+        LocalPersistence.save(favorites, filename: "favorites")
+    }
+
+    // MARK: - Category Counts
+
+    private func updateCategoryCounts() {
+        let request = NSFetchRequest<CategoryEntity>(entityName: "CategoryEntity")
+        guard let catEntities = try? context.fetch(request) else { return }
+        for catEntity in catEntities {
+            let countRequest = NSFetchRequest<ProductEntity>(entityName: "ProductEntity")
+            countRequest.predicate = NSPredicate(format: "categoryId == %@", catEntity.id)
+            let count = (try? context.count(for: countRequest)) ?? 0
+            catEntity.productCount = Int32(count)
+        }
+        stack.save()
+        fetchCategories()
+    }
+}