mirror of
https://github.com/openziti/zrok.git
synced 2024-11-09 01:34:05 +01:00
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package zrokdir
|
|
|
|
import (
|
|
"github.com/go-openapi/runtime"
|
|
httptransport "github.com/go-openapi/runtime/client"
|
|
"github.com/go-openapi/strfmt"
|
|
"github.com/openziti-test-kitchen/zrok/build"
|
|
"github.com/openziti-test-kitchen/zrok/rest_client_zrok"
|
|
"github.com/pkg/errors"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func (zrd *ZrokDir) Client() (*rest_client_zrok.Zrok, error) {
|
|
apiEndpoint, _ := zrd.ApiEndpoint()
|
|
apiUrl, err := url.Parse(apiEndpoint)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "error parsing api endpoint '%v'", zrd)
|
|
}
|
|
transport := httptransport.New(apiUrl.Host, "/api/v1", []string{apiUrl.Scheme})
|
|
transport.Producers["application/zrok.v1+json"] = runtime.JSONProducer()
|
|
transport.Consumers["application/zrok.v1+json"] = runtime.JSONConsumer()
|
|
|
|
zrok := rest_client_zrok.New(transport, strfmt.Default)
|
|
v, err := zrok.Metadata.Version(nil)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "error getting version from api endpoint '%v': %v", apiEndpoint, err)
|
|
}
|
|
if !strings.HasPrefix(string(v.Payload), build.Series) {
|
|
return nil, errors.Errorf("expected a '%v' version, received: '%v'", build.Series, v.Payload)
|
|
}
|
|
|
|
return zrok, nil
|
|
}
|
|
|
|
func (zrd *ZrokDir) ApiEndpoint() (apiEndpoint string, from string) {
|
|
apiEndpoint = "https://api.zrok.io"
|
|
from = "binary"
|
|
|
|
if zrd.Cfg != nil && zrd.Cfg.ApiEndpoint != "" {
|
|
apiEndpoint = zrd.Cfg.ApiEndpoint
|
|
from = "config"
|
|
}
|
|
|
|
env := os.Getenv("ZROK_API_ENDPOINT")
|
|
if env != "" {
|
|
apiEndpoint = env
|
|
from = "ZROK_API_ENDPOINT"
|
|
}
|
|
|
|
if zrd.Env != nil && zrd.Env.ApiEndpoint != "" {
|
|
apiEndpoint = zrd.Env.ApiEndpoint
|
|
from = "env"
|
|
}
|
|
|
|
return apiEndpoint, from
|
|
}
|