mirror of
https://github.com/openziti/zrok.git
synced 2024-11-22 08:03:49 +01:00
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package zrokEdgeSdk
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/openziti/edge-api/rest_management_api_client"
|
|
"github.com/openziti/edge-api/rest_management_api_client/config"
|
|
"github.com/openziti/edge-api/rest_model"
|
|
"github.com/openziti/zrok/sdk/golang/sdk"
|
|
"github.com/sirupsen/logrus"
|
|
"time"
|
|
)
|
|
|
|
type FrontendOptions struct {
|
|
AuthScheme sdk.AuthScheme
|
|
BasicAuthUsers []*sdk.AuthUserConfig
|
|
Oauth *sdk.OauthConfig
|
|
}
|
|
|
|
func CreateConfig(cfgTypeZId, envZId, shrToken string, options *FrontendOptions, edge *rest_management_api_client.ZitiEdgeManagement) (cfgZId string, err error) {
|
|
cfg := &sdk.FrontendConfig{
|
|
AuthScheme: options.AuthScheme,
|
|
}
|
|
if cfg.AuthScheme == sdk.Basic {
|
|
cfg.BasicAuth = &sdk.BasicAuthConfig{}
|
|
for _, authUser := range options.BasicAuthUsers {
|
|
cfg.BasicAuth.Users = append(cfg.BasicAuth.Users, &sdk.AuthUserConfig{Username: authUser.Username, Password: authUser.Password})
|
|
}
|
|
}
|
|
if cfg.AuthScheme == sdk.Oauth && options.Oauth != nil {
|
|
cfg.OauthAuth = &sdk.OauthConfig{
|
|
Provider: options.Oauth.Provider,
|
|
EmailDomains: options.Oauth.EmailDomains,
|
|
AuthorizationCheckInterval: options.Oauth.AuthorizationCheckInterval,
|
|
}
|
|
}
|
|
cfgCrt := &rest_model.ConfigCreate{
|
|
ConfigTypeID: &cfgTypeZId,
|
|
Data: cfg,
|
|
Name: &shrToken,
|
|
Tags: ZrokShareTags(shrToken),
|
|
}
|
|
cfgReq := &config.CreateConfigParams{
|
|
Config: cfgCrt,
|
|
Context: context.Background(),
|
|
}
|
|
cfgReq.SetTimeout(30 * time.Second)
|
|
cfgResp, err := edge.Config.CreateConfig(cfgReq, nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
logrus.Infof("created config '%v' for environment '%v'", cfgResp.Payload.Data.ID, envZId)
|
|
return cfgResp.Payload.Data.ID, nil
|
|
}
|
|
|
|
func DeleteConfig(envZId, shrToken string, edge *rest_management_api_client.ZitiEdgeManagement) error {
|
|
filter := fmt.Sprintf("tags.zrokShareToken=\"%v\"", shrToken)
|
|
limit := int64(0)
|
|
offset := int64(0)
|
|
listReq := &config.ListConfigsParams{
|
|
Filter: &filter,
|
|
Limit: &limit,
|
|
Offset: &offset,
|
|
Context: context.Background(),
|
|
}
|
|
listReq.SetTimeout(30 * time.Second)
|
|
listResp, err := edge.Config.ListConfigs(listReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, cfg := range listResp.Payload.Data {
|
|
deleteReq := &config.DeleteConfigParams{
|
|
ID: *cfg.ID,
|
|
Context: context.Background(),
|
|
}
|
|
deleteReq.SetTimeout(30 * time.Second)
|
|
_, err := edge.Config.DeleteConfig(deleteReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logrus.Infof("deleted config '%v' for '%v'", *cfg.ID, envZId)
|
|
}
|
|
return nil
|
|
}
|