c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cliclient
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/spf13/pflag"
|
|
|
|
kratos "github.com/ory/kratos/pkg/httpclient"
|
|
"github.com/ory/x/httpx"
|
|
)
|
|
|
|
const (
|
|
envKeyEndpoint = "KRATOS_ADMIN_URL"
|
|
FlagEndpoint = "endpoint"
|
|
)
|
|
|
|
type ContextKey int
|
|
|
|
const (
|
|
ClientContextKey ContextKey = iota + 1
|
|
)
|
|
|
|
type ClientContext struct {
|
|
Endpoint string
|
|
HTTPClient *http.Client
|
|
}
|
|
|
|
func NewClient(cmd *cobra.Command) (*kratos.APIClient, error) {
|
|
if f, ok := cmd.Context().Value(ClientContextKey).(func(cmd *cobra.Command) (*ClientContext, error)); ok {
|
|
cc, err := f(cmd)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conf := kratos.NewConfiguration()
|
|
conf.HTTPClient = cc.HTTPClient
|
|
conf.Servers = kratos.ServerConfigurations{{URL: cc.Endpoint}}
|
|
return kratos.NewAPIClient(conf), nil
|
|
} else if f != nil {
|
|
return nil, errors.Errorf("ClientContextKey was expected to be *client.OryKratos but it contained an invalid type %T ", f)
|
|
}
|
|
|
|
endpoint, err := cmd.Flags().GetString(FlagEndpoint)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
if endpoint == "" {
|
|
endpoint = os.Getenv(envKeyEndpoint)
|
|
}
|
|
|
|
if endpoint == "" {
|
|
return nil, errors.Errorf("you have to set the remote endpoint, try --help for details")
|
|
}
|
|
|
|
u, err := url.Parse(endpoint)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, `could not parse the endpoint URL "%s"`, endpoint)
|
|
}
|
|
|
|
conf := kratos.NewConfiguration()
|
|
conf.HTTPClient = httpx.NewResilientClient(
|
|
httpx.ResilientClientWithConnectionTimeout(10 * time.Second),
|
|
).StandardClient()
|
|
conf.Servers = kratos.ServerConfigurations{{URL: u.String()}}
|
|
return kratos.NewAPIClient(conf), nil
|
|
}
|
|
|
|
func RegisterClientFlags(flags *pflag.FlagSet) {
|
|
flags.StringP(FlagEndpoint, FlagEndpoint[:1], "", fmt.Sprintf("The URL of Ory Kratos' Admin API. Alternatively set using the %s environmental variable.", envKeyEndpoint))
|
|
}
|