mirror of
https://github.com/openziti/zrok.git
synced 2024-11-29 11:34:07 +01:00
89 lines
2.6 KiB
Go
89 lines
2.6 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"
|
|
"github.com/sirupsen/logrus"
|
|
"time"
|
|
)
|
|
|
|
type OauthOptions struct {
|
|
Provider string
|
|
EmailDomains []string
|
|
AuthorizationCheckInterval string
|
|
}
|
|
|
|
func CreateConfig(cfgTypeZId, envZId, shrToken string, authSchemeStr string, authUsers []*sdk.AuthUser, oauthOptions *OauthOptions, edge *rest_management_api_client.ZitiEdgeManagement) (cfgZId string, err error) {
|
|
authScheme, err := sdk.ParseAuthScheme(authSchemeStr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
cfg := &sdk.ProxyConfig{
|
|
AuthScheme: authScheme,
|
|
}
|
|
if cfg.AuthScheme == sdk.Basic {
|
|
cfg.BasicAuth = &sdk.BasicAuth{}
|
|
for _, authUser := range authUsers {
|
|
cfg.BasicAuth.Users = append(cfg.BasicAuth.Users, &sdk.AuthUser{Username: authUser.Username, Password: authUser.Password})
|
|
}
|
|
}
|
|
if cfg.AuthScheme == sdk.Oauth && oauthOptions != nil {
|
|
cfg.OauthAuth = &sdk.OauthAuth{
|
|
Provider: oauthOptions.Provider,
|
|
EmailDomains: oauthOptions.EmailDomains,
|
|
AuthorizationCheckInterval: oauthOptions.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
|
|
}
|