zrok/controller/unshare.go

143 lines
4.3 KiB
Go
Raw Normal View History

2022-07-27 18:17:37 +02:00
package controller
import (
"context"
"fmt"
"github.com/go-openapi/runtime/middleware"
2022-08-01 21:44:26 +02:00
"github.com/openziti-test-kitchen/zrok/controller/store"
2022-12-14 20:40:45 +01:00
"github.com/openziti-test-kitchen/zrok/controller/zrokEdgeSdk"
2022-07-27 19:38:35 +02:00
"github.com/openziti-test-kitchen/zrok/rest_model_zrok"
"github.com/openziti-test-kitchen/zrok/rest_server_zrok/operations/share"
2022-07-27 18:17:37 +02:00
"github.com/openziti/edge/rest_management_api_client"
edge_service "github.com/openziti/edge/rest_management_api_client/service"
2022-08-01 21:44:26 +02:00
"github.com/pkg/errors"
2022-07-27 18:17:37 +02:00
"github.com/sirupsen/logrus"
"time"
)
type unshareHandler struct{}
2022-08-12 17:03:15 +02:00
func newUnshareHandler() *unshareHandler {
return &unshareHandler{}
2022-08-12 17:03:15 +02:00
}
func (h *unshareHandler) Handle(params share.UnshareParams, principal *rest_model_zrok.Principal) middleware.Responder {
2022-08-01 21:44:26 +02:00
tx, err := str.Begin()
if err != nil {
logrus.Errorf("error starting transaction: %v", err)
return share.NewUnshareInternalServerError()
2022-08-01 21:44:26 +02:00
}
defer func() { _ = tx.Rollback() }()
edge, err := edgeClient()
2022-07-27 18:17:37 +02:00
if err != nil {
logrus.Error(err)
return share.NewUnshareInternalServerError()
2022-07-27 18:17:37 +02:00
}
shrToken := params.Body.ShrToken
2023-01-04 20:42:58 +01:00
shrZId, err := h.findShareZId(shrToken, edge)
2022-08-01 21:44:26 +02:00
if err != nil {
logrus.Error(err)
return share.NewUnshareNotFound()
2022-08-01 21:44:26 +02:00
}
2022-08-03 20:58:11 +02:00
var senv *store.Environment
if envs, err := str.FindEnvironmentsForAccount(int(principal.ID), tx); err == nil {
for _, env := range envs {
if env.ZId == params.Body.EnvZID {
2022-08-03 20:58:11 +02:00
senv = env
break
}
}
if senv == nil {
err := errors.Errorf("environment with id '%v' not found for '%v", params.Body.EnvZID, principal.Email)
2022-08-03 20:58:11 +02:00
logrus.Error(err)
return share.NewUnshareNotFound()
2022-08-03 20:58:11 +02:00
}
} else {
2022-09-09 16:20:05 +02:00
logrus.Errorf("error finding environments for account '%v': %v", principal.Email, err)
return share.NewUnshareNotFound()
2022-08-03 20:58:11 +02:00
}
var sshr *store.Share
2023-01-04 20:21:23 +01:00
if shrs, err := str.FindSharesForEnvironment(senv.Id, tx); err == nil {
for _, shr := range shrs {
2023-01-04 20:42:58 +01:00
if shr.ZId == shrZId {
2023-01-04 20:21:23 +01:00
sshr = shr
2022-08-01 21:44:26 +02:00
break
}
}
if sshr == nil {
2023-01-04 20:42:58 +01:00
err := errors.Errorf("share with id '%v' not found for '%v'", shrZId, principal.Email)
2022-08-01 21:44:26 +02:00
logrus.Error(err)
return share.NewUnshareNotFound()
2022-08-01 21:44:26 +02:00
}
} else {
2023-01-04 20:21:23 +01:00
logrus.Errorf("error finding shares for account '%v': %v", principal.Email, err)
return share.NewUnshareInternalServerError()
2022-08-01 21:44:26 +02:00
}
if sshr.Reserved == params.Body.Reserved {
2023-01-04 20:21:23 +01:00
// single tag-based share deallocator; should work regardless of sharing mode
2023-01-04 20:42:58 +01:00
if err := h.deallocateResources(senv, shrToken, shrZId, edge); err != nil {
logrus.Errorf("error unsharing ziti resources for '%v': %v", sshr, err)
return share.NewUnshareInternalServerError()
}
2022-07-27 18:17:37 +02:00
2023-01-04 20:21:23 +01:00
logrus.Debugf("deallocated share '%v'", shrToken)
if err := str.DeleteShare(sshr.Id, tx); err != nil {
2023-01-04 20:42:58 +01:00
logrus.Errorf("error deactivating share '%v': %v", shrZId, err)
return share.NewUnshareInternalServerError()
}
if err := tx.Commit(); err != nil {
2023-01-04 20:42:58 +01:00
logrus.Errorf("error committing transaction for '%v': %v", shrZId, err)
return share.NewUnshareInternalServerError()
}
} else {
2023-01-04 20:21:23 +01:00
logrus.Infof("share '%v' is reserved, skipping deallocation", shrToken)
}
return share.NewUnshareOK()
2022-07-27 18:17:37 +02:00
}
2023-01-04 20:42:58 +01:00
func (h *unshareHandler) findShareZId(shrToken string, edge *rest_management_api_client.ZitiEdgeManagement) (string, error) {
2023-01-04 20:21:23 +01:00
filter := fmt.Sprintf("name=\"%v\"", shrToken)
2022-08-01 21:44:26 +02:00
limit := int64(1)
offset := int64(0)
listReq := &edge_service.ListServicesParams{
2022-08-01 21:44:26 +02:00
Filter: &filter,
Limit: &limit,
Offset: &offset,
Context: context.Background(),
}
listReq.SetTimeout(30 * time.Second)
listResp, err := edge.Service.ListServices(listReq, nil)
if err != nil {
return "", err
}
if len(listResp.Payload.Data) == 1 {
return *(listResp.Payload.Data[0].ID), nil
}
2023-01-04 20:21:23 +01:00
return "", errors.Errorf("share '%v' not found", shrToken)
2022-08-01 21:44:26 +02:00
}
2022-11-22 20:24:41 +01:00
2023-01-04 20:42:58 +01:00
func (h *unshareHandler) deallocateResources(senv *store.Environment, shrToken, shrZId string, edge *rest_management_api_client.ZitiEdgeManagement) error {
2023-01-04 20:21:23 +01:00
if err := zrokEdgeSdk.DeleteServiceEdgeRouterPolicy(senv.ZId, shrToken, edge); err != nil {
2022-11-22 20:24:41 +01:00
return err
}
2023-01-04 20:21:23 +01:00
if err := zrokEdgeSdk.DeleteServicePolicyDial(senv.ZId, shrToken, edge); err != nil {
2022-11-22 20:24:41 +01:00
return err
}
2023-01-04 20:21:23 +01:00
if err := zrokEdgeSdk.DeleteServicePolicyBind(senv.ZId, shrToken, edge); err != nil {
2022-11-22 20:24:41 +01:00
return err
}
2023-01-04 20:21:23 +01:00
if err := zrokEdgeSdk.DeleteConfig(senv.ZId, shrToken, edge); err != nil {
2022-11-22 20:24:41 +01:00
return err
}
2023-01-04 20:42:58 +01:00
if err := zrokEdgeSdk.DeleteService(senv.ZId, shrZId, edge); err != nil {
2022-11-22 20:24:41 +01:00
return err
}
return nil
}