feat: add support for schedule specs and tasks

- Add a new function `GetSchedulesMapByIDs` to retrieve schedules by ID
- Create new schedule and schedule spec when creating a schedule task
- Delete schedule specs when deleting a schedule
- Add a new model `ActionScheduleSpec` for schedule specs

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

Author: appleboy

models/actions/schedule.go

@@ -38,6 +38,12 @@ func init() {
 	db.RegisterModel(new(ActionSchedule))
 }
 
+// GetSchedulesMapByIDs returns the schedules by given id slice.
+func GetSchedulesMapByIDs(ids []int64) (map[int64]*ActionSchedule, error) {
+	schedules := make(map[int64]*ActionSchedule, len(ids))
+	return schedules, db.GetEngine(db.DefaultContext).In("id", ids).Find(&schedules)
+}
+
 // CreateScheduleTask creates new schedule task.
 func CreateScheduleTask(ctx context.Context, rows []*ActionSchedule) error {
 	ctx, committer, err := db.TxContext(db.DefaultContext)
@@ -47,27 +53,42 @@ func CreateScheduleTask(ctx context.Context, rows []*ActionSchedule) error {
 	defer committer.Close()
 
 	if len(rows) > 0 {
-		if err = db.Insert(ctx, rows); err != nil {
-			return err
+		for _, row := range rows {
+			// create new schedule
+			if err = db.Insert(ctx, row); err != nil {
+				return err
+			}
+
+			// create new schedule spec
+			for _, spec := range row.Specs {
+				if err = db.Insert(ctx, &ActionScheduleSpec{
+					RepoID:     row.RepoID,
+					ScheduleID: row.ID,
+					Spec:       spec,
+				}); err != nil {
+					return err
+				}
+			}
 		}
 	}
 
 	return committer.Commit()
 }
 
 func DeleteScheduleTaskByRepo(ctx context.Context, id int64) error {
+	ctx, committer, err := db.TxContext(db.DefaultContext)
+	if err != nil {
+		return err
+	}
+	defer committer.Close()
+
 	if _, err := db.GetEngine(ctx).Delete(&ActionSchedule{RepoID: id}); err != nil {
 		return err
 	}
-	return nil
-}
 
-func UpdateSchedule(ctx context.Context, schedule *ActionSchedule, cols ...string) error {
-	sess := db.GetEngine(ctx).ID(schedule.ID)
-	if len(cols) > 0 {
-		sess.Cols(cols...)
+	if _, err := db.GetEngine(ctx).Delete(&ActionScheduleSpec{RepoID: id}); err != nil {
+		return err
 	}
-	_, err := sess.Update(schedule)
 
-	return err
+	return committer.Commit()
 }

models/actions/schedule_spec.go

@@ -0,0 +1,24 @@
+// Copyright 2023 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package actions
+
+import (
+	"code.gitea.io/gitea/models/db"
+	repo_model "code.gitea.io/gitea/models/repo"
+)
+
+// ActionScheduleSpec represents a schedule spec of a workflow file
+type ActionScheduleSpec struct {
+	ID         int64
+	RepoID     int64
+	Repo       *repo_model.Repository `xorm:"-"`
+	ScheduleID int64
+	Schedule   *ActionSchedule `xorm:"-"`
+
+	Spec string
+}
+
+func init() {
+	db.RegisterModel(new(ActionScheduleSpec))
+}

models/actions/schedule_spec_list.go

@@ -0,0 +1,91 @@
+// Copyright 2023 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package actions
+
+import (
+	"context"
+
+	"code.gitea.io/gitea/models/db"
+	repo_model "code.gitea.io/gitea/models/repo"
+	"code.gitea.io/gitea/modules/container"
+	"xorm.io/builder"
+)
+
+type SpecList []*ActionScheduleSpec
+
+func (specs SpecList) GetScheduleIDs() []int64 {
+	ids := make(container.Set[int64], len(specs))
+	for _, spec := range specs {
+		ids.Add(spec.ScheduleID)
+	}
+	return ids.Values()
+}
+
+func (specs SpecList) LoadSchedules() error {
+	scheduleIDs := specs.GetScheduleIDs()
+	schedules, err := GetSchedulesMapByIDs(scheduleIDs)
+	if err != nil {
+		return err
+	}
+	for _, spec := range specs {
+		spec.Schedule = schedules[spec.ScheduleID]
+	}
+	return nil
+}
+
+func (specs SpecList) GetRepoIDs() []int64 {
+	ids := make(container.Set[int64], len(specs))
+	for _, spec := range specs {
+		ids.Add(spec.RepoID)
+	}
+	return ids.Values()
+}
+
+func (specs SpecList) LoadRepos() error {
+	repoIDs := specs.GetRepoIDs()
+	repos, err := repo_model.GetRepositoriesMapByIDs(repoIDs)
+	if err != nil {
+		return err
+	}
+	for _, spec := range specs {
+		spec.Repo = repos[spec.RepoID]
+	}
+	return nil
+}
+
+type FindSpecOptions struct {
+	db.ListOptions
+	RepoID int64
+	GetAll bool
+}
+
+func (opts FindSpecOptions) toConds() builder.Cond {
+	cond := builder.NewCond()
+	if opts.RepoID > 0 {
+		cond = cond.And(builder.Eq{"repo_id": opts.RepoID})
+	}
+
+	return cond
+}
+
+func FindSpecs(ctx context.Context, opts FindSpecOptions) (SpecList, int64, error) {
+	e := db.GetEngine(ctx).Where(opts.toConds())
+	if !opts.GetAll && opts.PageSize > 0 && opts.Page >= 1 {
+		e.Limit(opts.PageSize, (opts.Page-1)*opts.PageSize)
+	}
+	var specs SpecList
+	total, err := e.Desc("id").FindAndCount(&specs)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	if err := specs.LoadSchedules(); err != nil {
+		return nil, 0, err
+	}
+	return specs, total, nil
+}
+
+func CountSpecs(ctx context.Context, opts FindSpecOptions) (int64, error) {
+	return db.GetEngine(ctx).Where(opts.toConds()).Count(new(ActionScheduleSpec))
+}

models/migrations/v1_20/v245.go

@@ -28,7 +28,15 @@ func AddActionScheduleTable(x *xorm.Engine) error {
 		Updated       timeutil.TimeStamp `xorm:"updated"`
 	}
 
+	type ActionScheduleSpec struct {
+		ID         int64
+		RepoID     int64
+		ScheduleID int64
+		Spec       string
+	}
+
 	return x.Sync(
 		new(ActionSchedule),
+		new(ActionScheduleSpec),
 	)
 }