From 6fcc569bf25d5d78246b7ce451cdfd5036e68202 Mon Sep 17 00:00:00 2001
From: Simon Ding <dingchuan91@gmail.com>
Date: Sun, 29 Sep 2024 15:45:31 +0800
Subject: [PATCH] feat: clean dangling episodes

---
 db/db.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/db/db.go b/db/db.go
index b781c46..804ea8b 100644
--- a/db/db.go
+++ b/db/db.go
@@ -219,7 +219,10 @@ func (c *Client) DeleteMedia(id int) error {
 		return err
 	}
 	_, err = c.ent.Media.Delete().Where(media.ID(id)).Exec(context.TODO())
-	return err
+	if err != nil {
+		return err
+	}
+	return c.CleanAllDanglingEpisodes()
 }
 
 func (c *Client) SaveEposideDetail(d *ent.Episode) (int, error) {
@@ -645,3 +648,9 @@ func (c *Client) GetMovingNamingFormat() string {
 	}
 	return s
 }
+
+
+func (c *Client) CleanAllDanglingEpisodes() error {
+	_, err := c.ent.Episode.Delete().Where(episode.MediaID(0)).Exec(context.Background())
+	return err
+}
\ No newline at end of file