mirror of
https://github.com/openziti/zrok.git
synced 2025-06-30 14:30:30 +02:00
/agent/status (#967)
This commit is contained in:
75
controller/agentRemoteStatus.go
Normal file
75
controller/agentRemoteStatus.go
Normal file
@ -0,0 +1,75 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/go-openapi/runtime/middleware"
|
||||
"github.com/openziti/zrok/agent/agentGrpc"
|
||||
"github.com/openziti/zrok/controller/agentController"
|
||||
"github.com/openziti/zrok/rest_model_zrok"
|
||||
"github.com/openziti/zrok/rest_server_zrok/operations/agent"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type agentRemoteStatusHandler struct{}
|
||||
|
||||
func newAgentRemoteStatusHandler() *agentRemoteStatusHandler {
|
||||
return &agentRemoteStatusHandler{}
|
||||
}
|
||||
|
||||
func (h *agentRemoteStatusHandler) Handle(params agent.RemoteStatusParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
||||
trx, err := str.Begin()
|
||||
if err != nil {
|
||||
logrus.Errorf("error starting transaction for '%v': %v", principal.Email, err)
|
||||
return agent.NewRemoteStatusInternalServerError()
|
||||
}
|
||||
defer trx.Rollback()
|
||||
|
||||
env, err := str.FindEnvironmentForAccount(params.Body.EnvZID, int(principal.ID), trx)
|
||||
if err != nil {
|
||||
logrus.Errorf("error finding environment '%v' for '%v' (%v)", params.Body.EnvZID, principal.Email, err)
|
||||
return agent.NewRemoteStatusUnauthorized()
|
||||
}
|
||||
|
||||
ae, err := str.FindAgentEnrollmentForEnvironment(env.Id, trx)
|
||||
if err != nil {
|
||||
logrus.Errorf("error finding agent enrollment for environment '%v' (%v): %v", params.Body.EnvZID, principal.Email, err)
|
||||
return agent.NewRemoteStatusBadGateway()
|
||||
}
|
||||
|
||||
acli, aconn, err := agentController.NewAgentClient(ae.Token, cfg.AgentController)
|
||||
if err != nil {
|
||||
logrus.Errorf("error creating agent client for '%v' (%v): %v", params.Body.EnvZID, principal.Email, err)
|
||||
return agent.NewRemoteStatusInternalServerError()
|
||||
}
|
||||
defer aconn.Close()
|
||||
|
||||
resp, err := acli.Status(context.Background(), &agentGrpc.StatusRequest{})
|
||||
if err != nil {
|
||||
logrus.Errorf("error retrieving remote agent status for '%v' (%v): %v", params.Body.EnvZID, principal.Email, err)
|
||||
return agent.NewRemoteStatusBadGateway()
|
||||
}
|
||||
|
||||
out := &agent.RemoteStatusOKBody{}
|
||||
for _, share := range resp.Shares {
|
||||
out.Shares = append(out.Shares, &agent.RemoteStatusOKBodySharesItems0{
|
||||
BackendEndpoint: share.BackendEndpoint,
|
||||
BackendMode: share.BackendMode,
|
||||
FrontendEndpoints: share.FrontendEndpoint,
|
||||
Open: !share.Closed,
|
||||
Reserved: share.Reserved,
|
||||
ShareMode: share.ShareMode,
|
||||
Status: share.Status,
|
||||
Token: share.Token,
|
||||
})
|
||||
}
|
||||
for _, access := range resp.Accesses {
|
||||
out.Accesses = append(out.Accesses, &agent.RemoteStatusOKBodyAccessesItems0{
|
||||
BindAddress: access.BindAddress,
|
||||
FrontendToken: access.FrontendToken,
|
||||
ResponseHeaders: access.ResponseHeaders,
|
||||
Token: access.Token,
|
||||
})
|
||||
}
|
||||
|
||||
return agent.NewRemoteStatusOK().WithPayload(out)
|
||||
}
|
@ -69,6 +69,7 @@ func Run(inCfg *config.Config) error {
|
||||
api.AgentEnrollHandler = newAgentEnrollHandler()
|
||||
api.AgentPingHandler = newAgentPingHandler()
|
||||
api.AgentRemoteShareHandler = newAgentRemoteShareHandler()
|
||||
api.AgentRemoteStatusHandler = newAgentRemoteStatusHandler()
|
||||
api.AgentRemoteUnshareHandler = newAgentRemoteUnshareHandler()
|
||||
api.AgentUnenrollHandler = newAgentUnenrollHandler()
|
||||
}
|
||||
|
Reference in New Issue
Block a user