119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package shared
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/AlecAivazis/survey/v2"
|
|
"github.com/cli/cli/internal/config"
|
|
"github.com/cli/cli/internal/run"
|
|
"github.com/cli/cli/pkg/cmd/ssh-key/add"
|
|
"github.com/cli/cli/pkg/prompt"
|
|
"github.com/cli/safeexec"
|
|
)
|
|
|
|
type sshContext struct {
|
|
configDir string
|
|
keygenExe string
|
|
}
|
|
|
|
func (c *sshContext) sshDir() (string, error) {
|
|
if c.configDir != "" {
|
|
return c.configDir, nil
|
|
}
|
|
dir, err := config.HomeDirPath(".ssh")
|
|
if err == nil {
|
|
c.configDir = dir
|
|
}
|
|
return dir, err
|
|
}
|
|
|
|
func (c *sshContext) localPublicKeys() ([]string, error) {
|
|
sshDir, err := c.sshDir()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return filepath.Glob(filepath.Join(sshDir, "*.pub"))
|
|
}
|
|
|
|
func (c *sshContext) findKeygen() (string, error) {
|
|
if c.keygenExe != "" {
|
|
return c.keygenExe, nil
|
|
}
|
|
|
|
keygenExe, err := safeexec.LookPath("ssh-keygen")
|
|
if err != nil && runtime.GOOS == "windows" {
|
|
// We can try and find ssh-keygen in a Git for Windows install
|
|
if gitPath, err := safeexec.LookPath("git"); err == nil {
|
|
gitKeygen := filepath.Join(filepath.Dir(gitPath), "..", "usr", "bin", "ssh-keygen.exe")
|
|
if _, err = os.Stat(gitKeygen); err == nil {
|
|
return gitKeygen, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
if err == nil {
|
|
c.keygenExe = keygenExe
|
|
}
|
|
return keygenExe, err
|
|
}
|
|
|
|
func (c *sshContext) generateSSHKey() (string, error) {
|
|
keygenExe, err := c.findKeygen()
|
|
if err != nil {
|
|
// give up silently if `ssh-keygen` is not available
|
|
return "", nil
|
|
}
|
|
|
|
var sshChoice bool
|
|
err = prompt.SurveyAskOne(&survey.Confirm{
|
|
Message: "Generate a new SSH key to add to your GitHub account?",
|
|
Default: true,
|
|
}, &sshChoice)
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not prompt: %w", err)
|
|
}
|
|
if !sshChoice {
|
|
return "", nil
|
|
}
|
|
|
|
sshDir, err := c.sshDir()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
keyFile := filepath.Join(sshDir, "id_ed25519")
|
|
if _, err := os.Stat(keyFile); err == nil {
|
|
return "", fmt.Errorf("refusing to overwrite file %s", keyFile)
|
|
}
|
|
|
|
if err := os.MkdirAll(filepath.Dir(keyFile), 0711); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var sshLabel string
|
|
var sshPassphrase string
|
|
err = prompt.SurveyAskOne(&survey.Password{
|
|
Message: "Enter a passphrase for your new SSH key (Optional)",
|
|
}, &sshPassphrase)
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not prompt: %w", err)
|
|
}
|
|
|
|
keygenCmd := exec.Command(keygenExe, "-t", "ed25519", "-C", sshLabel, "-N", sshPassphrase, "-f", keyFile)
|
|
return keyFile + ".pub", run.PrepareCmd(keygenCmd).Run()
|
|
}
|
|
|
|
func sshKeyUpload(httpClient *http.Client, hostname, keyFile string) error {
|
|
f, err := os.Open(keyFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
return add.SSHKeyUpload(httpClient, hostname, f, "GitHub CLI")
|
|
}
|