mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-08-28 02:44:59 +00:00
f6a767befb
This commit moves the cli out of the `ctl` package and into its own
`cli` package. It also adds some basic tests for expected input.
Finally, it fixes a bug which was causing intentional line feeds to be
ignored, which was a problem with the BULK INSERT command.
(cherry picked from commit cf72bfa16f)
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package fbcloud
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
authFlow = "USER_PASSWORD_AUTH"
|
|
cognitoURLTemplate = "https://cognito-idp.%s.amazonaws.com"
|
|
)
|
|
|
|
type cognitoParameters struct {
|
|
Email string `json:"USERNAME"`
|
|
Password string `json:"PASSWORD"`
|
|
}
|
|
|
|
type cognitoAuthRequest struct {
|
|
AuthParameters cognitoParameters `json:"AuthParameters"`
|
|
AuthFlow string `json:"AuthFlow"`
|
|
AppClientID string `json:"ClientId"`
|
|
}
|
|
|
|
type cognitoAuthResult struct {
|
|
IDToken string `json:"IdToken"`
|
|
}
|
|
|
|
type cognitoAuthResponse struct {
|
|
Result cognitoAuthResult `json:"AuthenticationResult"`
|
|
}
|
|
|
|
func authenticate(clientID, region, email, password string) (string, error) {
|
|
authPayload := cognitoAuthRequest{
|
|
AuthParameters: cognitoParameters{
|
|
Email: email,
|
|
Password: password,
|
|
},
|
|
AuthFlow: authFlow,
|
|
AppClientID: clientID,
|
|
}
|
|
|
|
data, err := json.Marshal(authPayload)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "marshaling json")
|
|
}
|
|
|
|
url := fmt.Sprintf(cognitoURLTemplate, region)
|
|
req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer(data))
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "creating authentication request object")
|
|
}
|
|
req.Header.Add("Content-Type", "application/x-amz-json-1.1")
|
|
req.Header.Add("X-Amz-Target", "AWSCognitoIdentityProviderService.InitiateAuth")
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "making request")
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
fullbod, err := io.ReadAll(resp.Body)
|
|
|
|
if resp.StatusCode != http.StatusOK || err != nil {
|
|
return "", errors.Errorf("HTTP status code=%d from Cognito authentication response. reading body: %v, body: '%s'", resp.StatusCode, err, fullbod)
|
|
}
|
|
|
|
var auth cognitoAuthResponse
|
|
err = json.Unmarshal(fullbod, &auth)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "decoding cognito auth response")
|
|
}
|
|
|
|
return auth.Result.IDToken, nil
|
|
}
|