mirror of https://github.com/go-gitea/gitea
add CLI command to register runner tokens (#23762)
This is a CLI command to generate new tokens for the runners to register with Fix https://github.com/go-gitea/gitea/issues/23643 --------- Co-authored-by: delvh <dev.lh@web.de>pull/24119/head^2
parent
1819c4b59b
commit
4014200021
@ -0,0 +1,56 @@ |
||||
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package cmd |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"code.gitea.io/gitea/modules/private" |
||||
"code.gitea.io/gitea/modules/setting" |
||||
|
||||
"github.com/urfave/cli" |
||||
) |
||||
|
||||
var ( |
||||
// CmdActions represents the available actions sub-commands.
|
||||
CmdActions = cli.Command{ |
||||
Name: "actions", |
||||
Usage: "", |
||||
Description: "Commands 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}] - leave empty for a global runner", |
||||
}, |
||||
}, |
||||
} |
||||
) |
||||
|
||||
func runGenerateActionsRunnerToken(c *cli.Context) error { |
||||
ctx, cancel := installSignals() |
||||
defer cancel() |
||||
|
||||
setting.InitProviderFromExistingFile() |
||||
setting.LoadCommonSettings() |
||||
|
||||
scope := c.String("scope") |
||||
|
||||
respText, extra := private.GenerateActionsRunnerToken(ctx, scope) |
||||
if extra.HasError() { |
||||
return handleCliResponseExtra(extra) |
||||
} |
||||
_, _ = fmt.Printf("%s\n", respText) |
||||
return nil |
||||
} |
@ -0,0 +1,27 @@ |
||||
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package private |
||||
|
||||
import ( |
||||
"context" |
||||
|
||||
"code.gitea.io/gitea/modules/setting" |
||||
) |
||||
|
||||
// Email structure holds a data for sending general emails
|
||||
type GenerateTokenRequest struct { |
||||
Scope string |
||||
} |
||||
|
||||
// GenerateActionsRunnerToken calls the internal GenerateActionsRunnerToken function
|
||||
func GenerateActionsRunnerToken(ctx context.Context, scope string) (string, ResponseExtra) { |
||||
reqURL := setting.LocalURL + "api/internal/actions/generate_actions_runner_token" |
||||
|
||||
req := newInternalRequest(ctx, reqURL, "POST", GenerateTokenRequest{ |
||||
Scope: scope, |
||||
}) |
||||
|
||||
resp, extra := requestJSONResp(req, &responseText{}) |
||||
return resp.Text, extra |
||||
} |
@ -0,0 +1,91 @@ |
||||
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package private |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
"net/http" |
||||
"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/context" |
||||
"code.gitea.io/gitea/modules/json" |
||||
"code.gitea.io/gitea/modules/log" |
||||
"code.gitea.io/gitea/modules/private" |
||||
"code.gitea.io/gitea/modules/util" |
||||
) |
||||
|
||||
// GenerateActionsRunnerToken generates a new runner token for a given scope
|
||||
func GenerateActionsRunnerToken(ctx *context.PrivateContext) { |
||||
var genRequest private.GenerateTokenRequest |
||||
rd := ctx.Req.Body |
||||
defer rd.Close() |
||||
|
||||
if err := json.NewDecoder(rd).Decode(&genRequest); err != nil { |
||||
log.Error("%v", err) |
||||
ctx.JSON(http.StatusInternalServerError, private.Response{ |
||||
Err: err.Error(), |
||||
}) |
||||
return |
||||
} |
||||
|
||||
owner, repo, err := parseScope(ctx, genRequest.Scope) |
||||
if err != nil { |
||||
log.Error("%v", err) |
||||
ctx.JSON(http.StatusInternalServerError, private.Response{ |
||||
Err: err.Error(), |
||||
}) |
||||
} |
||||
|
||||
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 { |
||||
err := fmt.Sprintf("error while creating runner token: %v", err) |
||||
log.Error("%v", err) |
||||
ctx.JSON(http.StatusInternalServerError, private.Response{ |
||||
Err: err, |
||||
}) |
||||
return |
||||
} |
||||
} else if err != nil { |
||||
err := fmt.Sprintf("could not get unactivated runner token: %v", err) |
||||
log.Error("%v", err) |
||||
ctx.JSON(http.StatusInternalServerError, private.Response{ |
||||
Err: err, |
||||
}) |
||||
return |
||||
} |
||||
|
||||
ctx.PlainText(http.StatusOK, token.Token) |
||||
} |
||||
|
||||
func parseScope(ctx *context.PrivateContext, scope string) (ownerID, repoID int64, err error) { |
||||
ownerID = 0 |
||||
repoID = 0 |
||||
if scope == "" { |
||||
return ownerID, repoID, nil |
||||
} |
||||
|
||||
ownerName, repoName, found := strings.Cut(scope, "/") |
||||
|
||||
u, err := user_model.GetUserByName(ctx, ownerName) |
||||
if err != nil { |
||||
return ownerID, repoID, err |
||||
} |
||||
|
||||
if !found { |
||||
return u.ID, repoID, nil |
||||
} |
||||
|
||||
r, err := repo_model.GetRepositoryByName(u.ID, repoName) |
||||
if err != nil { |
||||
return ownerID, repoID, err |
||||
} |
||||
repoID = r.ID |
||||
return ownerID, repoID, nil |
||||
} |
Loading…
Reference in new issue