82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package socket
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/FAU-CDI/wisski-distillery/internal/dis/component/server/admin/socket/actions"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/dis/component/server/admin/socket/proto"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
func (sockets *Sockets) Actions(ctx context.Context) proto.ActionMap {
|
|
logger := zerolog.Ctx(ctx)
|
|
actions := make(proto.ActionMap, len(sockets.dependencies.Actions)+len(sockets.dependencies.IActions))
|
|
|
|
// setup basic actions
|
|
for _, a := range sockets.dependencies.Actions {
|
|
name, action := sockets.regularAction(a)
|
|
if _, ok := actions[name]; ok {
|
|
logger.Warn().Str("name", name).Str("type", "regular").Msg("duplicate websocket action")
|
|
}
|
|
actions[name] = action
|
|
|
|
logger.Info().
|
|
Str("name", name).
|
|
Str("type", "regular").
|
|
Int("params", action.NumParams).
|
|
Str("scope", string(action.Scope)).
|
|
Str("scopeParam", action.ScopeParam).
|
|
Msg("registering websocket action")
|
|
}
|
|
|
|
// setup instance actions
|
|
for _, a := range sockets.dependencies.IActions {
|
|
name, action := sockets.instanceAction(a)
|
|
if _, ok := actions[name]; ok {
|
|
zerolog.Ctx(ctx).Warn().Str("name", name).Str("type", "instance").Msg("duplicate websocket action")
|
|
}
|
|
actions[name] = action
|
|
|
|
logger.Info().
|
|
Str("name", name).
|
|
Str("type", "instance").
|
|
Int("params", action.NumParams-1).
|
|
Str("scope", string(action.Scope)).
|
|
Str("scopeParam", action.ScopeParam).
|
|
Msg("registering websocket action")
|
|
}
|
|
|
|
return actions
|
|
}
|
|
|
|
func (sockets *Sockets) regularAction(a actions.WebsocketAction) (name string, action proto.Action) {
|
|
meta := a.Action()
|
|
|
|
return meta.Name, proto.Action{
|
|
NumParams: meta.NumParams,
|
|
Scope: meta.Scope,
|
|
ScopeParam: meta.ScopeParam,
|
|
|
|
Handle: a.Act,
|
|
}
|
|
}
|
|
|
|
func (sockets *Sockets) instanceAction(a actions.WebsocketInstanceAction) (name string, action proto.Action) {
|
|
meta := a.Action()
|
|
|
|
return meta.Name, proto.Action{
|
|
NumParams: meta.NumParams + 1,
|
|
Scope: meta.Scope,
|
|
ScopeParam: meta.ScopeParam,
|
|
|
|
Handle: func(ctx context.Context, in io.Reader, out io.Writer, params ...string) error {
|
|
instance, err := sockets.dependencies.Instances.WissKI(ctx, params[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return a.Act(ctx, instance, in, out, params[1:]...)
|
|
},
|
|
}
|
|
|
|
}
|