mirror of
https://github.com/turbot/steampipe.git
synced 2026-05-10 15:01:45 -04:00
76 lines
2.7 KiB
Go
76 lines
2.7 KiB
Go
package dashboardexecute
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/turbot/steampipe/pkg/dashboard/dashboardevents"
|
|
"github.com/turbot/steampipe/pkg/dashboard/dashboardtypes"
|
|
"github.com/turbot/steampipe/pkg/initialisation"
|
|
"github.com/turbot/steampipe/pkg/steampipeconfig/modconfig"
|
|
)
|
|
|
|
func GenerateSnapshot(ctx context.Context, target string, initData *initialisation.InitData, inputs map[string]any) (snapshot *dashboardtypes.SteampipeSnapshot, err error) {
|
|
w := initData.Workspace
|
|
|
|
parsedName, err := modconfig.ParseResourceName(target)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// no session for manual execution
|
|
sessionId := ""
|
|
errorChannel := make(chan error)
|
|
resultChannel := make(chan *dashboardtypes.SteampipeSnapshot)
|
|
dashboardEventHandler := func(ctx context.Context, event dashboardevents.DashboardEvent) {
|
|
handleDashboardEvent(ctx, event, resultChannel, errorChannel)
|
|
}
|
|
w.RegisterDashboardEventHandler(ctx, dashboardEventHandler)
|
|
// clear event handlers again in case another snapshot will be generated in this run
|
|
defer w.UnregisterDashboardEventHandlers()
|
|
|
|
// all runtime dependencies must be resolved before execution (i.e. inputs must be passed in)
|
|
Executor.interactive = false
|
|
Executor.ExecuteDashboard(ctx, sessionId, target, inputs, w, initData.Client)
|
|
|
|
select {
|
|
case err = <-errorChannel:
|
|
return nil, err
|
|
case snapshot = <-resultChannel:
|
|
// set the filename root of the snapshot
|
|
fileRootName, err := parsedName.ToFullNameWithMod(w.Mod.ShortName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
snapshot.FileNameRoot = fileRootName
|
|
// return the context error (if any) to ensure we respect cancellation
|
|
return snapshot, ctx.Err()
|
|
}
|
|
}
|
|
|
|
func handleDashboardEvent(_ context.Context, event dashboardevents.DashboardEvent, resultChannel chan *dashboardtypes.SteampipeSnapshot, errorChannel chan error) {
|
|
switch e := event.(type) {
|
|
case *dashboardevents.ExecutionError:
|
|
errorChannel <- e.Error
|
|
case *dashboardevents.ExecutionComplete:
|
|
log.Println("[TRACE] execution complete event", *e)
|
|
snap := ExecutionCompleteToSnapshot(e)
|
|
resultChannel <- snap
|
|
}
|
|
}
|
|
|
|
// ExecutionCompleteToSnapshot transforms the ExecutionComplete event into a SteampipeSnapshot
|
|
func ExecutionCompleteToSnapshot(event *dashboardevents.ExecutionComplete) *dashboardtypes.SteampipeSnapshot {
|
|
return &dashboardtypes.SteampipeSnapshot{
|
|
SchemaVersion: fmt.Sprintf("%d", dashboardtypes.SteampipeSnapshotSchemaVersion),
|
|
Panels: event.Panels,
|
|
Layout: event.Root.AsTreeNode(),
|
|
Inputs: event.Inputs,
|
|
Variables: event.Variables,
|
|
SearchPath: event.SearchPath,
|
|
StartTime: event.StartTime,
|
|
EndTime: event.EndTime,
|
|
Title: event.Root.GetTitle(),
|
|
}
|
|
}
|