From b04f4bb2d80fcde5bf64eb8f89aa92f305a89001 Mon Sep 17 00:00:00 2001
From: yystopf <yystopf@163.com>
Date: Thu, 30 May 2024 16:00:19 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=EF=BC=9Arerun=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91=E6=96=B0=E5=A2=9E=E4=B8=80=E6=9D=A1=E8=AE=B0=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/actions/run.go               | 92 +++++++++++++++++++++++++++++
 routers/hat/repo/actions/actions.go | 28 +++++++--
 2 files changed, 115 insertions(+), 5 deletions(-)
 create mode 100644 models/actions/run.go

diff --git a/models/actions/run.go b/models/actions/run.go
new file mode 100644
index 0000000..38067c8
--- /dev/null
+++ b/models/actions/run.go
@@ -0,0 +1,92 @@
+package actions
+
+import (
+	"context"
+
+	gitea_actions_model "code.gitea.io/gitea/models/actions"
+	"code.gitea.io/gitea/models/db"
+	gitea_repo_model "code.gitea.io/gitea/models/repo"
+	"xorm.io/builder"
+)
+
+func InsertRun(ctx context.Context, run *gitea_actions_model.ActionRun, jobs []*gitea_actions_model.ActionRunJob) error {
+	ctx, commiter, err := db.TxContext(ctx)
+	if err != nil {
+		return err
+	}
+	defer commiter.Close()
+
+	index, err := db.GetNextResourceIndex(ctx, "action_run_index", run.RepoID)
+	if err != nil {
+		return err
+	}
+	run.Index = index
+	run.ID = 0
+
+	if err := db.Insert(ctx, run); err != nil {
+		return err
+	}
+
+	if run.Repo == nil {
+		repo, err := gitea_repo_model.GetRepositoryByID(ctx, run.RepoID)
+		if err != nil {
+			return err
+		}
+		run.Repo = repo
+	}
+
+	if err := updateRepoRunsNumbers(ctx, run.Repo); err != nil {
+		return err
+	}
+
+	runJobs := make([]*gitea_actions_model.ActionRunJob, 0, len(jobs))
+	for _, job := range jobs {
+		runJobs = append(runJobs, &gitea_actions_model.ActionRunJob{
+			RunID:             run.ID,
+			RepoID:            run.RepoID,
+			OwnerID:           run.OwnerID,
+			CommitSHA:         run.CommitSHA,
+			IsForkPullRequest: run.IsForkPullRequest,
+			Name:              job.Name,
+			WorkflowPayload:   job.WorkflowPayload,
+			JobID:             job.JobID,
+			Needs:             job.Needs,
+			RunsOn:            job.RunsOn,
+			Status:            gitea_actions_model.StatusWaiting,
+		})
+
+	}
+	if err := db.Insert(ctx, runJobs); err != nil {
+		return err
+	}
+
+	if err := gitea_actions_model.IncreaseTaskVersion(ctx, run.OwnerID, run.RepoID); err != nil {
+		return err
+	}
+
+	return commiter.Commit()
+}
+
+func updateRepoRunsNumbers(ctx context.Context, repo *gitea_repo_model.Repository) error {
+	_, err := db.GetEngine(ctx).ID(repo.ID).
+		SetExpr("num_action_runs",
+			builder.Select("count(*)").From("action_run").
+				Where(builder.Eq{"repo_id": repo.ID}),
+		).
+		SetExpr("num_closed_action_runs",
+			builder.Select("count(*)").From("action_run").
+				Where(builder.Eq{
+					"repo_id": repo.ID,
+				}.And(
+					builder.In("status",
+						gitea_actions_model.StatusSuccess,
+						gitea_actions_model.StatusFailure,
+						gitea_actions_model.StatusCancelled,
+						gitea_actions_model.StatusSkipped,
+					),
+				),
+				),
+		).
+		Update(repo)
+	return err
+}
diff --git a/routers/hat/repo/actions/actions.go b/routers/hat/repo/actions/actions.go
index 5ff296f..337d851 100644
--- a/routers/hat/repo/actions/actions.go
+++ b/routers/hat/repo/actions/actions.go
@@ -18,6 +18,7 @@ import (
 	"code.gitea.io/gitea/modules/actions"
 	"code.gitea.io/gitea/modules/base"
 	"code.gitea.io/gitea/modules/container"
+	hat_actions_model "code.gitlink.org.cn/Gitlink/gitea_hat.git/models/actions"
 
 	"code.gitea.io/gitea/modules/context"
 	"code.gitea.io/gitea/modules/util"
@@ -426,16 +427,33 @@ func Rerun(ctx *context.APIContext) {
 		jobs = []*actions_model.ActionRunJob{job}
 	}
 
-	for _, j := range jobs {
-		if err := rerunJob(ctx, j); err != nil {
-			ctx.Error(http.StatusInternalServerError, "rerunJob", err.Error())
-			return
-		}
+	// if err := hat_actions_model.InsertRun(ctx, run, jobs); err != nil {
+	// 	ctx.Error(http.StatusInternalServerError, "InsertRun", err.Error())
+	// 	return
+	// }
+
+	// for _, j := range jobs {
+	if err := newRerunJob(ctx, run, jobs); err != nil {
+		ctx.Error(http.StatusInternalServerError, "rerunJob", err.Error())
+		return
 	}
+	// }
 
 	ctx.Status(http.StatusNoContent)
 }
 
+func newRerunJob(ctx *context.APIContext, run *actions_model.ActionRun, jobs []*actions_model.ActionRunJob) error {
+	if err := db.WithTx(ctx, func(ctx stdCtx.Context) error {
+		err := hat_actions_model.InsertRun(ctx, run, jobs)
+		return err
+	}); err != nil {
+		return err
+	}
+
+	actions_service.CreateCommitStatus(ctx, jobs...)
+	return nil
+}
+
 func rerunJob(ctx *context.APIContext, job *actions_model.ActionRunJob) error {
 	status := job.Status
 	if !status.IsDone() {