Fix $relationship.create() and $relationship.save() to accept a s…

…ingle model and an array of models. (#27)

.changeset/perfect-keys-allow.md

@@ -0,0 +1,5 @@
+---
+"superflare": patch
+---
+
+- Fix `$relationship.create()` and `$relationship.save()` to accept a single model and an array of models.

packages/superflare/src/relations/has-many.ts

@@ -13,7 +13,9 @@ export class HasMany extends Relation {
     super(query);
   }
 
-  save(models: any[]) {
+  save(models: any[] | any) {
+    models = models instanceof Array ? models : [models];
+
     return Promise.all(
       models.map(async (model) => {
         model[this.foreignKey as keyof Model] =
@@ -24,7 +26,9 @@ export class HasMany extends Relation {
     );
   }
 
-  create(attributeSets: Record<string, any>[]) {
+  create(attributeSets: Record<string, any>[] | Record<string, any>) {
+    attributeSets = attributeSets instanceof Array ? attributeSets : [attributeSets];
+
     return Promise.all(
       attributeSets.map(async (attributes) => {
         const model = new this.query.modelInstance.constructor(attributes);

packages/superflare/tests/model/has-many.test.ts

@@ -73,6 +73,19 @@ it("works", async () => {
 });
 
 it("saves", async () => {
+  const user = await User.create({
+    name: "John Doe",
+  });
+  const posts = await user.$posts().save(
+    new Post({
+      text: "Hello World",
+    }),
+  );
+
+  expect(posts[0].userId).toBe(user.id);
+});
+
+it("saves array", async () => {
   const user = await User.create({
     name: "John Doe",
   });
@@ -86,6 +99,19 @@ it("saves", async () => {
 });
 
 it("creates", async () => {
+  const user = await User.create({
+    name: "John Doe",
+  });
+  const posts = await user.$posts().create(
+    {
+      text: "Hello World",
+    },
+  );
+
+  expect(posts[0].userId).toBe(user.id);
+});
+
+it("creates array", async () => {
   const user = await User.create({
     name: "John Doe",
   });