Add attribute and image associations

Author: joshheald

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

@@ -42,6 +42,19 @@ extension PersistedProduct {
             stockStatusKey: stockStatusKey
         )
     }
+    
+    public 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: - PersistedProductAttribute Conversions

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

@@ -36,6 +36,19 @@ extension PersistedProductVariation {
             stockStatusKey: stockStatusKey
         )
     }
+
+    public 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: - PersistedProductVariationAttribute Conversions