Pass partial octant state to JS plugins

[vmware-archive#1512, vmware-archive#1513] Signed-off-by: vikram yadav <vyadav@pivotal.io>
xtreme-vikram-yadav · Mar 29, 2021 · feed139 · feed139
1 parent 559d039
commit feed139
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 1 deletion.
diff --git a/internal/api/action_request_manager.go b/internal/api/action_request_manager.go
@@ -46,6 +46,12 @@ func (a *ActionRequestManager) Handlers() []octant.ClientRequestHandler {
 func (a *ActionRequestManager) PerformAction(state octant.State, payload action.Payload) error {
 	ctx := ocontext.WithWebsocketClientID(context.TODO(), state.GetClientID())
 
+	pluginState := ocontext.PluginState{
+		Filters:   state.GetFilters(),
+		Namespace: state.GetNamespace(),
+	}
+	ctx = ocontext.WithPluginState(ctx, pluginState)
+
 	actionName, err := payload.String("action")
 	if err != nil {
 		// TODO: alert the user this action doesn't exist (GH#493)

diff --git a/internal/api/content_manager.go b/internal/api/content_manager.go
@@ -195,6 +195,11 @@ func (cm *ContentManager) generateContent(ctx context.Context, state octant.Stat
 	}
 
 	ctx = ocontext.WithWebsocketClientID(ctx, state.GetClientID())
+	pluginState := ocontext.PluginState{
+		Filters:   state.GetFilters(),
+		Namespace: state.GetNamespace(),
+	}
+	ctx = ocontext.WithPluginState(ctx, pluginState)
 
 	contentResponse, err := m.Content(ctx, modulePath, options)
 	if err != nil {

diff --git a/internal/context/context.go b/internal/context/context.go
@@ -5,7 +5,11 @@
 
 package context
 
-import "context"
+import (
+	"context"
+
+	"github.com/vmware-tanzu/octant/internal/octant"
+)
 
 type OctantContextKey string
 
@@ -33,3 +37,23 @@ func WebsocketClientIDFrom(ctx context.Context) string {
 func WithWebsocketClientID(ctx context.Context, id string) context.Context {
 	return context.WithValue(ctx, WebsocketClientIDKey, id)
 }
+
+type OctantPluginState string
+
+const PluginStateKey = OctantPluginState("clientState")
+
+type PluginState struct {
+	Filters   []octant.Filter `json:"filters"`
+	Namespace string          `json:"namespace"`
+}
+
+func WithPluginState(ctx context.Context, state PluginState) context.Context {
+	return context.WithValue(ctx, PluginStateKey, state)
+}
+
+func PluginStateFrom(ctx context.Context) PluginState {
+	if ctx.Value(PluginStateKey) == nil {
+		return PluginState{}
+	}
+	return ctx.Value(PluginStateKey).(PluginState)
+}
diff --git a/pkg/plugin/javascript.go b/pkg/plugin/javascript.go
@@ -238,6 +238,7 @@ func (t *jsPlugin) Content(ctx context.Context, contentPath string) (component.C
 
 	t.loop.RunOnLoop(func(vm *goja.Runtime) {
 		clientID := ocontext.WebsocketClientIDFrom(ctx)
+		ostate := ocontext.PluginStateFrom(ctx)
 
 		handler, err := vm.RunString("_concretePlugin.contentHandler")
 		if err != nil {
@@ -258,6 +259,10 @@ func (t *jsPlugin) Content(ctx context.Context, contentPath string) (component.C
 		if err := obj.Set("clientID", vm.ToValue(clientID)); err != nil {
 			errCh <- fmt.Errorf("unable to set clientID: %w", err)
 		}
+		if err := obj.Set("ostate", vm.ToValue(ostate)); err != nil {
+			errCh <- fmt.Errorf("unable to set octant state: %w", err)
+		}
+
 		s, err := cHandler(t.pluginClass, obj)
 		if err != nil {
 			errCh <- fmt.Errorf("calling contentHandler: %w", err)
@@ -457,6 +462,7 @@ func (t *jsPlugin) HandleAction(ctx context.Context, actionPath string, payload
 
 	t.loop.RunOnLoop(func(vm *goja.Runtime) {
 		clientID := ocontext.WebsocketClientIDFrom(ctx)
+		ostate := ocontext.PluginStateFrom(ctx)
 
 		handler, err := vm.RunString("_concretePlugin.actionHandler")
 		if err != nil {
@@ -486,6 +492,9 @@ func (t *jsPlugin) HandleAction(ctx context.Context, actionPath string, payload
 			errCh <- fmt.Errorf("unable to set clientID: %w", err)
 			return
 		}
+		if err := obj.Set("ostate", vm.ToValue(ostate)); err != nil {
+			errCh <- fmt.Errorf("unable to set octant state: %w", err)
+		}
 
 		s, err := cHandler(t.pluginClass, obj)
 		if err != nil {
@@ -570,6 +579,7 @@ func (t *jsPlugin) objectRequestCall(ctx context.Context, handlerName string, ob
 
 	t.loop.RunOnLoop(func(vm *goja.Runtime) {
 		clientID := ocontext.WebsocketClientIDFrom(ctx)
+		ostate := ocontext.PluginStateFrom(ctx)
 
 		handler, err := vm.RunString(fmt.Sprintf("_concretePlugin.%s", handlerName))
 		if err != nil {
@@ -592,6 +602,9 @@ func (t *jsPlugin) objectRequestCall(ctx context.Context, handlerName string, ob
 			errCh <- fmt.Errorf("unable to set clientID: %w", err)
 			return
 		}
+		if err := obj.Set("ostate", vm.ToValue(ostate)); err != nil {
+			errCh <- fmt.Errorf("unable to set octant state: %w", err)
+		}
 		s, err := cHandler(t.pluginClass, obj)
 		if err != nil {
 			errCh <- err