[Woo POS][Local catalog] Add relationship mapping and saving for POS models

Modules/Sources/Storage/GRDB/Model/PersistedProduct.swift

@@ -67,6 +67,9 @@ extension PersistedProduct: FetchableRecord, PersistableRecord {
         static let stockQuantity = Column(CodingKeys.stockQuantity)
         static let stockStatusKey = Column(CodingKeys.stockStatusKey)
     }
+
+    public static let images = hasMany(PersistedProductImage.self)
+    public static let attributes = hasMany(PersistedProductAttribute.self)
 }
 
 private extension PersistedProduct {

Modules/Sources/Storage/GRDB/Model/PersistedProductVariation.swift

@@ -55,6 +55,9 @@ extension PersistedProductVariation: FetchableRecord, PersistableRecord {
         static let stockQuantity = Column(CodingKeys.stockQuantity)
         static let stockStatusKey = Column(CodingKeys.stockStatusKey)
     }
+
+    public static let attributes = hasMany(PersistedProductVariationAttribute.self)
+    public static let image = hasOne(PersistedProductVariationImage.self)
 }
 
 

Modules/Sources/Yosemite/Model/Storage/PersistedProduct+Conversions.swift

@@ -3,7 +3,7 @@ import Storage
 
 // MARK: - PersistedProduct Conversions
 extension PersistedProduct {
-    public init(from posProduct: POSProduct) {
+    init(from posProduct: POSProduct) {
         self.init(
             id: posProduct.productID,
             siteID: posProduct.siteID,
@@ -22,7 +22,7 @@ extension PersistedProduct {
         )
     }
 
-    public func toPOSProduct(images: [ProductImage] = [], attributes: [ProductAttribute] = []) -> POSProduct {
+    func toPOSProduct(images: [ProductImage] = [], attributes: [ProductAttribute] = []) -> POSProduct {
         return POSProduct(
             siteID: siteID,
             productID: id,
@@ -42,11 +42,47 @@ extension PersistedProduct {
             stockStatusKey: stockStatusKey
         )
     }
+
+    func toPOSProduct(db: GRDBDatabaseConnection) throws -> POSProduct {
+        let images = try db.read { db in
+            return try request(for: PersistedProduct.images).fetchAll(db)
+        }
+        let attributes = try db.read { db in
+            return try request(for: PersistedProduct.attributes).fetchAll(db)
+        }
+        return toPOSProduct(
+            images: images.map { $0.toProductImage() },
+            attributes: attributes.map { $0.toProductAttribute(siteID: siteID) }
+        )
+    }
+
+}
+
+// MARK: - POSProduct Storage Extensions
+extension POSProduct {
+    public func save(to db: GRDBDatabaseConnection) throws {
+        try db.write { db in
+            let product = PersistedProduct(from: self)
+            try product.insert(db)
+
+            // Save related images
+            for image in self.images {
+                let persistedImage = PersistedProductImage(from: image, productID: self.productID)
+                try persistedImage.insert(db)
+            }
+
+            // Save related attributes
+            for attribute in self.attributes {
+                var persistedAttribute = PersistedProductAttribute(from: attribute, productID: self.productID)
+                try persistedAttribute.insert(db)
+            }
+        }
+    }
 }
 
 // MARK: - PersistedProductAttribute Conversions
 extension PersistedProductAttribute {
-    public init(from productAttribute: ProductAttribute, productID: Int64) {
+    init(from productAttribute: ProductAttribute, productID: Int64) {
         self.init(
             productID: productID,
             name: productAttribute.name,
@@ -57,7 +93,7 @@ extension PersistedProductAttribute {
         )
     }
 
-    public func toProductAttribute(siteID: Int64) -> ProductAttribute {
+    func toProductAttribute(siteID: Int64) -> ProductAttribute {
         return ProductAttribute(
             siteID: siteID,
             attributeID: 0,
@@ -72,7 +108,7 @@ extension PersistedProductAttribute {
 
 // MARK: - PersistedProductImage Conversions
 extension PersistedProductImage {
-    public init(from productImage: ProductImage, productID: Int64) {
+    init(from productImage: ProductImage, productID: Int64) {
         self.init(
             id: productImage.imageID,
             productID: productID,
@@ -84,7 +120,7 @@ extension PersistedProductImage {
         )
     }
 
-    public func toProductImage() -> ProductImage {
+    func toProductImage() -> ProductImage {
         return ProductImage(
             imageID: id,
             dateCreated: dateCreated,

Modules/Sources/Yosemite/Model/Storage/PersistedProductVariation+Conversions.swift

@@ -3,7 +3,7 @@ import Storage
 
 // MARK: - PersistedProductVariation Conversions
 extension PersistedProductVariation {
-    public init(from posProductVariation: POSProductVariation) {
+    init(from posProductVariation: POSProductVariation) {
         self.init(
             id: posProductVariation.productVariationID,
             siteID: posProductVariation.siteID,
@@ -19,7 +19,7 @@ extension PersistedProductVariation {
         )
     }
 
-    public func toPOSProductVariation(attributes: [ProductVariationAttribute] = [], image: ProductImage? = nil) -> POSProductVariation {
+    func toPOSProductVariation(attributes: [ProductVariationAttribute] = [], image: ProductImage? = nil) -> POSProductVariation {
         return POSProductVariation(
             siteID: siteID,
             productID: productID,
@@ -36,19 +36,55 @@ extension PersistedProductVariation {
             stockStatusKey: stockStatusKey
         )
     }
+
+    func toPOSProductVariation(db: GRDBDatabaseConnection) throws -> POSProductVariation {
+        let image = try db.read { db in
+            return try request(for: PersistedProductVariation.image).fetchOne(db)
+        }
+        let attributes = try db.read { db in
+            return try request(for: PersistedProductVariation.attributes).fetchAll(db)
+        }
+        return toPOSProductVariation(
+            attributes: attributes.map { $0.toProductVariationAttribute() },
+            image: image?.toProductImage()
+        )
+    }
+
+}
+
+// MARK: - POSProductVariation Storage Extensions
+extension POSProductVariation {
+    public func save(to db: GRDBDatabaseConnection) throws {
+        try db.write { db in
+            let variation = PersistedProductVariation(from: self)
+            try variation.insert(db)
+
+            // Save related image if present
+            if let image = self.image {
+                let persistedImage = PersistedProductVariationImage(from: image, productVariationID: self.productVariationID)
+                try persistedImage.insert(db)
+            }
+
+            // Save related attributes
+            for attribute in self.attributes {
+                var persistedAttribute = PersistedProductVariationAttribute(from: attribute, productVariationID: self.productVariationID)
+                try persistedAttribute.insert(db)
+            }
+        }
+    }
 }
 
 // MARK: - PersistedProductVariationAttribute Conversions
 extension PersistedProductVariationAttribute {
-    public init(from productVariationAttribute: ProductVariationAttribute, productVariationID: Int64) {
+    init(from productVariationAttribute: ProductVariationAttribute, productVariationID: Int64) {
         self.init(
             productVariationID: productVariationID,
             name: productVariationAttribute.name,
             option: productVariationAttribute.option
         )
     }
 
-    public func toProductVariationAttribute() -> ProductVariationAttribute {
+    func toProductVariationAttribute() -> ProductVariationAttribute {
         return ProductVariationAttribute(
             id: id ?? 0,
             name: name,