add CLI command to register runner tokens

cmd/actions.go

@@ -0,0 +1,100 @@
+// Copyright 2023 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package cmd
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"strings"
+
+	actions_model "code.gitea.io/gitea/models/actions"
+	repo_model "code.gitea.io/gitea/models/repo"
+	user_model "code.gitea.io/gitea/models/user"
+	"code.gitea.io/gitea/modules/util"
+
+	"github.com/urfave/cli"
+)
+
+// Cmdembedded represents the available extract sub-command.
+var (
+	CmdActions = cli.Command{
+		Name:        "actions",
+		Usage:       "",
+		Description: "Tasks for managing Gitea Actions",
+		Subcommands: []cli.Command{
+			subcmdActionsGenRunnerToken,
+		},
+	}
+
+	subcmdActionsGenRunnerToken = cli.Command{
+		Name:    "generate-runner-token",
+		Usage:   "Generate a new token for a runner to use to register with the server",
+		Action:  runGenerateActionsRunnerToken,
+		Aliases: []string{"grt"},
+		Flags: []cli.Flag{
+			cli.StringFlag{
+				Name:  "scope, s",
+				Value: "",
+				Usage: "{owner}[/{repo}]",
+			},
+		},
+	}
+)
+
+func runGenerateActionsRunnerToken(c *cli.Context) error {
+	ctx, cancel := installSignals()
+	defer cancel()
+
+	if err := initDB(ctx); err != nil {
+		return err
+	}
+
+	scope := c.String("scope")
+
+	owner, repo, err := parseScope(ctx, scope)
+	if err != nil {
+		return err
+	}
+
+	token, err := actions_model.GetUnactivatedRunnerToken(ctx, owner, repo)
+	if errors.Is(err, util.ErrNotExist) {
+		token, err = actions_model.NewRunnerToken(ctx, owner, repo)
+		if err != nil {
+			return fmt.Errorf("CreateRunnerToken: %s", err)
+		}
+	} else if err != nil {
+		return fmt.Errorf("GetUnactivatedRunnerToken: %s", err)
+	}
+
+	fmt.Printf("%s", token.Token)
+
+	return nil
+}
+
+func parseScope(ctx context.Context, scope string) (owner, repo int64, err error) {
+	owner = 0
+	repo = 0
+	if scope == "" {
+		return owner, repo, nil
+	}
+
+	before, after, found := strings.Cut(scope, "/")
+
+	u, err := user_model.GetUserByName(ctx, before)
+	if err != nil {
+		return owner, repo, nil
+	}
+
+	if !found {
+		return u.ID, repo, nil
+	}
+
+	r, err := repo_model.GetRepositoryByName(u.ID, after)
+	if err != nil {
+		return owner, repo, err
+	}
+	repo = r.ID
+	return owner, repo, nil
+}

main.go

@@ -75,6 +75,7 @@ arguments - which can alternatively be run by running the subcommand web.`
 		cmd.CmdDocs,
 		cmd.CmdDumpRepository,
 		cmd.CmdRestoreRepository,
+		cmd.CmdActions,
 	}
 	// Now adjust these commands to add our global configuration options