chore: kickoff release

harsh62 · web-flow · commit 91dd0b90ea23 · 2024-07-16T10:00:15.000-04:00
diff --git a/Amplify/Categories/DataStore/Model/Internal/Schema/ModelField+Association.swift b/Amplify/Categories/DataStore/Model/Internal/Schema/ModelField+Association.swift
@@ -88,7 +88,7 @@ import Foundation
 ///   directly by host applications. The behavior of this may change without warning.
 public enum ModelAssociation {
     case hasMany(associatedFieldName: String?, associatedFieldNames: [String] = [])
-    case hasOne(associatedFieldName: String?, targetNames: [String])
+    case hasOne(associatedFieldName: String?, associatedFieldNames: [String] = [], targetNames: [String])
     case belongsTo(associatedFieldName: String?, targetNames: [String])
 
     public static let belongsTo: ModelAssociation = .belongsTo(associatedFieldName: nil, targetNames: [])
@@ -108,14 +108,20 @@ public enum ModelAssociation {
         )
     }
 
-    @available(*, deprecated, message: "Use hasOne(associatedWith:targetNames:)")
+    @available(*, deprecated, message: "Use hasOne(associatedWith:associatedFields:targetNames:)")
     public static func hasOne(associatedWith: CodingKey?, targetName: String? = nil) -> ModelAssociation {
         let targetNames = targetName.map { [$0] } ?? []
         return .hasOne(associatedWith: associatedWith, targetNames: targetNames)
     }
 
-    public static func hasOne(associatedWith: CodingKey?, targetNames: [String] = []) -> ModelAssociation {
-        return .hasOne(associatedFieldName: associatedWith?.stringValue, targetNames: targetNames)
+    public static func hasOne(
+        associatedWith: CodingKey? = nil,
+        associatedFields: [CodingKey] = [],
+        targetNames: [String] = []) -> ModelAssociation {
+            return .hasOne(
+                associatedFieldName: associatedWith?.stringValue,
+                associatedFieldNames: associatedFields.map { $0.stringValue },
+                targetNames: targetNames)
     }
 
     @available(*, deprecated, message: "Use belongsTo(associatedWith:targetNames:)")
@@ -254,7 +260,7 @@ extension ModelField {
             let associatedModel = requiredAssociatedModelName
             switch association {
             case .belongsTo(let associatedKey, _),
-                    .hasOne(let associatedKey, _),
+                    .hasOne(let associatedKey, _, _),
                     .hasMany(let associatedKey, _):
                 // TODO handle modelName casing (convert to camelCase)
                 let key = associatedKey ?? associatedModel
diff --git a/Amplify/Categories/DataStore/Model/Internal/Schema/ModelSchema+Definition.swift b/Amplify/Categories/DataStore/Model/Internal/Schema/ModelSchema+Definition.swift
@@ -279,6 +279,21 @@ public enum ModelFieldDefinition {
                       association: .hasOne(associatedWith: associatedKey, targetNames: targetNames))
     }
 
+    public static func hasOne(_ key: CodingKey,
+                              is nullability: ModelFieldNullability = .required,
+                              isReadOnly: Bool = false,
+                              ofType type: Model.Type,
+                              associatedFields associatedKeys: [CodingKey],
+                              targetNames: [String] = []) -> ModelFieldDefinition {
+        return .field(key,
+                      is: nullability,
+                      isReadOnly: isReadOnly,
+                      ofType: .model(type: type),
+                      association: .hasOne(associatedWith: associatedKeys.first,
+                                           associatedFields: associatedKeys,
+                                           targetNames: targetNames))
+    }
+
     public static func belongsTo(_ key: CodingKey,
                                  is nullability: ModelFieldNullability = .required,
                                  isReadOnly: Bool = false,
diff --git a/AmplifyPlugins/API/Sources/AWSAPIPlugin/Core/AppSyncListProvider.swift b/AmplifyPlugins/API/Sources/AWSAPIPlugin/Core/AppSyncListProvider.swift
@@ -268,7 +268,7 @@ public class AppSyncListProvider<Element: Model>: ModelListProvider {
         }
         let defaultFieldName = modelSchema.name.camelCased() + field.pascalCased() + "Id"
         switch modelField.association {
-        case .belongsTo(_, let targetNames), .hasOne(_, let targetNames):
+        case .belongsTo(_, let targetNames), .hasOne(_, _, let targetNames):
             guard !targetNames.isEmpty else {
                 return [defaultFieldName]
 
diff --git a/AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginGen2GraphQLTests/Gen2_4/Customer4+Schema.swift b/AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginGen2GraphQLTests/Gen2_4/Customer4+Schema.swift
@@ -32,7 +32,7 @@ extension Customer4 {
     model.fields(
       .field(customer4.id, is: .required, ofType: .string),
       .field(customer4.name, is: .optional, ofType: .string),
-      .hasOne(customer4.activeCart, is: .optional, ofType: Cart4.self, associatedWith: Cart4.keys.customer),
+      .hasOne(customer4.activeCart, is: .optional, ofType: Cart4.self, associatedFields: [Cart4.keys.customer]),
       .field(customer4.createdAt, is: .optional, isReadOnly: true, ofType: .dateTime),
       .field(customer4.updatedAt, is: .optional, isReadOnly: true, ofType: .dateTime)
     )
diff --git a/AmplifyPlugins/Core/AWSPluginsCore/Model/Support/Model+GraphQL.swift b/AmplifyPlugins/Core/AWSPluginsCore/Model/Support/Model+GraphQL.swift
@@ -229,7 +229,7 @@ extension Model {
         let defaultFieldName = modelName.camelCased() + modelField.name.pascalCased() + "Id"
         if case let .belongsTo(_, targetNames) = modelField.association, !targetNames.isEmpty {
             return targetNames
-        } else if case let .hasOne(_, targetNames) = modelField.association,
+        } else if case let .hasOne(_, _, targetNames) = modelField.association,
                   !targetNames.isEmpty {
             return targetNames
         }
diff --git a/AmplifyPlugins/Core/AWSPluginsCore/Model/Support/QueryPredicate+GraphQL.swift b/AmplifyPlugins/Core/AWSPluginsCore/Model/Support/QueryPredicate+GraphQL.swift
@@ -131,7 +131,7 @@ extension QueryPredicateOperation: GraphQLFilterConvertible {
             }
             let targetName = targetNames.first ?? defaultFieldName
             return targetName
-        case .hasOne(_, let targetNames):
+        case .hasOne(_, _, let targetNames):
             guard targetNames.count == 1 else {
                 preconditionFailure("QueryPredicate not supported on associated field with composite key: \(field)")
             }
diff --git a/AmplifyPlugins/DataStore/Sources/AWSDataStorePlugin/Storage/SQLite/ModelSchema+SQLite.swift b/AmplifyPlugins/DataStore/Sources/AWSDataStorePlugin/Storage/SQLite/ModelSchema+SQLite.swift
@@ -82,7 +82,7 @@ extension ModelField: SQLColumn {
     var sqlName: String {
         if case let .belongsTo(_, targetNames) = association {
             return foreignKeySqlName(withAssociationTargets: targetNames)
-        } else if case let .hasOne(_, targetNames) = association {
+        } else if case let .hasOne(_, _, targetNames) = association {
             return foreignKeySqlName(withAssociationTargets: targetNames)
         }
         return name
diff --git a/AmplifyPlugins/DataStore/Sources/AWSDataStorePlugin/Storage/SQLite/Statement+Model.swift b/AmplifyPlugins/DataStore/Sources/AWSDataStorePlugin/Storage/SQLite/Statement+Model.swift
@@ -203,7 +203,7 @@ extension Statement: StatementModelConvertible {
 
     private func getTargetNames(field: ModelField) -> [String] {
         switch field.association {
-        case let .some(.hasOne(_, targetNames)):
+        case let .some(.hasOne(_, _, targetNames)):
             return targetNames
         case let .some(.belongsTo(_, targetNames)):
             return targetNames
diff --git a/AmplifyTests/CategoryTests/DataStore/ModelFieldAssociationTests.swift b/AmplifyTests/CategoryTests/DataStore/ModelFieldAssociationTests.swift
@@ -37,24 +37,37 @@ class ModelFieldAssociationTests: XCTestCase {
 
     func testHasOneWithCodingKeys() {
         let hasOne = ModelAssociation.hasOne(associatedWith: Comment.keys.post, targetNames: [])
-        guard case .hasOne(let fieldName, let target) = hasOne else {
+        guard case .hasOne(let fieldName, let fieldNames, let target) = hasOne else {
             XCTFail("Should create hasOne association")
             return
         }
         XCTAssertEqual(fieldName, Comment.keys.post.stringValue)
+        XCTAssertEqual(fieldNames, [])
         XCTAssertEqual(target, [])
     }
 
     func testHasOneWithCodingKeysWithTargetName() {
         let hasOne = ModelAssociation.hasOne(associatedWith: Comment.keys.post, targetNames: ["postID"])
-        guard case .hasOne(let fieldName, let target) = hasOne else {
+        guard case .hasOne(let fieldName, let fieldNames, let target) = hasOne else {
             XCTFail("Should create hasOne association")
             return
         }
         XCTAssertEqual(fieldName, Comment.keys.post.stringValue)
+        XCTAssertEqual(fieldNames, [])
         XCTAssertEqual(target, ["postID"])
     }
 
+    func testHasOneWithCodingKeysWithAssociatedFields() {
+        let hasOne = ModelAssociation.hasOne(associatedFields: [Comment.keys.post])
+        guard case .hasOne(let fieldName, let fieldNames, let target) = hasOne else {
+            XCTFail("Should create hasOne association")
+            return
+        }
+        XCTAssertEqual(fieldName, nil)
+        XCTAssertEqual(fieldNames, ["post"])
+        XCTAssertEqual(target, [])
+    }
+
     func testBelongsToWithTargetName() {
         let belongsTo = ModelAssociation.belongsTo(targetName: "postID")
         guard case .belongsTo(let fieldName, let target) = belongsTo else {

Original file line number	Diff line number	Diff line change
`@@ -268,7 +268,7 @@ public class AppSyncListProvider<Element: Model>: ModelListProvider {`
`268`	`268`	`}`
`269`	`269`	`let defaultFieldName = modelSchema.name.camelCased() + field.pascalCased() + "Id"`
`270`	`270`	`switch modelField.association {`
`271`		`- case .belongsTo(_, let targetNames), .hasOne(_, let targetNames):`
	`271`	`+ case .belongsTo(_, let targetNames), .hasOne(_, _, let targetNames):`
`272`	`272`	`guard !targetNames.isEmpty else {`
`273`	`273`	`return [defaultFieldName]`
`274`	`274`
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ extension Customer4 {`
`32`	`32`	`model.fields(`
`33`	`33`	`.field(customer4.id, is: .required, ofType: .string),`
`34`	`34`	`.field(customer4.name, is: .optional, ofType: .string),`
`35`		`- .hasOne(customer4.activeCart, is: .optional, ofType: Cart4.self, associatedWith: Cart4.keys.customer),`
	`35`	`+ .hasOne(customer4.activeCart, is: .optional, ofType: Cart4.self, associatedFields: [Cart4.keys.customer]),`
`36`	`36`	`.field(customer4.createdAt, is: .optional, isReadOnly: true, ofType: .dateTime),`
`37`	`37`	`.field(customer4.updatedAt, is: .optional, isReadOnly: true, ofType: .dateTime)`
`38`	`38`	`)`
Original file line number	Diff line number	Diff line change
`@@ -229,7 +229,7 @@ extension Model {`
`229`	`229`	`let defaultFieldName = modelName.camelCased() + modelField.name.pascalCased() + "Id"`
`230`	`230`	`if case let .belongsTo(_, targetNames) = modelField.association, !targetNames.isEmpty {`
`231`	`231`	`return targetNames`
`232`		`- } else if case let .hasOne(_, targetNames) = modelField.association,`
	`232`	`+ } else if case let .hasOne(_, _, targetNames) = modelField.association,`
`233`	`233`	`!targetNames.isEmpty {`
`234`	`234`	`return targetNames`
`235`	`235`	`}`
Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ extension QueryPredicateOperation: GraphQLFilterConvertible {`
`131`	`131`	`}`
`132`	`132`	`let targetName = targetNames.first ?? defaultFieldName`
`133`	`133`	`return targetName`
`134`		`- case .hasOne(_, let targetNames):`
	`134`	`+ case .hasOne(_, _, let targetNames):`
`135`	`135`	`guard targetNames.count == 1 else {`
`136`	`136`	`preconditionFailure("QueryPredicate not supported on associated field with composite key: \(field)")`
`137`	`137`	`}`
Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,7 @@ extension ModelField: SQLColumn {`
`82`	`82`	`var sqlName: String {`
`83`	`83`	`if case let .belongsTo(_, targetNames) = association {`
`84`	`84`	`return foreignKeySqlName(withAssociationTargets: targetNames)`
`85`		`- } else if case let .hasOne(_, targetNames) = association {`
	`85`	`+ } else if case let .hasOne(_, _, targetNames) = association {`
`86`	`86`	`return foreignKeySqlName(withAssociationTargets: targetNames)`
`87`	`87`	`}`
`88`	`88`	`return name`