feat: change history data field

simon-ding · Jul 15, 2024 · b2c4b69 · b2c4b69
1 parent f37a8ee
commit b2c4b69
Show file tree

Hide file tree

Showing 15 changed files with 178 additions and 142 deletions.
diff --git a/db/db.go b/db/db.go
@@ -382,11 +382,11 @@ func (c *Client) SetDefaultStorageByName(name string) error {
 
 func (c *Client) SaveHistoryRecord(h ent.History) (*ent.History, error) {
 	return c.ent.History.Create().SetSeriesID(h.SeriesID).SetEpisodeID(h.EpisodeID).SetDate(time.Now()).
-		SetCompleted(h.Completed).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetSaved(h.Saved).Save(context.TODO())
+		SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetSaved(h.Saved).Save(context.TODO())
 }
 
-func (c *Client) SetHistoryComplete(id int) error {
-	return c.ent.History.Update().Where(history.ID(id)).SetCompleted(true).Exec(context.TODO())
+func (c *Client) SetHistoryStatus(id int, status history.Status) error {
+	return c.ent.History.Update().Where(history.ID(id)).SetStatus(status).Exec(context.TODO())
 }
 
 func (c *Client) GetHistories() ent.Histories {
@@ -398,7 +398,8 @@ func (c *Client) GetHistories() ent.Histories {
 }
 
 func (c *Client) GetRunningHistories() ent.Histories {
-	h, err := c.ent.History.Query().Where(history.Completed(false)).All(context.TODO())
+	h, err := c.ent.History.Query().Where(history.Or(history.StatusEQ(history.StatusRunning), 
+		history.StatusEQ(history.StatusUploading))).All(context.TODO())
 	if err != nil {
 		return nil
 	}

diff --git a/ent/history.go b/ent/history.go
diff --git a/ent/history/history.go b/ent/history/history.go
diff --git a/ent/history/where.go b/ent/history/where.go
diff --git a/ent/history_create.go b/ent/history_create.go
diff --git a/ent/history_update.go b/ent/history_update.go
diff --git a/ent/migrate/schema.go b/ent/migrate/schema.go