mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-08-21 07:02:00 +02:00
Backport #26071 by @yardenshoham We are now: - Making sure there is no existing access token with the same name - Making sure the given scopes are valid (we already did this before but now we have a message) The logic is mostly taken froma12a5f3652/routers/api/v1/user/app.go (L101-L123)
Closes #26044 Signed-off-by: Yarden Shoham <git@yardenshoham.com> (cherry picked from commit43213b816d
)
92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
// Copyright 2023 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
auth_model "code.gitea.io/gitea/models/auth"
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
var microcmdUserGenerateAccessToken = cli.Command{
|
|
Name: "generate-access-token",
|
|
Usage: "Generate an access token for a specific user",
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "username,u",
|
|
Usage: "Username",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "token-name,t",
|
|
Usage: "Token name",
|
|
Value: "gitea-admin",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "raw",
|
|
Usage: "Display only the token value",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "scopes",
|
|
Value: "",
|
|
Usage: "Comma separated list of scopes to apply to access token",
|
|
},
|
|
},
|
|
Action: runGenerateAccessToken,
|
|
}
|
|
|
|
func runGenerateAccessToken(c *cli.Context) error {
|
|
if !c.IsSet("username") {
|
|
return fmt.Errorf("You must provide a username to generate a token for")
|
|
}
|
|
|
|
ctx, cancel := installSignals()
|
|
defer cancel()
|
|
|
|
if err := initDB(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
user, err := user_model.GetUserByName(ctx, c.String("username"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// construct token with name and user so we can make sure it is unique
|
|
t := &auth_model.AccessToken{
|
|
Name: c.String("token-name"),
|
|
UID: user.ID,
|
|
}
|
|
|
|
exist, err := auth_model.AccessTokenByNameExists(t)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exist {
|
|
return fmt.Errorf("access token name has been used already")
|
|
}
|
|
|
|
// make sure the scopes are valid
|
|
accessTokenScope, err := auth_model.AccessTokenScope(c.String("scopes")).Normalize()
|
|
if err != nil {
|
|
return fmt.Errorf("invalid access token scope provided: %w", err)
|
|
}
|
|
t.Scope = accessTokenScope
|
|
|
|
// create the token
|
|
if err := auth_model.NewAccessToken(t); err != nil {
|
|
return err
|
|
}
|
|
|
|
if c.Bool("raw") {
|
|
fmt.Printf("%s\n", t.Token)
|
|
} else {
|
|
fmt.Printf("Access token was successfully created: %s\n", t.Token)
|
|
}
|
|
|
|
return nil
|
|
}
|