mirror of
https://github.com/openziti/zrok.git
synced 2025-06-10 03:46:57 +02:00
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/openziti/edge-api/rest_management_api_client"
|
|
apiConfig "github.com/openziti/edge-api/rest_management_api_client/config"
|
|
"github.com/openziti/edge-api/rest_management_api_client/edge_router_policy"
|
|
"github.com/openziti/edge-api/rest_management_api_client/identity"
|
|
"github.com/openziti/zrok/controller/config"
|
|
"github.com/openziti/zrok/controller/zrokEdgeSdk"
|
|
"github.com/openziti/zrok/sdk/golang/sdk"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func Unbootstrap(cfg *config.Config) error {
|
|
edge, err := zrokEdgeSdk.Client(cfg.Ziti)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := unbootstrapEdgeRouterPolicies(edge); err != nil {
|
|
logrus.Errorf("error unbootstrapping edge router policies: %v", err)
|
|
}
|
|
if err := unbootstrapIdentities(edge); err != nil {
|
|
logrus.Errorf("error unbootstrapping identities: %v", err)
|
|
}
|
|
if err := unbootstrapConfigs(edge); err != nil {
|
|
logrus.Errorf("error unbootrapping configs: %v", err)
|
|
}
|
|
if err := unbootstrapConfigType(edge); err != nil {
|
|
logrus.Errorf("error unbootstrapping config type: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func unbootstrapEdgeRouterPolicies(edge *rest_management_api_client.ZitiEdgeManagement) error {
|
|
filter := "tags.zrok != null"
|
|
limit := int64(100)
|
|
offset := int64(0)
|
|
listReq := &edge_router_policy.ListEdgeRouterPoliciesParams{
|
|
Filter: &filter,
|
|
Limit: &limit,
|
|
Offset: &offset,
|
|
Context: context.Background(),
|
|
}
|
|
listResp, err := edge.EdgeRouterPolicy.ListEdgeRouterPolicies(listReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, erp := range listResp.Payload.Data {
|
|
logrus.Infof("found edge router policy: %v", *erp.ID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func unbootstrapIdentities(edge *rest_management_api_client.ZitiEdgeManagement) error {
|
|
filter := "tags.zrok != null"
|
|
limit := int64(100)
|
|
offset := int64(0)
|
|
listReq := &identity.ListIdentitiesParams{
|
|
Filter: &filter,
|
|
Limit: &limit,
|
|
Offset: &offset,
|
|
Context: context.Background(),
|
|
}
|
|
listResp, err := edge.Identity.ListIdentities(listReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, identity := range listResp.Payload.Data {
|
|
logrus.Infof("found identity: %v", *identity.ID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func unbootstrapConfigs(edge *rest_management_api_client.ZitiEdgeManagement) error {
|
|
filter := "tags.zrok != null"
|
|
limit := int64(100)
|
|
offset := int64(0)
|
|
listReq := &apiConfig.ListConfigsParams{
|
|
Filter: &filter,
|
|
Limit: &limit,
|
|
Offset: &offset,
|
|
Context: context.Background(),
|
|
}
|
|
listResp, err := edge.Config.ListConfigs(listReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, listCfg := range listResp.Payload.Data {
|
|
logrus.Infof("found config: %v", *listCfg.ID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func unbootstrapConfigType(edge *rest_management_api_client.ZitiEdgeManagement) error {
|
|
filter := fmt.Sprintf("name = \"%v\"", sdk.ZrokProxyConfig)
|
|
limit := int64(100)
|
|
offset := int64(0)
|
|
listReq := &apiConfig.ListConfigTypesParams{
|
|
Filter: &filter,
|
|
Limit: &limit,
|
|
Offset: &offset,
|
|
Context: context.Background(),
|
|
}
|
|
listResp, err := edge.Config.ListConfigTypes(listReq, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, listCfgType := range listResp.Payload.Data {
|
|
logrus.Infof("found config type: %v (%v)", *listCfgType.ID, *listCfgType.Name)
|
|
}
|
|
return nil
|
|
}
|