fix: pod delete relations

ElaBosak233 · Jun 2, 2024 · bb2c400 · bb2c400
1 parent 4b7932e
commit bb2c400
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/internal/model/challenge.go b/internal/model/challenge.go
@@ -74,8 +74,13 @@ func (c *Challenge) BeforeDelete(db *gorm.DB) (err error) {
 	db.Table("flags").Where("challenge_id = ?", c.ID).Delete(&Flag{})
 	db.Table("ports").Where("challenge_id = ?", c.ID).Delete(&Port{})
 	db.Table("envs").Where("challenge_id = ?", c.ID).Delete(&Env{})
-	db.Table("pods").Where("challenge_id = ?", c.ID).Delete(&Pod{})
 	db.Table("submissions").Where("challenge_id = ?", c.ID).Delete(&Submission{})
 	db.Table("game_challenges").Where("challenge_id = ?", c.ID).Delete(&GameChallenge{})
+
+	var pods []Pod
+	db.Table("pods").Where("challenge_id = ?", c.ID).Find(&pods)
+	for _, pod := range pods {
+		db.Table("pods").Where("id = ?", pod.ID).Delete(&Pod{})
+	}
 	return nil
 }
diff --git a/internal/model/pod.go b/internal/model/pod.go
@@ -1,5 +1,7 @@
 package model
 
+import "gorm.io/gorm"
+
 type Pod struct {
 	ID          uint       `json:"id"`
 	GameID      *uint      `gorm:"index;null;default:null" json:"game_id"`
@@ -19,3 +21,9 @@ func (p *Pod) Simplify() {
 		p.Challenge.Simplify()
 	}
 }
+
+func (p *Pod) BeforeDelete(db *gorm.DB) error {
+	db.Table("nats").Where("pod_id = ?", p.ID).Delete(&Nat{})
+	db.Table("flag_gens").Where("pod_id = ?", p.ID).Delete(&FlagGen{})
+	return nil
+}