fix(login): use ssh-agent instead of prompting for key passphrase

This commit is contained in:
Lunny Xiao 2026-08-22 00:03:46 -07:00
parent ee531914cd
commit a20f4ba441
8 changed files with 90 additions and 13 deletions

View file

@ -481,7 +481,7 @@ func (l *Login) Client(options ...gitea.ClientOption) *gitea.Client {
fmt.Fprintf(os.Stderr, "Failed to read SSH passphrase: %s\n", err) fmt.Fprintf(os.Stderr, "Failed to read SSH passphrase: %s\n", err)
os.Exit(1) os.Exit(1)
} }
options = append(options, gitea.UseSSHCert(l.SSHCertPrincipal, l.SSHKey, l.SSHPassphrase)) options = append(options, gitea.UseSSHCert(l.SSHCertPrincipal, l.SSHKeyPath(), l.SSHPassphrase))
} }
if l.SSHKeyFingerprint != "" { if l.SSHKeyFingerprint != "" {
@ -489,7 +489,7 @@ func (l *Login) Client(options ...gitea.ClientOption) *gitea.Client {
fmt.Fprintf(os.Stderr, "Failed to read SSH passphrase: %s\n", err) fmt.Fprintf(os.Stderr, "Failed to read SSH passphrase: %s\n", err)
os.Exit(1) os.Exit(1)
} }
options = append(options, gitea.UseSSHPubkey(l.SSHKeyFingerprint, l.SSHKey, l.SSHPassphrase)) options = append(options, gitea.UseSSHPubkey(l.SSHKeyFingerprint, l.SSHKeyPath(), l.SSHPassphrase))
} }
client, err := gitea.NewClient(l.URL, options...) client, err := gitea.NewClient(l.URL, options...)
@ -505,7 +505,7 @@ func (l *Login) Client(options ...gitea.ClientOption) *gitea.Client {
} }
func (l *Login) askForSSHPassphrase() error { func (l *Login) askForSSHPassphrase() error {
if ok, err := utils.IsKeyEncrypted(l.SSHKey); ok && err == nil && l.SSHPassphrase == "" { if ok, err := utils.IsKeyEncrypted(l.SSHKeyPath()); ok && err == nil && l.SSHPassphrase == "" {
return huh.NewInput(). return huh.NewInput().
Title("ssh-key is encrypted please enter the passphrase: "). Title("ssh-key is encrypted please enter the passphrase: ").
Validate(huh.ValidateNotEmpty()). Validate(huh.ValidateNotEmpty()).
@ -517,6 +517,15 @@ func (l *Login) askForSSHPassphrase() error {
return nil return nil
} }
// SSHKeyPath returns the on-disk SSH key to use, or an empty string when the
// login is configured to authenticate through a running ssh-agent.
func (l *Login) SSHKeyPath() string {
if l.SSHAgent {
return ""
}
return l.SSHKey
}
// GetSSHHost returns SSH host name // GetSSHHost returns SSH host name
func (l *Login) GetSSHHost() string { func (l *Login) GetSSHHost() string {
if l.SSHHost != "" { if l.SSHHost != "" {

View file

@ -90,6 +90,39 @@ func TestLoginClientWithSSHPubkeyDoesNotDeadlockOnFirstRequest(t *testing.T) {
assert.EqualValues(t, 1, signedIssueRequests.Load()) assert.EqualValues(t, 1, signedIssueRequests.Load())
} }
func TestLoginSSHKeyPath(t *testing.T) {
t.Parallel()
tests := []struct {
name string
login Login
want string
}{
{
name: "disk key",
login: Login{SSHKey: "/tmp/id_ed25519"},
want: "/tmp/id_ed25519",
},
{
name: "ssh agent ignores disk key",
login: Login{SSHKey: "/tmp/id_ed25519", SSHAgent: true},
want: "",
},
{
name: "ssh agent without disk key",
login: Login{SSHAgent: true},
want: "",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
t.Parallel()
assert.Equal(t, tt.want, tt.login.SSHKeyPath())
})
}
}
func writeTestSSHKey(t *testing.T) (string, string) { func writeTestSSHKey(t *testing.T) (string, string) {
t.Helper() t.Helper()

View file

@ -130,7 +130,7 @@ func CreateLogin(ctx stdctx.Context, name, token, user, passwd, otp, scopes, ssh
// so we just use the host // so we just use the host
login.SSHHost = serverURL.Host login.SSHHost = serverURL.Host
if len(sshKey) == 0 { if shouldFindSSHKey(sshKey, sshAgent) {
login.SSHKey, err = findSSHKey(ctx, client) login.SSHKey, err = findSSHKey(ctx, client)
if err != nil { if err != nil {
fmt.Printf("Warning: problem while finding a SSH key: %s\n", err) fmt.Printf("Warning: problem while finding a SSH key: %s\n", err)
@ -153,6 +153,10 @@ func CreateLogin(ctx stdctx.Context, name, token, user, passwd, otp, scopes, ssh
return nil return nil
} }
func shouldFindSSHKey(sshKey string, sshAgent bool) bool {
return sshKey == "" && !sshAgent
}
func shouldCheckTokenUniqueness(token string, sshAgent bool, sshKey, sshCertPrincipal, sshKeyFingerprint string) bool { func shouldCheckTokenUniqueness(token string, sshAgent bool, sshKey, sshCertPrincipal, sshKeyFingerprint string) bool {
if sshAgent || sshKey != "" || sshCertPrincipal != "" || sshKeyFingerprint != "" { if sshAgent || sshKey != "" || sshCertPrincipal != "" || sshKeyFingerprint != "" {
return false return false

View file

@ -55,3 +55,35 @@ func TestShouldCheckTokenUniqueness(t *testing.T) {
}) })
} }
} }
func TestShouldFindSSHKey(t *testing.T) {
tests := []struct {
name string
sshKey string
sshAgent bool
want bool
}{
{
name: "explicit key",
sshKey: "~/.ssh/id_ed25519",
want: false,
},
{
name: "ssh agent",
sshAgent: true,
want: false,
},
{
name: "no key configured",
want: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := shouldFindSSHKey(tt.sshKey, tt.sshAgent); got != tt.want {
t.Fatalf("expected %v, got %v", tt.want, got)
}
})
}
}

View file

@ -61,10 +61,9 @@ func remoteURLForPR(login *config.Login, pr *gitea.PullRequest) string {
if isRemoteDeleted(pr) { if isRemoteDeleted(pr) {
repo = pr.Base.Repository repo = pr.Base.Repository
} }
if len(login.SSHKey) != 0 { if login.SSHKeyPath() != "" || login.SSHAgent {
// login.SSHKey is nonempty, if user specified a key manually or we automatically // Use SSH when a key file is configured, or when the login authenticates
// found a matching private key on this machine during login creation. // through a running ssh-agent. In both cases we have a working SSH setup.
// this means, we are very likely to have a working ssh setup.
return repo.SSHURL return repo.SSHURL
} }
return repo.CloneURL return repo.CloneURL
@ -84,7 +83,7 @@ func doPRFetch(
if err != nil { if err != nil {
return "", err return "", err
} }
auth, err := local_git.GetAuthForURL(url, login.GetAccessToken(), login.SSHKey, callback) auth, err := local_git.GetAuthForURL(url, login.GetAccessToken(), login.SSHKeyPath(), callback)
if err != nil { if err != nil {
return "", err return "", err
} }

View file

@ -95,7 +95,7 @@ call me again with the --ignore-sha flag`, remoteBranch)
if urlErr != nil { if urlErr != nil {
return urlErr return urlErr
} }
auth, authErr := local_git.GetAuthForURL(url, login.GetAccessToken(), login.SSHKey, callback) auth, authErr := local_git.GetAuthForURL(url, login.GetAccessToken(), login.SSHKeyPath(), callback)
if authErr != nil { if authErr != nil {
return authErr return authErr
} }

View file

@ -211,7 +211,7 @@ func CreateAgitFlowPull(requestCtx stdctx.Context, ctx *context.TeaContext, remo
return err return err
} }
auth, err := local_git.GetAuthForURL(url, ctx.Login.GetAccessToken(), ctx.Login.SSHKey, callback) auth, err := local_git.GetAuthForURL(url, ctx.Login.GetAccessToken(), ctx.Login.SSHKeyPath(), callback)
if err != nil { if err != nil {
return err return err
} }

View file

@ -33,7 +33,7 @@ func RepoClone(
return nil, err return nil, err
} }
auth, err := local_git.GetAuthForURL(originURL, login.GetAccessToken(), login.SSHKey, callback) auth, err := local_git.GetAuthForURL(originURL, login.GetAccessToken(), login.SSHKeyPath(), callback)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -68,7 +68,7 @@ func RepoClone(
func cloneURL(repo *gitea.Repository, login *config.Login) (*url.URL, error) { func cloneURL(repo *gitea.Repository, login *config.Login) (*url.URL, error) {
urlStr := repo.CloneURL urlStr := repo.CloneURL
if login.SSHKey != "" { if login.SSHKeyPath() != "" || login.SSHAgent {
urlStr = repo.SSHURL urlStr = repo.SSHURL
} }
return local_git.ParseURL(urlStr) return local_git.ParseURL(urlStr)