mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-11-04 08:21:11 +00:00 
			
		
		
		
	Backport #25537 More fix for #24981 * #24981 Close #22361, #25552 * #22361 * #25552 There were many patches for Gitea's sub-commands to satisfy the facts: * Some sub-commands shouldn't output any log, otherwise the git protocol would be broken * Sometimes the users want to see "verbose" or "quiet" outputs That's a longstanding problem, and very fragile. This PR is only a quick patch for the problem. In the future, the sub-command system should be refactored to a clear solution. ---- Other changes: * Use `ReplaceAllWriters` to replace `RemoveAllWriters().AddWriters(writer)`, then it's an atomic operation. * Remove unnecessary `syncLevelInternal` calls, because `AddWriters/addWritersInternal` already calls it.
		
			
				
	
	
		
			78 lines
		
	
	
	
		
			2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			78 lines
		
	
	
	
		
			2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2018 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package cmd
 | 
						|
 | 
						|
import (
 | 
						|
	"errors"
 | 
						|
	"fmt"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/modules/log"
 | 
						|
	"code.gitea.io/gitea/modules/private"
 | 
						|
 | 
						|
	"github.com/urfave/cli"
 | 
						|
)
 | 
						|
 | 
						|
// CmdKeys represents the available keys sub-command
 | 
						|
var CmdKeys = cli.Command{
 | 
						|
	Name:   "keys",
 | 
						|
	Usage:  "This command queries the Gitea database to get the authorized command for a given ssh key fingerprint",
 | 
						|
	Before: PrepareConsoleLoggerLevel(log.FATAL),
 | 
						|
	Action: runKeys,
 | 
						|
	Flags: []cli.Flag{
 | 
						|
		cli.StringFlag{
 | 
						|
			Name:  "expected, e",
 | 
						|
			Value: "git",
 | 
						|
			Usage: "Expected user for whom provide key commands",
 | 
						|
		},
 | 
						|
		cli.StringFlag{
 | 
						|
			Name:  "username, u",
 | 
						|
			Value: "",
 | 
						|
			Usage: "Username trying to log in by SSH",
 | 
						|
		},
 | 
						|
		cli.StringFlag{
 | 
						|
			Name:  "type, t",
 | 
						|
			Value: "",
 | 
						|
			Usage: "Type of the SSH key provided to the SSH Server (requires content to be provided too)",
 | 
						|
		},
 | 
						|
		cli.StringFlag{
 | 
						|
			Name:  "content, k",
 | 
						|
			Value: "",
 | 
						|
			Usage: "Base64 encoded content of the SSH key provided to the SSH Server (requires type to be provided too)",
 | 
						|
		},
 | 
						|
	},
 | 
						|
}
 | 
						|
 | 
						|
func runKeys(c *cli.Context) error {
 | 
						|
	if !c.IsSet("username") {
 | 
						|
		return errors.New("No username provided")
 | 
						|
	}
 | 
						|
	// Check username matches the expected username
 | 
						|
	if strings.TrimSpace(c.String("username")) != strings.TrimSpace(c.String("expected")) {
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	content := ""
 | 
						|
 | 
						|
	if c.IsSet("type") && c.IsSet("content") {
 | 
						|
		content = fmt.Sprintf("%s %s", strings.TrimSpace(c.String("type")), strings.TrimSpace(c.String("content")))
 | 
						|
	}
 | 
						|
 | 
						|
	if content == "" {
 | 
						|
		return errors.New("No key type and content provided")
 | 
						|
	}
 | 
						|
 | 
						|
	ctx, cancel := installSignals()
 | 
						|
	defer cancel()
 | 
						|
 | 
						|
	setup(ctx, false)
 | 
						|
 | 
						|
	authorizedString, extra := private.AuthorizedPublicKeyByContent(ctx, content)
 | 
						|
	// do not use handleCliResponseExtra or cli.NewExitError, if it exists immediately, it breaks some tests like Test_CmdKeys
 | 
						|
	if extra.Error != nil {
 | 
						|
		return extra.Error
 | 
						|
	}
 | 
						|
	fmt.Println(strings.TrimSpace(authorizedString))
 | 
						|
	return nil
 | 
						|
}
 |