133 lines
2.9 KiB
Go
133 lines
2.9 KiB
Go
package list
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/cli/cli/v2/api"
|
|
"github.com/cli/cli/v2/internal/config"
|
|
"github.com/cli/cli/v2/internal/tableprinter"
|
|
"github.com/cli/cli/v2/internal/text"
|
|
"github.com/cli/cli/v2/pkg/cmd/ssh-key/shared"
|
|
"github.com/cli/cli/v2/pkg/cmdutil"
|
|
"github.com/cli/cli/v2/pkg/iostreams"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type ListOptions struct {
|
|
IO *iostreams.IOStreams
|
|
Config func() (config.Config, error)
|
|
HTTPClient func() (*http.Client, error)
|
|
}
|
|
|
|
func NewCmdList(f *cmdutil.Factory, runF func(*ListOptions) error) *cobra.Command {
|
|
opts := &ListOptions{
|
|
IO: f.IOStreams,
|
|
Config: f.Config,
|
|
HTTPClient: f.HttpClient,
|
|
}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "list",
|
|
Short: "Lists SSH keys in your GitHub account",
|
|
Aliases: []string{"ls"},
|
|
Args: cobra.ExactArgs(0),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if runF != nil {
|
|
return runF(opts)
|
|
}
|
|
return listRun(opts)
|
|
},
|
|
}
|
|
|
|
return cmd
|
|
}
|
|
|
|
func listRun(opts *ListOptions) error {
|
|
apiClient, err := opts.HTTPClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cfg, err := opts.Config()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
host, _ := cfg.Authentication().DefaultHost()
|
|
sshAuthKeys, authKeyErr := shared.UserKeys(apiClient, host, "")
|
|
if authKeyErr != nil {
|
|
printError(opts.IO.ErrOut, authKeyErr)
|
|
}
|
|
|
|
sshSigningKeys, signKeyErr := shared.UserSigningKeys(apiClient, host, "")
|
|
if signKeyErr != nil {
|
|
printError(opts.IO.ErrOut, signKeyErr)
|
|
}
|
|
|
|
if authKeyErr != nil && signKeyErr != nil {
|
|
return cmdutil.SilentError
|
|
}
|
|
|
|
sshKeys := append(sshAuthKeys, sshSigningKeys...)
|
|
|
|
if len(sshKeys) == 0 {
|
|
return cmdutil.NewNoResultsError("no SSH keys present in the GitHub account")
|
|
}
|
|
|
|
t := tableprinter.New(opts.IO, tableprinter.WithHeader("TITLE", "ID", "KEY", "TYPE", "ADDED"))
|
|
cs := opts.IO.ColorScheme()
|
|
now := time.Now()
|
|
|
|
for _, sshKey := range sshKeys {
|
|
id := strconv.Itoa(sshKey.ID)
|
|
createdAt := sshKey.CreatedAt.Format(time.RFC3339)
|
|
|
|
if t.IsTTY() {
|
|
t.AddField(sshKey.Title)
|
|
t.AddField(id)
|
|
t.AddField(sshKey.Key, tableprinter.WithTruncate(truncateMiddle))
|
|
t.AddField(sshKey.Type)
|
|
t.AddField(text.FuzzyAgoAbbr(now, sshKey.CreatedAt), tableprinter.WithColor(cs.Gray))
|
|
} else {
|
|
t.AddField(sshKey.Title)
|
|
t.AddField(sshKey.Key)
|
|
t.AddField(createdAt)
|
|
t.AddField(id)
|
|
t.AddField(sshKey.Type)
|
|
}
|
|
|
|
t.EndRow()
|
|
}
|
|
|
|
return t.Render()
|
|
}
|
|
|
|
func truncateMiddle(maxWidth int, t string) string {
|
|
if len(t) <= maxWidth {
|
|
return t
|
|
}
|
|
|
|
ellipsis := "..."
|
|
if maxWidth < len(ellipsis)+2 {
|
|
return t[0:maxWidth]
|
|
}
|
|
|
|
halfWidth := (maxWidth - len(ellipsis)) / 2
|
|
remainder := (maxWidth - len(ellipsis)) % 2
|
|
return t[0:halfWidth+remainder] + ellipsis + t[len(t)-halfWidth:]
|
|
}
|
|
|
|
func printError(w io.Writer, err error) {
|
|
fmt.Fprintln(w, "warning: ", err)
|
|
var httpErr api.HTTPError
|
|
if errors.As(err, &httpErr) {
|
|
if msg := httpErr.ScopesSuggestion(); msg != "" {
|
|
fmt.Fprintln(w, msg)
|
|
}
|
|
}
|
|
}
|