mirror of
https://github.com/openziti/zrok.git
synced 2024-11-07 16:54:23 +01:00
92 lines
3.0 KiB
Go
92 lines
3.0 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/go-openapi/runtime/middleware"
|
|
"github.com/openziti/zrok/controller/store"
|
|
"github.com/openziti/zrok/rest_model_zrok"
|
|
"github.com/openziti/zrok/rest_server_zrok/operations/metadata"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func overviewHandler(_ metadata.OverviewParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
|
tx, err := str.Begin()
|
|
if err != nil {
|
|
logrus.Errorf("error starting transaction: %v", err)
|
|
return metadata.NewOverviewInternalServerError()
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
envs, err := str.FindEnvironmentsForAccount(int(principal.ID), tx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding environments for '%v': %v", principal.Email, err)
|
|
return metadata.NewOverviewInternalServerError()
|
|
}
|
|
var envShrsList rest_model_zrok.EnvironmentSharesList
|
|
for _, env := range envs {
|
|
shrs, err := str.FindSharesForEnvironment(env.Id, tx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding shares for environment '%v': %v", env.ZId, err)
|
|
return metadata.NewOverviewInternalServerError()
|
|
}
|
|
envShrs := &rest_model_zrok.EnvironmentShares{
|
|
Environment: &rest_model_zrok.Environment{
|
|
Address: env.Address,
|
|
CreatedAt: env.CreatedAt.UnixMilli(),
|
|
Description: env.Description,
|
|
Host: env.Host,
|
|
UpdatedAt: env.UpdatedAt.UnixMilli(),
|
|
ZID: env.ZId,
|
|
},
|
|
}
|
|
var shrIds []int
|
|
for i := range shrs {
|
|
shrIds = append(shrIds, shrs[i].Id)
|
|
}
|
|
shrsLimited, err := str.FindSelectedLatestShareLimitjournal(shrIds, tx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding limited shares for environment '%v': %v", env.ZId, err)
|
|
return metadata.NewOverviewInternalServerError()
|
|
}
|
|
shrsLimitedMap := make(map[int]store.LimitJournalAction)
|
|
for i := range shrsLimited {
|
|
shrsLimitedMap[shrsLimited[i].ShareId] = shrsLimited[i].Action
|
|
}
|
|
for _, shr := range shrs {
|
|
feEndpoint := ""
|
|
if shr.FrontendEndpoint != nil {
|
|
feEndpoint = *shr.FrontendEndpoint
|
|
}
|
|
feSelection := ""
|
|
if shr.FrontendSelection != nil {
|
|
feSelection = *shr.FrontendSelection
|
|
}
|
|
beProxyEndpoint := ""
|
|
if shr.BackendProxyEndpoint != nil {
|
|
beProxyEndpoint = *shr.BackendProxyEndpoint
|
|
}
|
|
envShr := &rest_model_zrok.Share{
|
|
Token: shr.Token,
|
|
ZID: shr.ZId,
|
|
ShareMode: shr.ShareMode,
|
|
BackendMode: shr.BackendMode,
|
|
FrontendSelection: feSelection,
|
|
FrontendEndpoint: feEndpoint,
|
|
BackendProxyEndpoint: beProxyEndpoint,
|
|
Reserved: shr.Reserved,
|
|
CreatedAt: shr.CreatedAt.UnixMilli(),
|
|
UpdatedAt: shr.UpdatedAt.UnixMilli(),
|
|
}
|
|
if action, found := shrsLimitedMap[shr.Id]; found {
|
|
if action == store.LimitAction {
|
|
envShr.Limited = true
|
|
}
|
|
}
|
|
envShrs.Shares = append(envShrs.Shares, envShr)
|
|
}
|
|
envShrsList = append(envShrsList, envShrs)
|
|
}
|
|
return metadata.NewOverviewOK().WithPayload(&rest_model_zrok.Overview{
|
|
AccountLimited: false,
|
|
Environments: envShrsList,
|
|
})
|
|
}
|