初始提交: Gitea 项目代码

cmd/admin_regenerate.go

@@ -0,0 +1,44 @@
+// Copyright 2023 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package cmd
+
+import (
+	"context"
+
+	"gitea.dev/modules/graceful"
+	asymkey_service "gitea.dev/services/asymkey"
+	repo_service "gitea.dev/services/repository"
+
+	"github.com/urfave/cli/v3"
+)
+
+func newRegenerateHooksCommand() *cli.Command {
+	return &cli.Command{
+		Name:   "hooks",
+		Usage:  "Regenerate git-hooks",
+		Action: runRegenerateHooks,
+	}
+}
+
+func newRegenerateKeysCommand() *cli.Command {
+	return &cli.Command{
+		Name:   "keys",
+		Usage:  "Regenerate authorized_keys file",
+		Action: runRegenerateKeys,
+	}
+}
+
+func runRegenerateHooks(ctx context.Context, _ *cli.Command) error {
+	if err := initDB(ctx); err != nil {
+		return err
+	}
+	return repo_service.SyncRepositoryHooks(graceful.GetManager().ShutdownContext())
+}
+
+func runRegenerateKeys(ctx context.Context, _ *cli.Command) error {
+	if err := initDB(ctx); err != nil {
+		return err
+	}
+	return asymkey_service.RewriteAllPublicKeys(ctx)
+}