add disableMirrorActionsUnit

Author: Zettat123

cmd/doctor.go

@@ -14,14 +14,10 @@ import (
 	"code.gitea.io/gitea/models/db"
 	"code.gitea.io/gitea/models/migrations"
 	migrate_base "code.gitea.io/gitea/models/migrations/base"
-	repo_model "code.gitea.io/gitea/models/repo"
-	unit_model "code.gitea.io/gitea/models/unit"
 	"code.gitea.io/gitea/modules/container"
 	"code.gitea.io/gitea/modules/log"
-	"code.gitea.io/gitea/modules/optional"
 	"code.gitea.io/gitea/modules/setting"
 	"code.gitea.io/gitea/services/doctor"
-	repo_service "code.gitea.io/gitea/services/repository"
 
 	"github.com/urfave/cli/v2"
 	"xorm.io/xorm"
@@ -37,7 +33,6 @@ var CmdDoctor = &cli.Command{
 		cmdDoctorCheck,
 		cmdRecreateTable,
 		cmdDoctorConvert,
-		cmdDisableMirrorActionsUnit,
 	},
 }
 
@@ -143,48 +138,6 @@ func runRecreateTable(ctx *cli.Context) error {
 	})
 }
 
-var cmdDisableMirrorActionsUnit = &cli.Command{
-	Name:   "disable-mirror-actions-unit",
-	Usage:  "Disable the actions unit for all mirrors",
-	Action: runDisableMirrorActionsUnit,
-}
-
-func runDisableMirrorActionsUnit(_ *cli.Context) error {
-	stdCtx, cancel := installSignals()
-	defer cancel()
-
-	if err := initDB(stdCtx); err != nil {
-		return err
-	}
-
-	const pageSize = repo_model.RepositoryListDefaultPageSize
-	for page := 1; ; page++ {
-		repos, count, err := repo_model.SearchRepository(stdCtx, &repo_model.SearchRepoOptions{
-			ListOptions: db.ListOptions{
-				PageSize: pageSize,
-				Page:     page,
-			},
-			Mirror: optional.Some(true),
-		})
-		if err != nil {
-			return fmt.Errorf("SearchRepository: %w", err)
-		}
-		if len(repos) == 0 {
-			break
-		}
-		log.Info("Processing %d-%d of %d repos", (page-1)*pageSize+1, (page-1)*pageSize+len(repos), count)
-		for _, repo := range repos {
-			if err := repo_service.UpdateRepositoryUnits(stdCtx, repo, nil, []unit_model.Type{unit_model.TypeActions}); err != nil {
-				return err
-			}
-		}
-	}
-
-	log.Info("Finish processing")
-
-	return nil
-}
-
 func setupDoctorDefaultLogger(ctx *cli.Context, colorize bool) {
 	// Silence the default loggers
 	setupConsoleLogger(log.FATAL, log.CanColorStderr, os.Stderr)

services/doctor/actions.go

@@ -0,0 +1,69 @@
+package doctor
+
+import (
+	"context"
+	"fmt"
+
+	"code.gitea.io/gitea/models/db"
+	repo_model "code.gitea.io/gitea/models/repo"
+	unit_model "code.gitea.io/gitea/models/unit"
+	"code.gitea.io/gitea/modules/log"
+	"code.gitea.io/gitea/modules/optional"
+	repo_service "code.gitea.io/gitea/services/repository"
+)
+
+func disableMirrorActionsUnit(ctx context.Context, logger log.Logger, autofix bool) error {
+	var reposToFix []*repo_model.Repository
+
+	for page := 1; ; page++ {
+		repos, _, err := repo_model.SearchRepository(ctx, &repo_model.SearchRepoOptions{
+			ListOptions: db.ListOptions{
+				PageSize: repo_model.RepositoryListDefaultPageSize,
+				Page:     page,
+			},
+			Mirror: optional.Some(true),
+		})
+		if err != nil {
+			return fmt.Errorf("SearchRepository: %w", err)
+		}
+		if len(repos) == 0 {
+			break
+		}
+
+		for _, repo := range repos {
+			if repo.UnitEnabled(ctx, unit_model.TypeActions) {
+				reposToFix = append(reposToFix, repo)
+			}
+		}
+	}
+
+	if len(reposToFix) == 0 {
+		logger.Info("Found no mirror with actions unit enabled")
+	} else {
+		logger.Warn("Found %d mirrors with actions unit enabled", len(reposToFix))
+	}
+
+	if !autofix {
+		return nil
+	}
+
+	for _, repo := range reposToFix {
+		if err := repo_service.UpdateRepositoryUnits(ctx, repo, nil, []unit_model.Type{unit_model.TypeActions}); err != nil {
+			return err
+		}
+	}
+
+	logger.Info("Fixed %d mirrors with actions unit enabled", len(reposToFix))
+
+	return nil
+}
+
+func init() {
+	Register(&Check{
+		Title:     "Disable the actions unit for all mirrors",
+		Name:      "disable-mirror-actions-unit",
+		IsDefault: false,
+		Run:       disableMirrorActionsUnit,
+		Priority:  9,
+	})
+}