68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package add
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"github.com/cli/cli/v2/api"
|
|
"github.com/cli/cli/v2/internal/ghinstance"
|
|
)
|
|
|
|
var scopesError = errors.New("insufficient OAuth scopes")
|
|
|
|
func SSHKeyUpload(httpClient *http.Client, hostname string, keyFile io.Reader, title string) error {
|
|
url := ghinstance.RESTPrefix(hostname) + "user/keys"
|
|
|
|
keyBytes, err := ioutil.ReadAll(keyFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
payload := map[string]string{
|
|
"title": title,
|
|
"key": string(keyBytes),
|
|
}
|
|
|
|
payloadBytes, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(payloadBytes))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
resp, err := httpClient.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode == 404 {
|
|
return scopesError
|
|
} else if resp.StatusCode > 299 {
|
|
var httpError api.HTTPError
|
|
err := api.HandleHTTPError(resp)
|
|
if errors.As(err, &httpError) && isDuplicateError(&httpError) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
_, err = io.Copy(ioutil.Discard, resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func isDuplicateError(err *api.HTTPError) bool {
|
|
return err.StatusCode == 422 && len(err.Errors) == 1 &&
|
|
err.Errors[0].Field == "key" && err.Errors[0].Message == "key is already in use"
|
|
}
|