99 lines
2 KiB
Go
99 lines
2 KiB
Go
package shared
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/cli/cli/v2/api"
|
|
"github.com/cli/cli/v2/internal/ghinstance"
|
|
)
|
|
|
|
const (
|
|
// AuthenticationKey represents an SSH authentication key type.
|
|
AuthenticationKey = "authentication"
|
|
// SigningKey represents an SSH signing key type.
|
|
SigningKey = "signing"
|
|
)
|
|
|
|
type sshKey struct {
|
|
ID int
|
|
Key string
|
|
Title string
|
|
Type string
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// UserKeys fetches the user's SSH authentication keys.
|
|
func UserKeys(httpClient *http.Client, host, userHandle string) ([]sshKey, error) {
|
|
resource := "user/keys"
|
|
if userHandle != "" {
|
|
resource = fmt.Sprintf("users/%s/keys", userHandle)
|
|
}
|
|
url := fmt.Sprintf("%s%s?per_page=%d", ghinstance.RESTPrefix(host), resource, 100)
|
|
|
|
keys, err := getUserKeys(httpClient, url)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i := 0; i < len(keys); i++ {
|
|
keys[i].Type = AuthenticationKey
|
|
}
|
|
|
|
return keys, nil
|
|
}
|
|
|
|
// UserSigningKeys fetches the user's SSH signing keys.
|
|
func UserSigningKeys(httpClient *http.Client, host, userHandle string) ([]sshKey, error) {
|
|
resource := "user/ssh_signing_keys"
|
|
if userHandle != "" {
|
|
resource = fmt.Sprintf("users/%s/ssh_signing_keys", userHandle)
|
|
}
|
|
url := fmt.Sprintf("%s%s?per_page=%d", ghinstance.RESTPrefix(host), resource, 100)
|
|
|
|
keys, err := getUserKeys(httpClient, url)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i := 0; i < len(keys); i++ {
|
|
keys[i].Type = SigningKey
|
|
}
|
|
|
|
return keys, nil
|
|
}
|
|
|
|
func getUserKeys(httpClient *http.Client, url string) ([]sshKey, error) {
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode > 299 {
|
|
return nil, api.HandleHTTPError(resp)
|
|
}
|
|
|
|
b, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var keys []sshKey
|
|
err = json.Unmarshal(b, &keys)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return keys, nil
|
|
}
|