2022-07-25 23:05:44 +02:00
|
|
|
package controller
|
|
|
|
|
|
|
|
import (
|
2022-07-26 18:26:58 +02:00
|
|
|
"bytes"
|
|
|
|
"encoding/json"
|
2022-07-25 23:05:44 +02:00
|
|
|
"github.com/go-openapi/runtime/middleware"
|
2022-07-29 21:54:13 +02:00
|
|
|
"github.com/openziti-test-kitchen/zrok/controller/store"
|
2022-07-26 21:16:02 +02:00
|
|
|
"github.com/openziti-test-kitchen/zrok/rest_model_zrok"
|
2022-11-30 17:43:00 +01:00
|
|
|
"github.com/openziti-test-kitchen/zrok/rest_server_zrok/operations/environment"
|
2022-07-25 23:05:44 +02:00
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
)
|
|
|
|
|
2022-08-12 17:03:15 +02:00
|
|
|
type enableHandler struct {
|
|
|
|
}
|
|
|
|
|
2022-10-19 19:20:47 +02:00
|
|
|
func newEnableHandler() *enableHandler {
|
|
|
|
return &enableHandler{}
|
2022-08-12 17:03:15 +02:00
|
|
|
}
|
|
|
|
|
2022-11-30 17:43:00 +01:00
|
|
|
func (h *enableHandler) Handle(params environment.EnableParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
2022-07-29 22:03:53 +02:00
|
|
|
// start transaction early; if it fails, don't bother creating ziti resources
|
|
|
|
tx, err := str.Begin()
|
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("error starting transaction: %v", err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-29 22:03:53 +02:00
|
|
|
}
|
|
|
|
|
2022-10-19 19:20:47 +02:00
|
|
|
client, err := edgeClient()
|
2022-07-26 18:26:58 +02:00
|
|
|
if err != nil {
|
2022-07-26 22:21:49 +02:00
|
|
|
logrus.Errorf("error getting edge client: %v", err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-26 18:26:58 +02:00
|
|
|
}
|
2022-12-05 21:40:42 +01:00
|
|
|
ident, err := createEnvironmentIdentity(principal.Email, client)
|
2022-07-26 18:26:58 +02:00
|
|
|
if err != nil {
|
|
|
|
logrus.Error(err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-26 18:26:58 +02:00
|
|
|
}
|
2022-12-05 21:00:22 +01:00
|
|
|
envZId := ident.Payload.Data.ID
|
|
|
|
cfg, err := enrollIdentity(envZId, client)
|
2022-07-26 18:26:58 +02:00
|
|
|
if err != nil {
|
2022-07-26 22:21:49 +02:00
|
|
|
logrus.Error(err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-26 18:26:58 +02:00
|
|
|
}
|
2022-12-05 21:00:22 +01:00
|
|
|
if err := createEdgeRouterPolicy(envZId, envZId, client); err != nil {
|
2022-08-17 19:43:16 +02:00
|
|
|
logrus.Error(err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-08-17 19:43:16 +02:00
|
|
|
}
|
2022-08-03 20:25:27 +02:00
|
|
|
envId, err := str.CreateEnvironment(int(principal.ID), &store.Environment{
|
2022-10-19 18:24:43 +02:00
|
|
|
Description: params.Body.Description,
|
|
|
|
Host: params.Body.Host,
|
|
|
|
Address: realRemoteAddress(params.HTTPRequest),
|
2022-12-05 21:00:22 +01:00
|
|
|
ZId: envZId,
|
2022-08-03 20:25:27 +02:00
|
|
|
}, tx)
|
2022-07-29 21:54:13 +02:00
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("error storing created identity: %v", err)
|
|
|
|
_ = tx.Rollback()
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-29 21:54:13 +02:00
|
|
|
}
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
|
|
logrus.Errorf("error committing: %v", err)
|
2022-11-30 17:43:00 +01:00
|
|
|
return environment.NewEnableInternalServerError()
|
2022-07-29 21:54:13 +02:00
|
|
|
}
|
2022-11-08 21:07:18 +01:00
|
|
|
logrus.Infof("created environment for '%v', with ziti identity '%v', and database id '%v'", principal.Email, ident.Payload.Data.ID, envId)
|
2022-07-29 21:54:13 +02:00
|
|
|
|
2022-11-30 17:43:00 +01:00
|
|
|
resp := environment.NewEnableCreated().WithPayload(&rest_model_zrok.EnableResponse{
|
2022-12-05 21:00:22 +01:00
|
|
|
Identity: envZId,
|
2022-07-25 23:05:44 +02:00
|
|
|
})
|
2022-07-26 18:26:58 +02:00
|
|
|
|
|
|
|
var out bytes.Buffer
|
|
|
|
enc := json.NewEncoder(&out)
|
|
|
|
enc.SetEscapeHTML(false)
|
|
|
|
err = enc.Encode(&cfg)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
resp.Payload.Cfg = out.String()
|
|
|
|
|
|
|
|
return resp
|
|
|
|
}
|