From b6c2db0e83f96a0b637ed67da0f90fa51ed2c371 Mon Sep 17 00:00:00 2001
From: Hayden <64056131+hay-kot@users.noreply.github.com>
Date: Mon, 6 Mar 2023 11:46:01 -0900
Subject: [PATCH] refactor

---
 backend/internal/data/ent/schema/helpers.go |  1 -
 backend/internal/data/ent/schema/item.go    | 22 ++++++++++-----------
 2 files changed, 10 insertions(+), 13 deletions(-)
 delete mode 100644 backend/internal/data/ent/schema/helpers.go

diff --git a/backend/internal/data/ent/schema/helpers.go b/backend/internal/data/ent/schema/helpers.go
deleted file mode 100644
index b9e149c..0000000
--- a/backend/internal/data/ent/schema/helpers.go
+++ /dev/null
@@ -1 +0,0 @@
-package schema
diff --git a/backend/internal/data/ent/schema/item.go b/backend/internal/data/ent/schema/item.go
index b685898..344829f 100644
--- a/backend/internal/data/ent/schema/item.go
+++ b/backend/internal/data/ent/schema/item.go
@@ -99,6 +99,13 @@ func (Item) Fields() []ent.Field {
 
 // Edges of the Item.
 func (Item) Edges() []ent.Edge {
+	owned := func(s string, t any) ent.Edge {
+		return edge.To(s, t).
+			Annotations(entsql.Annotation{
+				OnDelete: entsql.Cascade,
+			})
+	}
+
 	return []ent.Edge{
 		edge.To("children", Item.Type).
 			From("parent").
@@ -108,17 +115,8 @@ func (Item) Edges() []ent.Edge {
 		edge.From("location", Location.Type).
 			Ref("items").
 			Unique(),
-		edge.To("fields", ItemField.Type).
-			Annotations(entsql.Annotation{
-				OnDelete: entsql.Cascade,
-			}),
-		edge.To("maintenance_entries", MaintenanceEntry.Type).
-			Annotations(entsql.Annotation{
-				OnDelete: entsql.Cascade,
-			}),
-		edge.To("attachments", Attachment.Type).
-			Annotations(entsql.Annotation{
-				OnDelete: entsql.Cascade,
-			}),
+		owned("fields", ItemField.Type),
+		owned("maintenance_entries", MaintenanceEntry.Type),
+		owned("attachments", Attachment.Type),
 	}
 }